Sharkey/src/server/api/endpoints/drive/files/update.ts

122 lines
2.6 KiB
TypeScript
Raw Normal View History

2018-11-01 20:32:24 +02:00
import $ from 'cafy'; import ID, { transform } from '../../../../../misc/cafy-id';
2018-03-29 14:32:18 +03:00
import DriveFolder from '../../../../../models/drive-folder';
import DriveFile, { validateFileName, pack } from '../../../../../models/drive-file';
2018-07-07 13:19:00 +03:00
import { publishDriveStream } from '../../../../../stream';
2018-11-02 06:47:44 +02:00
import define from '../../../define';
2018-09-25 15:09:38 +03:00
import Note from '../../../../../models/note';
2016-12-29 00:49:51 +02:00
2018-07-16 22:36:44 +03:00
export const meta = {
desc: {
2018-08-29 00:59:43 +03:00
'ja-JP': '指定したドライブのファイルの情報を更新します。',
'en-US': 'Update specified file of drive.'
2018-07-16 22:36:44 +03:00
},
requireCredential: true,
2018-07-19 20:40:37 +03:00
kind: 'drive-write',
params: {
2018-11-01 20:32:24 +02:00
fileId: {
validator: $.type(ID),
transform: transform,
2018-07-19 20:40:37 +03:00
desc: {
2018-08-29 00:59:43 +03:00
'ja-JP': '対象のファイルID'
2018-07-19 20:40:37 +03:00
}
2018-11-01 20:32:24 +02:00
},
2018-07-19 20:40:37 +03:00
2018-11-01 20:32:24 +02:00
folderId: {
validator: $.type(ID).optional.nullable,
transform: transform,
default: undefined as any,
2018-07-19 20:40:37 +03:00
desc: {
2018-08-29 00:59:43 +03:00
'ja-JP': 'フォルダID'
2018-07-19 20:40:37 +03:00
}
2018-11-01 20:32:24 +02:00
},
2018-07-19 20:40:37 +03:00
2018-11-01 20:32:24 +02:00
name: {
validator: $.str.optional.pipe(validateFileName),
default: undefined as any,
2018-07-19 20:40:37 +03:00
desc: {
2018-08-29 00:59:43 +03:00
'ja-JP': 'ファイル名',
'en-US': 'Name of the file'
2018-07-19 20:40:37 +03:00
}
2018-11-01 20:32:24 +02:00
},
2018-07-19 20:40:37 +03:00
2018-11-01 20:32:24 +02:00
isSensitive: {
validator: $.bool.optional,
default: undefined as any,
2018-07-19 20:40:37 +03:00
desc: {
2018-08-29 00:59:43 +03:00
'ja-JP': 'このメディアが「閲覧注意」(NSFW)かどうか',
'en-US': 'Whether this media is NSFW'
2018-07-19 20:40:37 +03:00
}
2018-11-01 20:32:24 +02:00
}
2018-07-19 20:40:37 +03:00
}
2018-07-16 22:36:44 +03:00
};
2018-11-02 06:47:44 +02:00
export default define(meta, (ps, user) => new Promise(async (res, rej) => {
2017-03-03 12:33:14 +02:00
// Fetch file
2016-12-29 00:49:51 +02:00
const file = await DriveFile
.findOne({
2018-07-19 20:40:37 +03:00
_id: ps.fileId,
2018-03-29 08:48:47 +03:00
'metadata.userId': user._id
2016-12-29 00:49:51 +02:00
});
if (file === null) {
return rej('file-not-found');
}
2018-07-19 20:40:37 +03:00
if (ps.name) file.filename = ps.name;
2016-12-29 00:49:51 +02:00
2018-08-18 21:36:51 +03:00
if (ps.isSensitive !== undefined) file.metadata.isSensitive = ps.isSensitive;
2017-03-03 12:33:14 +02:00
2018-07-19 20:40:37 +03:00
if (ps.folderId !== undefined) {
if (ps.folderId === null) {
2018-03-29 08:48:47 +03:00
file.metadata.folderId = null;
2016-12-29 00:49:51 +02:00
} else {
2017-03-03 12:33:14 +02:00
// Fetch folder
2017-02-12 18:49:17 +02:00
const folder = await DriveFolder
2016-12-29 00:49:51 +02:00
.findOne({
2018-07-19 20:40:37 +03:00
_id: ps.folderId,
2018-03-29 08:48:47 +03:00
userId: user._id
2016-12-29 00:49:51 +02:00
});
if (folder === null) {
2017-02-12 18:17:44 +02:00
return rej('folder-not-found');
2016-12-29 00:49:51 +02:00
}
2018-03-29 08:48:47 +03:00
file.metadata.folderId = folder._id;
2016-12-29 00:49:51 +02:00
}
}
await DriveFile.update(file._id, {
$set: {
filename: file.filename,
2018-07-19 20:40:37 +03:00
'metadata.folderId': file.metadata.folderId,
'metadata.isSensitive': file.metadata.isSensitive
}
2017-02-08 18:37:50 +02:00
});
2016-12-29 00:49:51 +02:00
2018-09-25 15:09:38 +03:00
// ドライブのファイルが非正規化されているドキュメントも更新
Note.find({
'_files._id': file._id
}).then(notes => {
notes.forEach(note => {
note._files[note._files.findIndex(f => f._id.equals(file._id))] = file;
2018-10-14 04:06:10 +03:00
Note.update({ _id: note._id }, {
$set: {
_files: note._files
}
2018-09-25 15:09:38 +03:00
});
});
});
2016-12-29 00:49:51 +02:00
// Serialize
2018-02-02 01:21:30 +02:00
const fileObj = await pack(file);
2017-11-06 08:18:45 +02:00
2016-12-29 00:49:51 +02:00
// Response
res(fileObj);
2018-10-23 01:01:43 +03:00
// Publish fileUpdated event
publishDriveStream(user._id, 'fileUpdated', fileObj);
2018-11-02 06:47:44 +02:00
}));