33 lines
992 B
Python
33 lines
992 B
Python
from flask import Blueprint, Response
|
|
from sqlalchemy import select
|
|
|
|
from l4d2web.auth import current_user, require_login
|
|
from l4d2web.db import session_scope
|
|
from l4d2web.models import Server
|
|
from l4d2web.services import l4d2_facade as facade
|
|
|
|
|
|
bp = Blueprint("logs", __name__)
|
|
|
|
|
|
def load_authorized_server(server_id: int) -> Server | None:
|
|
user = current_user()
|
|
if user is None:
|
|
return None
|
|
with session_scope() as db:
|
|
server = db.scalar(select(Server).where(Server.id == server_id, Server.user_id == user.id))
|
|
return server
|
|
|
|
|
|
@bp.get("/servers/<int:server_id>/logs/stream")
|
|
@require_login
|
|
def stream_server_logs(server_id: int) -> Response:
|
|
server = load_authorized_server(server_id)
|
|
if server is None:
|
|
return Response(status=404)
|
|
|
|
def generate():
|
|
for line in facade.stream_server_logs(server.name, lines=200, follow=True):
|
|
yield f"data: {line}\n\n"
|
|
|
|
return Response(generate(), mimetype="text/event-stream")
|