koishi-plugin-sy-bot 0.0.15 → 0.1.0

package/lib/function/cronTask.d.ts

@@ -0,0 +1,5 @@
+export declare function cronTask({ ctx, bot, config }: {
+    ctx: any;
+    bot: any;
+    config: any;
+}): Promise<void>;

package/lib/function/everydayShare.d.ts

@@ -0,0 +1 @@
+export declare function everyDayShare(obj: any): Promise<void>;

package/lib/index.d.ts

@@ -8,8 +8,19 @@ export interface Config {
     api: string;
     model: string;
     prompt: string;
-    guild: string;
+    guild: number;
     user: string;
+    dailyShares: Array<object>;
+    tasks: Task[];
+}
+export interface Task {
+    startTime: string;
+    endTime: string;
+    beforeTip: string;
+    afterTip: string;
+    lockName: string;
+    unlockName: string;
+    ifMute: boolean;
 }
 export declare const Config: Schema<Config>;
 export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -236,17 +236,17 @@ async function sendGroupMessageReaction(session, face_id, ifSet = true) {
   );
 }
 __name(sendGroupMessageReaction, "sendGroupMessageReaction");
-async function getAiChatResult({ config, user_query, ctx }) {
+async function getAiChatResult({ config, user_query, ctx, system_prompt }) {
   try {
     const logger = ctx.logger("sy-bot");
-    const { model: MODEL, api: API_KEY, prompt: system_prompt } = config;
+    const { model: MODEL, api: API_KEY, prompt: system_prompt2 } = config;
     const { data } = await import_axios.default.post(
       `https://api.qingyuntop.top/v1beta/models/${MODEL}?key=${API_KEY}`,
       {
         systemInstruction: {
           parts: [
             {
-              text: system_prompt
+              text: system_prompt2
             }
           ]
         },
@@ -277,7 +277,7 @@ async function getAiChatResult({ config, user_query, ctx }) {
       "过滤后的parts回复列表：\n",
       data.candidates[0].content.parts.filter((part) => part.text)
     );
-    const result = content.parts.filter((part) => part.text).map((part) => part.text).join("");
+    const result = content.parts.filter((part) => part.text && !part.thought).map((part) => part.text).join("");
     logger.info("AI回复的文字：", result);
     return result;
   } catch (error) {
@@ -335,7 +335,7 @@ async function getAiImgUnderstandMultiple({
       "过滤后的parts回复列表：\n",
       data.candidates[0].content.parts.filter((part) => part.text)
     );
-    const text = data.candidates[0].content.parts.filter((part) => part.text).map((p) => p.text).join("");
+    const text = data.candidates[0].content.parts.filter((part) => part.text && !part.thought).map((p) => p.text).join("");
     logger.info("解析出的text：", text);
     return text;
   } catch (error) {
@@ -365,7 +365,8 @@ async function sendOnlyText({ session, ctx, config, query }) {
   const raw_answer = await getAiChatResult({
     config,
     user_query: question,
-    ctx
+    ctx,
+    system_prompt: config.prompt
   });
   if (!raw_answer) {
     return "AI没有回复";
@@ -515,11 +516,15 @@ __name(registerCountdown, "registerCountdown");
 function registerResource(ctx, config) {
   ctx.command("resource [index:number]", "获取26,27做题本/习题册/讲义pdf").action(async (_, index) => {
     const resources = await ctx.database.get("resources");
-    if (!index) {
-      const top = "-------- 资源合集 \n----------------\n";
-      return top + resources.map((res, idx) => `${idx + 1} 《${res.title}》`).join(" \n");
-    } else {
-      return resources[index - 1] ? resources[index - 1].content : "没有这个哦，喵";
+    try {
+      if (!index) {
+        const top = "-------- 资源合集 \n----------------\n";
+        return top + resources.map((res, idx) => `${idx + 1} 《${res.title}》`).join(" \n");
+      } else {
+        return resources[index - 1] ? resources[index - 1].content : "没有这个哦，喵";
+      }
+    } catch (error) {
+      return error.message;
     }
   });
 }
@@ -560,7 +565,7 @@ function registerTip(ctx, config) {
     const tips = await ctx.database.get("tips");
     if (!index) {
       const top = "-------- 经验贴合集 \n----------------\n";
-      return top + tips.map((tip, idx) => `${idx} 《${tip.title}》`).join(" \n");
+      return top + tips.map((tip, idx) => `${idx + 1} 《${tip.title}》`).join(" \n");
     } else {
       return tips[index - 1] ? tips[index - 1].url : "没有这个哦，喵";
     }
@@ -610,22 +615,123 @@ function registerTables(ctx) {
 }
 __name(registerTables, "registerTables");
 
+// src/function/everydayShare.tsx
+async function everyDayShare(obj) {
+  const { ctx, bot, config } = obj;
+  config.dailyShares.forEach((share) => {
+    registerCronShare({
+      ctx,
+      bot,
+      config,
+      time: share.time,
+      user_query: share.question,
+      beforeTip: share.beforeTip,
+      system_prompt: share.system_prompt
+    });
+  });
+}
+__name(everyDayShare, "everyDayShare");
+function registerCronShare(obj) {
+  const { ctx, bot, config, time, beforeTip, user_query, system_prompt } = obj;
+  ctx.cron(time, async () => {
+    const logger = ctx.logger;
+    beforeTip && await bot.sendMessage(config.guild, beforeTip);
+    const raw_answer = await getAiChatResult({
+      config,
+      user_query,
+      system_prompt,
+      ctx
+    });
+    if (!raw_answer) {
+      logger.info("AI没有回复");
+    }
+    try {
+      logger.info("开始渲染markdown...");
+      const h_ans_img = await textToImage(ctx, raw_answer);
+      logger.info("markdown渲染完毕！");
+      await bot.sendMessage(config.guild, h_ans_img);
+    } catch (err) {
+      return "调用 textToImage 报错：" + (err instanceof Error ? err.message : String(err));
+    }
+  });
+}
+__name(registerCronShare, "registerCronShare");
+
+// src/function/cronTask.tsx
+async function cronTask({ ctx, bot, config }) {
+  const logger = ctx.logger;
+  for (const task of config.tasks) {
+    const {
+      startTime,
+      endTime,
+      beforeTip,
+      afterTip,
+      lockName,
+      unlockName,
+      ifMute
+    } = task;
+    ctx.cron(startTime, async () => {
+      await bot.sendMessage(config.guild.toString(), beforeTip);
+      await bot.muteChannel(config.guild.toString(), ifMute);
+      await bot.internal.setGroupName(config.guild, lockName);
+    });
+    ctx.cron(endTime, async () => {
+      await bot.sendMessage(config.guild.toString(), afterTip);
+      await bot.muteChannel(config.guild.toString(), false);
+      await bot.internal.setGroupName(config.guild, unlockName);
+    });
+  }
+}
+__name(cronTask, "cronTask");
+
 // src/index.ts
 var name = "sy-bot";
 var usage = "考研群答疑机器人";
 var inject = {
-  required: ["database", "puppeteer"]
+  required: ["database", "puppeteer", "cron"]
 };
 var Config = import_koishi3.Schema.object({
   api: import_koishi3.Schema.string().required().description("青云☁️ Gemini API（必填）"),
+  tasks: import_koishi3.Schema.array(
+    import_koishi3.Schema.object({
+      startTime: import_koishi3.Schema.string().default("0 0 23 * * *").description("开始 Cron"),
+      endTime: import_koishi3.Schema.string().default("0 0 7 * * *").description("结束 Cron"),
+      beforeTip: import_koishi3.Schema.string().description("开始公告").role("textarea"),
+      afterTip: import_koishi3.Schema.string().description("解放公告").role("textarea"),
+      lockName: import_koishi3.Schema.string().description("开始群名"),
+      unlockName: import_koishi3.Schema.string().description("解放群名"),
+      ifMute: import_koishi3.Schema.boolean().description("禁言")
+    })
+  ).description("定时禁言任务列表"),
   model: import_koishi3.Schema.string().required().description("模型名称（必填）"),
   prompt: import_koishi3.Schema.string().description("预设系统提示词（可以留空）"),
-  guild: import_koishi3.Schema.string().description("在目标群聊启用，留空表示广播所有群聊（填入QQ群号）"),
-  user: import_koishi3.Schema.string().required().description("在目标用户启用（填入管理员QQ号）")
+  guild: import_koishi3.Schema.number().required().description(
+    "在目标群聊启用，留空表示广播所有群聊（填入QQ群号）"
+  ),
+  user: import_koishi3.Schema.string().required().description("在目标用户启用（填入管理员QQ号）"),
+  dailyShares: import_koishi3.Schema.array(
+    import_koishi3.Schema.object({
+      time: import_koishi3.Schema.string().required().description(`时间字符串`),
+      question: import_koishi3.Schema.string().role("textarea", { rows: [2, 4] }).required().description(`提示词`),
+      beforeTip: import_koishi3.Schema.string().role("textarea", { rows: [2, 4] }).required().description(`发送前的提示语`),
+      system_prompt: import_koishi3.Schema.string().required().role("textarea", { rows: [2, 4] }).description(`系统提示词`)
+    })
+  )
 });
 function apply(ctx, config) {
+  const platform = `milky`;
+  const bot = ctx.bots.filter((bot2) => bot2.platform == platform)[0];
+  ctx.on("ready", () => {
+    ctx.bots.forEach((bot2) => {
+      ctx.logger.info("milky_bot:", bot2.platform);
+      ctx.logger.info("milky_bot:", bot2.selfId);
+      ctx.logger.info("milky_bot 加载成功！");
+    });
+  });
   registerCommand(ctx, config);
   registerTables(ctx);
+  cronTask({ ctx, bot, config });
+  everyDayShare({ ctx, bot, config });
 }
 __name(apply, "apply");
 // Annotate the CommonJS export names for ESM import in node:

package/lib/utils/utils.d.ts

@@ -7,10 +7,11 @@ import { Context, h, Session } from "koishi";
 export declare function getKaoyanCountdownText(targetTime: any): string;
 export declare function getPrompt(): void;
 export declare function sendGroupMessageReaction(session: any, face_id: any, ifSet?: boolean): Promise<void>;
-export declare function getAiChatResult({ config, user_query, ctx }: {
+export declare function getAiChatResult({ config, user_query, ctx, system_prompt }: {
     config: any;
     user_query: any;
     ctx: any;
+    system_prompt: any;
 }): Promise<any>;
 export declare function imageUrlToBase64(ctx: Context, url: string): Promise<string>;
 export declare function getAiImgUnderstandMultiple({ ctx, config, url_list, query, }: {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-sy-bot",
   "description": "考研群答疑Bot，接入Gemini。",
-  "version": "0.0.15",
+  "version": "0.1.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -28,6 +28,7 @@
     "@satorijs/element": "^3.1.8",
     "@vscode/markdown-it-katex": "^1.1.2",
     "axios": "^1.13.2",
+    "corn": "^0.0.1",
     "highlight": "^0.2.4",
     "highlight.js": "^11.11.1",
     "markdown-it": "^14.1.0",