#!/usr/bin/env python3
import streamlink
import tornado.web
import tornado.routing
import requests
import json
import os

providers = {}
providers["nrk"] = "https://tv.nrk.no"
providers["svt"] = "https://svtplay.se"
providers["youtube"] = "https://youtu.be"
providers["twitch"] = "https://twitch.tv"

proxies = {}
proxies["nrk"] = streamlink.Streamlink()
proxies["svt"] = streamlink.Streamlink()
proxies["youtube"] = streamlink.Streamlink()
proxies["twitch"] = streamlink.Streamlink()

for key in proxies:
    proxy = os.environ.get(key.upper() + "_PROXY")
    proxies[key].set_option("http-timeout", 2.0)
    if proxy is not None:
        proxies[key].set_option("https-proxy", proxy)

playlist = None
icecast_server = os.environ.get("ICECAST_SERVER")
stream_server = os.environ.get("STREAM_SERVER")
if icecast_server is not None and stream_server is not None:
    with open("/app/sources.json", "r") as f:
        data = json.loads(f.read())
        playlist = "#EXTM3U\n"
        for key in data:
            current = data[key]
            name = current["name"]
            radio = current["radio"]
            if radio:
                playlist += f'#EXTINF:0 radio="true", {name}\n'
                playlist += icecast_server + key + "\n"
            else:
                playlist += f'#EXTINF:0 radio="false", {name}\n'
                playlist += stream_server + key + "\n"

class MainHandler(tornado.web.RequestHandler):
    def handle_any(self):
        provider = self.get_query_argument("provider", None)
        endpoint = None
        if provider is not None and provider in providers.keys():
            proxy = proxies.get(provider)
            src = providers[provider] + self.request.uri
            try:
                resp = requests.get(src)
                if resp is not None:
                    src = resp.url
            except Exception as e:
                print(e)
            else:
                try:
                    streams = proxy.streams(src)
                    for key in reversed(streams):
                        stream = streams.get(key)
                        if hasattr(stream, "url"):
                            endpoint = stream.url
                            break
                except Exception as e:
                    print(str(e))
                    endpoint = None
        if endpoint is None:
            self.set_status(404, reason="stream not found")
        else:
            self.redirect(endpoint, status=303)
    def get(self):
        self.handle_any()
    def head(self):
        self.handle_any()

class FileHandler(tornado.web.RequestHandler):
    def get(self):
        self.set_header("Content-Type", "text/plain; charset=utf-8")
        self.write(playlist)

try:
    handlers = []
    handlers.append((tornado.routing.PathMatches("/sources.m3u8"), FileHandler))
    handlers.append((tornado.routing.AnyMatches(), MainHandler))
    app_web = tornado.web.Application(handlers)
    app_web.listen(8080)
    tornado.ioloop.IOLoop.current().start()
except KeyboardInterrupt:
    print()