mirror of
https://github.com/hanzoai/runtime.git
synced 2026-10-30 18:23:13 +00:00
zeekay
ea80dc3638
- Add main Dockerfile for production builds - Add Dockerfile.runtime for secure code execution - Add Dockerfile.runtime.fixed with improved configuration - Include .dockerignore for build optimization - Add integration documentation and publishing guides - Update API and CLI components for better runtime support - Enhance sandbox and container management features
91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
// Copyright 2025 Hanzo Industries Inc.
|
|
// SPDX-License-Identifier: AGPL-3.0
|
|
|
|
package mcp
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/hanzoai/runtime/cli/cmd/mcp/agents"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var InitCmd = &cobra.Command{
|
|
Use: "init [AGENT_NAME]",
|
|
Short: "Initialize Runtime MCP Server with an agent (currently supported: claude, windsurf, cursor)",
|
|
Args: cobra.MaximumNArgs(1),
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
if len(args) == 0 {
|
|
return fmt.Errorf("agent name is required")
|
|
}
|
|
|
|
homeDir, err := os.UserHomeDir()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var agentConfigFilePath, mcpLogFilePath string
|
|
|
|
switch args[0] {
|
|
case "claude":
|
|
agentConfigFilePath, mcpLogFilePath, err = agents.InitClaude(homeDir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case "cursor":
|
|
agentConfigFilePath, mcpLogFilePath, err = agents.InitCursor(homeDir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case "windsurf":
|
|
agentConfigFilePath, mcpLogFilePath, err = agents.InitWindsurf(homeDir)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
return fmt.Errorf("agent name %s is not supported", args[0])
|
|
}
|
|
|
|
return injectConfig(agentConfigFilePath, mcpLogFilePath)
|
|
},
|
|
}
|
|
|
|
func injectConfig(agentConfigFilePath, mcpLogFilePath string) error {
|
|
runtimeMcpConfig, err := getDayonaMcpConfig(mcpLogFilePath)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Read existing model config or create new one
|
|
var agentConfig map[string]interface{}
|
|
if agentConfigData, err := os.ReadFile(agentConfigFilePath); err == nil {
|
|
if err := json.Unmarshal(agentConfigData, &agentConfig); err != nil {
|
|
return err
|
|
}
|
|
} else if !os.IsNotExist(err) {
|
|
return err
|
|
} else {
|
|
agentConfig = make(map[string]interface{})
|
|
}
|
|
|
|
// Initialize or update mcpServers field
|
|
mcpServers, ok := agentConfig["mcpServers"].(map[string]interface{})
|
|
if !ok {
|
|
mcpServers = make(map[string]interface{})
|
|
}
|
|
|
|
// Add or update runtime-mcp configuration
|
|
mcpServers["runtime-mcp"] = runtimeMcpConfig
|
|
agentConfig["mcpServers"] = mcpServers
|
|
|
|
// Write back the updated config with indentation
|
|
updatedJSON, err := json.MarshalIndent(agentConfig, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return os.WriteFile(agentConfigFilePath, updatedJSON, 0644)
|
|
}
|