from fastapi import Depends, APIRouter, HTTPException
from sqlalchemy.orm import Session

from context.common import email_verify_code, phone_verify_code, auth_util, AdminUserTokenDataModel
from utils.verify_code_utils import PhoneVerifyType, EmailVerifyType
from .common import get_db
from .mods.user.models import User
from .mods.user import crud as user_crud
from .mods.user.router import router as user_router
from .mods.post.router import router as post_router
from .mods.department.router import router as department_router
from .mods.auth_rule.router import router as auth_rule_router
from . import schemas

router = APIRouter(prefix="/user")
router.include_router(user_router)
router.include_router(post_router)
router.include_router(department_router)
router.include_router(auth_rule_router)


@router.post("/get_login_verify_code", tags=["用户登录"], summary="获取登录验证码")
def get_login_verify_code(body: schemas.GetLoginVerifyCodeReq, db: Session = Depends(get_db)):
    try:
        account = body.account
        user = db.query(User).filter(User.email == account).first()
        if user:
            account_type = 'email'
        else:
            user = db.query(User).filter(User.phone == account).first()
            if user:
                account_type = "phone"
            else:
                raise HTTPException(detail="账号未录入系统", status_code=303)
        if account_type == 'email':
            email_verify_code.send_code(account, EmailVerifyType.login)
        if account_type == 'phone':
            phone_verify_code.send_code(account, PhoneVerifyType.login)
    except HTTPException as e:
        raise e
    except Exception as e:
        print(e)
        raise HTTPException(detail="验证码发送失败", status_code=403)
    return {"msg": f"验证码已发送至{'邮箱' if account_type == 'email' else '手机'},请查看", "state": 1}


@router.post('/login_by_verify_code', tags=["用户登录"], summary='验证码登录')
def login_by_verify_code(req: schemas.LoginByVerifyCode,
                         db: Session = Depends(get_db)):
    account = req.account
    if phone_verify_code.check_code(account, req.code, PhoneVerifyType.login):
        account_type = 'phone'
    else:
        if email_verify_code.check_code(account, req.code, EmailVerifyType.login):
            account_type = 'email'
        else:
            raise HTTPException(detail="验证码错误", status_code=303)
    user = None
    if account_type == 'phone':
        user = db.query(User).filter(User.phone == account).first()
    if account_type == 'email':
        user = db.query(User).filter(User.email == account).first()
    if not user:
        raise HTTPException(detail="账号未录入系统", status_code=303)
    auth_data = user_crud.get_user_auth(user)
    user_data = user.to_full_dict(relation_use_id=True)
    user_data["auth_data"] = auth_data
    token_data = auth_util.tokenDataModel(**user_data).dict()
    token = auth_util.create_token(token_data)
    return {'token': token, 'user_info': user_data}


@router.post('/refresh_token', tags=["用户登录"], summary='刷新token')
def refresh_token(token: AdminUserTokenDataModel = Depends(auth_util.token_data_depend), db: Session = Depends(get_db)):
    user = db.query(User).filter(User.id == token.id).first()
    if not user:
        raise HTTPException(detail="账号未录入系统", status_code=303)
    auth_data = user_crud.get_user_auth(user)
    user_data = user.to_full_dict(relation_use_id=True)
    user_data["auth_data"] = auth_data
    token_data = auth_util.tokenDataModel(**user_data).dict()
    token = auth_util.create_token(token_data)
    return {'token': token, 'user_info': user_data}