87 行
2.7 KiB
Python
87 行
2.7 KiB
Python
# SPDX-FileCopyrightText: 2022 n9k <https://git.076.ne.jp/ninya9k>
|
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
|
|
from collections import OrderedDict
|
|
|
|
from quart_compress import Compress
|
|
|
|
from anonstream.config import update_flask_from_toml
|
|
from anonstream.utils.captcha import create_captcha_factory, create_captcha_signer
|
|
from anonstream.quart import Quart
|
|
|
|
compress = Compress()
|
|
|
|
def create_app(toml_config):
|
|
app = Quart('anonstream', static_folder=None)
|
|
app.jinja_options['trim_blocks'] = True
|
|
app.jinja_options['lstrip_blocks'] = True
|
|
|
|
auth_password = update_flask_from_toml(toml_config, app.config)
|
|
print('Broadcaster username:', app.config['AUTH_USERNAME'])
|
|
print('Broadcaster password:', auth_password)
|
|
|
|
# Compress some responses
|
|
compress.init_app(app)
|
|
app.config.update({
|
|
'COMPRESS_MIN_SIZE': 2048,
|
|
'COMPRESS_LEVEL': 9,
|
|
})
|
|
|
|
# Global state: messages, users, captchas
|
|
app.messages_by_id = OrderedDict()
|
|
app.messages = app.messages_by_id.values()
|
|
|
|
app.users_by_token = {}
|
|
app.users = app.users_by_token.values()
|
|
|
|
app.captchas = OrderedDict()
|
|
app.captcha_factory = create_captcha_factory(app.config['CAPTCHA_FONTS'])
|
|
app.captcha_signer = create_captcha_signer(app.config['SECRET_KEY'])
|
|
|
|
app.failures = OrderedDict()
|
|
|
|
# State for tasks
|
|
app.users_update_buffer = set()
|
|
app.stream_title = None
|
|
app.stream_uptime = None
|
|
app.stream_viewership = None
|
|
app.last_info_task = None
|
|
|
|
# Background tasks' asyncio.sleep tasks, cancelled on shutdown
|
|
app.background_sleep = set()
|
|
|
|
# Queues for event socket clients
|
|
app.event_queues = set()
|
|
|
|
@app.after_serving
|
|
async def shutdown():
|
|
# Force all background tasks to finish
|
|
for task in app.background_sleep:
|
|
task.cancel()
|
|
|
|
@app.before_serving
|
|
async def startup():
|
|
# Start control server
|
|
if app.config['SOCKET_CONTROL_ENABLED']:
|
|
from anonstream.control.server import start_control_server_at
|
|
async def start_control_server():
|
|
return await start_control_server_at(
|
|
app.config['SOCKET_CONTROL_ADDRESS']
|
|
)
|
|
app.add_background_task(start_control_server)
|
|
|
|
# Start event server
|
|
if app.config['SOCKET_EVENT_ENABLED']:
|
|
from anonstream.events import start_event_server_at
|
|
async def start_event_server():
|
|
return await start_event_server_at(
|
|
app.config['SOCKET_EVENT_ADDRESS']
|
|
)
|
|
app.add_background_task(start_event_server)
|
|
|
|
# Create routes and background tasks
|
|
import anonstream.routes
|
|
import anonstream.tasks
|
|
|
|
return app
|