koishi-plugin-booth-get 6.0.0-beta.1 → 6.0.0-beta.2

package/lib/discount-tracker.js

@@ -8,6 +8,8 @@ class DiscountTracker {
     this.config = null;
     this.DISCOUNT_ITEMS_FILE = path.join(__dirname, "discount_items.json");
     this.DISCOUNT_GROUPS_FILE = path.join(__dirname, "discount_groups.json");
+    this.discountCache = new Map();
+    this.CACHE_DURATION = 24 * 60 * 60 * 1000;
   }
 
   init(ctx, config) {
@@ -19,6 +21,30 @@ class DiscountTracker {
     }
   }
 
+  cleanupExpiredCache() {
+    const now = Date.now();
+    for (const [id, timestamp] of this.discountCache.entries()) {
+      if (now - timestamp > this.CACHE_DURATION) {
+        this.discountCache.delete(id);
+      }
+    }
+  }
+
+  isCached(itemId) {
+    const timestamp = this.discountCache.get(itemId);
+    if (!timestamp) return false;
+    
+    if (Date.now() - timestamp > this.CACHE_DURATION) {
+      this.discountCache.delete(itemId);
+      return false;
+    }
+    return true;
+  }
+
+  addToCache(itemId) {
+    this.discountCache.set(itemId, Date.now());
+  }
+
   async loadJSON(file, def = {}) {
     try {
       const raw = await fs.readFile(file, "utf8");
@@ -123,78 +149,81 @@ class DiscountTracker {
     }
   }
 
-async notifyDiscountGroups(discountItems) {
-  const logger = this.ctx.logger("booth-discount");
-  const cardGenerator = require('./card-generator');
-  
-  for (const groupKey of this.config.targetGroups || []) {
-    const parts = groupKey.split(':');
-    if (parts.length < 3) continue;
-    const platform = parts[1];
-    const channelId = parts.slice(2).join(':');
+  async notifyDiscountGroups(discountItems) {
+    const logger = this.ctx.logger("booth-discount");
+    const cardGenerator = require('./card-generator');
     
-    for (const item of discountItems) {
-      try {
-        const html = cardGenerator.generateDiscountCardHTML(item);
-        const buffer = await cardGenerator.captureCardHTML(html, this.config);
-        if (!buffer) continue;
-        
-        const message = `🎉 发现折扣商品！\n` +
-                      `商品链接：${item.url}\n` +
-                      `原价：¥${item.original_price.toLocaleString()}\n` +
-                      `现价：¥${item.price.toLocaleString()}\n` +
-                      `折扣：${item.discount_rate}% OFF\n` +
-                      `节省：¥${(item.original_price - item.price).toLocaleString()}`;
-        
-        await this.ctx.bots[0].sendMessage(channelId, [message, import_koishi.h.image(buffer, "image/png")]);
-        
-        await new Promise(resolve => setTimeout(resolve, 1000));
-      } catch (e) {
-        logger.warn(`向群组 ${channelId} 推送折扣信息失败:`, e);
+    const itemsToPush = discountItems.slice(0, this.config.maxDiscountPushCount || 5);
+    
+    for (const groupKey of this.config.targetGroups || []) {
+      const parts = groupKey.split(':');
+      if (parts.length < 3) continue;
+      const platform = parts[1];
+      const channelId = parts.slice(2).join(':');
+      
+      for (const item of itemsToPush) {
+        try {
+          const html = cardGenerator.generateDiscountCardHTML(item);
+          const buffer = await cardGenerator.captureCardHTML(html, this.config);
+          if (!buffer) continue;
+          
+          const message = `🎉 发现折扣商品！\n` +
+                        `商品链接：${item.url}\n` +
+                        `原价：¥${item.original_price.toLocaleString()}\n` +
+                        `现价：¥${item.price.toLocaleString()}\n` +
+                        `折扣：${item.discount_rate}% OFF\n` +
+                        `节省：¥${(item.original_price - item.price).toLocaleString()}`;
+          
+          await this.ctx.bots[0].sendMessage(channelId, [message, import_koishi.h.image(buffer, "image/png")]);
+          
+          await new Promise(resolve => setTimeout(resolve, 1000));
+        } catch (e) {
+          logger.warn(`向群组 ${channelId} 推送折扣信息失败:`, e);
+        }
       }
     }
   }
-}
 
-startDiscountTracking() {
-  const logger = this.ctx.logger("booth-discount");
-  logger.info('启动折扣商品追踪器');
+  startDiscountTracking() {
+    const logger = this.ctx.logger("booth-discount");
+    logger.info('启动折扣商品追踪器');
 
-  this.ctx.setInterval(async () => {
-    try {
-      this.cleanupExpiredCache();
-      
-      logger.info('开始检查折扣商品...');
-      const discountItems = await this.searchDiscountItems();
-      const knownItems = await this.loadJSON(this.DISCOUNT_ITEMS_FILE);
-      const newDiscountItems = [];
-      
-      for (const item of discountItems) {
-        if (!this.isCached(item.id)) {
-          if (!knownItems[item.id] || knownItems[item.id].discount_rate !== item.discount_rate) {
-            newDiscountItems.push(item);
-            this.addToCache(item.id);
-            knownItems[item.id] = {
-              ...item,
-              first_seen: knownItems[item.id]?.first_seen || Date.now(),
-              last_updated: Date.now()
-            };
+    this.ctx.setInterval(async () => {
+      try {
+        this.cleanupExpiredCache();
+        
+        logger.info('开始检查折扣商品...');
+        const discountItems = await this.searchDiscountItems();
+        const knownItems = await this.loadJSON(this.DISCOUNT_ITEMS_FILE);
+        const newDiscountItems = [];
+        
+        for (const item of discountItems) {
+          if (!this.isCached(item.id)) {
+            if (!knownItems[item.id] || knownItems[item.id].discount_rate !== item.discount_rate) {
+              newDiscountItems.push(item);
+              this.addToCache(item.id);
+              knownItems[item.id] = {
+                ...item,
+                first_seen: knownItems[item.id]?.first_seen || Date.now(),
+                last_updated: Date.now()
+              };
+            }
           }
         }
+        
+        if (newDiscountItems.length > 0) {
+          logger.info(`发现 ${newDiscountItems.length} 个新的折扣商品`);
+          await this.saveJSON(this.DISCOUNT_ITEMS_FILE, knownItems);
+          await this.notifyDiscountGroups(newDiscountItems);
+        } else {
+          logger.info('未发现新的折扣商品');
+        }
+      } catch (error) {
+        logger.error('折扣商品检查失败:', error);
       }
-      
-      if (newDiscountItems.length > 0) {
-        logger.info(`发现 ${newDiscountItems.length} 个新的折扣商品`);
-        await this.saveJSON(this.DISCOUNT_ITEMS_FILE, knownItems);
-        await this.notifyDiscountGroups(newDiscountItems);
-      } else {
-        logger.info('未发现新的折扣商品');
-      }
-    } catch (error) {
-      logger.error('折扣商品检查失败:', error);
-    }
-  }, (this.config.discountCheckInterval || 60) * 60 * 1000);
-}
+    }, (this.config.discountCheckInterval || 60) * 60 * 1000);
+  }
+
   async addDiscountGroupSubscription(platform, channelId) {
     const groupKey = `group:${platform}:${channelId}`;
     const discountGroups = await this.loadJSON(this.DISCOUNT_GROUPS_FILE);

package/package.json

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