This commit is contained in:
syuilo 2018-11-02 13:47:44 +09:00
parent ddc3c5ba68
commit 174f8022eb
No known key found for this signature in database
GPG key ID: BDC4C49D06AB9D69
131 changed files with 512 additions and 917 deletions

View file

@ -1,10 +1,8 @@
import * as fs from 'fs';
const ms = require('ms');
import $ from 'cafy'; import ID, { transform } from '../../../../../misc/cafy-id';
import { validateFileName, pack } from '../../../../../models/drive-file';
import create from '../../../../../services/drive/add-file';
import { ILocalUser } from '../../../../../models/user';
import getParams from '../../../get-params';
import define from '../../../define';
export const meta = {
desc: {
@ -52,11 +50,7 @@ export const meta = {
}
};
export default async (file: any, params: any, user: ILocalUser): Promise<any> => {
if (file == null) {
throw 'file is required';
}
export default define(meta, (ps, user, app, file, cleanup) => new Promise(async (res, rej) => {
// Get 'name' parameter
let name = file.originalname;
if (name !== undefined && name !== null) {
@ -66,35 +60,24 @@ export default async (file: any, params: any, user: ILocalUser): Promise<any> =>
} else if (name === 'blob') {
name = null;
} else if (!validateFileName(name)) {
throw 'invalid name';
return rej('invalid name');
}
} else {
name = null;
}
function cleanup() {
fs.unlink(file.path, () => {});
}
const [ps, psErr] = getParams(meta, params);
if (psErr) {
cleanup();
throw psErr;
}
try {
// Create file
const driveFile = await create(user, file.path, name, null, ps.folderId, ps.force, false, null, null, ps.isSensitive);
cleanup();
// Serialize
return pack(driveFile);
res(pack(driveFile));
} catch (e) {
console.error(e);
cleanup();
throw e;
rej(e);
}
};
}));