48 行
1.4 KiB
Python
48 行
1.4 KiB
Python
import asyncio
|
|
|
|
from quart import current_app, request, render_template, make_response, redirect, websocket
|
|
|
|
from anonstream.segments import CatSegments
|
|
from anonstream.wrappers import with_token_from, auth_required
|
|
from anonstream.websocket import websocket_outbound, websocket_inbound
|
|
|
|
@current_app.route('/')
|
|
@with_token_from(request)
|
|
async def home(token):
|
|
return await render_template('home.html', token=token)
|
|
|
|
@current_app.route('/stream.mp4')
|
|
@with_token_from(request)
|
|
async def stream(token):
|
|
try:
|
|
cat_segments = CatSegments(current_app.segments_directory_cache, token)
|
|
except ValueError:
|
|
return 'offline', 404
|
|
response = await make_response(cat_segments.stream())
|
|
response.headers['Content-Type'] = 'video/mp4'
|
|
response.timeout = None
|
|
return response
|
|
|
|
@current_app.route('/login')
|
|
@auth_required
|
|
async def login():
|
|
return redirect('/')
|
|
|
|
@current_app.websocket('/live')
|
|
@with_token_from(websocket)
|
|
async def live(token):
|
|
queue = asyncio.Queue()
|
|
current_app.websockets[token] = queue
|
|
|
|
producer = websocket_outbound(queue)
|
|
consumer = websocket_inbound(
|
|
connected_websockets=current_app.websockets,
|
|
token=token,
|
|
secret=current_app.config['SECRET_KEY'],
|
|
chat=current_app.chat,
|
|
)
|
|
try:
|
|
await asyncio.gather(producer, consumer)
|
|
finally:
|
|
current_app.websockets.pop(token)
|