koishi-plugin-codec-tools 1.0.7 → 1.0.9

package/lib/index.js

@@ -18,6 +18,27 @@ exports.Config = koishi_1.Schema.object({})
     .i18n({
     'zh-CN': {},
 });
+async function extractImageFromSession(session) {
+    if (session.quote) {
+        try {
+            const quotedMsg = await session.bot.getMessage(session.channelId, session.quote.messageId);
+            if (quotedMsg && quotedMsg.elements) {
+                const quotedImage = quotedMsg.elements.find(el => el.type === 'image');
+                if (quotedImage) {
+                    return quotedImage.attrs.src;
+                }
+            }
+        }
+        catch (e) {
+            console.warn('获取引用消息失败:', e);
+        }
+    }
+    const currentImage = session.elements.find(el => el.type === 'image');
+    if (currentImage) {
+        return currentImage.attrs.src;
+    }
+    return null;
+}
 function apply(ctx, config) {
     const zhCNPath = path_1.default.join(__dirname, './locales/zh-CN.yml');
     const zhCNContent = fs_1.default.readFileSync(zhCNPath, 'utf8');
@@ -38,42 +59,42 @@ function apply(ctx, config) {
     ctx.command('decode-code')
         .action(async ({ session }) => {
         if (!session)
-            return;
-        const image = session.elements.find(el => el.type === 'image');
-        const replyWithQuote = (content) => {
-            return session.send([
-                (0, koishi_1.h)('quote', { id: session.messageId }),
-                content
-            ]);
-        };
-        if (!image)
-            return replyWithQuote(session.text('.no-image'));
-        const url = image.attrs.src;
-        const buffer = await ctx.http.get(url, { responseType: 'arraybuffer' });
-        const img = await jimp_1.default.read(Buffer.from(buffer));
-        const imageData = {
-            data: new Uint8ClampedArray(img.bitmap.data),
-            width: img.bitmap.width,
-            height: img.bitmap.height
-        };
-        const qrResult = (0, jsqr_1.default)(imageData.data, imageData.width, imageData.height);
-        if (qrResult)
-            return replyWithQuote(session.text('.qr-content', { content: qrResult.data }));
-        const hints = new Map();
-        hints.set(library_1.DecodeHintType.POSSIBLE_FORMATS, [
-            library_1.BarcodeFormat.QR_CODE, library_1.BarcodeFormat.CODE_128, library_1.BarcodeFormat.EAN_13,
-            library_1.BarcodeFormat.EAN_8, library_1.BarcodeFormat.UPC_A, library_1.BarcodeFormat.UPC_E,
-            library_1.BarcodeFormat.CODE_39, library_1.BarcodeFormat.ITF
-        ]);
-        const reader = new library_1.MultiFormatReader();
+            return '';
+        const imageUrl = await extractImageFromSession(session);
+        if (!imageUrl) {
+            return session.send(session.text('.no-image'), { quote: session.messageId });
+        }
         try {
+            const buffer = await ctx.http.get(imageUrl, { responseType: 'arraybuffer' });
+            const img = await jimp_1.default.read(Buffer.from(buffer));
+            if (img.bitmap.width > 1000 || img.bitmap.height > 1000) {
+                img.scaleToFit(800, 800);
+            }
+            img.contrast(0.2);
+            const imageData = {
+                data: new Uint8ClampedArray(img.bitmap.data),
+                width: img.bitmap.width,
+                height: img.bitmap.height
+            };
+            const qrResult = (0, jsqr_1.default)(imageData.data, imageData.width, imageData.height);
+            if (qrResult) {
+                return session.send(session.text('.qr-content', { content: qrResult.data }), { quote: session.messageId });
+            }
+            const hints = new Map();
+            hints.set(library_1.DecodeHintType.POSSIBLE_FORMATS, [
+                library_1.BarcodeFormat.QR_CODE, library_1.BarcodeFormat.CODE_128, library_1.BarcodeFormat.EAN_13,
+                library_1.BarcodeFormat.EAN_8, library_1.BarcodeFormat.UPC_A, library_1.BarcodeFormat.UPC_E,
+                library_1.BarcodeFormat.CODE_39, library_1.BarcodeFormat.ITF
+            ]);
+            const reader = new library_1.MultiFormatReader();
             const luminanceSource = new library_1.RGBLuminanceSource(imageData.data, imageData.width, imageData.height);
             const binaryBitmap = new library_1.BinaryBitmap(new library_1.HybridBinarizer(luminanceSource));
             const barcodeResult = reader.decode(binaryBitmap, hints);
-            return replyWithQuote(session.text('.barcode-content', { content: barcodeResult.text }));
+            return session.send(session.text('.barcode-content', { content: barcodeResult.text }), { quote: session.messageId });
         }
         catch (e) {
-            return replyWithQuote(session.text('.no-code'));
+            console.error('解析失败:', e);
+            return session.send(session.text('.no-code'), { quote: session.messageId });
         }
     });
 }

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-codec-tools",
   "description": "编码&解码 URL/Base64/Unicode 和条形码/二维码的 Koishi 插件",
-  "version": "1.0.7",
+  "version": "1.0.9",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [