NapCatQQ source snapshot

packages/napcat-onebot/helper/cqcode.ts

@@ -0,0 +1,91 @@
+import { OB11MessageData } from '@/napcat-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;
+}