koishi-plugin-booth-get 5.2.0 → 5.2.2

package/README.md

@@ -1 +1,8 @@
-简单获取booth.pm页面的插件
+简单获取booth.pm页面的插件
+适用用于“vrchat”“MMD模型”“周边”“游戏”“Live2D”“视频”
+在使用vrchat的搜索时由于booth.pm的特殊性《指令摊：位搜索》会以最新发布商品为第一检索
+默认情况下，插件会自动获取最新发布商品，如需获取其他商品，请自行修改插件代码
+============================================================
+后续更新会以卡片样式web版进行更新
+优化更好且简介的样式面板
+============================================================

package/lib/index.js

@@ -31,10 +31,9 @@ var QRCode = require("qrcode");
 var name = "booth-get";
 var inject = ["puppeteer"];
 var Config = import_koishi.Schema.object({
-  loadTimeout: import_koishi.Schema.natural().role("ms").description("加载页面的最长时间").default(import_koishi.Time.second * 5),
+  loadTimeout: import_koishi.Schema.natural().role("ms").description("加载页面的最长时间").default(import_koishi.Time.second * 10),
   idleTimeout: import_koishi.Schema.natural().role("ms").description("等待页面空闲的最长时间").default(import_koishi.Time.second * 30),
-  proxyServer: import_koishi.Schema.string().description("代理服务器地址").default(""),
-  Text: import_koishi.Schema.title("摊位卡片生成器").description("生成BOOTH商品的摊位卡片,由VRCBBS提供"),
+  proxyServer: import_koishi.Schema.string().description("代理服务器地址").default("61.216.156.222:60808"),
 
 }).description("booth-get");
 
@@ -403,10 +402,12 @@ async function captureCard(ctx, id) {
     page.on('request', (request) => request.continue());
     
     await page.setContent(html, { 
-      waitUntil: 'networkidle0',
-      timeout: ctx.config.loadTimeout
+      waitUntil: 'domcontentloaded',
+      timeout: ctx.config.loadTimeout || import_koishi.Time.second * 10
     });
     
+    await new Promise(resolve => setTimeout(resolve, 2000));
+    
     await page.setViewport({ width: 640, height: 1200 });
     const container = await page.$('.container');
     return await container.screenshot({
@@ -440,47 +441,25 @@ function apply(ctx, config) {
     ctx.command("摊位名称 <query>")
     .action(async ({ session }, query) => {
       if (!query) return "请输入商品名称";
-  
-      const searchUrl = `https://booth.pm/zh-cn/search/${encodeURIComponent(query)}?in_stock=true`;
+
+      const tags = ['Vrchat'];
+      const tagsParams = tags.map(tag => `tags[]=${encodeURIComponent(tag)}`).join('&');
+      const searchUrl = `https://booth.pm/zh-cn/search/${encodeURIComponent(query)}?${tagsParams}&min_price=0&in_stock=true`;
+      
       const page = await ctx.puppeteer.page();
       
       try {
-        await page.goto(searchUrl, { waitUntil: 'networkidle0', timeout: 10000 });
+        await page.goto(searchUrl, { waitUntil: 'networkidle0', timeout: ctx.config.loadTimeout || import_koishi.Time.second * 10 });
         const content = await page.content();
-  
-        const itemRegex = /item-card__wrap" id="item_(\d+)".*?<h2 class="item-card__title">([^<]+)<\/h2>/gs;
+        
+        const itemRegex = /item-card__wrap"[\s\S]*?id="item_(\d+)"/g;
         const matches = [...content.matchAll(itemRegex)];
         
         if (!matches.length) return "没有找到相关商品";
   
-        const items = matches.map(match => ({
-          id: match[1],
-          name: match[2].trim()
-        }));
-  
-        const exactMatches = items.filter(item => item.name === query);
-        if (exactMatches.length === 1) {
-
-          const buffer = await captureCard(ctx, exactMatches[0].id);
-          return buffer ? import_koishi.h.image(buffer, "image/png") : "卡片生成失败";
-        } else if (exactMatches.length > 1) {
-
-          const buffer = await captureCard(ctx, exactMatches[0].id);
-          return buffer ? import_koishi.h.image(buffer, "image/png") : "卡片生成失败";
-        }
-
-        const queryLower = query.toLowerCase();
-        const scoredItems = items.map(item => ({
-          ...item,
-          score: getSimilarity(queryLower, item.name.toLowerCase())
-        })).sort((a, b) => b.score - a.score);
-  
-        const bestMatch = scoredItems[0];
-        if (bestMatch.score < 0.3) {
-          return `没有找到完全匹配的商品，最接近的是：${bestMatch.name}`;
-        }
-  
-        const buffer = await captureCard(ctx, bestMatch.id);
+        const itemId = matches[0][1];
+        
+        const buffer = await captureCard(ctx, itemId);
         return buffer ? import_koishi.h.image(buffer, "image/png") : "卡片生成失败";
   
       } catch (error) {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-booth-get",
   "description": "通过url与名称检查摊位物品并反馈用户搜索的图片",
-  "version": "5.2.0",
+  "version": "5.2.2",
   "contributors": [
     "rixiang <1148147857@qq.com>"
   ],
@@ -27,7 +27,9 @@
     "koishi",
     "plugin",
     "booth",
-    "pngjs"
+    "pngjs",
+    "puppeteer",
+    "qrcode"
   ],
   "peerDependencies": {
     "koishi": "4.18.9"