roypur/stream-api
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

refactor upstream

Browse Source
This commit is contained in:
Roy Olav Purser 2021-05-14 11:31:30 +02:00
parent 7fd96c820d
commit e20b388a69
Signed by: roypur
GPG Key ID: E14D26A036F21656
2 changed files with 127 additions and 131 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
frontend/index.html
View File

@ -1,23 +1,24 @@
<html> <html>
<head> <head>
<meta name="viewport" content="width=device-width, initial-scale=1"> <meta name="viewport" content="width=device-width, initial-scale=1">
<meta property="og:title" content="Video Embed" /> <meta property="og:title" content="{{ data["og:title"] }}" />
<meta property="og:description" content="Video Embed" /> <meta property="og:description" content="{{ data["og:title"] }}" />
<meta property="og:type" content="video.other" /> <meta property="og:type" content="video.other" />
<meta property="og:image" content="https://i.ytimg.com/vi/3OJY5d96Ri8/hqdefault.jpg" /> <meta property="og:image" content="https://i.ytimg.com/vi/3OJY5d96Ri8/hqdefault.jpg" />
<meta property="og:url" content="{{ origin }}?provider={{ provider }}&render=true" /> <meta property="og:image:height" content="{{ data["og:image:height"] }}" />
<meta property="og:video:url" content="{{ origin }}?provider={{ provider }}" /> <meta property="og:image:width" content="{{ data["og:image:width"] }}" />
<meta property="og:video:secure_url" content="{{ origin }}?provider={{ provider }}" /> <meta property="og:url" content="{% raw data["render_url"] %}" />
<meta property="og:video:height" content="320" /> <meta property="og:video:url" content="{% raw data["render_url"] %}" />
<meta property="og:video:width" content="180" /> <meta property="og:video:secure_url" content="{% raw data["stream_url"] %}" />
<link rel="alternate" type="application/json+oembed" href="{{ origin }}?provider={{ provider }}&embed=true"> <meta property="og:video:height" content="{{ data["og:video:height"] }}" />
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/video.js/{{ videojs_version }}/video-js.css"> <meta property="og:video:width" content="{{ data["og:video:width"] }}" />
<link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@silvermine/videojs-chromecast@{{ chromecast_version }}/dist/silvermine-videojs-chromecast.css"> <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/video.js/{{ data["videojs_version"] }}/video-js.css">
<script src="https://cdnjs.cloudflare.com/ajax/libs/video.js/{{ videojs_version }}/video.min.js"></script> <link rel="stylesheet" href="https://cdn.jsdelivr.net/npm/@silvermine/videojs-chromecast@{{ data["chromecast_version"] }}/dist/silvermine-videojs-chromecast.css">
<script src="https://cdn.jsdelivr.net/npm/@silvermine/videojs-chromecast@{{ chromecast_version }}/dist/silvermine-videojs-chromecast.min.js"></script> <script src="https://cdnjs.cloudflare.com/ajax/libs/video.js/{{ data["videojs_version"] }}/video.min.js"></script>
<script src="https://cdn.jsdelivr.net/npm/@silvermine/videojs-chromecast@{{ data["chromecast_version"] }}/dist/silvermine-videojs-chromecast.min.js"></script>
<script src="https://www.gstatic.com/cv/js/sender/v1/cast_sender.js?loadCastFramework=1"></script> <script src="https://www.gstatic.com/cv/js/sender/v1/cast_sender.js?loadCastFramework=1"></script>
<script src="{{ script }}"></script> <script src="{{ data["script"] }}"></script>
<link rel="stylesheet" href="{{ custom_style }}"> <link rel="stylesheet" href="{{ data["custom_style"] }}">
</head> </head>
<body> <body>
</body> </body>

229
stream.py
View File

@ -18,6 +18,10 @@ providers["nrk"] = "https://tv.nrk.no"
providers["svt"] = "https://svtplay.se" providers["svt"] = "https://svtplay.se"
providers["youtube"] = "https://www.youtube.com/watch?v=" providers["youtube"] = "https://www.youtube.com/watch?v="
providers["twitch"] = "https://twitch.tv" providers["twitch"] = "https://twitch.tv"
playlist = None
icecast_server = os.environ.get("ICECAST_SERVER")
stream_server = os.environ.get("STREAM_SERVER")
proxy_server = os.environ.get("PROXY_SERVER")
class ProxyElem(): class ProxyElem():
def __init__(self, proxy): def __init__(self, proxy):
@ -35,7 +39,7 @@ class ProxyElem():
return session return session
def __repr__(self): def __repr__(self):
return str(self.proxy) return str(self.proxy)
proxies = {} proxies = {}
for key in providers: for key in providers:
proxies[key] = [] proxies[key] = []
@ -50,10 +54,69 @@ for key in providers:
for proxy in current: for proxy in current:
proxies[key].append(ProxyElem(proxy)) proxies[key].append(ProxyElem(proxy))
playlist = None class UpstreamHandler():
icecast_server = os.environ.get("ICECAST_SERVER") def __init__(self, handler):
stream_server = os.environ.get("STREAM_SERVER") self.render_url = None
proxy_server = os.environ.get("PROXY_SERVER") self.stream_url = None
self.proxy = None
self.upstream = None
self.upstream_safe = None
self.render = False
self.stream = True
provider = handler.get_query_argument("provider", None)
render_str = handler.get_query_argument("render", "false")
if render_str.lower() == "true":
self.render = True
self.stream = False
if provider in providers.keys():
path = handler.request.path
if provider == "youtube":
path = path.strip("/")
if isinstance(stream_server, str):
self.render_url = f'{stream_server}{handler.request.path}?provider={provider}&render=true'
self.stream_url = f'{stream_server}{handler.request.path}?provider={provider}'
else:
self.render_url = f'{handler.request.path}?provider={provider}&render=true'
self.stream_url = f'{handler.request.path}?provider={provider}'
src = providers[provider] + path
proxy_list = None
proxy_iter = None
proxy_list_orig = proxies.get(provider)
if isinstance(proxy_list_orig, list):
proxy_list = proxy_list_orig.copy()
proxy_iter = proxy_list_orig.copy()
if isinstance(proxy_list, list):
for i in proxy_iter:
current_list = proxy_list.copy()
current = proxy_list.pop()
proxy_list = [current] + proxy_list
try:
resp = requests.head(src, allow_redirects=True, proxies=current.req, timeout=2)
except Exception as e:
logger.info(e)
else:
if resp.url.lower().startswith("https://consent.youtube.com"):
self.upstream = src
self.upstream_safe = urllib.parse.quote(src)
else:
self.upstream = resp.url
self.upstream_safe = urllib.parse.quote(resp.url)
self.proxy = current
proxies[provider] = current_list
return
def meta(self):
data = {}
try:
embed_url = f'https://noembed.com/embed?url={self.upstream_safe}'
logger.info(embed_url)
resp = requests.get(embed_url)
data_raw = json.loads(resp.text)
if isinstance(data_raw, dict):
data = data_raw
except Exception as e:
logger.info(e)
return data
if icecast_server is not None and stream_server is not None: if icecast_server is not None and stream_server is not None:
try: try:
@ -156,136 +219,68 @@ def rewrite(current, provider, proxy):
ndata += links.pop(0) ndata += links.pop(0)
ndata += "\n" ndata += "\n"
return ndata return ndata
class MainHandler(tornado.web.RequestHandler): class MainHandler(tornado.web.RequestHandler):
def handle_any(self, write): def handle_any(self):
provider = self.get_query_argument("provider", None) handler = UpstreamHandler(self)
render = self.get_query_argument("render", "false") if handler.render:
embed = self.get_query_argument("embed", "false") self.handle_render(handler)
if isinstance(provider, str): elif handler.stream:
if render.lower() == "true": self.handle_stream(handler)
self.handle_render(provider)
elif embed.lower() == "true":
self.handle_embed(provider)
else:
self.handle_stream(provider, write)
else: else:
logger.info(f'provider missing {self.request.uri}') logger.info(f'provider missing {self.request.uri}')
self.set_status(404) self.set_status(404)
if write: self.write("Stream not found. (provider missing)")
self.write("Stream not found. (provider missing)")
def handle_render(self, provider): def handle_render(self, handler):
if template_js is not None and template_html is not None: if template_js is not None and template_html is not None:
origin = self.request.path rendered_js = template_js.generate(stream=handler.stream_url);
if stream_server is not None:
origin = f'{stream_server}{self.request.path}'
stream_path = f'{origin}?provider={provider}'
rendered_js = template_js.generate(stream=stream_path);
b64_js = str(base64.b64encode(rendered_js), "ascii") b64_js = str(base64.b64encode(rendered_js), "ascii")
script = f'data:text/javascript;charset=utf-8;base64,{b64_js}' script = f'data:text/javascript;charset=utf-8;base64,{b64_js}'
rendered_html = template_html.generate(script=script, videojs_version=videojs_version, chromecast_version=chromecast_version, custom_style=custom_style, provider=provider, origin=origin) meta = handler.meta()
data = {}
if meta is not None:
data["og:title"] = meta.get("title", "")
data["og:image"] = meta.get("thumbnail_url", "")
data["og:video:height"] = meta.get("height", "180")
data["og:video:width"] = meta.get("width", "320")
data["og:image:height"] = meta.get("thumbnail_height", "180")
data["og:image:width"] = meta.get("thumbnail_width", "320")
data["script"] = script
data["videojs_version"] = videojs_version
data["chromecast_version"] = chromecast_version
data["custom_style"] = custom_style
data["stream_url"] = handler.stream_url
data["render_url"] = handler.render_url
rendered_html = template_html.generate(data=data)
self.write(rendered_html) self.write(rendered_html)
else: else:
self.set_status(404) self.set_status(404)
self.write("HTML template missing.") self.write("HTML template missing.")
def handle_embed(self, provider):
width_str = self.get_query_argument("maxwidth", None)
height_str = self.get_query_argument("maxheight", None)
width = 320
height = 180
try:
if isinstance(width_str, str):
width_new = int(width_str)
if width_new >= 16:
width = width_new
else:
raise ValueError("invalid width")
if isinstance(height_str, str):
height_new = int(height_str)
if height_new >= 9:
height = height_new
else:
raise ValueError("invalid height")
except Exception:
pass
else:
if isinstance(width_str, str) and isinstance(height_str, str):
pass
elif isinstance(width_str, str):
width = int((width // 16) * 16)
height = int((width * 9) // 16)
elif isinstance(height_str, str):
height = int((height // 9) * 9)
width = int((height * 16) // 9)
origin = self.request.path
if stream_server is not None:
origin = f'{stream_server}{self.request.path}'
embed_json = {}
embed_json["version"] = "1.0"
embed_json["type"] = "video"
embed_json["width"] = width
embed_json["height"] = height
embed_json["provider_name"] = "stream.purser.it"
embed_json["provider_url"] = "https://stream.purser.it"
embed_json["html"] = str(template_embed.generate(origin=origin, provider=provider, width=width, height=height), "utf-8")
self.set_header("Content-Type", "application/json; charset=utf-8")
self.write(json.dumps(embed_json))
def handle_stream(self, provider, write): def handle_stream(self, handler):
upstream = None upstream = None
proxy = None if handler.proxy is not None:
if provider in providers.keys(): try:
path = self.request.path logger.info(proxy)
if provider == "youtube": streams = handler.proxy.stream().streams(handler.upstream)
path = path.strip("/") except Exception as e:
src = providers[provider] + path logger.info(e)
proxy_list = None else:
proxy_iter = None for key in reversed(streams):
proxy_list_orig = proxies.get(provider) stream = streams.get(key)
if isinstance(proxy_list_orig, list): logger.info(stream)
proxy_list = proxy_list_orig.copy() if hasattr(stream, "url"):
proxy_iter = proxy_list_orig.copy() upstream = stream.url
if isinstance(proxy_list, list):
for i in proxy_iter:
current_list = proxy_list.copy()
current = proxy_list.pop()
proxy_list = [current] + proxy_list
try:
resp = requests.head(src, allow_redirects=True, proxies=current.req, timeout=2)
if resp is not None:
logger.info(src)
src = resp.url
except Exception as e:
logger.info(e)
else:
proxies[provider] = current_list
proxy = current
break break
if proxy is not None:
try:
logger.info(proxy)
streams = proxy.stream().streams(src)
except Exception as e:
logger.info(e)
else:
for key in reversed(streams):
stream = streams.get(key)
logger.info(stream)
if hasattr(stream, "url"):
upstream = stream.url
break
else: else:
logger.info(f'invalid provider ({provider})') logger.info(f'invalid provider ({provider})')
self.set_status(404) self.set_status(404)
if write: self.write("Stream not found. (invalid provider)")
self.write("Stream not found. (invalid provider)")
return return
if upstream is None: if upstream is None:
logger.info(f'invalid upstream ({provider})') logger.info(f'invalid upstream ({provider})')
self.set_status(404) self.set_status(404)
if write: self.write("Stream not found. (invalid upstream)")
self.write("Stream not found. (invalid upstream)")
else: else:
ctype = upstream_type(upstream, proxy) ctype = upstream_type(upstream, proxy)
data = None data = None
@ -294,8 +289,8 @@ class MainHandler(tornado.web.RequestHandler):
else: else:
ldata = {} ldata = {}
ldata["upstream"] = upstream ldata["upstream"] = upstream
ldata["proxy"] = proxy.proxy ldata["proxy"] = handler.proxy.proxy
ldata["proxied"] = isinstance(proxy.proxy, str) ldata["proxied"] = isinstance(handler.proxy.proxy, str)
links = [ldata] links = [ldata]
if isinstance(proxy_server, str): if isinstance(proxy_server, str):
try: try:
@ -312,9 +307,9 @@ class MainHandler(tornado.web.RequestHandler):
self.set_header("Content-Type", "application/vnd.apple.mpegurl") self.set_header("Content-Type", "application/vnd.apple.mpegurl")
self.write(data) self.write(data)
def get(self): def get(self):
self.handle_any(True) self.handle_any()
def head(self): def head(self):
self.handle_any(False) self.handle_any()
class FileHandler(tornado.web.RequestHandler): class FileHandler(tornado.web.RequestHandler):
def get(self): def get(self):