window.browser = window.browser || window.chrome

import utils from "./utils.js"

const targets = /^https?:\/{2}([im]\.)?imgur\.(com|io)(\/|$)/

const frontends = new Array("rimgo")
const protocols = new Array("normal", "tor", "i2p", "loki")

let redirects = {}

for (let i = 0; i < frontends.length; i++) {
	redirects[frontends[i]] = {}
	for (let x = 0; x < protocols.length; x++) {
		redirects[frontends[i]][protocols[x]] = []
	}
}

function setRedirects(val) {
	browser.storage.local.get(["cloudflareBlackList", "offlineBlackList"], r => {
		redirects.rimgo = val
		rimgoNormalRedirectsChecks = [...redirects.rimgo.normal]
		for (const instance of [...r.cloudflareBlackList, ...r.offlineBlackList]) {
			const a = rimgoNormalRedirectsChecks.indexOf(instance)
			if (a > -1) rimgoNormalRedirectsChecks.splice(a, 1)
		}
		browser.storage.local.set({
			imgurRedirects: redirects,
			rimgoNormalRedirectsChecks,
			rimgoTorRedirectsChecks: [...redirects.rimgo.tor],
			rimgoI2pRedirectsChecks: [...redirects.rimgo.i2p],
			rimgoLokiRedirectsChecks: [...redirects.rimgo.loki],
		})
	})
}

let disableImgur,
	imgurRedirects,
	protocol,
	protocolFallback,
	rimgoNormalRedirectsChecks,
	rimgoNormalCustomRedirects,
	rimgoTorRedirectsChecks,
	rimgoTorCustomRedirects,
	rimgoI2pRedirectsChecks,
	rimgoI2pCustomRedirects,
	rimgoLokiCustomRedirects

function init() {
	return new Promise(async resolve => {
		browser.storage.local.get(
			[
				"disableImgur",
				"imgurRedirects",
				"protocol",
				"protocolFallback",
				"rimgoNormalRedirectsChecks",
				"rimgoNormalCustomRedirects",
				"rimgoTorRedirectsChecks",
				"rimgoTorCustomRedirects",
				"rimgoI2pRedirectsChecks",
				"rimgoI2pCustomRedirects",
				"rimgoLokiCustomRedirects",
			],
			r => {
				disableImgur = r.disableImgur
				imgurRedirects = r.imgurRedirects
				protocol = r.protocol
				protocolFallback = r.protocolFallback
				rimgoNormalRedirectsChecks = r.rimgoNormalRedirectsChecks
				rimgoNormalCustomRedirects = r.rimgoNormalCustomRedirects
				rimgoTorRedirectsChecks = r.rimgoTorRedirectsChecks
				rimgoTorCustomRedirects = r.rimgoTorCustomRedirects
				rimgoI2pRedirectsChecks = r.rimgoI2pRedirectsChecks
				rimgoI2pCustomRedirects = r.rimgoI2pCustomRedirects
				rimgoLokiCustomRedirects = r.rimgoLokiCustomRedirects
				resolve()
			}
		)
	})
}

init()
browser.storage.onChanged.addListener(init)

// https://imgur.com/gallery/s4WXQmn
// https://imgur.com/a/H8M4rcp
// https://imgur.com/gallery/gYiQLWy
// https://imgur.com/gallery/cTRwaJU
// https://i.imgur.com/CFSQArP.jpeg

function all() {
	return [
		...imgurRedirects.rimgo.normal,
		...imgurRedirects.rimgo.tor,
		...imgurRedirects.rimgo.i2p,
		...rimgoNormalCustomRedirects,
		...rimgoTorCustomRedirects,
		...rimgoI2pCustomRedirects,
		...rimgoLokiCustomRedirects,
	]
}

function redirect(url, type, initiator, disableOverride) {
	if (disableImgur && !disableOverride) return
	if (url.pathname == "/" && !disableOverride) return
	if (!["main_frame", "sub_frame", "xmlhttprequest", "other", "image", "media"].includes(type)) return
	if (initiator && (all().includes(initiator.origin) || targets.test(initiator.host))) return
	if (!targets.test(url.href)) return
	if (url.pathname.includes("delete/")) return

	let instancesList = []
	if (protocol == "loki") instancesList = [...rimgoLokiCustomRedirects]
	else if (protocol == "i2p") instancesList = [...rimgoI2pCustomRedirects, ...rimgoI2pRedirectsChecks]
	else if (protocol == "tor") instancesList = [...rimgoTorRedirectsChecks, ...rimgoTorCustomRedirects]
	if ((instancesList.length === 0 && protocolFallback) || protocol == "normal") {
		instancesList = [...rimgoNormalRedirectsChecks, ...rimgoNormalCustomRedirects]
	}
	if (instancesList.length === 0) {
		return
	}

	const randomInstance = utils.getRandomInstance(instancesList)
	return `${randomInstance}${url.pathname}${url.search}`
}

function reverse(url) {
	return new Promise(async resolve => {
		await init()
		const protocolHost = utils.protocolHost(url)
		if (!all().includes(protocolHost)) {
			resolve()
			return
		}
		resolve(`https://imgur.com${url.pathname}${url.search}`)
	})
}

function switchInstance(url, disableOverride) {
	return new Promise(async resolve => {
		await init()
		if (disableImgur && !disableOverride) {
			resolve()
			return
		}
		let protocolHost = utils.protocolHost(url)
		if (!all().includes(protocolHost)) {
			resolve()
			return
		}
		let instancesList = []
		if (protocol == "loki") instancesList = [...rimgoLokiCustomRedirects]
		else if (protocol == "i2p") instancesList = [...rimgoI2pCustomRedirects, ...rimgoI2pRedirectsChecks]
		else if (protocol == "tor") instancesList = [...rimgoTorRedirectsChecks, ...rimgoTorCustomRedirects]
		if ((instancesList.length === 0 && protocolFallback) || protocol == "normal") {
			instancesList = [...rimgoNormalRedirectsChecks, ...rimgoNormalCustomRedirects]
		}

		const i = instancesList.indexOf(protocolHost)
		if (i > -1) instancesList.splice(i, 1)
		if (instancesList.length === 0) {
			resolve()
			return
		}

		const randomInstance = utils.getRandomInstance(instancesList)
		resolve(`${randomInstance}${url.pathname}${url.search}`)
	})
}

function initDefaults() {
	return new Promise(resolve => {
		fetch("/instances/data.json")
			.then(response => response.text())
			.then(async data => {
				let dataJson = JSON.parse(data)
				for (let i = 0; i < frontends.length; i++) {
					redirects[frontends[i]] = dataJson[frontends[i]]
				}
				browser.storage.local.get(["cloudflareBlackList", "offlineBlackList"], async r => {
					rimgoNormalRedirectsChecks = [...redirects.rimgo.normal]
					for (const instance of [...r.cloudflareBlackList, ...r.offlineBlackList]) {
						const a = rimgoNormalRedirectsChecks.indexOf(instance)
						if (a > -1) rimgoNormalRedirectsChecks.splice(a, 1)
					}
					browser.storage.local.set(
						{
							disableImgur: false,
							imgurRedirects: redirects,

							rimgoNormalRedirectsChecks: rimgoNormalRedirectsChecks,
							rimgoNormalCustomRedirects: [],

							rimgoTorRedirectsChecks: [...redirects.rimgo.tor],
							rimgoTorCustomRedirects: [],

							rimgoI2pRedirectsChecks: [...redirects.rimgo.i2p],
							rimgoI2pCustomRedirects: [],

							rimgoLokiRedirectsChecks: [...redirects.rimgo.loki],
							rimgoLokiCustomRedirects: [],
						},
						() => resolve()
					)
				})
			})
	})
}

export default {
	setRedirects,
	redirect,
	reverse,
	initDefaults,
	switchInstance,
}