8 Commits

8 changed files with 78 additions and 183 deletions

View File

@@ -3,7 +3,7 @@
import logging as lg import logging as lg
if __name__ == "__main__": if __name__ == "__main__":
from judas_server.web.server import JudasWebServer from judas_server.web.web_server import JudasWebServer
lg.basicConfig( lg.basicConfig(
level=lg.DEBUG, level=lg.DEBUG,

View File

@@ -9,7 +9,7 @@
<body> <body>
<div id="wrapper"> <div id="wrapper">
<header> <header>
<h1><a href="{{ url_for('index') }}">judas</a></h1> <h1><a href="{{ url_for('index.index') }}">judas</a></h1>
<p><a class="button" href="{{ url_for('logout') }}">Logout</a></p> <p><a class="button" href="{{ url_for('logout') }}">Logout</a></p>
</header> </header>
<main> <main>

View File

@@ -9,11 +9,11 @@
<body> <body>
<div id="wrapper"> <div id="wrapper">
<header> <header>
<h1><a href="{{ url_for('index') }}">judas</a></h1> <h1><a href="{{ url_for('index.index') }}">judas</a></h1>
{% if logged %} {% if logged %}
<p><a class="button" href="{{ url_for('logout') }}">Logout</a></p> <p><a class="button" href="{{ url_for('auth.logout') }}">Logout</a></p>
{% else %} {% else %}
<p><a class="button" href="{{ url_for('login') }}">Login</a></p> <p><a class="button" href="{{ url_for('auth.login') }}">Login</a></p>
{% endif %} {% endif %}
</header> </header>
<main class="center"> <main class="center">
@@ -24,9 +24,9 @@
</div> </div>
<p style="color: #bf616a;"><strong>Notice:</strong> Please use this system responsibly and in accordance with all applicable laws and organizational policies.</p> <p style="color: #bf616a;"><strong>Notice:</strong> Please use this system responsibly and in accordance with all applicable laws and organizational policies.</p>
{% if logged %} {% if logged %}
<p><a class="button" href="{{ url_for('panel') }}">Go to panel</a></p> <p><a class="button" href="{{ url_for('panel.panel') }}">Go to panel</a></p>
{% else %} {% else %}
<p>Please <a href="{{ url_for('login')}}" class="link">log in</a> to manage your remote PCs.</p> <p>Please <a href="{{ url_for('auth.login')}}" class="link">log in</a> to manage your remote PCs.</p>
{% endif %} {% endif %}
</main> </main>
</div> </div>

View File

@@ -9,7 +9,7 @@
<body> <body>
<div id="wrapper"> <div id="wrapper">
<header> <header>
<h1><a href="{{ url_for('index') }}">judas</a></h1> <h1><a href="{{ url_for('index.index') }}">judas</a></h1>
</header> </header>
<main> <main>
<h1>Login</h1> <h1>Login</h1>

View File

@@ -9,8 +9,8 @@
<body> <body>
<div id="wrapper"> <div id="wrapper">
<header> <header>
<h1><a href="{{ url_for('index') }}">judas</a></h1> <h1><a href="{{ url_for('index.index') }}">judas</a></h1>
<p><a class="button" href="{{ url_for('logout') }}">Logout</a></p> <p><a class="button" href="{{ url_for('auth.logout') }}">Logout</a></p>
</header> </header>
<main> <main>
<h2 class="center">Select a PC to Control</h2> <h2 class="center">Select a PC to Control</h2>

View File

@@ -1,16 +0,0 @@
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>judas panel - stream</title>
</head>
<body>
<div id="wrapper">
<header>
<h1><a href="{{ url_for('index') }}">judas</a></h1>
<p><a class="button" href="{{ url_for('logout') }}">Logout</a></p>
</header>
</div>
</body>
</html>

View File

@@ -0,0 +1,29 @@
# -*- coding: utf-8 -*-
from __future__ import annotations
import flask_login
class User(flask_login.UserMixin):
"""Represents a user for authentication purposes."""
def __init__(self, id: str) -> None:
super().__init__()
self.id = id
def get_id(self) -> str:
"""Return the unique identifier for the user."""
return self.id
def __str__(self) -> str:
return f"User(id={self.id})"
def __repr__(self) -> str:
return f"User(id={self.id})"
def load_user(user_id):
if user_id == "admin":
return User("admin")
return None

View File

@@ -1,171 +1,53 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
from typing import Final from __future__ import annotations
import flask import logging as lg
import flask_login from flask import Flask
from flask_login import LoginManager
PASSWORD: Final[str] = "123" from judas_server.web.user import load_user
app = flask.Flask(__name__)
app.secret_key = "dildo"
login_manager = flask_login.LoginManager()
login_manager.init_app(app)
PC_DETAILS = { class JudasWebServer:
"PC1": { def __init__(self, secret_key: str) -> None:
"id": "PC1", self.logger: lg.Logger = lg.getLogger(
"status": "online", f"{__name__}.{self.__class__.__name__}"
"last_seen": "2023-10-01 12:00:00",
},
"PC2": {
"id": "PC2",
"status": "offline",
"last_seen": "2023-10-01 11:00:00",
},
"PC3": {
"id": "PC3",
"status": "offline",
"last_seen": "2023-10-01 11:00:00",
},
"PC4": {
"id": "PC4",
"status": "offline",
"last_seen": "2023-10-01 11:00:00",
},
"PC5": {
"id": "PC5",
"status": "offline",
"last_seen": "2023-10-01 11:00:00",
},
"PC6": {
"id": "PC6",
"status": "offline",
"last_seen": "2023-10-01 11:00:00",
},
"PC7": {
"id": "PC7",
"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() -> flask.Response | str:
"""Renders the index page with a link to the login page."""
return flask.render_template(
"index.html", logged=flask_login.current_user.is_authenticated
) )
self.logger.debug("Initializing JudasWebServer...")
self.app: Flask = Flask(
@app.route("/login", methods=["GET", "POST"]) __name__, static_folder="static", template_folder="templates"
def login() -> flask.Response | 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 flask.render_template(
"login.html",
error="Invalid password. Please try again.",
) )
return flask.render_template("login.html") self.app.secret_key = secret_key
# hard-code password
self.app.config["PASSWORD"] = "123"
@app.route("/logout") # extensions
@flask_login.login_required self.login_manager: LoginManager = LoginManager()
def logout() -> str: self.app.config["LoginManager"] = self.login_manager
"""Logs out the current user."""
flask_login.logout_user()
return flask.redirect(flask.url_for("index"))
self.configure_extensions()
self.init_routes()
@app.route("/panel") def configure_extensions(self) -> None:
@flask_login.login_required self.logger.debug("Configuring extensions...")
def panel() -> str: self.login_manager.init_app(self.app)
"""Renders the main panel page with PC details. self.login_manager.user_loader(load_user)
self.login_manager.login_view = "auth.login"
Returns: # TODO: add login page
Rendered HTML template with PC details.
"""
return flask.render_template(
"panel.html",
username=flask_login.current_user.id,
pcs=PC_DETAILS,
)
def init_routes(self) -> None:
self.logger.debug("Initializing routes...")
from judas_server.web.routes import auth_bp, index_bp, panel_bp
@app.route("/details/<pc_id>") self.app.register_blueprint(index_bp)
@flask_login.login_required self.app.register_blueprint(auth_bp)
def details(pc_id: str) -> str: self.app.register_blueprint(panel_bp)
"""Renders the details page for a specific PC.
Args: def run(self, host: str = "0.0.0.0", port: int = 5000) -> None:
pc_id: The ID of the PC to display details for. self.logger.info(f"Starting server on {host}:{port}...")
self.app.run(host=host, port=port)
Returns: self.logger.info("Server stopped.")
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(
"stream.html",
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)