mioku-plugin-admin 2.3.5 → 3.0.1

package/commands/personal.ts

@@ -1,18 +1,5 @@
-import { MiokiContext } from "mioki";
-import type {
-  MessageEvent,
-  RecvAtElement,
-  RecvElement,
-  RecvFaceElement,
-  RecvFileElement,
-  RecvForwardElement,
-  RecvImageElement,
-  RecvJsonElement,
-  RecvRecordElement,
-  RecvReplyElement,
-  RecvTextElement,
-  RecvVideoElement,
-} from "napcat-sdk";
+import type { Bot, CommandDefinition, MessageEvent, MessageInput, MessageTarget, MiokuContext, MessageSegment } from "mioku";
+import {createGroupRef} from "mioku";
 import { extractImageUrl } from "../config";
 import { replyAdminErrorNotice } from "./notice";
 
@@ -23,80 +10,70 @@ function parseProfileSex(value: string): 0 | 1 | 2 {
   return 0;
 }
 
-function toSendSegment(ctx: MiokiContext, seg: RecvElement): any | null {
-  if (seg.type === "text") {
-    const text = String((seg as RecvTextElement).text || "");
-    return text ? ctx.segment.text(text) : null;
-  }
-
-  if (seg.type === "image" || seg.type === "record" || seg.type === "video") {
-    const mediaSeg = seg as
-      | RecvImageElement
-      | RecvRecordElement
-      | RecvVideoElement;
-    const source = String(mediaSeg.url || mediaSeg.file || "").trim();
-    if (!source) return null;
-    if (seg.type === "image") {
-      return ctx.segment.image(source);
+function toSendSegment(ctx: MiokuContext, seg: MessageSegment): MessageSegment | null {
+  const data = seg.data as Record<string, unknown>;
+  switch (seg.type) {
+    case "text": {
+      const text = String(data.text ?? "").trim();
+      return text ? ctx.segment.text(text) : null;
     }
-    if (seg.type === "record") {
-      return (ctx.segment as any).record(source);
+    case "image": {
+      const source = String(data.url ?? data.file ?? "").trim();
+      return source ? ctx.segment.image(source) : null;
     }
-    return (ctx.segment as any).video(source);
-  }
-
-  if (seg.type === "file") {
-    const fileSeg = seg as RecvFileElement;
-    const source = String(fileSeg.url || fileSeg.file || "").trim();
-    if (!source) return null;
-    return (ctx.segment as any).file(source);
-  }
-
-  if (seg.type === "at") {
-    const qq = (seg as RecvAtElement).qq;
-    return qq == null ? null : ctx.segment.at(String(qq));
-  }
-
-  if (seg.type === "face") {
-    const id = (seg as RecvFaceElement).id;
-    return id == null ? null : ctx.segment.face(Number(id));
-  }
-
-  if (seg.type === "reply") {
-    const id = (seg as RecvReplyElement).id;
-    return id == null ? null : ctx.segment.reply(String(id));
-  }
-
-  if (seg.type === "forward") {
-    const id = (seg as RecvForwardElement).id;
-    return id == null
-      ? null
-      : ((ctx.segment as any).forward?.(String(id)) ?? null);
-  }
-
-  if (seg.type === "json") {
-    const data = (seg as RecvJsonElement).data;
-    return ctx.segment.json ? ctx.segment.json(data) : null;
+    case "record": {
+      const source = String(data.file ?? data.url ?? "").trim();
+      return source ? ctx.segment.raw("record", { file: source }) : null;
+    }
+    case "video": {
+      const source = String(data.file ?? data.url ?? "").trim();
+      return source ? ctx.segment.raw("video", { file: source }) : null;
+    }
+    case "file": {
+      const source = String(data.file ?? data.url ?? "").trim();
+      return source ? ctx.segment.raw("file", { file: source }) : null;
+    }
+    case "at": {
+      const target = String(data.qq ?? data.target ?? "");
+      return target && target !== "all" ? ctx.segment.at(target) : null;
+    }
+    case "face": {
+      const id = data.id;
+      return id == null ? null : ctx.segment.raw("face", { id: String(id) });
+    }
+    case "reply": {
+      const id = data.message_id ?? data.id;
+      return id == null ? null : ctx.segment.reply(String(id));
+    }
+    case "forward": {
+      const id = data.id;
+      return id == null ? null : ctx.segment.raw("forward", { id: String(id) });
+    }
+    case "json": {
+      return ctx.segment.raw("json", data);
+    }
+    default:
+      return seg;
   }
-
-  return null;
 }
 
 function normalizeIncomingSegments(
-  ctx: MiokiContext,
-  segments: RecvElement[],
-): any[] {
+  ctx: MiokuContext,
+  segments: readonly MessageSegment[],
+): MessageSegment[] {
   if (!Array.isArray(segments)) return [];
-  return segments.map((seg) => toSendSegment(ctx, seg)).filter(Boolean);
+  return segments
+    .map((seg) => toSendSegment(ctx, seg))
+    .filter((seg): seg is MessageSegment => seg !== null);
 }
 
 function buildForwardPayloadAfterCommand(
-  ctx: MiokiContext,
-  message: RecvElement[],
+  ctx: MiokuContext,
+  message: readonly MessageSegment[],
   commandPattern: RegExp,
   fallbackText?: string,
-): any[] {
-  const payload: any[] = [];
+): MessageSegment[] {
+  const payload: MessageSegment[] = [];
   let stripped = false;
 
   for (const seg of message) {
@@ -107,7 +84,7 @@ function buildForwardPayloadAfterCommand(
       }
       continue;
     }
-    const original = String((seg as RecvTextElement).text || "");
+    const original = String((seg.data as Record<string, unknown>).text ?? "");
     if (!stripped) {
       const nextText = original.replace(commandPattern, "");
       if (nextText !== original) {
@@ -134,110 +111,72 @@ function buildForwardPayloadAfterCommand(
   return payload;
 }
 
-function toForwardMessages(bot: any, nodes: any[]): any[] {
-  const normalizeElements = (elements: any[]): any[] => {
-    if (typeof bot?.normalizeSendable === "function") {
-      return bot.normalizeSendable(elements);
-    }
-    return elements.map((element: any) => {
-      if (
-        element &&
-        typeof element === "object" &&
-        "type" in element &&
-        "data" in element
-      ) {
-        return element;
-      }
-      if (element && typeof element === "object" && "type" in element) {
-        const { type, ...data } = element;
-        return { type, data };
-      }
-      return element;
-    });
-  };
-
-  return nodes.map((node: any) => {
-    const rawNode =
-      node && typeof node === "object" && "type" in node && "data" in node
-        ? { type: node.type, ...node.data }
-        : node;
-    if (!rawNode || rawNode.type !== "node") {
-      return normalizeElements([rawNode])[0];
-    }
-
-    const content = Array.isArray(rawNode.content) ? rawNode.content : [];
-    if ("id" in rawNode && rawNode.id) {
-      return {
-        type: "node",
-        data: {
-          user_id: rawNode.user_id,
-          nickname: rawNode.nickname,
-          id: rawNode.id,
-        },
-      };
-    }
-
-    return {
-      type: "node",
-      data: {
-        user_id: rawNode.user_id,
-        nickname: rawNode.nickname,
-        content: normalizeElements(content),
-      },
-    };
-  });
-}
-
 async function sendForwardByEvent(options: {
-  bot: any;
-  event: any;
-  messages: any[];
+  bot: Bot;
+  event: MessageEvent;
+  messages: readonly unknown[];
 }): Promise<void> {
   const { bot, event, messages } = options;
   const chunkSize = 50;
 
-  for (let i = 0; i < messages.length; i += chunkSize) {
-    const chunk = messages.slice(i, i + chunkSize);
-    if (event?.message_type === "group" && event?.group_id) {
-      await bot.api("send_group_forward_msg", {
-        group_id: event.group_id,
-        messages: chunk,
-      });
-      continue;
-    }
-
-    await bot.api("send_private_forward_msg", {
-      user_id: event.user_id,
-      messages: chunk,
-    });
+  const nodes = (messages as Array<{
+    type?: string;
+    data?: { user_id?: string; nickname?: string; content?: unknown };
+  }>)
+    .map((node) => ({
+      user_id: String(node?.data?.user_id ?? event.user_id ?? ""),
+      nickname:
+        String(node?.data?.nickname || "") ||
+        String(node?.data?.user_id ?? event.user_id ?? "转发"),
+      content: Array.isArray(node?.data?.content)
+        ? node.data.content
+        : [node?.data?.content],
+    }))
+    .filter((node) => node.content.length > 0)
+    .map((node) => ({
+      user_id: node.user_id,
+      nickname: node.nickname,
+      content: node.content as MessageInput,
+    }));
+
+  const target: MessageTarget =
+    event.message_type === "group" && event.group_id
+      ? { type: "group", group_id: event.group_id }
+      : { type: "private", user_id: event.user_id ?? "" };
+
+  for (let i = 0; i < nodes.length; i += chunkSize) {
+    await bot.sendForward(target, nodes.slice(i, i + chunkSize));
   }
 }
 
-export function registerPersonalCommands(ctx: MiokiContext) {
-  ctx.handle("message", async (event: MessageEvent) => {
-    const text = ctx.text(event)?.trim();
-    if (!text) return;
-    if (event.user_id === event.self_id) return;
-
-    const isMaster = ctx.isOwner?.(event) ?? false;
-
-    const selfId = event.self_id;
-    const bot = ctx.pickBot(selfId);
-    if (!bot) return;
-
-    const isGroup = event.message_type === "group";
+export function registerPersonalCommands(ctx: MiokuContext) {
+  const register = (command: CommandDefinition) => {
+    const { handler, ...rest } = command;
+    ctx.command({
+      ...rest,
+      prefixes: false,
+      handler: async (c) => {
+        if (c.event.user_id === c.event.self_id) return;
+        await handler(c);
+      },
+    });
+  };
 
-    if (text.startsWith("/改头像")) {
-      if (!isMaster) {
-        ctx.logger.warn("[admin] 改头像功能仅主人可用");
-        return;
-      }
+  register({
+    name: "/改头像",
+    match: /^\/改头像/,
+    permission: "master",
+    description: "修改Bot头像",
+    handler: async ({ event }) => {
+      const bot = event.bot;
+      if (!bot) return;
       const imageUrl = extractImageUrl(event.message);
       if (!imageUrl) {
-        return event.reply("图片呢图片呢～", true);
+        await event.reply("图片呢图片呢～", true);
+        return;
       }
       try {
-        await bot.api("set_qq_avatar", { file: imageUrl });
+        await bot.setAvatar(imageUrl);
         await event.reply("done");
       } catch (err) {
         await replyAdminErrorNotice({
@@ -247,22 +186,25 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           fallbackMessage: `出错了，笨蛋～ ${String(err)}`,
           error: err,
         });
-        return;
       }
-      return;
-    }
+    },
+  });
 
-    if (text.startsWith("/改昵称")) {
-      if (!isMaster) {
-        ctx.logger.warn("[admin] 改昵称功能仅主人可用");
-        return;
-      }
-      const nickname = text.replace(/^\/改昵称\s*/, "").trim();
+  register({
+    name: "/改昵称",
+    match: /^\/改昵称/,
+    permission: "master",
+    description: "修改Bot昵称",
+    handler: async ({ event, body }) => {
+      const bot = event.bot;
+      if (!bot) return;
+      const nickname = body.replace(/^\/改昵称\s*/, "").trim();
       if (!nickname) {
-        return event.reply("想改成什么昵称呀～", true);
+        await event.reply("想改成什么昵称呀～", true);
+        return;
       }
       try {
-        await bot.api("set_qq_profile", { nickname });
+        await bot.setProfile({ nickname });
         await event.reply("done");
       } catch (err) {
         await replyAdminErrorNotice({
@@ -273,20 +215,24 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           error: err,
         });
       }
-      return;
-    }
+    },
+  });
 
-    if (text.startsWith("/改签名")) {
-      if (!isMaster) {
-        ctx.logger.warn("[admin] 改签名功能仅主人可用");
-        return;
-      }
-      const personalNote = text.replace(/^\/改签名\s*/, "").trim();
+  register({
+    name: "/改签名",
+    match: /^\/改签名/,
+    permission: "master",
+    description: "修改Bot个性签名",
+    handler: async ({ event, body }) => {
+      const bot = event.bot;
+      if (!bot) return;
+      const personalNote = body.replace(/^\/改签名\s*/, "").trim();
       if (!personalNote) {
-        return event.reply("想改成什么签名呀～", true);
+        await event.reply("想改成什么签名呀～", true);
+        return;
       }
       try {
-        await bot.api("set_qq_profile", { personal_note: personalNote });
+        await bot.setProfile({ personal_note: personalNote });
         await event.reply("done");
       } catch (err) {
         await replyAdminErrorNotice({
@@ -297,18 +243,20 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           error: err,
         });
       }
-      return;
-    }
+    },
+  });
 
-    if (text.startsWith("/改性别")) {
-      if (!isMaster) {
-        ctx.logger.warn("[admin] 改性别功能仅主人可用");
-        return;
-      }
-      const genderText = text.replace(/^\/改性别\s*/, "").trim();
-      const sex = parseProfileSex(genderText);
+  register({
+    name: "/改性别",
+    match: /^\/改性别/,
+    permission: "master",
+    description: "修改Bot性别",
+    handler: async ({ event, body }) => {
+      const bot = event.bot;
+      if (!bot) return;
+      const sex = parseProfileSex(body.replace(/^\/改性别\s*/, ""));
       try {
-        await bot.api("set_qq_profile", { sex });
+        await bot.setProfile({ sex });
         await event.reply("done");
       } catch (err) {
         await replyAdminErrorNotice({
@@ -319,15 +267,19 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           error: err,
         });
       }
-      return;
-    }
+    },
+  });
 
-    if (text.startsWith("/删好友")) {
-      if (!isMaster) {
-        ctx.logger.warn("[admin] 删好友功能仅主人可用");
-        return;
-      }
-      const qq = parseInt(text.replace(/^\/删好友\s*/, "").trim(), 10);
+  register({
+    name: "/删好友",
+    match: /^\/删好友(?:\s|$)/,
+    permission: "master",
+    description: "删除好友",
+    usage: "/删好友 qq号",
+    handler: async ({ event, body }) => {
+      const bot = event.bot;
+      if (!bot) return;
+      const qq = parseInt(body.replace(/^\/删好友\s*/, "").trim(), 10);
       if (!qq) {
         await replyAdminErrorNotice({
           ctx,
@@ -338,7 +290,7 @@ export function registerPersonalCommands(ctx: MiokiContext) {
         return;
       }
       try {
-        await bot.api("delete_friend", { user_id: qq });
+        await bot.deleteFriend(qq);
         await event.reply("done");
       } catch (err) {
         await replyAdminErrorNotice({
@@ -349,15 +301,19 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           error: err,
         });
       }
-      return;
-    }
+    },
+  });
 
-    if (text.startsWith("/退群")) {
-      if (!isMaster) {
-        ctx.logger.warn("[admin] 退群功能仅主人可用");
-        return;
-      }
-      const targetGroup = parseInt(text.replace(/^\/退群\s*/, "").trim(), 10);
+  register({
+    name: "/退群",
+    match: /^\/退群(?:\s|$)/,
+    permission: "master",
+    description: "退出群聊",
+    usage: "/退群 群号",
+    handler: async ({ event, body }) => {
+      const bot = event.bot;
+      if (!bot) return;
+      const targetGroup = parseInt(body.replace(/^\/退群\s*/, "").trim(), 10);
       if (!targetGroup) {
         await replyAdminErrorNotice({
           ctx,
@@ -368,10 +324,7 @@ export function registerPersonalCommands(ctx: MiokiContext) {
         return;
       }
       try {
-        await bot.api("set_group_leave", {
-          group_id: targetGroup,
-          is_dismiss: false,
-        });
+        await createGroupRef(bot, String(targetGroup)).leave(false);
         await event.reply("done");
       } catch (err) {
         await replyAdminErrorNotice({
@@ -382,15 +335,19 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           error: err,
         });
       }
-      return;
-    }
+    },
+  });
 
-    if (text.startsWith("/发好友")) {
-      if (!isMaster) {
-        ctx.logger.warn("[admin] 发好友功能仅主人可用");
-        return;
-      }
-      const matched = text.match(/^\/发好友\s*(\d+)\s*([\s\S]*)$/);
+  register({
+    name: "/发好友",
+    match: /^\/发好友(?:\s|$)/,
+    permission: "master",
+    description: "给好友发送私聊消息",
+    usage: "/发好友 qq号 内容",
+    handler: async ({ event, body }) => {
+      const bot = event.bot;
+      if (!bot) return;
+      const matched = body.match(/^\/发好友\s*(\d+)\s*([\s\S]*)$/);
       if (!matched) {
         await replyAdminErrorNotice({
           ctx,
@@ -420,7 +377,7 @@ export function registerPersonalCommands(ctx: MiokiContext) {
         return;
       }
       try {
-        await bot.sendPrivateMsg(targetUser, payload);
+        await bot.sendMessage({ type: "private", user_id: targetUser}, payload);
         await event.reply("done");
       } catch (err) {
         await replyAdminErrorNotice({
@@ -431,15 +388,19 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           error: err,
         });
       }
-      return;
-    }
+    },
+  });
 
-    if (text.startsWith("/发群聊")) {
-      if (!isMaster) {
-        ctx.logger.warn("[admin] 发群聊功能仅主人可用");
-        return;
-      }
-      const matched = text.match(/^\/发群聊\s*(\d+)\s*([\s\S]*)$/);
+  register({
+    name: "/发群聊",
+    match: /^\/发群聊(?:\s|$)/,
+    permission: "master",
+    description: "给群聊发送消息",
+    usage: "/发群聊 群号 内容",
+    handler: async ({ event, body }) => {
+      const bot = event.bot;
+      if (!bot) return;
+      const matched = body.match(/^\/发群聊\s*(\d+)\s*([\s\S]*)$/);
       if (!matched) {
         await replyAdminErrorNotice({
           ctx,
@@ -469,7 +430,7 @@ export function registerPersonalCommands(ctx: MiokiContext) {
         return;
       }
       try {
-        await bot.sendGroupMsg(targetGroup, payload);
+        await bot.sendMessage({ type: "group", group_id: targetGroup}, payload);
         await event.reply("done");
       } catch (err) {
         await replyAdminErrorNotice({
@@ -480,20 +441,23 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           error: err,
         });
       }
-      return;
-    }
+    },
+  });
 
-    if (text === "/全部好友") {
-      if (!isMaster) {
-        ctx.logger.warn("[admin] 全部好友功能仅主人可用");
-        return;
-      }
-      if (isGroup) {
+  register({
+    name: "/全部好友",
+    match: /^\/全部好友$/,
+    permission: "master",
+    description: "获取全部好友列表",
+    handler: async ({ event }) => {
+      const bot = event.bot;
+      if (!bot) return;
+      if (event.message_type === "group") {
         await event.reply("在私聊使用试试看吧～", true);
         return;
       }
       try {
-        const friendList: any[] = await bot.api("get_friend_list");
+        const friendList = await bot.getFriendList();
         if (!Array.isArray(friendList) || friendList.length === 0) {
           await replyAdminErrorNotice({
             ctx,
@@ -503,9 +467,9 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           });
           return;
         }
-        const nodes = friendList.map((friend: any) =>
-          ctx.segment.node({
-            user_id: String(friend.user_id),
+        const nodes = friendList.map((friend) =>
+          ctx.segment.raw("node", {
+            user_id: friend.user_id,
             nickname:
               friend.nickname || friend.remark || String(friend.user_id),
             content: [
@@ -518,11 +482,7 @@ export function registerPersonalCommands(ctx: MiokiContext) {
             ],
           }),
         );
-        await sendForwardByEvent({
-          bot,
-          event,
-          messages: toForwardMessages(bot, nodes),
-        });
+        await sendForwardByEvent({ bot, event, messages: nodes });
       } catch (err) {
         await replyAdminErrorNotice({
           ctx,
@@ -532,20 +492,23 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           error: err,
         });
       }
-      return;
-    }
+    },
+  });
 
-    if (text === "/全部群聊") {
-      if (!isMaster) {
-        ctx.logger.warn("[admin] 全部群聊功能仅主人可用");
-        return;
-      }
-      if (isGroup) {
+  register({
+    name: "/全部群聊",
+    match: /^\/全部群聊$/,
+    permission: "master",
+    description: "获取全部群聊列表",
+    handler: async ({ event }) => {
+      const bot = event.bot;
+      if (!bot) return;
+      if (event.message_type === "group") {
         await event.reply("在私聊使用试试看吧～", true);
         return;
       }
       try {
-        const groupList: any[] = await bot.api("get_group_list");
+        const groupList = await bot.getGroupList();
         if (!Array.isArray(groupList) || groupList.length === 0) {
           await replyAdminErrorNotice({
             ctx,
@@ -555,10 +518,10 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           });
           return;
         }
-        const nodes = groupList.map((group: any) =>
-          ctx.segment.node({
-            user_id: String(selfId),
-            nickname: String(selfId),
+        const nodes = groupList.map((group) =>
+          ctx.segment.raw("node", {
+            user_id: event.self_id ?? "",
+            nickname: String(event.self_id ?? ""),
             content: [
               ctx.segment.image(
                 `https://p.qlogo.cn/gh/${group.group_id}/${group.group_id}/640/`,
@@ -569,11 +532,7 @@ export function registerPersonalCommands(ctx: MiokiContext) {
             ],
           }),
         );
-        await sendForwardByEvent({
-          bot,
-          event,
-          messages: toForwardMessages(bot, nodes),
-        });
+        await sendForwardByEvent({ bot, event, messages: nodes });
       } catch (err) {
         await replyAdminErrorNotice({
           ctx,
@@ -583,7 +542,6 @@ export function registerPersonalCommands(ctx: MiokiContext) {
           error: err,
         });
       }
-      return;
-    }
+    },
   });
 }