Cloud Manage MCP Server

Instructions

通用的实例电源管理函数（支持所有云平台）

Args:
    provider (str): 云服务提供商 ('digitalocean', 'vultr', 'alibaba')
    instance_id (str): 实例ID
    action (str): 操作类型 ('power_on', 'power_off', 'reboot', 'shutdown')
    ip_confirmation (str): 确认IP地址
    name_confirmation (str): 确认实例名称
    operation_confirmation (str): 确认操作类型
    
Returns:
    Dict: 操作结果

Implementation Reference

• main.py:220-323 (handler)

  Primary handler and registration for the 'manage_instance_power' MCP tool. Validates provider and action, dispatches to provider-specific power management functions in DigitalOcean, Vultr, and Alibaba providers with triple confirmation security checks.

  @mcp.tool()
  def manage_instance_power(
      provider: str, 
      instance_id: str, 
      action: str,
      ip_confirmation: str = "", 
      name_confirmation: str = "", 
      operation_confirmation: str = ""
  ) -> Dict:
      """
      通用的实例电源管理函数（支持所有云平台）
      
      Args:
          provider (str): 云服务提供商 ('digitalocean', 'vultr', 'alibaba')
          instance_id (str): 实例ID
          action (str): 操作类型 ('power_on', 'power_off', 'reboot', 'shutdown')
          ip_confirmation (str): 确认IP地址
          name_confirmation (str): 确认实例名称
          operation_confirmation (str): 确认操作类型
          
      Returns:
          Dict: 操作结果
      """
      provider_name = provider.lower()
      
      # AWS不支持电源管理
      if provider_name == 'aws':
          return {
              'error': 'AWS平台仅支持只读查询，不允许执行电源管理操作',
              'provider': 'aws',
              'suggestion': '如需管理AWS实例，请使用AWS控制台或CLI'
          }
      
      if provider_name not in ['digitalocean', 'vultr', 'alibaba']:
          return {
              'error': f'不支持的云服务提供商: {provider_name}',
              'supported_providers': ['digitalocean', 'vultr', 'alibaba']
          }
      
      if action not in ['power_on', 'power_off', 'reboot', 'shutdown']:
          return {
              'error': f'不支持的操作类型: {action}',
              'supported_actions': ['power_on', 'power_off', 'reboot', 'shutdown']
          }
      
      provider_obj = PROVIDERS[provider_name]
      provider_info = get_cloud_provider_info(provider_name)
      
      # 检查提供商是否可用
      if not getattr(provider_obj, 'available', False):
          error_msg = getattr(provider_obj, 'error', '提供商不可用')
          return {
              'error': f'{provider_info["name"]} 提供商不可用: {error_msg}',
              'provider': provider_name
          }
      
      print(f"🎯 {provider_info['name']} 电源管理: {action} for {instance_id}")
      
      try:
          # 调用对应提供商的电源管理方法
          if provider_name == 'digitalocean':
              droplet_id = int(instance_id) if instance_id.isdigit() else None
              if not droplet_id:
                  return {'error': 'DigitalOcean Droplet ID必须是数字'}
                  
              if action == 'power_on':
                  return provider_obj.power_on_droplet(droplet_id, ip_confirmation, name_confirmation, operation_confirmation)
              elif action == 'power_off':
                  return provider_obj.power_off_droplet(droplet_id, ip_confirmation, name_confirmation, operation_confirmation)
              elif action == 'reboot':
                  return provider_obj.reboot_droplet(droplet_id, ip_confirmation, name_confirmation, operation_confirmation)
              elif action == 'shutdown':
                  return provider_obj.shutdown_droplet(droplet_id, ip_confirmation, name_confirmation, operation_confirmation)
                  
          elif provider_name == 'vultr':
              if action == 'power_on':
                  return provider_obj.power_on_instance(instance_id, ip_confirmation, name_confirmation, operation_confirmation)
              elif action == 'power_off':
                  return provider_obj.power_off_instance(instance_id, ip_confirmation, name_confirmation, operation_confirmation)
              elif action == 'reboot':
                  return provider_obj.reboot_instance(instance_id, ip_confirmation, name_confirmation, operation_confirmation)
              elif action == 'shutdown':
                  # Vultr可能不支持优雅关闭，使用强制关闭
                  return provider_obj.power_off_instance(instance_id, ip_confirmation, name_confirmation, operation_confirmation)
                  
          elif provider_name == 'alibaba':
              if action == 'power_on':
                  return provider_obj.power_on_instance(instance_id, ip_confirmation, name_confirmation, operation_confirmation)
              elif action == 'power_off':
                  return provider_obj.power_off_instance(instance_id, ip_confirmation, name_confirmation, operation_confirmation)
              elif action == 'reboot':
                  return provider_obj.reboot_instance(instance_id, ip_confirmation, name_confirmation, operation_confirmation)
              elif action == 'shutdown':
                  # 阿里云使用power_off作为关闭操作
                  return provider_obj.power_off_instance(instance_id, ip_confirmation, name_confirmation, operation_confirmation)
          
      except Exception as e:
          return {
              'error': f'执行 {action} 操作时发生错误: {str(e)}',
              'provider': provider_name,
              'instance_id': instance_id,
              'action': action
          }