koishi-plugin-cocoyyy-console 1.0.4 → 1.0.6

package/lib/index.js

@@ -344,6 +344,16 @@ var tagFuncMenuList = [
     description: "标签取消别名",
     command: "unbind [标签] [别名]"
   },
+  {
+    name: "addstart",
+    description: "开始给标签批量添加图片",
+    command: "addstart [标签]"
+  },
+  {
+    name: "addend",
+    description: "结束给标签批量添加图片",
+    command: "addend"
+  },
   {
     name: "taglist",
     description: "获取标签列表",
@@ -469,6 +479,46 @@ async function getImgCountByTag(tag_id) {
   return count;
 }
 __name(getImgCountByTag, "getImgCountByTag");
+async function startBatchAddImg(ctx, tag_name, session, savePath2) {
+  try {
+    const guild_id = session.guildId;
+    let tag = await getTag(tag_name, guild_id);
+    if (tag == null) {
+      throw new Error(`not found tag '${tag_name}'!`);
+    }
+    if (tag.get("status") !== 0) {
+      logger.info(`[startBatchAddImg Info]:tag '${tag_name}' is deleted`);
+      throw new Error(`tag '${tag_name}' is deleted!`);
+    }
+    if (savePath2 == null) {
+      throw new Error("saving path config wrong!");
+    }
+    if (uid_tag_map.has(session.userId)) {
+      throw new Error("already started");
+    }
+    let key = session.guildId + "_" + session.userId;
+    uid_tag_map.set(key, tag_name);
+    return { result: true, error: null };
+  } catch (e) {
+    logger.error("[startBatchAddImg Error]: " + e?.message || String(e));
+    return { result: false, error: e?.message || String(e) };
+  }
+}
+__name(startBatchAddImg, "startBatchAddImg");
+async function endBatchAddImg(ctx, session) {
+  try {
+    let key = session.guildId + "_" + session.userId;
+    if (!uid_tag_map.has(key)) {
+      throw new Error("not start recording");
+    }
+    uid_tag_map.delete(key);
+    return { result: true, error: null };
+  } catch (e) {
+    logger.error("[startBatchAddImg Error]: " + e?.message || String(e));
+    return { result: false, error: e?.message || String(e) };
+  }
+}
+__name(endBatchAddImg, "endBatchAddImg");
 
 // src/services/tag_func/tag_service.ts
 async function findTagByName(tag_name, guild_id) {
@@ -618,6 +668,7 @@ __name(getTagList, "getTagList");
 
 // src/services/tag_func/tag_commands.ts
 var import_url = require("url");
+var uid_tag_map = /* @__PURE__ */ new Map();
 function registerTagCommands(ctx, connected, savePath2) {
   ctx.command("maketag <参数>", "创建一个标签").check(({ session }) => {
     if (!connected) return;
@@ -678,16 +729,48 @@ function registerTagCommands(ctx, connected, savePath2) {
       return `取消绑定失败：${exec.error}`;
     return `取消绑定成功`;
   });
-  ctx.command("taglist", "获取标签列表").check(({ session }) => {
+  ctx.command("addstart <参数>", "开始给标签批量添加图片").check(({ session }) => {
     if (!connected) return;
     if (!is_at_bot(session)) return;
   }).action(async ({ session }, ...args) => {
+    const tag = args?.[0];
+    if (!tag) return "请提供正确格式，如：@bot addstart [标签]";
+    let exec = await startBatchAddImg(ctx, tag, session, savePath2);
+    if (!exec.result)
+      return exec.error;
+    return "[开始记录图片]";
+  });
+  ctx.command("addend", "结束给标签批量添加图片").check(({ session }) => {
+    if (!connected) return;
+    if (!is_at_bot(session)) return;
+  }).action(async ({ session }) => {
+    let exec = await endBatchAddImg(ctx, session);
+    if (!exec.result)
+      return exec.error;
+    return "[停止记录图片]";
+  });
+  ctx.command("taglist", "获取标签列表").check(({ session }) => {
+    if (!connected) return;
+    if (!is_at_bot(session)) return;
+  }).action(async ({ session }) => {
     let exec = await getTagList(session.guildId);
     if (exec == null)
       return `查询失败`;
     return `查询成功!
 ${exec.map((item) => "标签:" + item.tag + "	黑历史数:" + item.count + "\n别名:" + (item.alias ? item.alias : "无")).join("\n\n")}`;
   });
+  ctx.middleware(async (session, next) => {
+    let key = session.guildId + "_" + session.userId;
+    if (!uid_tag_map.has(key)) return await next();
+    const tag = uid_tag_map.get(key);
+    const img = session.content;
+    const srcMatch = img.match(/src="([^"]+)"/);
+    if (!srcMatch) return "请给出图片，如需停止记录图片，请输入@bot addend [标签]";
+    let exec = await saveImg(ctx, tag, session.guildId, img, savePath2);
+    if (!exec.result)
+      return `保存失败：${exec.error}`;
+    return "保存成功";
+  });
 }
 __name(registerTagCommands, "registerTagCommands");
 

package/lib/services/tag_func/img_service.d.ts

@@ -1,8 +1,16 @@
-import { Context } from "koishi";
+import { Context, Session } from "koishi";
 declare function saveImg(ctx: Context, tag_name: string, guild_id: string, imgurl: string, savePath: string): Promise<{
     result: boolean;
     error: string | null;
 }>;
 declare function randomImgByTag(tag_name: string, guild_id: string): Promise<string | null>;
 declare function getImgCountByTag(tag_id: number): Promise<number>;
-export { saveImg, randomImgByTag, getImgCountByTag };
+declare function startBatchAddImg(ctx: Context, tag_name: string, session: Session, savePath: string): Promise<{
+    result: boolean;
+    error: string | null;
+}>;
+declare function endBatchAddImg(ctx: Context, session: Session): Promise<{
+    result: boolean;
+    error: string | null;
+}>;
+export { saveImg, randomImgByTag, getImgCountByTag, startBatchAddImg, endBatchAddImg };

package/lib/services/tag_func/tag_commands.d.ts

@@ -1,3 +1,4 @@
 import { Context } from 'koishi';
+declare let uid_tag_map: Map<string, string>;
 declare function registerTagCommands(ctx: Context, connected: boolean, savePath: string | null): void;
-export { registerTagCommands };
+export { uid_tag_map, registerTagCommands };

package/lib/services/test_command.d.ts

@@ -0,0 +1 @@
+export {};

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-cocoyyy-console",
   "description": "for self using, function console",
-  "version": "1.0.4",
+  "version": "1.0.6",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "contributors": [