feat: add sandbox_stop tool for managing container lifecycle

- Introduce `sandbox_stop` tool to stop and remove running container sandboxes.
- Update README.md to document the new tool and its parameters.
- Implement functionality in `stop-container.go` to handle container stopping and removal with error handling.

Author: VikashLoomba

README.md

@@ -95,6 +95,15 @@ Copy a single file to the sandboxed filesystem.
 - `local_src_file` (string, required): Path to a file in the local file system
 - `dest_path` (string, optional): Path to save the file in the sandbox environment
 
+#### `sandbox_stop`
+Stop and remove a running container sandbox.
+
+**Parameters:**
+- `container_id` (string, required): ID of the container to stop and remove
+
+**Description:**
+Gracefully stops the specified container with a 10-second timeout and removes it along with its volumes.
+
 #### Container Logs Resource
 A dynamic resource that provides access to container logs.
 

src/code-sandbox-mcp/main.go

@@ -142,6 +142,18 @@ func main() {
 		),
 	)
 
+	// Stop and remove a container
+	stopContainerTool := mcp.NewTool("sandbox_stop",
+		mcp.WithDescription(
+			"Stop and remove a running container sandbox. \n"+
+				"Gracefully stops the specified container and removes it along with its volumes.",
+		),
+		mcp.WithString("container_id",
+			mcp.Required(),
+			mcp.Description("ID of the container to stop and remove"),
+		),
+	)
+
 	// Register dynamic resource for container logs
 	// Dynamic resource example - Container Logs by ID
 	containerLogsTemplate := mcp.NewResourceTemplate(
@@ -158,6 +170,7 @@ func main() {
 	s.AddTool(writeFileTool, tools.WriteFile)
 	s.AddTool(execTool, tools.Exec)
 	s.AddTool(copyFileTool, tools.CopyFile)
+	s.AddTool(stopContainerTool, tools.StopContainer)
 	switch *transport {
 	case "stdio":
 		if err := server.ServeStdio(s); err != nil {

src/code-sandbox-mcp/tools/stop-container.go

@@ -0,0 +1,54 @@
+package tools
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/docker/docker/api/types/container"
+	"github.com/docker/docker/client"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+// StopContainer stops and removes a container by its ID
+func StopContainer(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	// Get the container ID from the request
+	containerId, ok := request.Params.Arguments["container_id"].(string)
+	if !ok || containerId == "" {
+		return mcp.NewToolResultText("Error: container_id is required"), nil
+	}
+
+	// Stop and remove the container
+	if err := stopAndRemoveContainer(ctx, containerId); err != nil {
+		return mcp.NewToolResultText(fmt.Sprintf("Error: %v", err)), nil
+	}
+
+	return mcp.NewToolResultText(fmt.Sprintf("Successfully stopped and removed container: %s", containerId)), nil
+}
+
+// stopAndRemoveContainer stops and removes a Docker container
+func stopAndRemoveContainer(ctx context.Context, containerId string) error {
+	cli, err := client.NewClientWithOpts(
+		client.FromEnv,
+		client.WithAPIVersionNegotiation(),
+	)
+	if err != nil {
+		return fmt.Errorf("failed to create Docker client: %w", err)
+	}
+	defer cli.Close()
+
+	// Stop the container with a timeout
+	timeout := 10 // seconds
+	if err := cli.ContainerStop(ctx, containerId, container.StopOptions{Timeout: &timeout}); err != nil {
+		return fmt.Errorf("failed to stop container: %w", err)
+	}
+
+	// Remove the container
+	if err := cli.ContainerRemove(ctx, containerId, container.RemoveOptions{
+		RemoveVolumes: true,
+		Force:         true,
+	}); err != nil {
+		return fmt.Errorf("failed to remove container: %w", err)
+	}
+
+	return nil
+}