support fetch hydro

Author: pandadtdyy

package.json

@@ -6,12 +6,12 @@
     "packages/*"
   ],
   "scripts": {
-    "start:client": "node -r ./register.js packages/server/index.ts --client --debug",
-    "start:server": "node -r ./register.js packages/server/index.ts --debug",
+    "dev:client": "node -r ./register.js packages/server/index.ts --client --debug",
+    "dev:server": "node -r ./register.js packages/server/index.ts --debug",
     "lint": "eslint packages --ext js,ts,tsx,jsx",
-    "build": "yarn build:ui:prod && node -r ./register.js build.ts",
+    "build": "yarn build:ui && node -r ./register.js build.ts",
     "build:server": "node -r ./register.js build.ts",
-    "build:pkg": "yarn build:ui:prod && node -r ./register.js build.ts && pkg dist/xcpc-tools.js --targets linux,macos,win --out-path dist/pkg"
+    "build:pkg": "yarn build:ui && node -r ./register.js build.ts && pkg dist/xcpc-tools.js --targets linux,macos,win --out-path dist/pkg"
   },
   "devDependencies": {
     "@expo-google-fonts/noto-color-emoji": "^0.2.3",

packages/server/service/fetcher.ts

@@ -2,7 +2,7 @@
 import { Context, Service } from 'cordis';
 import superagent from 'superagent';
 import { config } from '../config';
-import { Logger } from '../utils';
+import { Logger, mongoId } from '../utils';
 
 const logger = new Logger('fetcher');
 const fetch = (url: string, type: 'get' | 'post' = 'get') => superagent[type](new URL(url, config.server).toString())
@@ -11,10 +11,10 @@ export interface IBasicFetcher {
     contest: Record<string, any>
     cron(): Promise<void>
     contestInfo(): Promise<boolean>
-    getToken?(username: string, password: string): Promise<void>
-    teamInfo?(): Promise<void>
-    balloonInfo?(all: boolean): Promise<void>
-    setBalloonDone?(bid: string): Promise<void>
+    getToken(username: string, password: string): Promise<void>
+    teamInfo(): Promise<void>
+    balloonInfo(all: boolean): Promise<void>
+    setBalloonDone(bid: string): Promise<void>
 }
 class BasicFetcher extends Service implements IBasicFetcher {
     contest: any;
@@ -60,7 +60,7 @@ class BasicFetcher extends Service implements IBasicFetcher {
     }
 }
 
-class DomJudgeFetcher extends BasicFetcher {
+class DOMjudgeFetcher extends BasicFetcher {
     async contestInfo() {
         let contest;
         if (!config.contestId) {
@@ -141,10 +141,96 @@ class DomJudgeFetcher extends BasicFetcher {
     }
 }
 
+class HydroFetcher extends BasicFetcher {
+    async contestInfo() {
+        const ids = config.contestId.split('/');
+        const [domainId, contestId] = ids.length === 2 ? ids : ['system', config.contestId];
+        const { body } = await fetch(`/d/${domainId}/contest/${contestId}`);
+        if (!body || !body.tdoc) {
+            logger.error('Contest not found');
+            return false;
+        }
+        const contest = body.tdoc;
+        contest.freeze_time = contest.lockAt;
+        const old = this?.contest?._id;
+        this.contest = {
+            info: contest, id: contest._id, name: contest.title, domainId,
+        };
+        logger.info(`Connected to ${contest.name}(id=${contest.id})`);
+        return old === this.contest.id;
+    }
+
+    async getToken(username, password) {
+        const res = await fetch('/login', 'post').send({ uname: username, password, rememberme: 'on' })
+            .redirects(0).ok((i) => i.status === 302);
+        if (!res) throw new Error('Failed to get token');
+        config.token = `Bearer ${res.header['set-cookie'][0].split(';')[0].split('=')[1]}`;
+    }
+
+    async teamInfo() {
+        const { body } = await fetch(`/d/${this.contest.domainId}/contest/${this.contest.id}/user`);
+        if (!body || !body.length) return;
+        const teams = body.tsdocs.filter((t) => body.udict[t.uid]).map((t) => (body.udict[t.uid]));
+        for (const team of teams) {
+            await this.ctx.db.teams.update({ id: team._id }, { $set: team }, { upsert: true });
+        }
+        logger.debug(`Found ${teams.length} teams`);
+    }
+
+    async balloonInfo(all) {
+        if (all) logger.info('Sync all balloons...');
+        const { body } = await fetch(`/d/${this.contest.domainId}/contest/${this.contest.id}/balloon?todo=${all ? 'false' : 'true'}`);
+        if (!body || !body.length) return;
+        const balloons = body;
+        for (const balloon of balloons) {
+            const teamTotal = await this.ctx.db.balloon.find({ teamid: balloon.teamid, time: { $lt: (balloon.time * 1000).toFixed(0) } });
+            const encourage = teamTotal.length < (config.freezeEncourage ?? 0);
+            const totalDict = {};
+            for (const t of teamTotal) {
+                totalDict[t.problem] = t.contestproblem;
+            }
+            const shouldPrint = this.contest.info.freeze_time ? (balloon.time * 1000) < this.contest.info.freeze_time || encourage : true;
+            if (!shouldPrint && !balloon.done) await this.setBalloonDone(balloon.balloonid);
+            const contestproblem = {
+                id: String.fromCharCode(this.contest.pids.indexOf(balloon.pid) + 65),
+                name: body.pdict[balloon.pid].title,
+                rgb: this.contest.balloon[balloon.pid].color,
+                color: this.contest.balloon[balloon.pid].name,
+            };
+            await this.ctx.db.balloon.update({ balloonid: balloon.balloonid }, {
+                $set: {
+                    balloonid: balloon._id,
+                    time: mongoId(balloon._id).timestamp,
+                    problem: contestproblem.id,
+                    contestproblem,
+                    team: body.udict[balloon.uid].displayName,
+                    teamid: balloon.uid,
+                    location: body.udict[balloon.uid].studentId,
+                    affiliation: body.udict[balloon.uid].school,
+                    awards: balloon.first ? 'First of Problem' : (
+                        this.contest.info.freeze_time && (balloon.time * 1000) > this.contest.info.freeze_time
+                            && encourage ? 'Encourage Balloon' : ''
+                    ),
+                    done: balloon.sent,
+                    total: totalDict,
+                    printDone: balloon.done ? 1 : 0,
+                    shouldPrint,
+                },
+            }, { upsert: true });
+        }
+        logger.debug(`Found ${balloons.length} balloons`);
+    }
+
+    async setBalloonDone(bid) {
+        await fetch(`/d/${this.contest.domainId}/contest/${this.contest.id}/balloon`, 'post').send({ balloon: bid });
+        logger.debug(`Balloon ${bid} set done`);
+    }
+}
+
 const fetcherList = {
     server: BasicFetcher,
-    domjudge: DomJudgeFetcher,
-    hydro: BasicFetcher, // TODO: HydroFetcher
+    domjudge: DOMjudgeFetcher,
+    hydro: HydroFetcher,
 };
 
 export async function apply(ctx) {

packages/server/utils/index.ts

@@ -84,6 +84,17 @@ export function sleep(timeout: number) {
     });
 }
 
+// https://github.com/andrasq/node-mongoid-js/blob/master/mongoid.js
+export function mongoId(idstring: string) {
+    if (typeof idstring !== 'string') idstring = String(idstring);
+    return {
+        timestamp: parseInt(idstring.slice(0, 0 + 8), 16),
+        machineid: parseInt(idstring.slice(8, 8 + 6), 16),
+        pid: parseInt(idstring.slice(14, 14 + 4), 16),
+        sequence: parseInt(idstring.slice(18, 18 + 6), 16),
+    };
+}
+
 export * as fs from 'fs-extra';
 export * as yaml from 'js-yaml';
 export { Logger };

packages/ui/build.ts

@@ -101,14 +101,14 @@ logger.info('Building...');
 (async () => {
   if (dev) {
     const server = new WebpackDevServer({
-      port: 8890,
+      port: 8080,
       compress: true,
       hot: true,
       server: 'http',
       allowedHosts: 'all',
       proxy: {
         context: (p) => p !== '/ws',
-        target: process.env.TOOLS_API || 'http://localhost:8889',
+        target: process.env.TOOLS_API || 'http://localhost:5283',
       },
       client: {
         webSocketURL: 'auto://0.0.0.0:0/ws',

packages/ui/package.json

@@ -4,8 +4,8 @@
     "author": "panda",
     "main": "index.ts",
     "scripts": {
-        "build:ui:dev": "node -r @hydrooj/register build.ts --watch --dev",
-        "build:ui:prod": "node -r @hydrooj/register build.ts --production"
+        "dev:ui": "node -r @hydrooj/register build.ts --watch --dev",
+        "build:ui": "node -r @hydrooj/register build.ts --production"
     },
     "preferUnplugged": true,
     "devDependencies": {