import { marked } from 'marked'
import Highlight, { defaultProps, Language, } from 'prism-react-renderer'
import { renderToStaticMarkup } from 'react-dom/server'

//@ts-ignore
delete defaultProps.theme
// import linkStyles from '../components/link/link.module.css'

const renderer = new marked.Renderer()

const convertHtmlEntities = (str: string) => {
    const quot = '"'
    const apos = '''
    const amp = '&'
    const nbsp = ' '
    const lt = '<'
    const gt = '>'
    const code = '<code>'
    const endCode = '</code>'
    const combinedRegex = new RegExp(`${code}|${endCode}|${quot}|${apos}|${amp}|${nbsp}|${lt}|${gt}`, 'g')

    return str.replace(combinedRegex, (match) => {
        switch (match) {
            case quot:
                return '"'
            case apos:
                return "'"
            case amp:
                return '&'
            case nbsp:
                return ' '
            case lt:
                return '<'
            case gt:
                return '>'
            case code:
            case endCode:
                return '`'
            default:
                return match
        }
    })
}

renderer.heading = (text, level, _, slugger) => {
    const id = slugger.slug(text)
    const Component = `h${level}`



    return renderToStaticMarkup(
        //@ts-ignore
        <Component>
            <a href={`#${id}`} id={id} style={{ color: "inherit" }} >
                {convertHtmlEntities(text)}
            </a>
        </Component>
    )
}

// renderer.link = (href, _, text) => {
//     const isHrefLocal = href?.startsWith('/') || href?.startsWith('#')
//     if (isHrefLocal) {
//         return renderToStaticMarkup(
//             <a href={href || ''}>
//                 {text}
//             </a>
//         )
//     }

//     // dirty hack
//     // if text contains elements, render as html
//     return `<a href='${href}' target="_blank" rel="noopener noreferrer">${convertHtmlEntities(text)}</a>`
// }


renderer.image = function (href, _, text) {
    return `<Image loading="lazy" src="${href}" alt="${text}" layout="fill" />`
}

renderer.checkbox = () => ''
renderer.listitem = (text, task, checked) => {
    if (task) {
        return `<li class="reset"><span class="check">&#8203;<input type="checkbox" disabled ${checked ? 'checked' : ''
            } /></span><span>${text}</span></li>`
    }

    return `<li>${text}</li>`
}

renderer.code = (code: string, language: string) => {
    return renderToStaticMarkup(
        <pre>
            {/* {title && <code>{title} </code>} */}
            {/* {language && title && <code style={{}}> {language} </code>} */}
            <Code
                language={language}
                // title={title}
                code={code}
            // highlight={highlight}
            />
        </pre>
    )
}

marked.setOptions({
    gfm: true,
    breaks: true,
    headerIds: true,
    renderer,
})

const markdown = (markdown: string) => marked(markdown)

export default markdown

const Code = ({ code, language, highlight, title, ...props }: {
    code: string,
    language: string,
    highlight?: string,
    title?: string,
}) => {
    if (!language)
        return (
            <>
                <code {...props} dangerouslySetInnerHTML={{ __html: code }
                } />
            </>
        )

    const highlightedLines = highlight
        //@ts-ignore
        ? highlight.split(',').reduce((lines, h) => {
            if (h.includes('-')) {
                // Expand ranges like 3-5 into [3,4,5]
                const [start, end] = h.split('-').map(Number)
                const x = Array(end - start + 1)
                    .fill(undefined)
                    .map((_, i) => i + start)
                return [...lines, ...x]
            }

            return [...lines, Number(h)]
        }, [])
        : ''

    // https://mdxjs.com/guides/syntax-harkedighlighting#all-together
    return (
        <>
            <Highlight {...defaultProps} code={code.trim()} language={language as Language} >
                {({ className, style, tokens, getLineProps, getTokenProps }) => (
                    <code className={className} style={{ ...style }}>
                        {
                            tokens.map((line, i) => (
                                <div
                                    key={i}
                                    {...getLineProps({ line, key: i })}
                                    style={
                                        //@ts-ignore
                                        highlightedLines.includes((i + 1).toString())
                                            ? {
                                                background: 'var(--highlight)',
                                                margin: '0 -1rem',
                                                padding: '0 1rem',
                                            }
                                            : undefined
                                    }
                                >
                                    {
                                        line.map((token, key) => (
                                            <span key={key} {...getTokenProps({ token, key })} />
                                        ))
                                    }
                                </div>
                            ))}
                    </code>
                )}
            </Highlight>
        </>
    )
}