Cloud Manage MCP Server

Instructions

Implementation Reference

• main.py:432-444 (registration)

  MCP tool registration and thin handler for shutdown_digitalocean_droplet, delegates to digitalocean_provider.shutdown_droplet

  @mcp.tool()
  def shutdown_digitalocean_droplet(
      droplet_id: int, 
      ip_confirmation: str = "", 
      name_confirmation: str = "", 
      operation_confirmation: str = ""
  ) -> Dict:
      """
      优雅关闭DigitalOcean Droplet（需要三次确认）
      """
      return digitalocean_provider.shutdown_droplet(
          droplet_id, ip_confirmation, name_confirmation, operation_confirmation
      )

• providers/digitalocean_provider.py:246-258 (handler)

  Provider class method that handles the shutdown_digitalocean_droplet logic by calling the generic power operation with 'shutdown' action

  def shutdown_droplet(
      self, 
      droplet_id: int, 
      ip_confirmation: str = "", 
      name_confirmation: str = "", 
      operation_confirmation: str = ""
  ) -> Dict:
      """
      优雅关闭Droplet（需要三次确认）
      """
      return self._execute_power_operation(
          droplet_id, 'shutdown', ip_confirmation, name_confirmation, operation_confirmation
      )

• providers/digitalocean_provider.py:274-356 (handler)

  Core handler implementing the shutdown logic: triple security confirmation, API call to DigitalOcean droplet_actions.post with type='shutdown', and response handling

  def _execute_power_operation(
      self, 
      droplet_id: int, 
      operation: str, 
      ip_confirmation: str, 
      name_confirmation: str, 
      operation_confirmation: str
  ) -> Dict:
      """
      执行电源操作的通用函数
      """
      if not self.available:
          return {
              'error': f'DigitalOcean服务不可用: {getattr(self, "error", "未知错误")}',
              'provider': 'digitalocean'
          }
      
      # 首先获取droplet信息
      try:
          droplet_response = self.client.droplets.get(droplet_id)
          droplet = droplet_response.get("droplet", {})
          
          if not droplet:
              return {
                  'error': f'未找到ID为 {droplet_id} 的Droplet',
                  'provider': 'digitalocean'
              }
          
          # 格式化droplet信息用于确认
          droplet_info = self._format_droplet_for_confirmation(droplet)
          
      except Exception as e:
          return {
              'error': f'获取Droplet信息时发生错误: {str(e)}',
              'provider': 'digitalocean'
          }
      
      # 检查是否提供了确认信息
      if not ip_confirmation or not name_confirmation or not operation_confirmation:
          # 返回确认要求
          return require_triple_confirmation(droplet_info, operation)
      
      # 验证确认信息
      security = SecurityConfirmation()
      is_valid, error_message = security.validate_power_operation(
          droplet_info, operation, ip_confirmation, name_confirmation, operation_confirmation
      )
      
      if not is_valid:
          return {
              'error': f'确认验证失败: {error_message}',
              'provider': 'digitalocean',
              'requires_confirmation': True
          }
      
      # 执行实际操作
      try:
          action_data = {"type": operation}
          response = self.client.droplet_actions.post(droplet_id=droplet_id, body=action_data)
          
          action = response.get("action", {})
          
          return {
              'provider': 'digitalocean',
              'droplet_id': droplet_id,
              'operation_success': True,
              'action': {
                  'id': action.get("id"),
                  'status': action.get("status"),
                  'type': action.get("type"),
                  'started_at': action.get("started_at"),
                  'resource_id': action.get("resource_id")
              },
              'message': f'已成功提交 {operation} 操作，操作ID: {action.get("id")}',
              'confirmation_validated': True
          }
          
      except Exception as e:
          return {
              'error': f'执行 {operation} 操作时发生错误: {str(e)}',
              'provider': 'digitalocean'
          }