kt-template-admin/apps/web-antdv-next/src/api/qqbot/index.ts

import type { Recordable } from '@vben/types';

import { encryptPassword } from '#/api/core/auth';
import { requestClient } from '#/api/request';

export namespace QqbotApi {
  export type PluginTriggerMode = 'command' | 'event';
  export type OneBotStatus = 'offline' | 'online';
  export type WebuiStatus = 'offline' | 'online' | 'unknown';
  export type QqLoginStatus =
    | 'offline'
    | 'online'
    | 'qrcode_expired'
    | 'qrcode_pending'
    | 'unknown';

  export interface PageResult<T> {
    list: T[];
    pageNo?: number;
    pageSize?: number;
    total: number;
  }

  export interface DashboardSummary {
    accountTotal: number;
    bus: {
      connected: boolean;
      mode: string;
      url: string;
    };
    conversationTotal: number;
    enabledRuleTotal: number;
    messageTotal: number;
    onlineTotal: number;
    runtime: {
      enabled: boolean;
      path: string;
      sessions: string[];
    };
    sendFailedTotal: number;
    sendSuccessTotal: number;
  }

  export interface Account {
    clientRole?: string;
    connectStatus: 'offline' | 'online';
    connectionMode: 'reverse-ws';
    containerStatus?: AccountNapcatRuntime['containerStatus'];
    createTime?: string;
    enabled: boolean;
    id: string;
    lastConnectedAt?: string;
    lastError?: string;
    lastHeartbeatAt?: string;
    name: string;
    napcat?: AccountNapcatRuntime | null;
    oneBotStatus?: OneBotStatus;
    qqLoginMessage?: null | string;
    qqLoginStatus?: QqLoginStatus;
    remark?: string;
    selfId: string;
    webuiStatus?: WebuiStatus;
  }

  export interface AccountNapcatRuntime {
    bindStatus?: 'bound' | 'disabled' | 'pending';
    containerId?: string;
    containerName?: string;
    containerOnline?: boolean;
    containerStatus?: 'creating' | 'error' | 'running' | 'stopped';
    profileStatus?: 'drift' | 'failed' | 'ok' | 'unknown';
    recoveryState?: 'idle' | 'password' | 'quick' | 'suspended';
    riskMode?: 'cooldown' | 'manual_only' | 'normal';
    runtimeProfile?: {
      desktopProfileVersion?: string;
      imageDigest?: string;
      imageRef?: string;
      locale?: string;
      shmSize?: string;
    };
    lastCheckedAt?: string;
    lastError?: string;
    lastLoginAt?: string;
    lastStartedAt?: string;
    oneBotOnline?: boolean;
    qqLoginMessage?: null | string;
    qqLoginStatus?: QqLoginStatus;
    webuiOnline?: boolean | null;
    webuiPort?: null | number;
  }

  export interface AccountBody {
    accessToken?: string;
    connectionMode?: 'reverse-ws';
    enabled?: boolean;
    encryptedLoginPassword?: string;
    id?: string;
    loginPassword?: string;
    name?: string;
    remark?: string;
    selfId: string;
  }

  export interface Rule {
    cooldownMs: number;
    enabled: boolean;
    id: string;
    keyword: string;
    lastHitAt?: string;
    matchType: 'equals' | 'keyword' | 'regex';
    name: string;
    priority: number;
    remark?: string;
    replyContent: string;
    targetType: 'all' | 'channel' | 'group' | 'private';
  }

  export interface RuleBody {
    cooldownMs?: number;
    enabled?: boolean;
    id?: string;
    keyword: string;
    matchType: 'equals' | 'keyword' | 'regex';
    name?: string;
    priority?: number;
    remark?: string;
    replyContent: string;
    targetType?: 'all' | 'channel' | 'group' | 'private';
  }

  export interface Conversation {
    createTime?: string;
    id: string;
    lastMessageText?: string;
    lastMessageTime?: string;
    messageCount: number;
    selfId: string;
    targetId: string;
    targetName?: string;
    targetType: 'channel' | 'group' | 'private';
  }

  export interface Message {
    direction: 'inbound' | 'outbound';
    eventTime: string;
    id: string;
    messageText: string;
    messageType: 'channel' | 'group' | 'private';
    senderNickname?: string;
    selfId: string;
    targetId: string;
    userId: string;
  }

  export interface SendLog {
    action: string;
    createTime?: string;
    errorMessage?: string;
    id: string;
    messageText: string;
    selfId: string;
    status: 'failed' | 'pending' | 'success';
    targetId: string;
    targetType: 'channel' | 'group' | 'private';
  }

  export interface PermissionConfig {
    allowlistEnabled: boolean;
    blocklistEnabled: boolean;
  }

  export interface Permission {
    enabled: boolean;
    id: string;
    preciseUser: boolean;
    remark?: string;
    selfId?: string;
    targetId: string;
    targetType: 'channel' | 'group' | 'private' | 'qq';
    userId?: string;
  }

  export interface PermissionBody {
    enabled?: boolean;
    id?: string;
    preciseUser?: boolean;
    remark?: string;
    selfId?: string;
    targetId: string;
    targetType: 'channel' | 'group' | 'private' | 'qq';
    userId?: string;
  }

  export interface Command {
    aliases: string[];
    code: string;
    cooldownMs: number;
    defaultParams?: Recordable<any>;
    enabled: boolean;
    errorTemplate?: string;
    id: string;
    lastHitAt?: string;
    name: string;
    operationKey: string;
    parserKey: 'ff14Price' | 'plain';
    pluginKey: string;
    prefixes: string[];
    priority: number;
    remark?: string;
    replyTemplate?: string;
    targetType: 'all' | 'channel' | 'group' | 'private';
  }

  export interface CommandBody {
    aliases?: string | string[];
    code: string;
    cooldownMs?: number;
    defaultParams?: Recordable<any> | string;
    enabled?: boolean;
    errorTemplate?: string;
    id?: string;
    name: string;
    operationKey: string;
    parserKey?: 'ff14Price' | 'plain';
    pluginKey: string;
    prefixes?: string | string[];
    priority?: number;
    remark?: string;
    replyTemplate?: string;
    targetType?: 'all' | 'channel' | 'group' | 'private';
  }

  export interface CommandTestResult {
    command?: Command;
    input?: Recordable<any>;
    matched: boolean;
    message?: string;
    output?: Recordable<any>;
    replyText?: string;
  }

  export interface Plugin {
    description?: string;
    key: string;
    name: string;
    operationCount: number;
    triggerMode: PluginTriggerMode;
    version: string;
  }

  export interface PluginOperation {
    cacheTtlMs?: number;
    description?: string;
    inputSchema?: Recordable<any>;
    key: string;
    name: string;
    outputSchema?: Recordable<any>;
    pluginKey: string;
    triggerMode: PluginTriggerMode;
  }

  export interface PluginOperationQuery extends Recordable<any> {
    pageNo?: number;
    pageSize?: number;
    pluginKey?: string;
    triggerMode?: PluginTriggerMode;
  }

  export interface PluginHealth {
    checkedAt: string;
    message?: string;
    name?: string;
    pluginKey?: string;
    status: 'degraded' | 'healthy' | 'offline';
    triggerMode?: PluginTriggerMode;
  }

  export interface EventPlugin {
    accountName?: string;
    bound: boolean;
    connectStatus?: string;
    description?: string;
    key: string;
    name: string;
    remark?: string;
    selfId: string;
    triggerType: 'message';
    version: string;
  }

  export type Query = Recordable<any>;
}

export function getQqbotDashboardSummary() {
  return requestClient.get<QqbotApi.DashboardSummary>(
    '/qqbot/dashboard/summary',
  );
}

export function getQqbotAccountList(params: QqbotApi.Query) {
  return requestClient.get<QqbotApi.PageResult<QqbotApi.Account>>(
    '/qqbot/account/list',
    { params },
  );
}

export function getQqbotEnabledAccounts() {
  return requestClient.get<QqbotApi.Account[]>('/qqbot/account/enabled');
}

export async function createQqbotAccount(data: QqbotApi.AccountBody) {
  return requestClient.post<string>(
    '/qqbot/account/save',
    await buildAccountRequest(data),
  );
}

export async function updateQqbotAccount(data: QqbotApi.AccountBody) {
  return requestClient.post<boolean>(
    '/qqbot/account/update',
    await buildAccountRequest(data),
  );
}

async function buildAccountRequest(data: QqbotApi.AccountBody) {
  const { loginPassword, ...payload } = data;
  const password =
    loginPassword === undefined || loginPassword === null
      ? ''
      : `${loginPassword}`;
  if (password.trim()) {
    payload.encryptedLoginPassword = await encryptPassword(password);
  }
  return payload;
}

export function deleteQqbotAccount(id: string) {
  return requestClient.post<{ deletedContainers: number }>(
    `/qqbot/account/delete?id=${id}`,
  );
}

export function bindQqbotAccountCommand(selfId: string, commandId: string) {
  const params = new URLSearchParams({ commandId, selfId });
  return requestClient.post<boolean>(`/qqbot/account/bind/command?${params}`);
}

export function unbindQqbotAccountCommand(selfId: string, commandId: string) {
  const params = new URLSearchParams({ commandId, selfId });
  return requestClient.post<boolean>(`/qqbot/account/unbind/command?${params}`);
}

export function bindQqbotAccountRule(selfId: string, ruleId: string) {
  const params = new URLSearchParams({ ruleId, selfId });
  return requestClient.post<boolean>(`/qqbot/account/bind/rule?${params}`);
}

export function unbindQqbotAccountRule(selfId: string, ruleId: string) {
  const params = new URLSearchParams({ ruleId, selfId });
  return requestClient.post<boolean>(`/qqbot/account/unbind/rule?${params}`);
}

export function kickQqbotAccount(selfId: string) {
  return requestClient.post<{ count: number }>(
    `/qqbot/account/kick?selfId=${selfId}`,
  );
}

export function getQqbotRuleList(params: QqbotApi.Query) {
  return requestClient.get<QqbotApi.PageResult<QqbotApi.Rule>>(
    '/qqbot/rule/list',
    { params },
  );
}

export function createQqbotRule(data: QqbotApi.RuleBody) {
  return requestClient.post<string>('/qqbot/rule/save', data);
}

export function updateQqbotRule(data: QqbotApi.RuleBody) {
  return requestClient.post<boolean>('/qqbot/rule/update', data);
}

export function deleteQqbotRule(id: string) {
  return requestClient.post<boolean>(`/qqbot/rule/delete?id=${id}`);
}

export function toggleQqbotRule(id: string, enabled: boolean) {
  return requestClient.post<boolean>(
    `/qqbot/rule/toggle?id=${id}&enabled=${enabled}`,
  );
}

export function getQqbotConversationList(params: QqbotApi.Query) {
  return requestClient.get<QqbotApi.PageResult<QqbotApi.Conversation>>(
    '/qqbot/conversation/list',
    { params },
  );
}

export function getQqbotMessageList(params: QqbotApi.Query) {
  return requestClient.get<QqbotApi.PageResult<QqbotApi.Message>>(
    '/qqbot/message/list',
    { params },
  );
}

export function getQqbotSendLogList(params: QqbotApi.Query) {
  return requestClient.get<QqbotApi.PageResult<QqbotApi.SendLog>>(
    '/qqbot/send/log/list',
    { params },
  );
}

export function sendQqbotPrivate(data: {
  message: string;
  selfId?: string;
  userId: string;
}) {
  return requestClient.post('/qqbot/send/private', data);
}

export function sendQqbotGroup(data: {
  groupId: string;
  message: string;
  selfId?: string;
}) {
  return requestClient.post('/qqbot/send/group', data);
}

export function getQqbotPermissionList(
  kind: 'allowlist' | 'blocklist',
  params: QqbotApi.Query,
) {
  return requestClient.get<QqbotApi.PageResult<QqbotApi.Permission>>(
    `/qqbot/permission/${kind}`,
    { params },
  );
}

export function getQqbotPermissionConfig() {
  return requestClient.get<QqbotApi.PermissionConfig>(
    '/qqbot/permission/config',
  );
}

export function updateQqbotPermissionConfig(
  data: Partial<QqbotApi.PermissionConfig>,
) {
  return requestClient.post<QqbotApi.PermissionConfig>(
    '/qqbot/permission/config',
    data,
  );
}

export function createQqbotPermission(
  kind: 'allowlist' | 'blocklist',
  data: QqbotApi.PermissionBody,
) {
  return requestClient.post<string>(`/qqbot/permission/${kind}/save`, data);
}

export function updateQqbotPermission(
  kind: 'allowlist' | 'blocklist',
  data: QqbotApi.PermissionBody,
) {
  return requestClient.post<boolean>(`/qqbot/permission/${kind}/update`, data);
}

export function deleteQqbotPermission(
  kind: 'allowlist' | 'blocklist',
  id: string,
) {
  return requestClient.post<boolean>(
    `/qqbot/permission/${kind}/delete?id=${id}`,
  );
}

export function getQqbotCommandList(params: QqbotApi.Query) {
  return requestClient.get<QqbotApi.PageResult<QqbotApi.Command>>(
    '/qqbot/command/list',
    { params },
  );
}

export function createQqbotCommand(data: QqbotApi.CommandBody) {
  return requestClient.post<string>('/qqbot/command/save', data);
}

export function updateQqbotCommand(data: QqbotApi.CommandBody) {
  return requestClient.post<boolean>('/qqbot/command/update', data);
}

export function deleteQqbotCommand(id: string) {
  return requestClient.post<boolean>(`/qqbot/command/delete?id=${id}`);
}

export function toggleQqbotCommand(id: string, enabled: boolean) {
  return requestClient.post<boolean>(
    `/qqbot/command/toggle?id=${id}&enabled=${enabled}`,
  );
}

export function testQqbotCommand(data: {
  commandId?: string;
  selfId?: string;
  targetId?: string;
  targetType?: 'channel' | 'group' | 'private';
  text: string;
  userId?: string;
}) {
  return requestClient.post<QqbotApi.CommandTestResult>(
    '/qqbot/command/test',
    data,
  );
}