LLOneBot/src/onebot11/cqcode.ts

import {OB11MessageData} from "./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, any> = {}
    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 h(type: string, data: any) {
    return {
        type,
        data,
    }
}

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


export function encodeCQCode(data: OB11MessageData) {
    const CQCodeEscapeText = (text: string) => {
        return text.replace(/\&/g, '&amp;')
            .replace(/\[/g, '&#91;')
            .replace(/\]/g, '&#93;')
            
    };

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

    if (data.type === 'text') {
        return CQCodeEscapeText(data.data.text);
    }

    let result = '[CQ:' + data.type;
    for (const name in data.data) {
        const value = data.data[name];
        result += `,${name}=${CQCodeEscape(value)}`;
    }
    result += ']';
    return result;
}

// const result = parseCQCode("[CQ:at,qq=114514]早上好啊[CQ:image,file=http://baidu.com/1.jpg,type=show,id=40004]")
// const result = parseCQCode("好好好")
// console.log(JSON.stringify(result))