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

114 lines
2.4 KiB
TypeScript
Raw Normal View History

2017-11-06 06:37:00 +01:00
import * as mongodb from 'mongodb';
2018-02-02 00:06:01 +01:00
import deepcopy = require('deepcopy');
import { pack as packFolder } from './drive-folder';
import config from '../../conf';
2017-11-06 06:37:00 +01:00
import monkDb, { nativeDbConn } from '../../db/mongodb';
2017-01-17 01:12:33 +01:00
2018-02-02 00:06:01 +01:00
const DriveFile = monkDb.get<IDriveFile>('drive_files.files');
2018-02-02 00:06:01 +01:00
export default DriveFile;
2016-12-28 23:49:51 +01:00
2017-11-06 06:37:00 +01:00
const getGridFSBucket = async (): Promise<mongodb.GridFSBucket> => {
2017-11-06 08:32:01 +01:00
const db = await nativeDbConn();
2017-11-06 06:37:00 +01:00
const bucket = new mongodb.GridFSBucket(db, {
bucketName: 'drive_files'
2017-11-06 08:32:01 +01:00
});
return bucket;
};
2017-11-06 06:37:00 +01:00
2017-11-06 08:32:01 +01:00
export { getGridFSBucket };
2017-11-06 06:37:00 +01:00
2018-02-02 00:06:01 +01:00
export type IDriveFile = {
_id: mongodb.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-02-02 00:21:30 +01:00
metadata: {
properties: any;
user_id: mongodb.ObjectID;
folder_id: mongodb.ObjectID;
}
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
/**
* Pack a drive file for API response
*
* @param {any} file
* @param {any} options?
* @return {Promise<any>}
*/
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
if (mongodb.ObjectID.prototype.isPrototypeOf(file)) {
_file = await DriveFile.findOne({
_id: file
});
} else if (typeof file === 'string') {
_file = await DriveFile.findOne({
_id: new mongodb.ObjectID(file)
});
} else {
_file = deepcopy(file);
}
if (!_file) return reject('invalid file arg.');
// rendered target
let _target: any = {};
_target.id = _file._id;
_target.created_at = _file.uploadDate;
_target.name = _file.filename;
_target.type = _file.contentType;
_target.datasize = _file.length;
_target.md5 = _file.md5;
_target = Object.assign(_target, _file.metadata);
_target.url = `${config.drive_url}/${_target.id}/${encodeURIComponent(_target.name)}`;
if (_target.properties == null) _target.properties = {};
if (opts.detail) {
if (_target.folder_id) {
// Populate folder
_target.folder = await packFolder(_target.folder_id, {
detail: true
});
}
/*
if (_target.tags) {
// Populate tags
_target.tags = await _target.tags.map(async (tag: any) =>
await serializeDriveTag(tag)
);
}
*/
}
resolve(_target);
});