feat(web_server.py): add web_server.py

judas_server/web/web_server.py

@@ -0,0 +1,152 @@
+# -*- coding: utf-8 -*-
+
+from typing import Final
+
+import flask
+import flask_login
+
+PASSWORD: Final[str] = "123"
+
+app = flask.Flask(__name__)
+app.secret_key = "dildo"
+
+login_manager = flask_login.LoginManager()
+login_manager.init_app(app)
+
+
+PC_DETAILS = {
+    "PC1": {
+        "id": "PC1",
+        "status": "Online",
+        "last_seen": "2023-10-01 12:00:00",
+    },
+    "PC2": {
+        "id": "PC2",
+        "status": "Offline",
+        "last_seen": "2023-10-01 11:00:00",
+    },
+}
+
+
+class User(flask_login.UserMixin):
+    """Represents a user for authentication purposes."""
+
+    def __init__(self, id: str):
+        super().__init__()
+        self.id = id
+
+
+@login_manager.user_loader
+def load_user(user_id: str) -> User | None:
+    """Loads a user by user_id.
+
+    Args:
+        user_id: The ID of the user.
+
+    Returns:
+        The User object if found, else None.
+    """
+    if user_id == "admin":
+        return User("admin")
+    return None
+
+
+@app.route("/")
+def index() -> str:
+    """Renders the index page with a link to the login page."""
+    if flask_login.current_user.is_authenticated:
+        return flask.render_template(
+            "index.html", logged=True, username=flask_login.current_user.id
+        )
+    else:
+        return flask.render_template(
+            "index.html", logged=False, login_url=flask.url_for("login")
+        )
+
+
+@app.route("/login", methods=["GET", "POST"])
+def login() -> str:
+    """Handles user login via password form."""
+    if flask.request.method == "POST":
+        password = flask.request.form.get("password", "")
+        if password == PASSWORD:
+            user = User("admin")
+            flask_login.login_user(user)
+            return flask.redirect(flask.url_for("panel"))
+        else:
+            return "Invalid password", 401
+    return """
+        <form method="post">
+            Password: <input type="password" name="password">
+            <input type="submit" value="Login">
+        </form>
+    """
+
+
+@app.route("/logout")
+@flask_login.login_required
+def logout() -> str:
+    """Logs out the current user."""
+    flask_login.logout_user()
+    return flask.redirect(flask.url_for("index"))
+
+
+@app.route("/panel")
+@flask_login.login_required
+def panel() -> str:
+    """Renders the main panel page with PC details.
+
+    Returns:
+        Rendered HTML template with PC details.
+    """
+    return flask.render_template(
+        "panel.html",
+        username=flask_login.current_user.id,
+        pcs=PC_DETAILS,
+    )
+
+
+@app.route("/details/<pc_id>")
+@flask_login.login_required
+def details(pc_id: str) -> str:
+    """Renders the details page for a specific PC.
+
+    Args:
+        pc_id: The ID of the PC to display details for.
+
+    Returns:
+        Rendered HTML template with PC details.
+    """
+    return flask.render_template(
+        "details.html",
+        username=flask_login.current_user.id,
+        pc=PC_DETAILS[pc_id],
+    )
+
+
+@app.route("/stream/<pc_id>")
+@flask_login.login_required
+def stream(pc_id: str) -> str:
+    """Renders the stream page for a specific PC.
+
+    Args:
+        pc_id: The ID of the PC to stream from.
+
+    Returns:
+        Rendered HTML template for streaming.
+    """
+    return flask.render_template(
+        logged=True,
+        username=flask_login.current_user.id,
+        pc=PC_DETAILS[pc_id],
+    )
+
+
+@app.route("/stream_feed/<pc_id>")
+@flask_login.login_required
+def stream_feed(pc_id: str) -> flask.Response:
+    return flask.Response(mimetype="multipart/x-mixed-replace; boundary=frame")
+
+
+if __name__ == "__main__":
+    app.run(debug=True)