import MarkdownIt from 'markdown-it'
import { escapeAttrValue, FilterXSS, safeAttrValue, whiteList } from 'xss'

export const configuredXss = new FilterXSS({
	whiteList: {
		...whiteList,
		summary: [],
		h1: ['id'],
		h2: ['id'],
		h3: ['id'],
		h4: ['id'],
		h5: ['id'],
		h6: ['id'],
		kbd: ['id'],
		input: ['checked', 'disabled', 'type'],
		iframe: ['width', 'height', 'allowfullscreen', 'frameborder', 'start', 'end'],
		img: [...(whiteList.img || []), 'usemap', 'style', 'align'],
		map: ['name'],
		area: [...(whiteList.a || []), 'coords'],
		a: [...(whiteList.a || []), 'rel'],
		td: [...(whiteList.td || []), 'style'],
		th: [...(whiteList.th || []), 'style'],
		picture: [],
		source: ['media', 'sizes', 'src', 'srcset', 'type'],
		p: [...(whiteList.p || []), 'align'],
		div: [...(whiteList.p || []), 'align'],
	},
	css: {
		whiteList: {
			'image-rendering': /^pixelated$/,
			'text-align': /^center|left|right$/,
			float: /^left|right$/,
		},
	},
	onIgnoreTagAttr: (tag, name, value) => {
		// Allow iframes from acceptable sources
		if (tag === 'iframe' && name === 'src') {
			const allowedSources = [
				{
					url: /^https?:\/\/(www\.)?youtube(-nocookie)?\.com\/embed\/[a-zA-Z0-9_-]{11}/,
					allowedParameters: [/start=\d+/, /end=\d+/],
				},
				{
					url: /^https?:\/\/(www\.)?discord\.com\/widget/,
					allowedParameters: [/id=\d{18,19}/],
				},
			]

			const url = new URL(value)

			for (const source of allowedSources) {
				if (!source.url.test(url.href)) {
					continue
				}

				const newSearchParams = new URLSearchParams()
				url.searchParams.forEach((value, key) => {
					if (!source.allowedParameters.some((param) => param.test(`${key}=${value}`))) {
						newSearchParams.delete(key)
					}
				})

				url.search = newSearchParams.toString()
				return `${name}="${escapeAttrValue(url.toString())}"`
			}
		}

		// For Highlight.JS
		if (name === 'class' && ['pre', 'code', 'span'].includes(tag)) {
			const allowedClasses: string[] = []
			for (const className of value.split(/\s/g)) {
				if (className.startsWith('hljs-') || className.startsWith('language-')) {
					allowedClasses.push(className)
				}
			}
			return `${name}="${escapeAttrValue(allowedClasses.join(' '))}"`
		}
	},
	safeAttrValue(tag, name, value, cssFilter) {
		if (
			(tag === 'img' || tag === 'video' || tag === 'audio' || tag === 'source') &&
			(name === 'src' || name === 'srcset') &&
			!value.startsWith('data:')
		) {
			try {
				const url = new URL(value)

				if (url.hostname.includes('wsrv.nl')) {
					url.searchParams.delete('errorredirect')
					url.searchParams.delete('default')
				}

				const allowedHostnames = [
					'imgur.com',
					'i.imgur.com',
					'cdn-raw.modrinth.com',
					'cdn.modrinth.com',
					'staging-cdn-raw.modrinth.com',
					'staging-cdn.modrinth.com',
					'github.com',
					'raw.githubusercontent.com',
					'img.shields.io',
					'i.postimg.cc',
					'wsrv.nl',
					'cf.way2muchnoise.eu',
					'bstats.org',
				]

				if (!allowedHostnames.includes(url.hostname)) {
					return safeAttrValue(
						tag,
						name,
						`https://wsrv.nl/?url=${encodeURIComponent(
							url.toString().replaceAll('&amp;', '&'),
						)}&n=-1`,
						cssFilter,
					)
				}
				return safeAttrValue(tag, name, url.toString(), cssFilter)
			} catch {
				/* empty */
			}
		}

		return safeAttrValue(tag, name, value, cssFilter)
	},
})

export const md = (options = {}) => {
	const md = new MarkdownIt('default', {
		html: true,
		linkify: true,
		breaks: false,
		...options,
	})

	const defaultLinkOpenRenderer =
		md.renderer.rules.link_open ||
		function (tokens, idx, options, _env, self) {
			return self.renderToken(tokens, idx, options)
		}

	md.linkify.set({
		fuzzyLink: false,
		fuzzyIP: false,
	})

	md.renderer.rules.link_open = function (tokens, idx, options, env, self) {
		const token = tokens[idx]
		const index = token.attrIndex('href')

		if (token.attrs && index !== -1) {
			const href = token.attrs[index][1]

			try {
				const url = new URL(href)
				const allowedHostnames = ['modrinth.com']

				if (allowedHostnames.includes(url.hostname)) {
					return defaultLinkOpenRenderer(tokens, idx, options, env, self)
				}
			} catch {
				/* empty */
			}
		}

		tokens[idx].attrSet('rel', 'noopener nofollow ugc')

		return defaultLinkOpenRenderer(tokens, idx, options, env, self)
	}

	return md
}

export const renderString = (string: string) => configuredXss.process(md().render(string))
