Megaraptor MCP

Instructions

Implementation Reference

• src/megaraptor_mcp/tools/deployment.py:1269-1362 (handler)

  The handler function `generate_server_config` implementation in `src/megaraptor_mcp/tools/deployment.py`.

  async def generate_server_config(
      deployment_id: str,
      output_format: str = "yaml",
  ) -> list[TextContent]:
      """Generate Velociraptor server configuration file.
  
      Args:
          deployment_id: The deployment to generate config for
          output_format: Output format - 'yaml' or 'json'
  
      Returns:
          Server configuration content.
      """
      try:
          from ..deployment.security import CertificateManager
          from ..deployment.deployers import DockerDeployer
  
          # Get deployment info
          deployer = DockerDeployer()
          info = await deployer.get_status(deployment_id)
  
          if not info:
              return [TextContent(
                  type="text",
                  text=json.dumps({
                      "error": f"Deployment not found: {deployment_id}",
                      "hint": "Use list_deployments tool to see available deployments"
                  }, indent=2)
              )]
  
          # Load certificates
          cert_manager = CertificateManager()
          bundle = cert_manager.load_bundle(deployment_id)
  
          if not bundle:
              return [TextContent(
                  type="text",
                  text=json.dumps({
                      "error": "Certificate bundle not found"
                  }, indent=2)
              )]
  
          # Generate config
          import yaml
          config = {
              "version": {"name": "megaraptor-deployment"},
              "Client": {
                  "server_urls": [info.server_url.replace("/api/", "") + ":8000/"],
                  "ca_certificate": bundle.ca_cert,
              },
              "API": {
                  "bind_address": "0.0.0.0:8889",
                  "bind_scheme": "https",
              },
              "GUI": {
                  "bind_address": "0.0.0.0:8889",
                  "bind_scheme": "https",
                  "public_url": info.server_url,
              },
              "Frontend": {
                  "bind_address": "0.0.0.0:8000",
              },
              "ca_certificate": bundle.ca_cert,
          }
  
          if output_format == "json":
              output = json.dumps(config, indent=2)
          else:
              output = yaml.dump(config, default_flow_style=False)
  
          return [TextContent(
              type="text",
              text=output
          )]
  
      except ImportError as e:
          return [TextContent(
              type="text",
              text=json.dumps({
                  "error": f"Missing dependency: {str(e)}",
                  "hint": "Install required packages with: pip install megaraptor-mcp[deployment]"
              }, indent=2)
          )]
  
      except Exception:
          # Generic errors - don't expose internals
          return [TextContent(
              type="text",
              text=json.dumps({
                  "error": "Operation failed",
                  "hint": "Check deployment configuration and try again"
              }, indent=2)
          )]