koishi-plugin-booth-get 5.2.1 → 5.2.3

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");
 
@@ -348,7 +347,7 @@ async function getBoothItem(id) {
       fetch(`https://booth.pm/zh-cn/items/${id}.json`),
       fetch(`https://accounts.booth.pm/wish_lists.json?item_ids%5B%5D=${id}`)
     ]);
-    
+
     const itemData = await itemRes.json();
     const wishData = await wishRes.json();
 
@@ -361,7 +360,7 @@ async function getBoothItem(id) {
       category: itemData.category?.name,
       parent_category: itemData.category?.parent?.name,
       author: itemData.shop?.name,
-      author_thumbnail_url: itemData.shop?.icon?.thumb?.original || itemData.shop?.icon?.small?.original,
+      author_thumbnail_url: itemData.shop?.thumbnail_url,
       likes: wishData.wishlists_counts[id] || 0,
       tags: itemData.tags
     };
@@ -394,19 +393,21 @@ async function captureCard(ctx, id) {
   if (!item) return null;
 
   const relatedItems = await fetchRelatedItems(item.author);
-  
+
   const html = generateCardHTML(item, relatedItems);
 
   const page = await ctx.puppeteer.page();
   try {
     await page.setRequestInterception(true);
     page.on('request', (request) => request.continue());
-    
-    await page.setContent(html, { 
-      waitUntil: 'networkidle0',
-      timeout: ctx.config.loadTimeout
+
+    await page.setContent(html, {
+      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({
@@ -437,52 +438,30 @@ function apply(ctx, config) {
       }
     });
 
-    ctx.command("摊位名称 <query>")
+  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) {
         logger.error("搜索失败:", error);
         return "搜索失败，请检查商品名称或稍后再试";
@@ -490,28 +469,28 @@ function apply(ctx, config) {
         await page.close();
       }
     });
-  
+
   function getSimilarity(a, b) {
     if (a === b) return 1;
     if (a.length < 2 || b.length < 2) return 0;
-  
+
     const bigramsA = new Set();
     for (let i = 0; i < a.length - 1; i++) {
       bigramsA.add(a.substring(i, i + 2));
     }
-  
+
     let matches = 0;
     for (let i = 0; i < b.length - 1; i++) {
       if (bigramsA.has(b.substring(i, i + 2))) matches++;
     }
-  
+
     return (2 * matches) / (a.length + b.length - 2);
   }
 
   ctx.middleware(async (session, next) => {
     const boothUrlRegex = /https:\/\/booth.pm\/[\w-]+\/items\/(\d+)/;
     const match = session.content.match(boothUrlRegex);
-    
+
     if (match) {
       try {
         const buffer = await captureCard(ctx, match[1]);

package/package.json

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