from flask import Flask, jsonify, request
from flask_cors import CORS
from flask_jwt_extended import JWTManager, get_jwt_identity, verify_jwt_in_request
from flask_jwt_extended.exceptions import JWTDecodeError, NoAuthorizationError, InvalidHeaderError
from config import Config
from models import db
from routes.auth import auth_bp
from routes.tokens import tokens_bp
from routes.dashboard import dashboard_bp
from routes.logs import logs_bp
from routes.recharge import recharge_bp
from routes.user import user_bp
from routes.models import models_bp
import logging

# 配置日志
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

def create_app():
    app = Flask(__name__)
    app.config.from_object(Config)
    
    # 请求前钩子：记录请求信息（用于调试）
    @app.before_request
    def log_request():
        if request.path.startswith('/api/') and request.path != '/api/auth/login' and request.path != '/api/auth/register':
            auth_header = request.headers.get('Authorization', 'None')
            logger.info(f"[{request.method}] {request.path}")
            logger.info(f"Authorization Header: {auth_header[:80] if auth_header != 'None' else 'Missing'}")
    
    # 初始化扩展
    db.init_app(app)
    # CORS配置：允许所有来源，并支持credentials
    CORS(app, 
         resources={r"/api/*": {
             "origins": "*",
             "methods": ["GET", "POST", "PUT", "DELETE", "OPTIONS"],
             "allow_headers": ["Content-Type", "Authorization"],
             "supports_credentials": True
         }},
         supports_credentials=True)
    jwt = JWTManager(app)
    
    # JWT错误处理器
    @jwt.expired_token_loader
    def expired_token_callback(jwt_header, jwt_payload):
        logger.warning(f"Token expired: {jwt_payload}")
        return jsonify({'error': 'Token已过期，请重新登录'}), 401
    
    @jwt.invalid_token_loader
    def invalid_token_callback(error):
        auth_header = request.headers.get('Authorization', 'None')
        logger.error(f"Invalid token error: {str(error)}")
        logger.error(f"Authorization header: {auth_header[:100] if auth_header != 'None' else 'Missing'}")
        return jsonify({
            'error': '无效的Token',
            'detail': str(error),
            'hint': '请检查Authorization头格式是否正确（应为: Bearer <token>）'
        }), 422
    
    @jwt.unauthorized_loader
    def missing_token_callback(error):
        logger.warning(f"Missing token: {str(error)}")
        auth_header = request.headers.get('Authorization', 'None')
        logger.warning(f"Authorization header: {auth_header[:100] if auth_header != 'None' else 'Missing'}")
        return jsonify({
            'error': '缺少认证Token，请先登录',
            'detail': str(error)
        }), 401
    
    @jwt.needs_fresh_token_loader
    def token_not_fresh_callback(jwt_header, jwt_payload):
        logger.warning(f"Token not fresh: {jwt_payload}")
        return jsonify({'error': 'Token需要刷新'}), 401
    
    # 注册蓝图
    app.register_blueprint(auth_bp, url_prefix='/api/auth')
    app.register_blueprint(tokens_bp, url_prefix='/api/tokens')
    app.register_blueprint(dashboard_bp, url_prefix='/api/dashboard')
    app.register_blueprint(logs_bp, url_prefix='/api/logs')
    app.register_blueprint(recharge_bp, url_prefix='/api/recharge')
    app.register_blueprint(user_bp, url_prefix='/api/user')
    app.register_blueprint(models_bp, url_prefix='/api/models')
    
    # 创建数据库表
    with app.app_context():
        db.create_all()
    
    return app

if __name__ == '__main__':
    app = create_app()
    app.run(debug=True, host='0.0.0.0', port=5000)