add print for hydro

Author: undefined-moe

packages/server/handler/client.ts

@@ -81,6 +81,7 @@ class ClientPrintDoneHandler extends Handler {
         if (!code) throw new ValidationError('Code', null, 'Code not found');
         if (code.printer !== params.cid) throw new BadRequestError('Client', null, 'Client not match');
         await this.ctx.db.code.updateOne({ _id: params.tid }, { $set: { done: 1, doneAt: new Date().getTime() } });
+        await this.ctx.fetcher.setPrintDone(code.id!);
         await this.ctx.parallel('print/doneTask', client._id, `${client._id}#${params.printer || 'unknown'}`);
         this.response.body = { code: 1 };
         logger.info(`Client ${client.name} connected, print task ${code.tid}#${code._id} completed.`);

packages/server/handler/printer.ts

@@ -91,7 +91,7 @@ class CodeHandler extends Handler {
         fs.writeFileSync(path.resolve(process.cwd(), 'data/codes', `${team}#${res._id}`), codeFile);
         this.response.body = `The code has been submitted. Code Print ID: ${team}#${res._id}`;
         logger.info(`Team(${team}): ${tname} submitted code. Code Print ID: ${team}#${res._id}`);
-        await this.ctx.parallel('print/newTask');
+        await this.ctx.parallel('print/newTask', 1);
         if (tname.length > 40) {
             logger.warn(`Team ${tname} name is too long, may cause overflow!`);
             this.response.body += ', your team name is too long, may cause print failed!';

packages/server/service/fetcher.ts

@@ -2,7 +2,8 @@
 import { Context, Service } from 'cordis';
 import superagent from 'superagent';
 import { config } from '../config';
-import { Logger, mongoId, sleep } from '../utils';
+import { fs, Logger, mongoId, sleep } from '../utils';
+import path from 'node:path';
 
 const logger = new Logger('fetcher');
 const fetch = (url: string, type: 'get' | 'post' = 'get') => {
@@ -57,6 +58,7 @@ class BasicFetcher extends Service implements IBasicFetcher {
         const first = await this.contestInfo();
         if (first) await this.teamInfo();
         await this.balloonInfo(first);
+        await this.printInfo(first);
     }
 
     async contestInfo() {
@@ -81,6 +83,14 @@ class BasicFetcher extends Service implements IBasicFetcher {
     async setBalloonDone(bid) {
         this.logger.debug(`Balloon ${bid} set done`);
     }
+
+    async printInfo(all) {
+        this.logger.debug('Found 0 prints in Server Mode');
+    }
+
+    async setPrintDone(pid) {
+        this.logger.debug(`Print ${pid} set done`);
+    }
 }
 
 class DOMjudgeFetcher extends BasicFetcher {
@@ -177,6 +187,7 @@ class HydroFetcher extends BasicFetcher {
         }
         const contest = body.tdoc;
         contest.freeze_time = contest.lockAt;
+        delete contest.content;
         const old = this?.contest?.id;
         this.contest = {
             info: contest, id: contest._id, name: contest.title, domainId,
@@ -252,6 +263,39 @@ class HydroFetcher extends BasicFetcher {
         await fetch(`/d/${this.contest.domainId}/contest/${this.contest.id}/balloon`, 'post').send({ operation: 'done', balloon: bid });
         this.logger.debug(`Balloon ${bid} set done`);
     }
+
+    async printInfo(all) {
+        const doFetch = async () => {
+            const { body } = await fetch(`/d/${this.contest.domainId}/contest/${this.contest.id}/print`, 'post').send({ operation: 'allocate_print_task' });
+            return body;
+        }
+        let { task, udoc } = await doFetch();
+        let cnt = 0;
+        while (task) {
+            const res = await this.ctx.db.code.insert({
+                id: task._id,
+                tid: task.owner,
+                team: `${udoc.school ? `${udoc.school}: ` : ''}${udoc.displayName || udoc.uname}`,
+                location: udoc.studentId,
+                filename: task.title,
+                lang: task.title.split('.').pop() || 'txt',
+                createdAt: new Date(parseInt(task._id.substring(0, 8), 16) * 1000).getTime(),
+                printer: '',
+                done: task.status === 'printed' ? 1 : 0,
+            });
+            await fs.ensureDir(path.resolve(process.cwd(), 'data/codes'));
+            await fs.writeFile(path.resolve(process.cwd(), 'data/codes', `${task.owner}#${res._id}`), task.content);
+            logger.info(`Team(${task.owner}): ${udoc.displayName || udoc.uname} submitted code. Code Print ID: ${task.owner}#${res._id}`);
+            cnt++;
+            ({ task, udoc } = await doFetch());
+        }
+        await this.ctx.parallel('print/newTask', cnt);
+    }
+
+    async setPrintDone(pid) {
+        await fetch(`/d/${this.contest.domainId}/contest/${this.contest.id}/print`, 'post').send({ operation: 'update_print_task', taskId: pid, status: 'printed' });
+        this.logger.debug(`Print ${pid} set done`);
+    }
 }
 
 const fetcherList = {

packages/server/utils/index.ts

@@ -19,7 +19,7 @@ export * as yaml from 'js-yaml';
 
 export function StaticHTML(context, randomHash) {
     // eslint-disable-next-line max-len
-    return `<html><head><meta charset="UTF-8"><meta name="viewport" content="width=device-width,initial-scale=1"><title>@Hydro/XCPC-TOOLS</title></head><body><div id="root"></div><script>window.Context=JSON.parse('${JSON.stringify(context)}')</script><script src="/main.js?${randomHash}"></script></body></html>`;
+    return `<html><head><meta charset="UTF-8"><meta name="viewport" content="width=device-width,initial-scale=1"><title>@Hydro/XCPC-TOOLS</title></head><body><div id="root"></div><script>window.Context=JSON.parse('${JSON.stringify(context).replace(/\\/g, '\\\\').replace(/'/g, '\\\'')}')</script><script src="/main.js?${randomHash}"></script></body></html>`;
 }
 
 export function decodeBinary(file: string, name: string) {

packages/server/utils/metrics.ts

@@ -8,7 +8,7 @@ declare module 'cordis' {
         metrics: Registry;
     }
     interface Events {
-        'print/newTask': () => void;
+        'print/newTask': (count: number) => void;
         'print/sendTask': (client: string) => void;
         'print/doneTask': (client: string, printer: string) => void;
         'balloon/newTask': (count: number) => void;
@@ -41,7 +41,7 @@ export function createMetricsRegistry(ctx: Context) {
     const printTaskCounter = createMetric(Counter, 'xcpc_printcount', 'printcount', {
         labelNames: ['status', 'client', 'printer'],
     });
-    ctx.on('print/newTask', () => printTaskCounter.inc({ status: 'new' }));
+    ctx.on('print/newTask', (count) => printTaskCounter.inc({ status: 'new' }, count));
 
     ctx.on('print/sendTask', (client) => printTaskCounter.inc({ status: 'sent', client }));