import jwt
import os
from datetime import datetime, timedelta
from dotenv import load_dotenv

load_dotenv()
SECRET_KEY = os.getenv("JWT_SECRET_KEY")
ALGORITHM = "HS256"

def create_token(user_id: str):
    if not SECRET_KEY:
        print("❌ Error: JWT_SECRET_KEY not found in .env")
        return
    
    payload = {
        "sub": user_id,
        "exp": datetime.utcnow() + timedelta(hours=24)
    }
    token = jwt.encode(payload, SECRET_KEY, algorithm=ALGORITHM)
    print(f"✅ Token generated for {user_id}:\n{token}")
    return token

if __name__ == "__main__":
    create_token("admin_user")