"""
API路由模块 - 处理所有API端点请求
"""
from flask import Blueprint, request, jsonify
from models.api_key import ApiKeyManager

# 创建API蓝图
api_bp = Blueprint('api', __name__, url_prefix='/api')

@api_bp.route('/keys', methods=['GET'])
def get_api_keys():
    """获取所有API密钥"""
    return jsonify(ApiKeyManager.get_all_keys())

@api_bp.route('/keys', methods=['POST'])
def add_api_key():
    """添加新的API密钥"""
    data = request.json
    
    new_key = ApiKeyManager.add_key(
        platform=data.get("platform"),
        name=data.get("name"),
        key=data.get("key")
    )
    
    return jsonify({"success": True, "key": new_key})

@api_bp.route('/keys/<key_id>', methods=['DELETE'])
def delete_api_key(key_id):
    """删除指定的API密钥"""
    success = ApiKeyManager.delete_key(key_id)
    
    return jsonify({"success": success})

@api_bp.route('/keys/bulk-delete', methods=['POST'])
def bulk_delete_api_keys():
    """批量删除多个API密钥"""
    data = request.json
    key_ids = data.get("ids", [])
    
    if not key_ids:
        return jsonify({"success": False, "error": "没有提供要删除的密钥ID"}), 400
    
    deleted_count = ApiKeyManager.bulk_delete(key_ids)
    
    return jsonify({
        "success": True, 
        "deleted_count": deleted_count,
        "message": f"成功删除 {deleted_count} 个API密钥"
    })

@api_bp.route('/keys/bulk-add', methods=['POST'])
def bulk_add_api_keys():
    """批量添加多个API密钥"""
    data = request.json
    keys_data = data.get("keys", [])
    
    if not keys_data:
        return jsonify({"success": False, "error": "没有提供要添加的密钥数据"}), 400
    
    added_keys = ApiKeyManager.bulk_add_keys(keys_data)
    
    return jsonify({
        "success": True,
        "added_count": len(added_keys),
        "keys": added_keys,
        "message": f"成功添加 {len(added_keys)} 个API密钥"
    })

@api_bp.route('/keys/update/<key_id>', methods=['POST', 'GET'])
def update_api_key_status(key_id):
    """更新API密钥状态"""
    from core.update import update
    try:
        print(f"正在更新密钥ID: {key_id}")
        result = update(key_id)
        print(f"更新结果: {result}")
        return jsonify({"success": True, "data": result})
    except Exception as e:
        import traceback
        error_msg = str(e)
        trace = traceback.format_exc()
        print(f"更新密钥时出错: {error_msg}\n{trace}")
        return jsonify({"success": False, "error": error_msg, "traceback": trace}), 500

@api_bp.route('/keys/<key_id>', methods=['PUT'])
def edit_api_key(key_id):
    """更新API密钥信息"""
    data = request.json
    
    updated_key = ApiKeyManager.update_key(
        key_id=key_id,
        name=data.get("name"),
        key=data.get("key")
    )
    
    if not updated_key:
        return jsonify({"success": False, "error": "未找到指定的密钥"}), 404
    
    return jsonify({"success": True, "key": updated_key})