Weishen/Hydro
2
0 Fork 0
mirror of https://gitcode.com/gh_mirrors/hy/Hydro.git synced 2026-10-20 01:38:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

core: storage: avoid special chars in filename

Browse Source
This commit is contained in:
undefined committed 2023-05-01 21:44:02 +08:00
1 parent ada5366334
commit 8f30975dc9
1 file changed
+6 -2
Download Patch File Download Diff File Expand all files Collapse all files
packages/hydrooj/src/model/storage.ts
+6 -2
View File
@@ -13,16 +13,20 @@ import * as system from './system';
export class StorageModel { export class StorageModel {
static coll = db.collection('storage'); static coll = db.collection('storage');
static generateId(ext: string) {
return `${nanoid(3).replace(/[_-]/g, '0')}/${nanoid().replace(/[_-]/g, '0')}${ext}`;
}
static async put(path: string, file: string | Buffer | Readable, owner?: number) { static async put(path: string, file: string | Buffer | Readable, owner?: number) {
const meta = {}; const meta = {};
await StorageModel.del([path]); await StorageModel.del([path]);
meta['Content-Type'] = (path.endsWith('.ans') || path.endsWith('.out')) meta['Content-Type'] = (path.endsWith('.ans') || path.endsWith('.out'))
? 'text/plain' ? 'text/plain'
: lookup(path) || 'application/octet-stream'; : lookup(path) || 'application/octet-stream';
let _id = `${nanoid(3)}/${nanoid()}${extname(path)}`; let _id = StorageModel.generateId(extname(path));
// Make sure id is not used // Make sure id is not used
// eslint-disable-next-line no-await-in-loop // eslint-disable-next-line no-await-in-loop
while (await StorageModel.coll.findOne({ _id })) _id = `${nanoid(3)}/${nanoid()}${extname(path)}`; while (await StorageModel.coll.findOne({ _id })) _id = StorageModel.generateId(extname(path));
await storage.put(_id, file, meta); await storage.put(_id, file, meta);
const { metaData, size, etag } = await storage.getMeta(_id); const { metaData, size, etag } = await storage.getMeta(_id);
await StorageModel.coll.insertOne({ await StorageModel.coll.insertOne({