WIP: Add light/dark theme with toggle in navbar (dark theme styling incomplete - dont care for now)

routes/auth.py

@@ -7,25 +7,26 @@ from jinja2_fragments import render_block
 auth = Blueprint('auth', __name__)
 
 class User(UserMixin):
-    def __init__(self, id, username, password_hash, created_at):
+    def __init__(self, id, username, password_hash, created_at, theme_preference='light'):
         self.id = id
         self.username = username
         self.password_hash = password_hash
         self.created_at = created_at
+        self.theme_preference = theme_preference
 
     @staticmethod
     def get(user_id):
         user_data = db.get_user(int(user_id))
 
         if user_data:
-            return User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'])
+            return User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'], theme_preference=user_data.get('theme_preference', 'light'))
         return None
 
 @login_manager.user_loader
 def load_user(user_id):
     user_data = db.get_user(int(user_id))
     if user_data:
-        return User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'])
+        return User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'], theme_preference=user_data.get('theme_preference', 'light'))
     return None
 
 @auth.route('/login', methods=['GET', 'POST'])
@@ -45,7 +46,7 @@ def login():
     if not check_password_hash(user_data['password_hash'], password):
         return render_template("login.html", error="Invalid username or password")
     
-    user = User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'])
+    user = User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'], theme_preference=user_data.get('theme_preference', 'light'))
     
     login_user(user)
 
@@ -72,7 +73,7 @@ def signup():
     hashed_password = generate_password_hash(password)
     user_data = db.create_new_user(username, hashed_password)
 
-    user = User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'])
+    user = User(id=str(user_data['id']), username=user_data['username'], password_hash=user_data['password_hash'], created_at=user_data['created_at'], theme_preference=user_data.get('theme_preference', 'light'))
     login_user(user)
 
     return redirect(url_for('home.index'))

routes/settings.py

@@ -62,3 +62,15 @@ def delete_api_key(key_id):
     user_id = current_user.id
     db.delete_api_key(user_id, key_id)
     return "", 200
+
+@settings.route("/theme", methods=["POST"])
+@login_required
+def toggle_theme():
+    user_id = current_user.id
+    theme = request.form.get("theme")
+    if theme in ['light', 'dark']:
+        db.update_user_theme_preference(user_id, theme)
+        # Return empty string as we'll handle the UI update via client-side JS or just let the class toggle persist
+        # Actually, for HTMX we might want to return something or just 200 OK.
+        return "", 200
+    return "Invalid theme", 400