feat: implement JWT authentication for user management

game_collection/requirements.txt

@@ -1,2 +1,3 @@
 Flask
 Flask-SQLAlchemy
+PyJWT

game_collection/user_management.py

@@ -1,23 +1,36 @@
 from flask import Blueprint, request, jsonify, g
 from models import db, User
-from datetime import datetime
+from datetime import datetime, timedelta
+import jwt
 from werkzeug.security import generate_password_hash, check_password_hash
+import os
 
 user_bp = Blueprint('user', __name__)
+SECRET_KEY = os.environ.get('SECRET_KEY', 'your_secret_key')  # Set a secret key for JWT
+
+def generate_token(user):
+    payload = {
+        'user_id': user.id,
+        'exp': datetime.utcnow() + timedelta(days=1)  # Token expires in 1 day
+    }
+    return jwt.encode(payload, SECRET_KEY, algorithm='HS256')
 
 def authenticate():
-    username = request.headers.get('X-Username')
-    password = request.headers.get('X-Password')
-    if username and password:
-        user = User.query.filter_by(username=username).first()
-        if user and check_password_hash(user.password, password):
-            g.user = user
+    token = request.headers.get('Authorization')
+    if token:
+        try:
+            payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])
+            g.user = User.query.get(payload['user_id'])
             return True
+        except jwt.ExpiredSignatureError:
+            return False
+        except jwt.InvalidTokenError:
+            return False
     return False
 
 @user_bp.before_request
 def before_request():
-    if request.endpoint not in ['user.login_user']:
+    if request.endpoint not in ['user.login_user', 'user.create_user']:
         if not authenticate():
             return jsonify({'message': 'Unauthorized access!'}), 401
 
@@ -42,7 +55,8 @@ def login_user():
     if user and check_password_hash(user.password, data['password']):
         user.last_login = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
         db.session.commit()
-        return jsonify({'message': 'Login successful!', 'role': user.role}), 200
+        token = generate_token(user)
+        return jsonify({'message': 'Login successful!', 'token': token, 'role': user.role}), 200
     return jsonify({'message': 'Invalid credentials!'}), 401
 
 @user_bp.route('/users', methods=['GET'])