NapCatQQ/src/onebot11/log.ts

import { OB11Message } from '@/onebot11/types';
import { log } from '@/common/utils/log';
import { getGroup, getGroupMember, selfInfo } from '@/core/data';
import exp from 'constants';
import { Group } from '@/core';
import chalk from 'chalk';

const spSegColor = chalk.blue;// for special segment
const spColor = chalk.cyan;// for special

// todo: 应该放到core去用RawMessage解析打印
export async function logMessage(ob11Message: OB11Message) {
  const isSelfSent = ob11Message.sender.user_id.toString() === selfInfo.uin;
  let prefix = '';
  let group: Group | undefined;
  if (isSelfSent) {
    prefix = '发送消息 ';
    if (ob11Message.message_type === 'private') {
      prefix += '给私聊 ';
      prefix += `${ob11Message.target_id}`;
    }
    else {
      prefix += '给群聊 ';
    }
  }
  if (ob11Message.message_type === 'group') {
    group = await getGroup(ob11Message.group_id!);
    prefix += `群[${group?.groupName}(${ob11Message.group_id})] `;
  }
  let msgChain = '';
  if (Array.isArray(ob11Message.message)) {
    const msgParts = [];
    for (const segment of ob11Message.message) {
      if (segment.type === 'text') {
        msgParts.push(segment.data.text);
      }
      else if (segment.type === 'at') {
        const groupMember = await getGroupMember(ob11Message.group_id!, segment.data.qq!);
        msgParts.push(spSegColor(`[@${groupMember?.cardName || groupMember?.nick}(${segment.data.qq})]`));
      }
      else if (segment.type === 'reply') {
        msgParts.push(spSegColor(`[回复消息|id:${segment.data.id}]`));
      }
      else if (segment.type === 'image') {
        msgParts.push(spSegColor(`[图片|${segment.data.url}]`));
      }
      else if (segment.type === 'face') {
        msgParts.push(spSegColor(`[表情|id:${segment.data.id}]`));
      }
      else if (segment.type === 'mface') {
        // @ts-expect-error 商城表情 url
        msgParts.push(spSegColor(`[商城表情|${segment.data.url}]`));
      }
      else if (segment.type === 'record') {
        msgParts.push(spSegColor(`[语音|${segment.data.file}]`));
      }
      else if (segment.type === 'file') {
        msgParts.push(spSegColor(`[文件|${segment.data.file}]`));
      }
      else if (segment.type === 'json') {
        msgParts.push(spSegColor(`[json|${JSON.stringify(segment.data)}]`));
      }
      else if (segment.type === 'markdown') {
        msgParts.push(spSegColor(`[markdown|${segment.data.content}]`));
      }
      else {
        msgParts.push(spSegColor(`[未实现|${JSON.stringify(segment)}]`));
      }
    }
    msgChain = msgParts.join(' ');
  }
  else {
    msgChain = ob11Message.message;
  }
  let msgString = `${prefix}${ob11Message.sender.nickname}(${ob11Message.sender.user_id}): ${msgChain}`;
  if (isSelfSent) {
    msgString = `${prefix}: ${msgChain}`;
  }
  log(msgString);
}

export async function logNotice(ob11Notice: any) {
  log(spColor('[Notice]'), ob11Notice);
}

export async function logRequest(ob11Request: any) {
  log(spColor('[Request]'), ob11Request);
}