"""
限流器使用示例
"""
from flask import jsonify
from iti.applications.extensions.limit import limiter


def create_example_app():
    """创建示例应用"""
    from iti.applications import create_app
    return create_app('dev')


app = create_example_app()


@app.route('/')
def index():
    """首页 - 使用默认限流"""
    return jsonify({'message': 'Hello, World!'})


@app.route('/api/public')
@limiter.limit("10 per minute")  # 自定义限流：每分钟10次
def public_api():
    """公共 API - 自定义限流"""
    return jsonify({'message': 'Public API', 'data': 'some data'})


@app.route('/api/private')
@limiter.limit("5 per minute")  # 更严格的限流：每分钟5次
def private_api():
    """私有 API - 更严格的限流"""
    return jsonify({'message': 'Private API', 'data': 'sensitive data'})


@app.route('/api/unlimited')
@limiter.exempt  # 豁免限流
def unlimited_api():
    """无限制 API - 豁免限流"""
    return jsonify({'message': 'Unlimited API', 'data': 'no limits'})


@app.route('/api/conditional')
@limiter.limit("20 per hour", per_method=True)  # 按方法限流
def conditional_api():
    """条件限流 API - 按方法限流"""
    return jsonify({'message': 'Conditional API', 'data': 'method specific'})


@app.route('/api/user-specific')
@limiter.limit("100 per hour", key_func=lambda: "user_123")  # 自定义键函数
def user_specific_api():
    """用户特定 API - 自定义键函数"""
    return jsonify({'message': 'User specific API', 'user': 'user_123'})


@app.route('/api/rate-limit-info')
def rate_limit_info():
    """获取限流信息"""
    if limiter is None:
        return jsonify({'message': 'Rate limiting is disabled'})
    
    return jsonify({
        'message': 'Rate limiting is enabled',
        'default_limits': limiter.default_limits,
        'storage_uri': limiter.storage_uri,
        'key_func': limiter.key_func.__name__ if limiter.key_func else None
    })


if __name__ == '__main__':
    print("🚀 启动限流器示例应用")
    print("📊 访问 http://localhost:5000/api/rate-limit-info 查看限流配置")
    print("🔒 访问其他端点测试限流功能")
    app.run(debug=True)