from qutebrowser.api import interceptor
import operator, typing

invid = 'invidious.himiko.cloud'
nitter = 'nitter.database.red'
reddit = 'teddit.net'

o = operator.methodcaller
s = 'setHost'
i = interceptor

MAP = {
	"reddit.com": o(s, reddit),
	"www.reddit.com": o(s, reddit),
	"old.reddit.com": o(s, reddit),

	"twitter.com": o(s, nitter),
	"api.twitter.com": o(s, nitter),
	"mobile.twitter.com": o(s, nitter),
	"platform.twitter.com": o(s, nitter),
	"www.platform.twitter.com": o(s, nitter),

	"youtu.be": o(s, invid),
	"youtube.com": o(s, invid),
	"www.youtube.com": o(s, invid),

	"www.instagram.com": o(s, 'bibliogram.art'),
        "www.amazon.com": o(s, 'smile.amazon.com'),
}
def f(info: i.Request):
	if (info.resource_type != i.ResourceType.main_frame or
			info.request_url.scheme() in {"data", "blob"}):
		return
	url = info.request_url
	redir = MAP.get(url.host())
	if redir is not None and redir(url) is not False:
		info.redirect(url)
i.register(f)