Utils.py

import jwt
import bcrypt
from datetime import datetime, timedelta, timezone


PRIVATE_KEY_PATH = ".keys/private.pem"
PUBLIC_KEY_PATH = ".keys/public.pem"
EXPIRE_MINUTES = 30


with open(PRIVATE_KEY_PATH, "r") as file:
    PRIVATE_KEY = file.read()

with open(PUBLIC_KEY_PATH, "r") as file:
    PUBLIC_KEY = file.read()


def create_jwt(
    data: dict, expires_delta: timedelta = timedelta(minutes=EXPIRE_MINUTES)
):
    payload = data.copy()
    payload["exp"] = datetime.now(timezone.utc) + expires_delta
    payload["iat"] = datetime.now(timezone.utc)
    payload["iss"] = "pi.daazed.dev"
    return jwt.encode(payload, PRIVATE_KEY, algorithm="RS256")


def verify_jwt(token: str):
    try:
        payload = jwt.decode(token, PUBLIC_KEY, algorithms=["RS256"])
        return payload
    except jwt.ExpiredSignatureError:
        raise ValueError("TOKEN EXPIRED")
    except jwt.InvalidTokenError:
        raise ValueError("INVALID TOKEN")


def verify_password(password, hashed_password):
    return bcrypt.checkpw(
        bytes(password, encoding="utf-8"),
        hashed_password,
    )


def hash_password(password):
    return bcrypt.hashpw(
        bytes(password, encoding="utf-8"),
        bcrypt.gensalt(),
    )