import './shared';

import { getMatch, type Match } from '~/lib/match';
import { Redirect, UrlReferer } from '~/lib/settings';

chrome.webRequest.onBeforeSendHeaders.addListener(
	// eslint-disable-next-line @typescript-eslint/ban-ts-comment
	// @ts-ignore
	async (details) => {
		const referer = await UrlReferer.get(new URL(details.url).hostname);
		if (!referer) return;

		await UrlReferer.delete(new URL(details.url).hostname);

		details.requestHeaders!.push({
			name: 'Referer',
			value: `https://${referer}/`
		});

		return { requestHeaders: details.requestHeaders };
	},
	{ urls: ['<all_urls>'], types: ['xmlhttprequest'] },
	['blocking', 'requestHeaders']
);

chrome.webRequest.onBeforeRedirect.addListener(
	async (details) => {
		// check if redirects origins from a previous redirect
		if ((await Redirect.get()) == null) {
			let match: Match | null;
			if ((match = await getMatch(new URL(details.url).hostname)) !== null) {
				await Redirect.set(match);
			}
		} else {
			await Redirect.delete();
		}
	},
	{ urls: ['<all_urls>'], types: ['main_frame', 'sub_frame'] }
);