import { pinyin } from 'pinyin-pro';

const slugify = (s) => {
    s = s?.replace(/[#&'()+,/:[\]|’“”、「」・（）．：｜]/g, '-');
    return encodeURIComponent(
        pinyin(s, {
            nonZh: 'consecutive',
            toneType: 'none',
            type: 'array',
            v: true,
        })
            .join(' ')
            .trim()
            .toLowerCase()
            .replace(/\s+/g, '-')
            .replace(/-+/g, '-')
            .replace(/(^-|-$)/g, '')
    );
};

export default {
    rules: (list) => list.filter((e) => e),
    handler: async (data) => {
        const content = data.split('\n');
        let lastH2 = '';
        let lastH3 = '';

        for (let i = 0; i < content.length; i++) {
            if (content[i].startsWith('## ')) {
                lastH2 = content[i].match(`## (([^{])*)`)?.[1].trim();

                content[i] = `## ${lastH2} {#${slugify(lastH2)}}`;
            } else if (content[i].startsWith('### ')) {
                lastH3 = content[i].match(`### (([^{])*)`)?.[1].trim();

                content[i] = `### ${lastH3} {#${slugify(lastH2)}-${slugify(lastH3)}}`;
            } else if (content[i].startsWith('#### ')) {
                const title = content[i].match(`#### (([^{])*)`)?.[1].trim();

                content[i] = `#### ${title} {#${slugify(lastH2)}-${slugify(lastH3)}-${slugify(title)}}`;
            }
        }

        return Promise.resolve(content.join('\n'));
    },
};