From fcdc40497fe0593bc2437ea596a2d288ea211e77 Mon Sep 17 00:00:00 2001
From: Roy Olav Purser <admin@royolav.net>
Date: Sun, 9 May 2021 11:37:19 +0200
Subject: [PATCH] add proxy to upstream

---
 stream.py | 20 +++++++++++---------
 1 file changed, 11 insertions(+), 9 deletions(-)

diff --git a/stream.py b/stream.py
index 1d33235..41ebc7a 100755
--- a/stream.py
+++ b/stream.py
@@ -15,16 +15,18 @@ providers["twitch"] = "https://twitch.tv"
 
 proxies = {}
 proxies_raw = {}
+proxies_req = {}
 
 for key in providers:
     proxies[key] = streamlink.Streamlink()
     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)
-        splitted = proxy.split(":", 1)
-        if len(splitted) == 2:
-            proxies_raw[key] = splitted[1].lstrip("/")
+        proxies[key].set_option("https-proxy", "socks5://" + proxy)
+        proxies_raw[key] = proxy
+        proxies_req[key] = {}
+        proxies_req[key]["http"] = "socks5://" + proxy
+        proxies_req[key]["https"] = "socks5://" + proxy
 
 playlist = None
 icecast_server = os.environ.get("ICECAST_SERVER")
@@ -46,8 +48,10 @@ if icecast_server is not None and stream_server is not None:
                 playlist += f'#EXTINF:0 radio="false", {name}\n'
                 playlist += stream_server + key + "\n"
 
-def rewrite(current, proxy):
-    resp = requests.head(current)
+def rewrite(current, provider):
+    proxy_req = proxies_req.get(provider)
+    proxy = proxies.get(provider)
+    resp = requests.head(current, proxies=proxy_req)
     ctype = resp.headers.get("Content-Type")
     print(ctype)
     if ctype is None:
@@ -79,10 +83,8 @@ class MainHandler(tornado.web.RequestHandler):
         provider = self.get_query_argument("provider", None)
         upstream = None
         proxy = None
-        proxy_raw = None
         if provider is not None and provider in providers.keys():
             proxy = proxies.get(provider)
-            proxy_raw = proxies_raw.get(provider)
             src = providers[provider] + self.request.uri
             try:
                 resp = requests.get(src)
@@ -106,7 +108,7 @@ class MainHandler(tornado.web.RequestHandler):
             if write:
                 self.write("stream not found")
         else:
-            data = rewrite(upstream, proxy_raw)
+            data = rewrite(upstream, provider)
             if data is None:
                 self.redirect(upstream, status=303)
             else: