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

framework: auto cleanup uploaded files (#818)

Browse Source
This commit is contained in:
undefined committed 2025-01-03 01:01:37 +08:00
1 parent 8eed68e78f
commit 566ebd9bb2
10 files changed
+40 -24
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

packages/fps-importer/index.ts
+4 -4
View File
@@ -82,12 +82,12 @@ class FpsProblemImportHandler extends Handler {
}
async post({ domainId }) {
if (!this.request.files.file) throw new ValidationError('file');
const file = this.request.files.file;
if (!file) throw new ValidationError('file');
const tasks = [];
try {
const file = await fs.stat(this.request.files.file.filepath);
if (file.size > SystemModel.get('import-fps.limit')) throw new FileTooLargeError();
const content = fs.readFileSync(this.request.files.file.filepath, 'utf-8');
const content = await fs.readFile(file.filepath, 'utf-8');
const result = await xml2js.parseStringPromise(content);
tasks.push(result);
} catch (e) {
@@ -95,7 +95,7 @@ class FpsProblemImportHandler extends Handler {
console.log(e);
let zip: AdmZip;
try {
zip = new AdmZip(this.request.files.file.filepath);
zip = new AdmZip(file.filepath);
} catch (err) {
throw new ValidationError('zip', null, err.message);
}
packages/hydrooj/src/handler/contest.ts
+1 -2
View File
@@ -476,8 +476,7 @@ export class ContestManagementHandler extends ContestManagementBaseHandler {
}
const file = this.request.files?.file;
if (!file) throw new ValidationError('file');
const f = statSync(file.filepath);
const size = Math.sum((this.tdoc.files || []).map((i) => i.size)) + f.size;
const size = Math.sum((this.tdoc.files || []).map((i) => i.size)) + file.size;
if (size >= system.get('limit.contest_files_size')) {
throw new FileLimitExceededError('size');
}
packages/hydrooj/src/handler/homework.ts
+1 -2
View File
@@ -273,8 +273,7 @@ export class HomeworkFilesHandler extends Handler {
}
const file = this.request.files?.file;
if (!file) throw new ValidationError('file');
const f = statSync(file.filepath);
const size = Math.sum((this.tdoc.files || []).map((i) => i.size)) + f.size;
const size = Math.sum((this.tdoc.files || []).map((i) => i.size)) + file.size;
if (size >= system.get('limit.contest_files_size')) {
throw new FileLimitExceededError('size');
}
packages/hydrooj/src/handler/misc.ts
+1 -2
View File
@@ -53,8 +53,7 @@ export class FilesHandler extends Handler {
}
const file = this.request.files?.file;
if (!file) throw new ValidationError('file');
const f = statSync(file.filepath);
const size = Math.sum((this.user._files || []).map((i) => i.size)) + f.size;
const size = Math.sum((this.user._files || []).map((i) => i.size)) + file.size;
if (size >= system.get('limit.user_files_size')) {
if (!this.user.hasPriv(PRIV.PRIV_UNLIMITED_QUOTA)) throw new FileLimitExceededError('size');
}
packages/hydrooj/src/handler/problem.ts
+4 -3
View File
@@ -705,13 +705,14 @@ export class ProblemFilesHandler extends ProblemDetailHandler {
@post('filename', Types.Filename, true)
@post('type', Types.Range(['testdata', 'additional_file']), true)
async postUploadFile(domainId: string, filename: string, type = 'testdata') {
if (!this.request.files.file) throw new ValidationError('file');
filename ||= this.request.files.file.originalFilename || String.random(16);
const file = this.request.files.file;
if (!file) throw new ValidationError('file');
filename ||= file.originalFilename || String.random(16);
const files = [];
if (filename.endsWith('.zip') && type === 'testdata') {
let zip: AdmZip;
try {
zip = new AdmZip(this.request.files.file.filepath);
zip = new AdmZip(file.filepath);
} catch (e) {
throw new ValidationError('zip', null, e.message);
}
packages/hydrooj/src/handler/training.ts
+1 -2
View File
@@ -261,8 +261,7 @@ export class TrainingFilesHandler extends Handler {
}
const file = this.request.files?.file;
if (!file) throw new ValidationError('file');
const f = statSync(file.filepath);
const size = Math.sum((this.tdoc.files || []).map((i) => i.size)) + f.size;
const size = Math.sum((this.tdoc.files || []).map((i) => i.size)) + file.size;
if (size >= system.get('limit.contest_files_size')) {
throw new FileLimitExceededError('size');
}
packages/hydrooj/src/service/watcher.ts
+1 -1
View File
@@ -102,7 +102,7 @@ export default class Watcher extends Service {
let hasUpdate = false;
while (index < pending.length) {
const filename = pending[index];
const { children } = require.cache[filename];
const children = require.cache[filename]?.children || [];
let isDeclined = true;
let isAccepted = false;
for (const { filename } of children) {
packages/import-hoj/index.ts
+4 -4
View File
@@ -124,10 +124,10 @@ class ImportHojHandler extends Handler {
}
async post({ domainId }) {
if (!this.request.files.file) throw new ValidationError('file');
const stat = await fs.stat(this.request.files.file.filepath);
if (stat.size > 128 * 1024 * 1024) throw new ValidationError('file', 'File too large');
await this.fromFile(domainId, this.request.files.file.filepath);
const file = this.request.files.file;
if (!file) throw new ValidationError('file');
if (file.size > 128 * 1024 * 1024) throw new ValidationError('file', 'File too large');
await this.fromFile(domainId, file.filepath);
this.response.redirect = this.url('problem_main');
}
}
packages/import-qduoj/index.ts
+4 -4
View File
@@ -129,10 +129,10 @@ class ImportQduojHandler extends Handler {
}
async post({ domainId }) {
if (!this.request.files.file) throw new ValidationError('file');
const stat = await fs.stat(this.request.files.file.filepath);
if (stat.size > 256 * 1024 * 1024) throw new FileTooLargeError('256m');
await this.fromFile(domainId, this.request.files.file.filepath);
const file = this.request.files.file;
if (!file) throw new ValidationError('file');
if (file.size > 256 * 1024 * 1024) throw new FileTooLargeError('256m');
await this.fromFile(domainId, file.filepath);
this.response.redirect = this.url('problem_main');
}
}