refactor: switch to stdio-based server and update language configurations

- Modify main.go to use stdio-based server instead of SSE server
- Update Go language configuration in types.go with newer image and run command
- Simplify installer configuration for code-sandbox-mcp
- Add logging for command execution in run_code.go
- Adjust Go code running process in tools/run_code.go

Author: VikashLoomba

src/code-sandbox-mcp/installer/install.go

@@ -68,10 +68,9 @@ func InstallConfig() error {
 			Env:     map[string]string{},
 		}
 	} else {
-		command = "/usr/bin/env"
 		config.MCPServers["code-sandbox-mcp"] = MCPServer{
-			Command: command,
-			Args:    []string{execPath},
+			Command: execPath,
+			Args:    []string{},
 			Env:     map[string]string{},
 		}
 	}

src/code-sandbox-mcp/languages/types.go

@@ -32,10 +32,10 @@ var SupportedLanguages = map[Language]LanguageConfig{
 		RunCommand:      []string{"python3", "-c"},
 	},
 	Go: {
-		Image:           "golang:1.21-alpine",
+		Image:           "docker.io/library/golang:1.23.6-bookworm",
 		DependencyFiles: []string{"go.mod"},
-		InstallCommand:  []string{"go mod init &&", "go", "mod", "download"},
-		RunCommand:      []string{"go", "run", "."},
+		InstallCommand:  []string{"go mod init &&", "go", "mod", "tidy"},
+		RunCommand:      []string{"go", "run", "main.go"},
 	},
 	NodeJS: {
 		Image:           "oven/bun:debian",

src/code-sandbox-mcp/main.go

@@ -53,10 +53,10 @@ func init() {
 }
 
 func main() {
-	port := flag.String("port", "9520", "Port to listen on")
+	// port := flag.String("port", "9520", "Port to listen on")
 	flag.Parse()
 	s := server.NewMCPServer("code-sandbox-mcp", "v1.0.0", server.WithLogging(), server.WithResourceCapabilities(true, true))
-	sseServer := server.NewSSEServer(s, fmt.Sprintf("http://localhost:%s", *port))
+	// sseServer := server.NewSSEServer(s, fmt.Sprintf("http://localhost:%s", *port))
 	// Register a tool to run code in a docker container
 	runCodeTool := mcp.NewTool("run_code",
 		mcp.WithDescription(
@@ -113,8 +113,8 @@ func main() {
 	s.AddTool(runCodeTool, tools.RunCodeSandbox)
 	s.AddTool(runProjectTool, tools.RunProjectSandbox)
 
-	if err := sseServer.Start(fmt.Sprintf(":%s", *port)); err != nil {
-		fmt.Printf("(sse) Server error: %v\n", err)
+	if err := server.ServeStdio(s); err != nil {
+		fmt.Printf("(stdio) Server error: %v", err)
 	}
 
 }

src/code-sandbox-mcp/tools/run_code.go

@@ -5,6 +5,7 @@ import (
 	"context"
 	"fmt"
 	"io"
+	"log"
 	"os"
 	"path/filepath"
 	"strings"
@@ -48,7 +49,7 @@ func RunCodeSandbox(ctx context.Context, request mcp.CallToolRequest) (*mcp.Call
 	escapedCode := strings.ToValidUTF8(code, "")
 	if parsed == languages.Go {
 		// For Go, we need to write the code to a file
-		cmd = []string{"/bin/sh", "-c", "go mod init sandbox && go mod tidy && go run main.go"}
+		cmd = []string{"go", "run", "main.go"}
 	} else if parsed == languages.Python {
 		// For Python leverage something like https://github.com/tliron/py4go to run pipreqs (https://github.com/bndr/pipreqs)
 		// natively to generate requirements.txt
@@ -62,7 +63,7 @@ func RunCodeSandbox(ctx context.Context, request mcp.CallToolRequest) (*mcp.Call
 			"progressToken": progressToken,
 		},
 	)
-
+	log.Println("Running Command: ", cmd)
 	logs, err := runInDocker(ctx, progressToken, cmd, config.Image, escapedCode, parsed, nil)
 	server.SendNotificationToClient(
 		"notifications/progress",
@@ -94,7 +95,6 @@ func runInDocker(ctx context.Context, progressToken mcp.ProgressToken, cmd []str
 		return "", fmt.Errorf("failed to pull Docker image %s: %w", dockerImage, err)
 	}
 	io.Copy(os.Stdout, reader)
-
 	// Create container config
 	config := &container.Config{
 		Image: dockerImage,