import base64
from functools import wraps

from core.error import ArcError, NoAccess
from core.sql import Connect
from core.user import UserAuth, UserLogin
from flask import Blueprint, jsonify, request
from setting import Config

from .func import arc_try, error_return

bp = Blueprint('auth', __name__, url_prefix='/auth')


@bp.route('/login', methods=['POST'])  # 登录接口
@arc_try
def login():
    headers = request.headers
    if 'AppVersion' in headers:  # 版本检查
        if Config.ALLOW_APPVERSION:
            if headers['AppVersion'] not in Config.ALLOW_APPVERSION:
                raise NoAccess('Wrong app version.', 1203)

    request.form['grant_type']
    with Connect() as c:
        id_pwd = headers['Authorization']
        id_pwd = base64.b64decode(id_pwd[6:]).decode()
        name, password = id_pwd.split(':', 1)
        if 'DeviceId' in headers:
            device_id = headers['DeviceId']
        else:
            device_id = 'low_version'

        user = UserLogin(c)
        user.login(name, password, device_id, request.remote_addr)

        return jsonify({"success": True, "token_type": "Bearer", 'user_id': user.user_id, 'access_token': user.token})


def auth_required(request):
    # arcaea登录验证，写成了修饰器
    def decorator(view):
        @wraps(view)
        def wrapped_view(*args, **kwargs):

            headers = request.headers

            if 'AppVersion' in headers:  # 版本检查
                if Config.ALLOW_APPVERSION:
                    if headers['AppVersion'] not in Config.ALLOW_APPVERSION:
                        return error_return(NoAccess('Wrong app version.', 1203))

            with Connect() as c:
                try:
                    user = UserAuth(c)
                    user.token = headers['Authorization'][7:]
                    return view(user.token_get_id(), *args, **kwargs)
                except ArcError as e:
                    return error_return(e)

            return error_return()

        return wrapped_view
    return decorator