from fastapi import APIRouter, Depends, Response, Request
from sqlalchemy.orm import Session
from app.models.database import get_db
from app.models.user import UserInfo,UserDB, PhoneLoginRequest, generate_user_code
from app.models.order import ShippingOrderDB, OrderStatus
from app.core.response import success_response, error_response, ResponseModel
from app.core.wechat import WeChatClient,generate_random_string
from app.core.security import create_access_token, set_jwt_cookie
from pydantic import BaseModel
import json
import time
from datetime import datetime, timezone
from app.api.deps import get_current_user
from app.core.config import settings
import random
import string

router = APIRouter()

class PhoneNumberRequest(BaseModel):
    login_code: str  # 登录凭证
    phone_code: str  # 手机号验证码
    referral_code: str = None  # 推荐码（可选）

@router.post("/phone-login", response_model=ResponseModel)
async def wechat_phone_login(
    request: PhoneNumberRequest,
    db: Session = Depends(get_db),
    response: Response = None
):
    """通过微信手机号登录/注册"""
    try:
        # 初始化微信客户端
        wechat = WeChatClient()
        
        # 获取用户 openid
        session_info = await wechat.code2session(request.login_code)
        openid = session_info["openid"]
        
        # 获取用户手机号
        phone_info = await wechat.get_phone_number(request.phone_code)
        
        if not phone_info:
            return error_response(code=400, message="获取手机号失败")
        
        # 打印调试信息
        print(f"获取到的手机号信息: {phone_info}")
        
        phone = phone_info['phone_number']
        if not phone:
            return error_response(code=400, message="手机号为空")
        
        # 查找或创建用户
        user = db.query(UserDB).filter(UserDB.phone == phone).first()
        if not user:
            # 生成用户编码
            user_code = generate_user_code(db)
            
            user = UserDB(
                nickname=f"蜂友{phone[-4:]}",
                phone=phone,
                user_code=user_code,
                referral_code=request.referral_code,
                openid=openid  # 保存 openid
            )
            db.add(user)
            db.flush()
            
            # 发放优惠券
            from app.api.endpoints.user import issue_register_coupons
            issue_register_coupons(db, user.userid)
            
            db.commit()
            db.refresh(user)
        else:
            # 更新现有用户的 openid
            user.openid = openid
            db.commit()
        
        # 创建访问令牌
        access_token = create_access_token(
            data={"phone": user.phone, "userid": user.userid}
        )
        
        # 设置JWT cookie
        if response:
            set_jwt_cookie(response, access_token)
        
        return success_response(
            message="登录成功",
            data={
                "user": UserInfo.model_validate(user),
                "access_token": access_token,
                "token_type": "bearer"
            }
        )
    except Exception as e:
        db.rollback()
        return error_response(code=500, message=f"登录失败: {str(e)}")