support windows for print code

pandadtdyy · pandadtdyy · commit f6cc2f8e4cc8 · 2024-06-11T14:53:44.000+08:00
diff --git a/packages/server/config.ts b/packages/server/config.ts
@@ -20,12 +20,13 @@ if (!fs.existsSync(configPath)) {
         const serverConfigDefault = `\
 type: server # server | domjudge | hydro
 viewPass: ${String.random(8)} # use admin / viewPass to login
+secretRoute: ${String.random(12)}
+seatFile: /home/icpc/Desktop/seat.txt
+# if type is server, the following is not needed
 server: 
 token: 
 username: 
 password: 
-secretRoute: ${String.random(12)}
-seatFile: /home/icpc/Desktop/seat.txt
 `;
 
         const clientConfigDefault = yaml.dump({
@@ -55,16 +56,15 @@ const serverSchema = Schema.intersect([
     }).description('Basic Config'),
     Schema.union([
         Schema.object({
-            type: Schema.const('domjudge').required(),
-            server: Schema.string().role('url').required(),
-            username: Schema.string().required(),
-            password: Schema.string().required(),
-        }).description('DomJudge Fetcher Config'),
-        Schema.object({
-            type: Schema.const('hydro').required(),
+            type: Schema.union([
+                Schema.const('domjudge'),
+                Schema.const('hydro'),
+            ] as const).required(),
             server: Schema.string().role('url').required(),
-            token: Schema.string().required(),
-        }).description('Hydro Fetcher Config'),
+            token: Schema.string(),
+            username: Schema.string(),
+            password: Schema.string(),
+        }).description('Fetcher Config'),
         Schema.object({
             type: Schema.const('server').required(),
         }).description('Server Mode Config'),
diff --git a/packages/server/handler/commands.ts b/packages/server/handler/commands.ts
@@ -1,57 +1,11 @@
 /* eslint-disable no-empty-pattern */
 /* eslint-disable no-await-in-loop */
-import child from 'child_process';
-import fs from 'fs';
-import { homedir } from 'os';
 import { Context } from 'cordis';
 import { param, Types } from '@hydrooj/framework';
 import { config } from '../config';
-import { Logger } from '../utils';
+import { executeOnHost } from '../utils';
 import { AuthHandler } from './misc';
 
-const logger = new Logger('handler/commands');
-
-async function asyncCommand(command: string | string[], timeout = 10000) {
-    let result = '';
-    const proc = typeof command === 'string' ? child.exec(command) : child.spawn(command[0], command.slice(1));
-    proc.stdout?.on('data', (d) => {
-        result += d.toString();
-    });
-    proc.stderr?.on('data', (d: Buffer) => {
-        logger.error(' STDERR ', d.toString());
-    });
-    if (!timeout) {
-        proc.on('exit', () => { });
-        setTimeout(() => proc.kill(), 1000);
-        return '';
-    }
-    return await new Promise<string>((resolve, reject) => {
-        const t = setTimeout(() => {
-            proc.kill();
-            reject(new Error('timeout'));
-        }, timeout);
-        proc.on('exit', (code) => {
-            clearTimeout(t);
-            if (code === 0) {
-                resolve(result.replace(/\r/g, ''));
-            } else {
-                reject(code);
-            }
-        });
-    });
-}
-
-const keyfile = fs.existsSync(`${homedir()}.ssh/id_rsa`) ? '.ssh/id_rsa' : '.ssh/id_ed25519';
-
-async function executeOnHost(host: string, command: string, timeout = 10000) {
-    logger.info('executing', command, 'on', host);
-    return await asyncCommand([
-        'ssh', '-o', 'StrictHostKeyChecking no', '-o', `IdentityFile ~/${keyfile}`,
-        `root@${host}`,
-        'bash', '-c', `'echo $(echo ${Buffer.from(command).toString('base64')} | base64 -d | bash)'`,
-    ], timeout);
-}
-
 class CommandsHandler extends AuthHandler {
     async get() {
         this.response.body = '';
diff --git a/packages/server/service/fetcher.ts b/packages/server/service/fetcher.ts
@@ -11,6 +11,7 @@ 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>
@@ -24,7 +25,12 @@ class BasicFetcher extends Service implements IBasicFetcher {
     }
 
     async cron() {
+        if (config.type === 'server') return;
         logger.info('Fetching contest info...');
+        if (!config.token) {
+            if (config.username && config.password) await this.getToken(config.username, config.password);
+            else throw new Error('No token or username/password provided');
+        }
         const first = await this.contestInfo();
         if (first) await this.teamInfo();
         await this.balloonInfo(first);
@@ -36,6 +42,10 @@ class BasicFetcher extends Service implements IBasicFetcher {
         return old === this.contest.id;
     }
 
+    async getToken(username, password) {
+        config.token = `Basic ${Buffer.from(`${username}:${password}`).toString('base64')}`;
+    }
+
     async teamInfo() {
         logger.debug('Found 0 teams');
     }
diff --git a/packages/server/utils/commandRunner.ts b/packages/server/utils/commandRunner.ts
@@ -0,0 +1,48 @@
+/* eslint-disable @typescript-eslint/no-loop-func */
+import child from 'child_process';
+import fs from 'fs';
+import { homedir } from 'os';
+import { Logger } from './index';
+
+const logger = new Logger('runner');
+
+export async function asyncCommand(command: string | string[], timeout = 10000) {
+    let result = '';
+    const proc = typeof command === 'string' ? child.exec(command) : child.spawn(command[0], command.slice(1));
+    proc.stdout?.on('data', (d) => {
+        result += d.toString();
+    });
+    proc.stderr?.on('data', (d: Buffer) => {
+        logger.error(' STDERR ', d.toString());
+    });
+    if (!timeout) {
+        proc.on('exit', () => { });
+        setTimeout(() => proc.kill(), 1000);
+        return '';
+    }
+    return await new Promise<string>((resolve, reject) => {
+        const t = setTimeout(() => {
+            proc.kill();
+            reject(new Error('timeout'));
+        }, timeout);
+        proc.on('exit', (code) => {
+            clearTimeout(t);
+            if (code === 0) {
+                resolve(result.replace(/\r/g, ''));
+            } else {
+                reject(code);
+            }
+        });
+    });
+}
+
+const keyfile = fs.existsSync(`${homedir()}.ssh/id_rsa`) ? '.ssh/id_rsa' : '.ssh/id_ed25519';
+
+export async function executeOnHost(host: string, command: string, timeout = 10000) {
+    logger.info('executing', command, 'on', host);
+    return await asyncCommand([
+        'ssh', '-o', 'StrictHostKeyChecking no', '-o', `IdentityFile ~/${keyfile}`,
+        `root@${host}`,
+        'bash', '-c', `'echo $(echo ${Buffer.from(command).toString('base64')} | base64 -d | bash)'`,
+    ], timeout);
+}
diff --git a/packages/server/utils/index.ts b/packages/server/utils/index.ts
@@ -1,6 +1,5 @@
 /* eslint-disable @typescript-eslint/no-loop-func */
 /* eslint-disable no-await-in-loop */
-import { spawn } from 'child_process';
 import { gunzipSync } from 'zlib';
 import { decode } from 'base16384';
 import Logger from 'reggol';
@@ -94,46 +93,11 @@ export function StaticHTML(context, randomHash) {
     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>`;
 }
 
-// wait for undefined to write
-export async function remoteRunner(user: string, target: string, targetPort: string, timeout = 10, RETRY = 3, command) {
-    let log = '';
-    const defaultCommand = `-o ConnectTimeout=${timeout} -o StrictHostKeyChecking=no -P ${targetPort}`;
-    const cmds = {
-        exec: [defaultCommand, `${user}@${target}`, command],
-        upload: [defaultCommand, command.from, `${user}@${target}:${command.to}`],
-        download: [defaultCommand, `${user}@${target}:${command.from.replace('{target}', target)}`, command.to.replace('{target}', target)],
-    };
-    let retry = 0;
-    while (retry < RETRY) {
-        const child = spawn(command.type === 'exec' ? 'ssh' : 'scp', cmds[command.type]);
-        // 输出命令行执行的结果
-        let success = false;
-        child.stdout.on('data', (data) => {
-            success = true;
-            log += data;
-        });
-        child.stderr.on('data', (data) => {
-            success = false;
-            log += data;
-        });
-        // 执行命令行错误
-        child.on('error', (err) => {
-            log += err;
-            return { success: false, log };
-        });
-        // 命令行执行结束
-        child.on('close', (e) => {
-            if (e === 0) return { success: true, log };
-            if (success) return { success: true, log };
-            log += `retry ${retry} times`;
-            retry++;
-        });
-    }
-    return { success: false, log };
-}
-
 export function decodeBinary(file: string) {
     if (process.env.NODE_ENV === 'development') return Buffer.from(file, 'base64');
     const buf = decode(file);
     return gunzipSync(buf);
 }
+
+export * from './commandRunner';
+export * from './printers';
diff --git a/packages/server/utils/printers.ts b/packages/server/utils/printers.ts
@@ -0,0 +1,40 @@
+import fs from 'fs';
+import winPrint from '@myteril/node-win-printer';
+import unixPrint from 'unix-print';
+import path from 'path';
+
+let winPrinter: winPrint.PDFPrinter;
+
+if (process.platform === 'win32') {
+    const execPath = [
+        './SumatraPDF.exe',
+        path.resolve(__dirname, 'SumatraPDF.exe'),
+        path.resolve(process.cwd(), 'SumatraPDF.exe'),
+        'C:\\Program Files\\SumatraPDF\\SumatraPDF.exe',
+        'C:\\Program Files (x86)\\SumatraPDF\\SumatraPDF.exe',
+    ];
+    const sumatraPdfPath = execPath.find((p) => fs.existsSync(p));
+    if (!sumatraPdfPath) {
+        throw new Error('SumatraPDF not found, please install it on https://www.sumatrapdfreader.org/download-free-pdf-viewer');
+    }
+    winPrinter = new winPrint.PDFPrinter({
+        sumatraPdfPath,
+    });
+}
+
+export async function getPrinters() {
+    if (process.platform === 'win32') {
+        return winPrint.getPrinters();
+    }
+    return unixPrint.getPrinters();
+}
+
+export async function print(printer, file) {
+    if (process.platform === 'win32') {
+        return winPrinter.print({
+            file,
+            printer,
+        });
+    }
+    return unixPrint.print(printer, file);
+}
diff --git a/packages/ui/app/pages/Commands.tsx b/packages/ui/app/pages/Commands.tsx
@@ -1,8 +1,9 @@
 import React from 'react';
+import { Button, Card, Divider, Group, Textarea, Title } from '@mantine/core';
 import { notifications } from '@mantine/notifications';
 
 export default function Commands() {
-  const [command, setCommand] = React.useState('')
+  const [command, setCommand] = React.useState('');
 
   const operation = async (op: string) => {
     try {
@@ -20,15 +21,30 @@ export default function Commands() {
       console.error(e);
       notifications.show({ title: 'Error', message: 'Failed to update balloon', color: 'red' });
     }
-  }
+  };
 
   return (
-    <div>
-      <textarea rows={10} cols={100} style={{ fontFamily: 'monospace' }} value={command} onChange={(ev) => setCommand(ev.target.value)} />
-      <button onClick={() => fetch('/commands', { method: 'POST', body: JSON.stringify({ command }) })}>Send</button>
-      <button onClick={() => operation('reboot')}>Reboot All</button>
-      <button onClick={() => operation('set_hostname')}>Update Hostname</button>
-      <button onClick={() => operation('show_ids')}>Show IDS</button>
-    </div >
+    <Card shadow="sm" padding="lg" radius="md" withBorder>
+      <Title order={3}>Send Commands to Monitoring Computer</Title>
+      <Textarea
+        label="Command"
+        my="md"
+        rows={10}
+        cols={100}
+        style={{ fontFamily: 'monospace' }}
+        value={command}
+        onChange={(ev) => setCommand(ev.target.value)}
+      />
+      <Group justify="center" my="md">
+        <Button onClick={() => fetch('/commands', { method: 'POST', body: JSON.stringify({ command }) })}>Send</Button>
+      </Group>
+      <Divider my="md" />
+      <Title order={3}>Quick Commands</Title>
+      <Group my="md" justify="space-start">
+        <Button onClick={() => operation('reboot')}>Reboot All</Button>
+        <Button onClick={() => operation('set_hostname')}>Update Hostname</Button>
+        <Button onClick={() => operation('show_ids')}>Show IDS</Button>
+      </Group>
+    </Card>
   );
 }
