from jose import jwt

from settings import settings


def encode(data):
    return jwt.encode(data, settings.secret_key)


def decode(token):
    return jwt.decode(token, settings.secret_key)


def validate_frontend(token):
    try:
        data = decode(token)
        assert data["type"] == "frontend"
        return True

    except Exception as e:
        print(e)
        return False


def validate_computer(token):
    try:
        data = decode(token)
        assert data["type"] == "computer"
        return True

    except Exception as e:
        print(e)
        return False