from flask import Blueprint, request, jsonify
from flask_jwt_extended import create_access_token, jwt_required, get_jwt_identity
from app import redis_client, LogService
from app.models.passenger_lib import Passenger, isAvailable
from presenter import PassengerPresenter
from utils import StateCode, create_response, checkMobile, checkIdCard, checkBankCard

register_bp = Blueprint('register', __name__)

#注册模块
@register_bp.route('/register', methods=['POST'])
def register():
    #获取到前端用户注册的数据
    data = request.json
    print(data)
    account = data.get('account') #用户名
    password = data.get('password') #密码
    mobile_no = data.get('mobileNo') #手机号
    mobile_code = data.get('mobileCode') #验证码
    id_card_no = data.get('idCardNo') #身份证号
    bank_card_no = data.get('bankCard') #银行卡号

    #判断账号和密码是否为空
    if not account or not password:
        return jsonify(create_response(StateCode.PARAMS_ERROR)), 400

    #判断手机号是否可行
    if not checkMobile(mobile_no):
        return jsonify(create_response(StateCode.MOBILE_ERROR)), 400

    #判断验证码是否正确(俩个参数，手机号和验证码)
    if not verifyCode(mobile_no, mobile_code):
        return jsonify(create_response(StateCode.MOBILE_CODE_ERROR)), 400

    #判断身份证号码是否可行
    if not checkIdCard(id_card_no):
        return jsonify(create_response(StateCode.ID_CARD_ERROR)), 400

    # 判断账号是否被注册过
    if isAvailable(account):
        return jsonify(create_response(StateCode.USER_ALREADY_EXISTS)), 400

    #判断银行卡号是否可行
    if not checkBankCard(bank_card_no):
        return jsonify(create_response(StateCode.BANK_CARD_ERROR)), 400

    new_passenger = Passenger.create(data=data)
    # 为新创建的乘客生成访问令牌
    access_token = create_access_token(identity=new_passenger.id)
    user_presenter = PassengerPresenter(new_passenger, {"token": access_token}).as_dict()
    LogService.log()
    return jsonify(create_response(StateCode.SUCCESS, data=user_presenter)), 200


@register_bp.route('/auth', methods=['GET'])
@jwt_required()
def auth():
    current_user = get_jwt_identity()
    user = Passenger.query.get(current_user)
    if not user:
        return jsonify(create_response(StateCode.USER_NOT_FOUND)), 400
    user_presenter = PassengerPresenter(user, {}).as_dict()
    LogService.log()

    return jsonify(create_response(code=StateCode.SUCCESS, data=user_presenter)), 200

#校验验证码
def verifyCode(mobile_no, code):
    #获取到随机产生的验证码，再判断用户输入的验证码是否正确
    #这里的用到了redis,Redis是一种开源的、基于内存的键值存储，通常用于缓存、分布式存储 和 高效数据处理，
    #所以通过get方法获取到对应的值即验证码
    stored_code = redis_client.get(mobile_no)
    return stored_code == code