refactor: make quick action apis instance methods

2024-11-21 09:36:35 +00:00 · 2024-08-26 14:14:49 +08:00 · 2024-08-26 14:14:49 +08:00 · 4f98c0d045
commit 4f98c0d045
parent c254441d40
1 changed files with 95 additions and 0 deletions
--- a/src/onebot/api/quick-action.ts
+++ b/src/onebot/api/quick-action.ts
@ -0,0 +1,95 @@
+import {
+    NapCatOneBot11Adapter,
+    OB11Message, OB11MessageAt,
+    OB11MessageData,
+    OB11MessageReply,
+    QuickAction,
+    QuickActionEvent, QuickActionFriendRequest, QuickActionGroupMessage, QuickActionGroupRequest,
+} from '@/onebot';
+import { ChatType, GroupRequestOperateTypes, NapCatCore, Peer } from '@/core';
+import { OB11FriendRequestEvent } from '@/onebot/event/request/OB11FriendRequest';
+import { OB11GroupRequestEvent } from '@/onebot/event/request/OB11GroupRequest';
+import { normalize } from '@/onebot/action/msg/SendMsg';
+import { isNull } from '@/common/utils/helper';
+
+export class OneBotQuickActionApi {
+    constructor(
+        public obContext: NapCatOneBot11Adapter,
+        public core: NapCatCore
+    ) {}
+
+    handleQuickOperation(context: QuickActionEvent, quickAction: QuickAction) {
+        if (context.post_type === 'message') {
+            this.handleMsg(context as OB11Message, quickAction)
+                .catch(this.core.context.logger.logError);
+        }
+        if (context.post_type === 'request') {
+            const friendRequest = context as OB11FriendRequestEvent;
+            const groupRequest = context as OB11GroupRequestEvent;
+            if ((friendRequest).request_type === 'friend') {
+                this.handleFriendRequest(friendRequest, quickAction)
+                    .catch(this.core.context.logger.logError);
+            } else if (groupRequest.request_type === 'group') {
+                this.handleGroupRequest(groupRequest, quickAction)
+                    .catch(this.core.context.logger.logError);
+            }
+        }
+    }
+
+    async handleMsg(msg: OB11Message, quickAction: QuickAction) {
+        const reply = quickAction.reply;
+        const peer: Peer = {
+            chatType: ChatType.KCHATTYPEC2C,
+            peerUid: await this.core.apis.UserApi.getUidByUinV2(msg.user_id.toString()) as string,
+        };
+        if (msg.message_type == 'private') {
+            if (msg.sub_type === 'group') {
+                peer.chatType = ChatType.KCHATTYPETEMPC2CFROMGROUP;
+            }
+        } else {
+            peer.chatType = ChatType.KCHATTYPETEMPC2CFROMGROUP;
+            peer.peerUid = msg.group_id!.toString();
+        }
+        if (reply) {
+            // let group: Group | undefined;
+            let replyMessage: OB11MessageData[] = [];
+
+            if (msg.message_type == 'group') {
+                // group = await core.apis.GroupApi.getGroup(msg.group_id!.toString());
+                replyMessage.push({
+                    type: 'reply',
+                    data: {
+                        id: msg.message_id.toString(),
+                    },
+                } as OB11MessageReply);
+                if ((quickAction as QuickActionGroupMessage).at_sender) {
+                    replyMessage.push({
+                        type: 'at',
+                        data: {
+                            qq: msg.user_id.toString(),
+                        },
+                    } as OB11MessageAt);
+                }
+            }
+            replyMessage = replyMessage.concat(normalize(reply, quickAction.auto_escape));
+            const { sendElements, deleteAfterSentFiles } = await this.obContext.apis.MsgApi.createSendElements(replyMessage, peer);
+            this.obContext.apis.MsgApi.sendMsgWithOb11UniqueId(peer, sendElements, deleteAfterSentFiles, false).then().catch(this.core.context.logger.logError);
+        }
+    }
+
+    async handleGroupRequest(request: OB11GroupRequestEvent, quickAction: QuickActionGroupRequest) {
+        if (!isNull(quickAction.approve)) {
+            this.core.apis.GroupApi.handleGroupRequest(
+                request.flag,
+                quickAction.approve ? GroupRequestOperateTypes.approve : GroupRequestOperateTypes.reject,
+                quickAction.reason,
+            ).catch(this.core.context.logger.logError);
+        }
+    }
+
+    async handleFriendRequest(request: OB11FriendRequestEvent, quickAction: QuickActionFriendRequest) {
+        if (!isNull(quickAction.approve)) {
+            this.core.apis.FriendApi.handleFriendRequest(request.flag, !!quickAction.approve).then().catch(this.core.context.logger.logError);
+        }
+    }
+}