koishi-plugin-booth-get 5.2.4 → 5.2.5

package/lib/index.js

@@ -438,13 +438,27 @@ function apply(ctx, config) {
       }
     });
 
-  ctx.command("摊位名称 <query>")
-    .action(async ({ session }, query) => {
+  ctx.command("摊位名称 <query:text>")
+    .option('author', '-a <author> 指定作者名称')
+    .action(async ({ session, options }, query) => {
       if (!query) return "请输入搜索关键词";
 
+      let searchQuery = query;
+      let authorFilter = options.author;
+      
+      if (!authorFilter && query.includes(' ')) {
+        const parts = query.split(' ');
+        if (parts.length >= 2) {
+          authorFilter = parts.pop();
+          searchQuery = parts.join(' ');
+        }
+      }
+
+      let searchUrl = `https://booth.pm/zh-cn/search/${encodeURIComponent(searchQuery)}?in_stock=true`;
+      
       const tags = ['3Dモデル', 'Vrchat'];
       const tagsParams = tags.map(tag => `tags[]=${encodeURIComponent(tag)}`).join('&');
-      const searchUrl = `https://booth.pm/zh-cn/search/${encodeURIComponent(query)}?${tagsParams}&min_price=4500`;
+      searchUrl += `&${tagsParams}&min_price=4500`;
 
       const page = await ctx.puppeteer.page();
       
@@ -473,10 +487,30 @@ function apply(ctx, config) {
         }
         
         const content = await page.content();
-        const regex = /item-card__wrap"[\s\S]*?id="item_(\d+)"/;
-        const match = content.match(regex);
+        
+        const itemRegex = /item-card__wrap"[\s\S]*?id="item_(\d+)"[\s\S]*?<h2[^>]*class="[^"]*item-card__title[^"]*"[^>]*>([^<]+)<\/h2>/g;
+        let matches = [];
+        let match;
+        
+        while ((match = itemRegex.exec(content)) !== null) {
+          matches.push({
+            id: match[1],
+            title: match[2].trim()
+          });
+        }
+        
+        if (matches.length === 0) {
+          const simpleRegex = /item-card__wrap"[\s\S]*?id="item_(\d+)"/g;
+          let simpleMatch;
+          while ((simpleMatch = simpleRegex.exec(content)) !== null) {
+            matches.push({
+              id: simpleMatch[1],
+              title: '未知商品'
+            });
+          }
+        }
 
-        if (!match) {
+        if (matches.length === 0) {
           if (content.includes('検索結果はありません') || 
               content.includes('没有找到') || 
               content.includes('検索条件に合致する作品は見つかりませんでした') ||
@@ -486,10 +520,30 @@ function apply(ctx, config) {
           return "没有找到相关商品";
         }
 
-        const itemId = match[1];
+        let selectedItemId;
+        if (authorFilter) {
+          for (const item of matches) {
+            try {
+              const itemDetail = await getBoothItem(item.id);
+              if (itemDetail && itemDetail.author && 
+                  itemDetail.author.toLowerCase().includes(authorFilter.toLowerCase())) {
+                selectedItemId = item.id;
+                break;
+              }
+            } catch (err) {
+              continue;
+            }
+          }
+          
+          if (!selectedItemId) {
+            return `找不到作者"${authorFilter}"的相关商品`;
+          }
+        } else {
+          selectedItemId = matches[0].id;
+        }
 
         try {
-          const buffer = await captureCard(ctx, itemId);
+          const buffer = await captureCard(ctx, selectedItemId);
           if (!buffer) {
             return "卡片生成失败";
           }
@@ -528,22 +582,17 @@ function apply(ctx, config) {
 
   ctx.middleware(async (session, next) => {
     const boothUrlRegex = /https:\/\/booth.pm\/[\w-]+\/items\/(\d+)/;
+    const boothAuthorUrlRegex = /https:\/\/([\w-]+)\.booth\.pm\/items(?:\/(\d+))?/;
     const match = session.content.match(boothUrlRegex);
+    const authorMatch = session.content.match(boothAuthorUrlRegex);
 
     if (match) {
-      if (session.onebot) {
-        session.onebot._sned = session.send;
-        session.send = function (...args) {
-          return this._sned.apply(this, args);
-        };
-      }
-      
       const itemId = match[1];
       try {
         const buffer = await captureCard(ctx, itemId);
         if (buffer) {
           await session.send(import_koishi.h.image(buffer, "image/png"));
-          return; 
+          return ""; 
         } else {
           return "商品解析失败";
         }
@@ -551,6 +600,26 @@ function apply(ctx, config) {
         logger.warn("链接解析失败:", error);
         return "商品解析失败";
       }
+    } else if (authorMatch) {
+      const authorName = authorMatch[1];
+      const itemId = authorMatch[2]; 
+      
+      try {
+        if (itemId) {
+          const buffer = await captureCard(ctx, itemId);
+          if (buffer) {
+            await session.send(import_koishi.h.image(buffer, "image/png"));
+            return "";
+          } else {
+            return "商品解析失败";
+          }
+        } else {
+          return `检测到作者"${authorName}"的主页链接。请使用"摊位名称 <商品名> ${authorName}"来搜索该作者的商品。`;
+        }
+      } catch (error) {
+        logger.warn("作者链接解析失败:", error);
+        return "作者链接解析失败";
+      }
     }
     return next();
   });

package/package.json

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