This commit is contained in:
syuilo
2017-11-07 21:04:32 +09:00
parent 307f0389ac
commit a85d5ac4bb
4 changed files with 51 additions and 5 deletions

View File

@ -13,13 +13,13 @@ import { Duplex } from 'stream';
const log = debug('misskey:register-drive-file');
const addToGridFS = (name, binary, metadata): Promise<any> => new Promise(async (resolve, reject) => {
const addToGridFS = (name, binary, type, metadata): Promise<any> => new Promise(async (resolve, reject) => {
const dataStream = new Duplex();
dataStream.push(binary);
dataStream.push(null);
const bucket = await getGridFSBucket();
const writeStream = bucket.openUploadStream(name, { metadata });
const writeStream = bucket.openUploadStream(name, { contentType: type, metadata });
writeStream.once('finish', (doc) => { resolve(doc); });
writeStream.on('error', reject);
dataStream.pipe(writeStream);
@ -144,10 +144,9 @@ export default (
}
// Create DriveFile document
const file = await addToGridFS(name, data, {
const file = await addToGridFS(name, data, mime, {
user_id: user._id,
folder_id: folder !== null ? folder._id : null,
type: mime,
comment: comment,
properties: properties
});

View File

@ -48,6 +48,7 @@ export default (
_target.id = _file._id;
_target.created_at = _file.uploadDate;
_target.name = _file.filename;
_target.type = _file.contentType;
_target = Object.assign(_target, _file.metadata);