import base64
import datetime
import uuid
from typing import Optional

import jwt

SECRET_KEY = 'IrOYFjXtQPuofoXEpsR+2VsvjnaCWkPzvfmym1qNmcI='
"""
自定义密钥，生成方法见 generate_secret_key() 函数。该密钥应当在应用程序服务器启动时更新，并使用动态生成的密钥。
"""

PRIVATE_ISS = 'hǎi_tén_education_technology_co_ltd'
"""
签发者签名。
"""


def generate_secret_key():
    """
    生成如 IrOYFjXtQPuofoXEpsR+2VsvjnaCWkPzvfmym1qNmcI= 的随机串。
    """
    return base64.b64encode(uuid.uuid4().bytes + uuid.uuid4().bytes).decode()


def reset_secret_key():
    """
    重置加密密钥。

    :return: 加密密钥
    """
    global SECRET_KEY
    SECRET_KEY = generate_secret_key()
    return SECRET_KEY


def get_secret_key():
    """
    取得加密密钥。

    :return: 加密密钥
    """
    global SECRET_KEY
    return SECRET_KEY


def encode_token(exp: Optional[datetime.datetime] = None, **kwargs):
    """
    对用户信息加密生成令牌。

    :param exp: 过期时间，不传默认 7 天过期
    :param kwargs: 要加入到数据部分的参数
    :return: 加密后的 token 内容
    """
    try:
        iat = datetime.datetime.now(datetime.timezone.utc)
        exp = exp if exp else iat + datetime.timedelta(days=7)
        # 即 JWT 三部分中的载荷部分
        # 过期时间最后是和 UTC 作比较，所以设置的时候使用 datetime.datetime.now()
        payload = {
            'iss': PRIVATE_ISS,     # 签发者
            'iat': iat,             # 签发时间
            'exp': exp,             # 过期时间，这里设置7天
            'params': {}            # 参数，存放用户自定义数据
        }

        payload['params'].update(kwargs)

        # 开始进行加密，返回字符串，传入例如密钥，指定加密算法
        # encode 返回的是 bytes，需要 decode() 得到 str,不转换的话，在封装json的时候报错
        auth_token = jwt.encode(payload, SECRET_KEY, algorithm='HS256')
        if isinstance(auth_token, bytes):
            auth_token = auth_token.decode()
        return auth_token
    except Exception as e:
        raise e


def decode_token(auth_token: str):
    """
    解码 token 并从中提取用户信息，进行验证。

    :param auth_token: 令牌
    """
    # 如果需要关闭过期时间的验证，可以在 options 中使用 verify_exp
    # jwt.decode(auth_token, secret_key, issuer=private_iss, algorithms=['HS256'], options={'verify_exp': False})
    # 传入了密钥和算法，和加密是对应的，因此密钥一定不要泄露
    token_payload = jwt.decode(auth_token, SECRET_KEY, issuer=PRIVATE_ISS, algorithms=['HS256'])
    return token_payload