OPS MCP Server

Implementation Reference

• server_monitor/main.py:67-70 (handler)

  Registration and execution handler for the '_list_available_tools' MCP tool. This is a wrapper function decorated with @mcp.tool() that calls the helper list_available_tools(mcp) since the helper requires the MCP instance.

  # 特殊处理list_available_tools，因为它需要mcp实例
  @mcp.tool()
  def _list_available_tools():
      return list_available_tools(mcp)

• server_monitor/tools/utils.py:6-141 (helper)

  Helper function list_available_tools that implements the core logic for listing available tools by introspecting MCP tools or falling back to a hardcoded list of tool descriptions including schemas.

  def list_available_tools(mcp_instance) -> List[ToolInfo]:
      """列出所有可用的工具及其描述"""
      tools = []
  
      # 获取所有被装饰为工具的函数
      for tool_name in dir(mcp_instance.tool):
          if tool_name.startswith("__"):  # 跳过内部属性
              continue
  
          tool_func = getattr(mcp_instance.tool, tool_name, None)
          if callable(tool_func) and hasattr(tool_func, "__doc__") and tool_func.__doc__:
              # 获取参数信息
              params = []
              if hasattr(tool_func, "__annotations__"):
                  for param_name, param_type in tool_func.__annotations__.items():
                      if param_name != "return":
                          # 尝试获取默认值
                          default_value = None
                          if hasattr(tool_func, "__defaults__") and tool_func.__defaults__:
                              # 计算参数在默认值元组中的索引
                              param_index = list(tool_func.__annotations__.keys()).index(param_name) - len(tool_func.__annotations__.keys()) + len(tool_func.__defaults__)
                              if 0 <= param_index < len(tool_func.__defaults__):
                                  default_value = tool_func.__defaults__[param_index]
  
                          params.append({
                              "name": param_name,
                              "type": str(param_type),
                              "default": default_value
                          })
  
              tools.append({
                  "name": tool_name,
                  "description": tool_func.__doc__.strip(),
                  "parameters": params
              })
  
      # 手动列出所有工具，确保返回所有已定义的函数
      tool_descriptions = [
          {"name": "get_memory_info", "description": "获取本地服务器内存信息", "parameters": []},
          {"name": "remote_server_inspection", "description": "执行远程服务器巡检", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "inspection_modules", "type": "list[str]", "default": ["cpu", "memory", "disk"]},
              {"name": "timeout", "type": "int", "default": 30}
          ]},
          {"name": "get_system_load", "description": "获取系统负载信息", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "timeout", "type": "int", "default": 30}
          ]},
          {"name": "list_available_tools", "description": "列出所有可用的工具及其描述", "parameters": []},
          {"name": "monitor_processes", "description": "监控远程服务器进程，返回占用资源最多的进程", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "top_n", "type": "int", "default": 10},
              {"name": "sort_by", "type": "str", "default": "cpu"},
              {"name": "timeout", "type": "int", "default": 30}
          ]},
          {"name": "check_service_status", "description": "检查指定服务的运行状态", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "services", "type": "list[str]", "default": []},
              {"name": "timeout", "type": "int", "default": 30}
          ]},
          {"name": "inspect_network", "description": "检查网络接口和连接状态", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "timeout", "type": "int", "default": 30}
          ]},
          {"name": "analyze_logs", "description": "分析服务器日志文件中的错误和警告", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "log_file", "type": "str", "default": "/var/log/syslog"},
              {"name": "pattern", "type": "str", "default": "error|fail|critical"},
              {"name": "lines", "type": "int", "default": 100},
              {"name": "timeout", "type": "int", "default": 30}
          ]},
          {"name": "backup_critical_files", "description": "备份重要系统配置文件", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "files", "type": "list[str]", "default": ["/etc/passwd", "/etc/shadow", "/etc/fstab", "/etc/hosts"]},
              {"name": "backup_dir", "type": "str", "default": "/tmp/backup"},
              {"name": "timeout", "type": "int", "default": 60}
          ]},
          {"name": "security_vulnerability_scan", "description": "执行基础安全漏洞扫描", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "scan_type", "type": "str", "default": "basic"},
              {"name": "timeout", "type": "int", "default": 60}
          ]},
          {"name": "check_ssh_risk_logins", "description": "检查SSH登录风险，包括失败尝试和可疑IP", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "log_file", "type": "str", "default": "/var/log/auth.log"},
              {"name": "threshold", "type": "int", "default": 5},
              {"name": "timeout", "type": "int", "default": 30}
          ]},
          {"name": "check_firewall_config", "description": "检查防火墙配置和开放端口", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "timeout", "type": "int", "default": 30}
          ]},
          {"name": "get_os_details", "description": "获取操作系统详细信息", "parameters": [
              {"name": "hostname", "type": "str", "default": None},
              {"name": "username", "type": "str", "default": None},
              {"name": "password", "type": "str", "default": ""},
              {"name": "port", "type": "int", "default": 22},
              {"name": "timeout", "type": "int", "default": 30}
          ]}
      ]
  
      # 如果自动检测的工具不足，则使用手动定义的工具列表
      if len(tools) < len(tool_descriptions):
          return tool_descriptions
  
      return tools