core: fix config hande

packages/hydrooj/src/model/problem.ts

@@ -1,6 +1,7 @@
 import type { Readable } from 'stream';
 import { ObjectID, FilterQuery } from 'mongodb';
 import { Dictionary, escapeRegExp, pick } from 'lodash';
+import { streamToBuffer } from '@hydrooj/utils/lib/utils';
 import { STATUS } from './builtin';
 import * as document from './document';
 import { buildProjection } from '../utils';
@@ -248,6 +249,16 @@ export class ProblemModel {
     }
 }
 
+bus.on('problem/addTestdata', async (domainId, docId, name) => {
+    if (!['config.yaml', 'config.yml', 'Config.yaml', 'Config.yml'].includes(name)) return;
+    const buf = await storage.get(`problem/${domainId}/${docId}/testdata/${name}`);
+    await ProblemModel.edit(domainId, docId, { config: (await streamToBuffer(buf)).toString() });
+});
+bus.on('problem/delTestdata', async (domainId, docId, names) => {
+    if (!names.includes('config.yaml')) return;
+    await ProblemModel.edit(domainId, docId, { config: '' });
+});
+
 ProblemModel.extend((docId, pid) => ({
     _id: new ObjectID(),
     domainId: 'system',