koishi-plugin-booth-get 2.0.0 → 3.1.0

package/README.md

@@ -8,4 +8,19 @@
 
 内容展示：
 
-https://www.freeimg.cn/i/2024/09/01/66d47ad049577.png
+https://www.freeimg.cn/i/2024/09/01/66d47ad049577.png
+
+更新内容：
+
+2.0.0版本更新自定义代理
+
+3.0.0版本更新ws服务（目前还在测试阶段，预测5.0.0版本与大家见面）
+
+
+感谢大家的喜欢与支持
+
+最后更新预测6.0.0版本，本版本将优化反馈图片
+
+图片预览
+
+https://www.freeimg.cn/i/2024/09/15/66e626ea2f552.webp

package/lib/index.js

@@ -33,6 +33,8 @@ var Config = import_koishi.Schema.object({
   loadTimeout: import_koishi.Schema.natural().role("ms").description("加载页面的最长时间。当一个页面等待时间超过这个值时，如果此页面主体已经加载完成，则会发送一条提示消息“正在加载中，请稍等片刻”并继续等待加载；否则会直接提示“无法打开页面”并终止加载。").default(import_koishi.Time.second * 5),
   idleTimeout: import_koishi.Schema.natural().role("ms").description("等待页面空闲的最长时间。当一个页面等待时间超过这个值时，将停止进一步的加载并立即发送截图。").default(import_koishi.Time.second * 30),
   proxyServer: import_koishi.Schema.string().description("自定义代理服务器地址。例如：http://127.0.0.1:1080").default(""),
+  wsServer: import_koishi.Schema.string().description("WebSocket服务器地址。").default(""),
+  enableWs: import_koishi.Schema.boolean().description("是否启用WebSocket连接。").default(false),
 }).description("booth-get");
 function apply(ctx, config) {
   const logger = ctx.logger("screenshot");
@@ -41,6 +43,28 @@ function apply(ctx, config) {
   const booth_url = "https://booth.pm/zh-cn/items/";
   const maxSize = 1024 * 1024;
   const launchOptions = { ...ctx.puppeteer.config.launchOptions };
+
+  if (config.enableWs && config.wsServer) {
+    const WebSocket = require('ws');
+    let wsUrl = config.wsServer;
+    if (!wsUrl.startsWith('ws://') && !wsUrl.startsWith('wss://')) {
+      wsUrl = 'ws://' + wsUrl; 
+    }
+    const ws = new WebSocket(wsUrl);
+    ws.on('open', function open() {
+      console.log('Connected to WebSocket server');
+      ws.send('Hello, server!');
+    });
+    ws.on('message', function incoming(data) {
+      console.log('Received message:', data);
+    });
+    ws.on('error', function handleErrors(error) {
+      console.error('WebSocket error:', error);
+    });
+    ws.on('close', function close() {
+      console.log('Disconnected from WebSocket server');
+    });
+  }
   
   if (proxyServer) {
     launchOptions.args = launchOptions.args || [];
@@ -103,6 +127,7 @@ function apply(ctx, config) {
       return "截图失败。";
     }).finally(() => page.close());
   });
+
 ctx.command("摊位名称 <query>").action(async ({ session }, query) => {
   if (!query)
     return "请输入搜索关键词";
@@ -179,6 +204,23 @@ ctx.middleware((session, next) => {
 
   return next();
 });
+function sendItemImage(itemId) {
+  fetch(`http://127.0.0.1:8080/generate_image/${itemId}`)
+      .then(response => {
+          if (response.ok) {
+              return response.blob();
+          } else {
+              throw new Error('Image generation failed');
+          }
+      })
+      .then(blob => {
+          const imageUrl = URL.createObjectURL(blob);
+          console.log(imageUrl);
+      })
+      .catch(error => {
+          console.error(error);
+      });
+}
 }
 __name(apply, "apply");
 0 && (module.exports = {
@@ -186,4 +228,13 @@ __name(apply, "apply");
   apply,
   inject,
   name
-});
+});
+async function triggerImageGeneration() {
+  try {
+    const response = await fetch('http://localhost:8000/generate');
+    const data = await response.json();
+    console.log(data.message); // Log the response from the server
+  } catch (error) {
+    console.error('Error calling the API:', error);
+  }
+}

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-booth-get",
   "description": "通过url与名称检查摊位物品并反馈用户搜索的图片",
-  "version": "2.0.0",
+  "version": "3.1.0",
   "contributors": [
     "rixiang <1148147857@qq.com>"
   ],
@@ -29,7 +29,7 @@
     "booth"
   ],
   "peerDependencies": {
-    "koishi": "4.17.12"
+    "koishi": "4.18.4"
   },
   "dependencies": {
     "atsc": "^2.1.0",