SnowCatQQ/packages/snowcat-onebot/helper/cqcode.ts

import { OB11MessageData } from '@/snowcat-onebot/types';

const pattern = /\[CQ:(\w+)((,\w+=[^,\]]*)*)]/;

function unescape (source: string) {
  return String(source)
    .replace(/[/g, '[')
    .replace(/]/g, ']')
    .replace(/,/g, ',')
    .replace(/&/g, '&');
}

function from (source: string) {
  const capture = pattern.exec(source);
  if (!capture) return null;
  const [, type, attrs] = capture;
  const data: Record<string, unknown> = {};
  if (attrs) {
    attrs.slice(1).split(',').forEach((str) => {
      const index = str.indexOf('=');
      data[str.slice(0, index)] = unescape(str.slice(index + 1));
    });
  }
  return { type, data, capture };
}

function convert (type: string, data: unknown) {
  return {
    type,
    data,
  };
}

export function decodeCQCode (source: string): OB11MessageData[] {
  const elements: Array<unknown> = [];
  let result: ReturnType<typeof from>;
  while ((result = from(source))) {
    const { type, data, capture } = result;
    if (type) {
      if (capture.index) {
        elements.push(convert('text', { text: unescape(source.slice(0, capture.index)) }));
      }
      elements.push(convert(type, data));
      source = source.slice(capture.index + capture[0].length);
    }
  }
  if (source) elements.push(convert('text', { text: unescape(source) }));
  return elements as OB11MessageData[];
}

function CQCodeEscapeText (text: string) {
  return text.replace(/&/g, '&amp;')
    .replace(/\[/g, '&#91;')
    .replace(/]/g, '&#93;');
}

function CQCodeEscape (text: string) {
  return text.replace(/&/g, '&amp;')
    .replace(/\[/g, '&#91;')
    .replace(/]/g, '&#93;')
    .replace(/,/g, '&#44;');
}

export function encodeCQCode (data: OB11MessageData) {
  if (data.type === 'text') {
    return CQCodeEscapeText(data.data.text);
  }

  let result = '[CQ:' + data.type;
  for (const name in data.data) {
    const value = (data.data as Record<string, unknown>)[name];
    if (value === undefined) {
      continue;
    }
    try {
      let text: string | undefined;
      if (typeof value === 'object' && value?.toString === Object.prototype.toString) {
        text = JSON.stringify(value);
      } else {
        text = value?.toString();
      }
      if (text) {
        result += `,${name}=${CQCodeEscape(text)}`;
      }
    } catch (error) {
      console.error(`Error encoding CQCode for ${name}:`, error);
    }
  }
  result += ']';
  return result;
}