from functools import wraps
from flask import request, jsonify, g
from flask_jwt_extended import jwt_required, get_jwt_identity

def validate_user_post_request(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        data = request.get_json()
        if not data:
            return jsonify({"message": "No input data provided"}), 400

        # Check required fields
        required_fields = ['username', 'password', 'email', 'location']
        if not all(field in data for field in required_fields):
            return jsonify({"message": "Please check your data, you missing some props; visit our docs https://git.dayanhub.com/kfir"}), 400

        # Validate 'username'
        if not isinstance(data['username'], str) or not data['username'].strip():
            return jsonify({"message": "Invalid username"}), 400

        # Validate 'password'
        if not isinstance(data['password'], str) or not data['password'].strip():
            return jsonify({"message": "Invalid password"}), 400

        # Validate 'email'
        if not isinstance(data['email'], str) or not data['email'].strip():
            return jsonify({"message": "Invalid email"}), 400
        
        # Validate 'location'
        if not isinstance(data['location'], str) or not data['location'].strip():
            return jsonify({"message": "Invalid location"}), 400

        return f(*args, **kwargs)
    return decorated_function


def validate_user_login_request(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        data = request.get_json()
        if not data:
            return jsonify({"message": "No input data provided"}), 400

        # Check required fields
        required_fields = ['email', 'password']
        if not all(field in data for field in required_fields):
            return jsonify({"message": "Please check your data, you missing some props; visit our docs https://git.dayanhub.com/kfir"}), 400

        # Validate 'email'
        if not isinstance(data['email'], str) or not data['email'].strip():
            return jsonify({"message": "Invalid email"}), 400

        # Validate 'password'
        if not isinstance(data['password'], str) or not data['password'].strip():
            return jsonify({"message": "Invalid password"}), 400

        return f(*args, **kwargs)
    return decorated_function

def authenticate_user(f):
    @wraps(f)
    @jwt_required(locations=["cookies"])  # Specify to look for the token in cookies
    def decorated_function(*args, **kwargs):
        # Get user identity from JWT
        user_id = get_jwt_identity()
        if user_id:
            g.user_id = user_id
        else:
            return jsonify({"error": "Invalid session token"}), 401

        return f(*args, **kwargs)
    return decorated_function