diff --git a/app.py b/app.py index 94c7c1c..969dca0 100644 --- a/app.py +++ b/app.py @@ -3,7 +3,7 @@ import os import json import time import uuid -from flask import Flask, render_template, request, jsonify, send_from_directory +from flask import Flask, render_template, request, jsonify, redirect, make_response app = Flask(__name__) LAYOUT_DIR = os.path.join(os.path.dirname(__file__), "layouts") @@ -17,7 +17,6 @@ def get_layout(): if os.path.exists(CURRENT_LAYOUT): with open(CURRENT_LAYOUT) as f: return json.load(f) - # First run — copy default default = os.path.join(LAYOUT_DIR, "default.json") with open(default) as f: layout = json.load(f) @@ -28,9 +27,36 @@ def save_layout(layout): with open(CURRENT_LAYOUT, "w") as f: json.dump(layout, f, indent=2) +def get_user(): + return request.cookies.get("bfa_user", "") + @app.route("/") def index(): - return render_template("editor.html") + if not get_user(): + return redirect("login") + return render_template("editor.html", user=get_user()) + +@app.route("/login", methods=["GET", "POST"]) +def login(): + layout = get_layout() + users = layout.get("users", ["Richard", "Rob", "Guido"]) + if request.method == "POST": + name = request.form.get("user", "").strip() + if name: + resp = make_response(redirect("./")) + resp.set_cookie("bfa_user", name, max_age=86400*30) + # Add user to layout if new + if name not in users: + layout.setdefault("users", []).append(name) + save_layout(layout) + return resp + return render_template("login.html", users=users) + +@app.route("/logout") +def logout(): + resp = make_response(redirect("login")) + resp.delete_cookie("bfa_user") + return resp @app.route("/api/layout", methods=["GET"]) def api_get_layout(): @@ -49,7 +75,7 @@ def api_add_comment(): comment = { "id": str(uuid.uuid4())[:8], "target": data["target"], - "user": data["user"], + "user": data.get("user", get_user()), "time": time.strftime("%Y-%m-%dT%H:%M:%S"), "text": data["text"] } @@ -75,12 +101,12 @@ def api_upload_photo(): f = request.files["file"] fname = f"{int(time.time())}_{f.filename}" f.save(os.path.join(PHOTO_DIR, fname)) - return jsonify({"filename": fname, "url": f"/static/photos/{fname}"}) + return jsonify({"filename": fname, "url": f"static/photos/{fname}"}) @app.route("/api/photos", methods=["GET"]) def api_list_photos(): photos = sorted(os.listdir(PHOTO_DIR)) if os.path.exists(PHOTO_DIR) else [] - return jsonify([{"filename": p, "url": f"/static/photos/{p}"} for p in photos if not p.startswith(".")]) + return jsonify([{"filename": p, "url": f"static/photos/{p}"} for p in photos if not p.startswith(".")]) if __name__ == "__main__": app.run(host="0.0.0.0", port=5001, debug=True) diff --git a/templates/editor.html b/templates/editor.html index 1719347..2c6d003 100644 --- a/templates/editor.html +++ b/templates/editor.html @@ -170,8 +170,8 @@ body{font-family:'Segoe UI',system-ui,sans-serif;background:var(--bg);color:var( HMI Design Tool