rudeshark.net/src/models/drive-file.ts

183 lines
4.5 KiB
TypeScript
Raw Normal View History

2018-04-11 21:22:06 +02:00
import * as mongo from 'mongodb';
2018-05-26 09:04:47 +02:00
import * as deepcopy from 'deepcopy';
2018-02-02 00:06:01 +01:00
import { pack as packFolder } from './drive-folder';
2018-04-02 06:15:53 +02:00
import config from '../config';
2018-03-29 13:32:18 +02:00
import monkDb, { nativeDbConn } from '../db/mongodb';
2018-04-11 21:22:06 +02:00
import Note, { deleteNote } from './note';
import MessagingMessage, { deleteMessagingMessage } from './messaging-message';
import User from './user';
import DriveFileThumbnail, { deleteDriveFileThumbnail } from './drive-file-thumbnail';
2017-01-17 01:12:33 +01:00
2018-03-29 07:48:47 +02:00
const DriveFile = monkDb.get<IDriveFile>('driveFiles.files');
2018-05-12 20:13:55 +02:00
DriveFile.createIndex('md5');
2018-04-03 16:45:13 +02:00
DriveFile.createIndex('metadata.uri', { sparse: true, unique: true });
2018-02-02 00:06:01 +01:00
export default DriveFile;
2016-12-28 23:49:51 +01:00
2018-04-17 12:55:58 +02:00
export const DriveFileChunk = monkDb.get('driveFiles.chunks');
export const getDriveFileBucket = async (): Promise<mongo.GridFSBucket> => {
2017-11-06 08:32:01 +01:00
const db = await nativeDbConn();
2018-04-11 21:22:06 +02:00
const bucket = new mongo.GridFSBucket(db, {
2018-03-29 07:48:47 +02:00
bucketName: 'driveFiles'
2017-11-06 08:32:01 +01:00
});
return bucket;
};
2017-11-06 06:37:00 +01:00
2018-04-03 16:45:13 +02:00
export type IMetadata = {
properties: any;
2018-04-11 21:22:06 +02:00
userId: mongo.ObjectID;
_user: any;
2018-04-11 21:22:06 +02:00
folderId: mongo.ObjectID;
2018-04-03 16:45:13 +02:00
comment: string;
uri?: string;
url?: string;
2018-04-17 13:04:19 +02:00
deletedAt?: Date;
isMetaOnly?: boolean;
2018-04-03 16:45:13 +02:00
};
2018-02-02 00:06:01 +01:00
export type IDriveFile = {
2018-04-11 21:22:06 +02:00
_id: mongo.ObjectID;
2018-02-02 00:21:30 +01:00
uploadDate: Date;
md5: string;
filename: string;
2018-02-04 06:52:33 +01:00
contentType: string;
2018-04-03 16:45:13 +02:00
metadata: IMetadata;
2018-02-02 00:06:01 +01:00
};
2016-12-28 23:49:51 +01:00
export function validateFileName(name: string): boolean {
return (
(name.trim().length > 0) &&
(name.length <= 200) &&
(name.indexOf('\\') === -1) &&
(name.indexOf('/') === -1) &&
(name.indexOf('..') === -1)
);
}
2018-02-02 00:06:01 +01:00
2018-04-11 21:22:06 +02:00
/**
* DriveFileを物理削除します
*/
export async function deleteDriveFile(driveFile: string | mongo.ObjectID | IDriveFile) {
let d: IDriveFile;
// Populate
if (mongo.ObjectID.prototype.isPrototypeOf(driveFile)) {
d = await DriveFile.findOne({
_id: driveFile
});
} else if (typeof driveFile === 'string') {
d = await DriveFile.findOne({
_id: new mongo.ObjectID(driveFile)
});
} else {
d = driveFile as IDriveFile;
}
if (d == null) return;
// このDriveFileを添付しているNoteをすべて削除
await Promise.all((
await Note.find({ mediaIds: d._id })
).map(x => deleteNote(x)));
// このDriveFileを添付しているMessagingMessageをすべて削除
await Promise.all((
await MessagingMessage.find({ fileId: d._id })
).map(x => deleteMessagingMessage(x)));
// このDriveFileがアバターやバナーに使われていたらそれらのプロパティをnullにする
const u = await User.findOne({ _id: d.metadata.userId });
if (u) {
2018-04-15 04:57:33 +02:00
if (u.avatarId && u.avatarId.equals(d._id)) {
2018-04-11 21:22:06 +02:00
await User.update({ _id: u._id }, { $set: { avatarId: null } });
}
2018-04-15 04:57:33 +02:00
if (u.bannerId && u.bannerId.equals(d._id)) {
2018-04-11 21:22:06 +02:00
await User.update({ _id: u._id }, { $set: { bannerId: null } });
}
}
// このDriveFileのDriveFileThumbnailをすべて削除
await Promise.all((
await DriveFileThumbnail.find({ 'metadata.originalId': d._id })
).map(x => deleteDriveFileThumbnail(x)));
2018-04-15 04:57:33 +02:00
// このDriveFileのチャンクをすべて削除
2018-04-17 12:55:58 +02:00
await DriveFileChunk.remove({
2018-04-15 04:57:33 +02:00
files_id: d._id
});
2018-04-11 21:22:06 +02:00
// このDriveFileを削除
await DriveFile.remove({
_id: d._id
});
}
2018-02-02 00:06:01 +01:00
/**
* Pack a drive file for API response
*/
export const pack = (
file: any,
options?: {
detail: boolean
}
) => new Promise<any>(async (resolve, reject) => {
const opts = Object.assign({
detail: false
}, options);
let _file: any;
// Populate the file if 'file' is ID
2018-04-11 23:20:53 +02:00
if (mongo.ObjectID.prototype.isPrototypeOf(file)) {
2018-02-02 00:06:01 +01:00
_file = await DriveFile.findOne({
_id: file
});
} else if (typeof file === 'string') {
_file = await DriveFile.findOne({
2018-04-11 23:20:53 +02:00
_id: new mongo.ObjectID(file)
2018-02-02 00:06:01 +01:00
});
} else {
_file = deepcopy(file);
}
if (!_file) return reject('invalid file arg.');
// rendered target
let _target: any = {};
_target.id = _file._id;
2018-03-29 07:48:47 +02:00
_target.createdAt = _file.uploadDate;
2018-02-02 00:06:01 +01:00
_target.name = _file.filename;
_target.type = _file.contentType;
_target.datasize = _file.length;
_target.md5 = _file.md5;
_target = Object.assign(_target, _file.metadata);
_target.src = _file.metadata.url;
_target.url = _file.metadata.isMetaOnly ? _file.metadata.url : `${config.drive_url}/${_target.id}/${encodeURIComponent(_target.name)}`;
_target.isRemote = _file.metadata.isMetaOnly;
2018-02-02 00:06:01 +01:00
if (_target.properties == null) _target.properties = {};
if (opts.detail) {
2018-03-29 07:48:47 +02:00
if (_target.folderId) {
2018-02-02 00:06:01 +01:00
// Populate folder
2018-03-29 07:48:47 +02:00
_target.folder = await packFolder(_target.folderId, {
2018-02-02 00:06:01 +01:00
detail: true
});
}
/*
if (_target.tags) {
// Populate tags
_target.tags = await _target.tags.map(async (tag: any) =>
await serializeDriveTag(tag)
);
}
*/
}
resolve(_target);
});