feat: Phase 1 - 核心功能实现

app/__init__.py

@@ -0,0 +1,117 @@
+"""
+PIT Router Flask 应用工厂
+"""
+from flask import Flask
+from flask_socketio import SocketIO
+import logging
+
+from app.config import config
+from app.extensions import (
+    db, migrate, jwt, login_manager, cors, limiter, init_redis
+)
+
+# Socket.IO 实例
+socketio = SocketIO(cors_allowed_origins="*", async_mode='threading')
+
+
+def create_app(config_name='default'):
+    """创建 Flask 应用实例"""
+    app = Flask(__name__)
+    app.config.from_object(config[config_name])
+    
+    # 初始化扩展
+    _init_extensions(app)
+    
+    # 注册蓝图
+    _register_blueprints(app)
+    
+    # 注册 Socket.IO 事件
+    _register_socketio_events()
+    
+    # 配置日志
+    _configure_logging(app)
+    
+    # 注册错误处理
+    _register_error_handlers(app)
+    
+    # 健康检查端点
+    @app.route('/health')
+    def health_check():
+        return {'status': 'ok', 'service': 'pit-router'}
+    
+    return app
+
+
+def _init_extensions(app):
+    """初始化 Flask 扩展"""
+    db.init_app(app)
+    migrate.init_app(app, db)
+    jwt.init_app(app)
+    login_manager.init_app(app)
+    cors.init_app(app)
+    limiter.init_app(app)
+    socketio.init_app(app)
+    
+    # 初始化 Redis
+    init_redis(app)
+    
+    # 创建数据库表
+    with app.app_context():
+        db.create_all()
+
+
+def _register_blueprints(app):
+    """注册蓝图"""
+    from app.routes.auth import auth_bp
+    from app.routes.sessions import sessions_bp
+    from app.routes.agents import agents_bp
+    from app.routes.gateways import gateways_bp
+    from app.routes.messages import messages_bp
+    from app.routes.stats import stats_bp
+    
+    app.register_blueprint(auth_bp, url_prefix='/api/auth')
+    app.register_blueprint(sessions_bp, url_prefix='/api/sessions')
+    app.register_blueprint(agents_bp, url_prefix='/api/agents')
+    app.register_blueprint(gateways_bp, url_prefix='/api/gateways')
+    app.register_blueprint(messages_bp, url_prefix='/api/messages')
+    app.register_blueprint(stats_bp, url_prefix='/api/stats')
+
+
+def _register_socketio_events():
+    """注册 Socket.IO 事件处理器"""
+    from app.socketio.handlers import register_handlers
+    register_handlers(socketio)
+
+
+def _configure_logging(app):
+    """配置日志"""
+    log_level = app.config.get('LOG_LEVEL', 'INFO')
+    logging.basicConfig(
+        level=getattr(logging, log_level),
+        format='%(asctime)s [%(levelname)s] %(name)s: %(message)s'
+    )
+
+
+def _register_error_handlers(app):
+    """注册错误处理器"""
+    from flask import jsonify
+    
+    @app.errorhandler(400)
+    def bad_request(error):
+        return jsonify({'error': 'Bad Request', 'message': str(error)}), 400
+    
+    @app.errorhandler(401)
+    def unauthorized(error):
+        return jsonify({'error': 'Unauthorized', 'message': str(error)}), 401
+    
+    @app.errorhandler(403)
+    def forbidden(error):
+        return jsonify({'error': 'Forbidden', 'message': str(error)}), 403
+    
+    @app.errorhandler(404)
+    def not_found(error):
+        return jsonify({'error': 'Not Found', 'message': str(error)}), 404
+    
+    @app.errorhandler(500)
+    def internal_error(error):
+        return jsonify({'error': 'Internal Server Error', 'message': str(error)}), 500