support windows printer & balloon receipt

pandadtdyy · pandadtdyy · commit baf5b152a6d8 · 2024-06-12T04:16:59.000+08:00
diff --git a/packages/server/client/balloon.ts b/packages/server/client/balloon.ts
@@ -49,6 +49,30 @@ const logger = new Logger('fetcher');
 let timer = null;
 let printer = null;
 
+async function getReceiptStatus(receipt) {
+    const lp = receipt.split('/').pop();
+    const oldPrinter = printer;
+    printer = {
+        printer: lp,
+        info: fs.readFileSync(`/sys/class/usb/${lp}/device/ieee1284_id`, 'utf8').trim(),
+    };
+    if (!oldPrinter || oldPrinter.info === printer.info) return;
+    logger.info('Printer changed:', printer.printer, printer.info);
+    const usbDevices = fs.readdirSync('/dev/usb');
+    for (const f of usbDevices) {
+        if (f.startsWith('lp')) {
+            const lpid = fs.readFileSync(`/sys/class/usb/${f}/device/ieee1284_id`, 'utf8').trim();
+            if (lpid === oldPrinter.info) {
+                logger.info('Printer found:', f, ':', lpid);
+                oldPrinter.printer = f;
+                printer = oldPrinter;
+                break;
+            }
+        }
+    }
+    if (oldPrinter.info !== printer.info) throw Error('Printer not found, please check the printer connection.');
+}
+
 async function printBalloon(doc) {
     const bReceipt = receiptText(
         doc.balloonid,
@@ -60,6 +84,7 @@ async function printBalloon(doc) {
         doc.total ? Object.keys(doc.total).map((k) => `- ${k}: ${doc.total[k].color}`).join('\n') : 'N/A',
     );
     if (printer) {
+        await getReceiptStatus(printer.printer);
         fs.writeFileSync(path.resolve(printer), bReceipt);
     }
 }
@@ -86,6 +111,6 @@ async function fetchTask(c) {
 }
 
 export async function apply() {
-    printer = config.balloon;
+    await getReceiptStatus(config.balloon);
     if (config.token && config.server && config.balloon) await fetchTask(config);
 }
diff --git a/packages/server/client/printer.ts b/packages/server/client/printer.ts
@@ -1,11 +1,10 @@
 /* eslint-disable no-await-in-loop */
 import path from 'path';
-// https://www.sumatrapdfreader.org//dl//rel/3.1.2/SumatraPDF-3.1.2.zip
-import winPrint from '@myteril/node-win-printer';
 import superagent from 'superagent';
-import { getPrinters, print } from 'unix-print';
 import { config, saveConfig } from '../config';
-import { fs, Logger, sleep } from '../utils';
+import {
+    fs, getPrinters, initWinPrinter, Logger, print, sleep,
+} from '../utils';
 import { createTypstCompiler, generateTypst } from './typst';
 
 let compiler;
@@ -14,41 +13,6 @@ const logger = new Logger('printer');
 
 let timer = null;
 
-async function fetchTask(c) {
-    if (timer) clearTimeout(timer);
-    logger.info('Fetching Task from tools server...');
-    try {
-        const printersInfo = await getPrinters();
-        const { body } = await superagent.post(`${c.server}/client/${c.token}/print`)
-            .send({
-                printers: config.printers,
-                printersInfo: JSON.stringify(printersInfo.map((p) => ({
-                    printer: p.printer,
-                    status: p.status,
-                    description: p.description,
-                }))),
-            });
-        if (body.setPrinter) {
-            config.printers = body.setPrinter;
-            saveConfig();
-            logger.info(`Printer set to ${config.printers}`);
-        }
-        if (body.doc) {
-            logger.info(`Print task ${body.doc.tid}#${body.doc._id}...`);
-            await printFile(body.doc);
-            await superagent.post(`${c.server}/client/${c.token}/doneprint/${body.doc._id}`);
-            logger.info(`Print task ${body.doc.tid}#${body.doc._id} completed.`);
-        } else {
-            logger.info('No print task, sleeping...');
-            await sleep(5000);
-        }
-    } catch (e) {
-        logger.error(e);
-        await sleep(5000);
-    }
-    timer = setTimeout(() => fetchTask(c), 3000);
-}
-
 export async function ConvertCodeToPDF(code, lang, filename, team, location) {
     compiler ||= await createTypstCompiler();
     const typst = generateTypst(team, location, filename, lang);
@@ -77,14 +41,14 @@ export async function printFile(doc) {
                 const randomP = printers[Math.floor(Math.random() * printers.length)];
                 if (randomP.status === 'idle') {
                     logger.info(`Printing ${_id} on ${randomP.printer}`);
-                    await print(path.resolve(process.cwd(), `data/${tid}#${_id}.pdf`), randomP.printer, ['-P', '1-5']);
+                    await print(path.resolve(process.cwd(), `data/${tid}#${_id}.pdf`), randomP.printer, 1, 5);
                     return;
                 }
                 for (const printer of printers.filter((p) => p.printer !== randomP.printer)) {
                     logger.info(`Checking ${printer.printer} ${printer.status}`);
                     if (printer.status === 'idle') {
                         logger.info(`Printing ${_id} on ${printer.printer}`);
-                        await print(path.resolve(process.cwd(), `data/${tid}#${_id}.pdf`), printer.printer, ['-P', '1-5']);
+                        await print(path.resolve(process.cwd(), `data/${tid}#${_id}.pdf`), printer.printer, 1, 5);
                         return;
                     }
                 }
@@ -97,7 +61,50 @@ export async function printFile(doc) {
     }
 }
 
+async function fetchTask(c) {
+    if (timer) clearTimeout(timer);
+    logger.info('Fetching Task from tools server...');
+    try {
+        const printersInfo = await getPrinters();
+        const { body } = await superagent.post(`${c.server}/client/${c.token}/print`)
+            .send({
+                printers: config.printers,
+                printersInfo: JSON.stringify(printersInfo.map((p) => ({
+                    printer: p.printer,
+                    status: p.status,
+                    description: p.description,
+                }))),
+            });
+        if (body.setPrinter) {
+            config.printers = body.setPrinter;
+            saveConfig();
+            logger.info(`Printer set to ${config.printers}`);
+        }
+        if (body.doc) {
+            logger.info(`Print task ${body.doc.tid}#${body.doc._id}...`);
+            await printFile(body.doc);
+            await superagent.post(`${c.server}/client/${c.token}/doneprint/${body.doc._id}`);
+            logger.info(`Print task ${body.doc.tid}#${body.doc._id} completed.`);
+        } else {
+            logger.info('No print task, sleeping...');
+            await sleep(5000);
+        }
+    } catch (e) {
+        logger.error(e);
+        await sleep(5000);
+    }
+    timer = setTimeout(() => fetchTask(c), 3000);
+}
+
 export async function apply() {
     compiler = await createTypstCompiler();
+    if (process.platform === 'win32') {
+        try {
+            initWinPrinter();
+        } catch (e) {
+            logger.error(e);
+            process.exit(1);
+        }
+    }
     if (config.token && config.server && config.printers?.length) await fetchTask(config);
 }
diff --git a/packages/server/config.ts b/packages/server/config.ts
@@ -28,11 +28,25 @@ token:
 username: 
 password: 
 `;
-
+        let printers = [];
+        if (isClient) {
+            printers = await getPrinters().then((r) => r.map((p) => p.printer)).catch(() => []);
+            logger.info(printers.length, 'printers found:', printers.map((p) => p.printer).join(', '));
+            if (process.platform === 'linux') {
+                const usbDevices = fs.readdirSync('/dev/usb');
+                for (const f of usbDevices) {
+                    if (f.startsWith('lp')) {
+                        const lpid = fs.readFileSync(`/sys/class/usbmisc/${f}/device/ieee1284_id`, 'utf8').trim();
+                        logger.info('USB Printer found:', f, ':', lpid);
+                        logger.info('If you want to use this printer for balloon print, set balloon:', f, 'in config.yaml.');
+                    }
+                }
+            } else logger.info('If you want to use balloon client, please run this on linux.');
+        }
         const clientConfigDefault = yaml.dump({
             server: '',
             balloon: '',
-            printers: await getPrinters().then((r) => r.map((p) => p.printer)).catch(() => []),
+            printers,
             token: '',
         });
         fs.writeFileSync(configPath, isClient ? clientConfigDefault : serverConfigDefault);
diff --git a/packages/server/utils/printers.ts b/packages/server/utils/printers.ts
@@ -1,11 +1,19 @@
 import fs from 'fs';
+import path from 'path';
 import winPrint from '@myteril/node-win-printer';
 import unixPrint from 'unix-print';
-import path from 'path';
 
 let winPrinter: winPrint.PDFPrinter;
 
-if (process.platform === 'win32') {
+export interface Printer {
+    printer: string;
+    description?: string;
+    status?: string;
+    alerts?: string;
+    connection?: string;
+}
+
+export function initWinPrinter() {
     const execPath = [
         './SumatraPDF.exe',
         path.resolve(__dirname, 'SumatraPDF.exe'),
@@ -15,26 +23,38 @@ if (process.platform === 'win32') {
     ];
     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');
+        throw new Error(`SumatraPDF not found, please install it on https://www.sumatrapdfreader.org/download-free-pdf-viewer,
+or direct download from https://www.sumatrapdfreader.org/dl/rel/3.1.2/SumatraPDF-3.1.2.zip`);
     }
     winPrinter = new winPrint.PDFPrinter({
         sumatraPdfPath,
     });
 }
 
-export async function getPrinters() {
+const windowsPrinterStatus = {
+    3: 'idle',
+    4: 'printing',
+};
+
+export async function getPrinters(): Promise<Printer[]> {
     if (process.platform === 'win32') {
-        return winPrint.getPrinters();
+        const winprinters = await winPrint.getPrinters();
+        return winprinters.map((p: any) => ({
+            printer: p.DriverName,
+            description: p.Caption,
+            status: windowsPrinterStatus[p.PrinterStatus] ? windowsPrinterStatus[p.PrinterStatus] : 'unknown',
+        }));
     }
     return unixPrint.getPrinters();
 }
 
-export async function print(printer, file) {
+export async function print(printer: string, file: string, startPage?: number, endPage?: number) {
     if (process.platform === 'win32') {
         return winPrinter.print({
             file,
             printer,
+            pages: startPage && endPage ? [{ start: startPage, end: endPage }] : undefined,
         });
     }
-    return unixPrint.print(printer, file);
+    return unixPrint.print(printer, file, startPage && endPage ? ['-P', `${startPage}-${endPage}`] : []);
 }
