feat: add sandbox disk and network tools

Author: BhautikChudasama

README.md

@@ -221,6 +221,9 @@ The server exposes **85+ tools** organized into the following categories:
 | **GitHub** | `InstallGithubApp`, `ListConnectedGithubAccounts`, `ListGithubRepositories`, `ListGithubRepositoryBranches`, `GetGithubRepositoryContent` | GitHub integration for repo-based deployments |
 | **Account** | `GetCurrentUser`, `GetQuotas`, `GetSupportedProjectTypes` | User info, quotas, and platform capabilities |
 | **Transfers** | `TransferProject`, `GetProjectTransferUri`, `ListProjectTransferHistory` | Transfer project ownership between accounts |
+| **Sandboxes** | `CreateSandbox`, `UpdateSandbox`, `ExecSandbox`, `DeleteSandbox` | Create, update, execute commands in, and destroy sandbox VMs |
+| **Sandbox Disks** | `CreateDisk`, `ListDisks`, `GetDisk`, `DeleteDisk`, `AttachSandboxDisk`, `DetachSandboxDisk`, `ListSandboxDisks` | Register S3 disks and attach them to sandboxes |
+| **Sandbox Networks** | `CreateNetwork`, `ListNetworks`, `GetNetwork`, `DeleteNetwork`, `AttachSandboxNetwork`, `DetachSandboxNetwork` | Create private sandbox networks and manage membership |
 
 ---
 

handlers/AttachSandboxDisk.go

@@ -0,0 +1,28 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+
+	mcputils "github.com/NodeOps-app/createos-mcp/helpers"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+type AttachSandboxDiskParams struct {
+	ID   string                 `json:"id"`
+	Body map[string]interface{} `json:"body"`
+}
+
+func AttachSandboxDiskHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	authInfo, args, err := handleRequest(ctx, request)
+	if err != nil {
+		return nil, err
+	}
+
+	params, err := mcputils.ParamsParser[AttachSandboxDiskParams](args)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse parameters: %w", err)
+	}
+
+	return makeSandboxPostRequest(fmt.Sprintf("/v1/sandboxes/%s/disks", params.ID), params.Body, authInfo)
+}

handlers/AttachSandboxNetwork.go

@@ -0,0 +1,28 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+
+	mcputils "github.com/NodeOps-app/createos-mcp/helpers"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+type AttachSandboxNetworkParams struct {
+	ID   string                 `json:"id"`
+	Body map[string]interface{} `json:"body"`
+}
+
+func AttachSandboxNetworkHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	authInfo, args, err := handleRequest(ctx, request)
+	if err != nil {
+		return nil, err
+	}
+
+	params, err := mcputils.ParamsParser[AttachSandboxNetworkParams](args)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse parameters: %w", err)
+	}
+
+	return makeSandboxPostRequest(fmt.Sprintf("/v1/sandboxes/%s/networks", params.ID), params.Body, authInfo)
+}

handlers/CreateDisk.go

@@ -0,0 +1,27 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+
+	mcputils "github.com/NodeOps-app/createos-mcp/helpers"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+type CreateDiskParams struct {
+	Body map[string]interface{} `json:"body"`
+}
+
+func CreateDiskHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	authInfo, args, err := handleRequest(ctx, request)
+	if err != nil {
+		return nil, err
+	}
+
+	params, err := mcputils.ParamsParser[CreateDiskParams](args)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse parameters: %w", err)
+	}
+
+	return makeSandboxPostRequest("/v1/disks", params.Body, authInfo)
+}

handlers/CreateNetwork.go

@@ -0,0 +1,27 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+
+	mcputils "github.com/NodeOps-app/createos-mcp/helpers"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+type CreateNetworkParams struct {
+	Body map[string]interface{} `json:"body"`
+}
+
+func CreateNetworkHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	authInfo, args, err := handleRequest(ctx, request)
+	if err != nil {
+		return nil, err
+	}
+
+	params, err := mcputils.ParamsParser[CreateNetworkParams](args)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse parameters: %w", err)
+	}
+
+	return makeSandboxPostRequest("/v1/networks", params.Body, authInfo)
+}

handlers/DeleteDisk.go

@@ -0,0 +1,27 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+
+	mcputils "github.com/NodeOps-app/createos-mcp/helpers"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+type DeleteDiskParams struct {
+	IDOrName string `json:"id_or_name"`
+}
+
+func DeleteDiskHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	authInfo, args, err := handleRequest(ctx, request)
+	if err != nil {
+		return nil, err
+	}
+
+	params, err := mcputils.ParamsParser[DeleteDiskParams](args)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse parameters: %w", err)
+	}
+
+	return makeSandboxDeleteRequest(fmt.Sprintf("/v1/disks/%s", params.IDOrName), authInfo)
+}

handlers/DeleteNetwork.go

@@ -0,0 +1,27 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+
+	mcputils "github.com/NodeOps-app/createos-mcp/helpers"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+type DeleteNetworkParams struct {
+	ID string `json:"id"`
+}
+
+func DeleteNetworkHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	authInfo, args, err := handleRequest(ctx, request)
+	if err != nil {
+		return nil, err
+	}
+
+	params, err := mcputils.ParamsParser[DeleteNetworkParams](args)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse parameters: %w", err)
+	}
+
+	return makeSandboxDeleteRequest(fmt.Sprintf("/v1/networks/%s", params.ID), authInfo)
+}

handlers/DetachSandboxDisk.go

@@ -0,0 +1,37 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+
+	mcputils "github.com/NodeOps-app/createos-mcp/helpers"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+type DetachSandboxDiskParams struct {
+	ID        string `json:"id"`
+	DiskID    string `json:"disk_id"`
+	MountPath string `json:"mount_path"`
+}
+
+func DetachSandboxDiskHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	authInfo, args, err := handleRequest(ctx, request)
+	if err != nil {
+		return nil, err
+	}
+
+	params, err := mcputils.ParamsParser[DetachSandboxDiskParams](args)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse parameters: %w", err)
+	}
+
+	queryParams := map[string]string{
+		"mount_path": params.MountPath,
+	}
+
+	return makeSandboxDeleteRequestWithQuery(
+		fmt.Sprintf("/v1/sandboxes/%s/disks/%s", params.ID, params.DiskID),
+		queryParams,
+		authInfo,
+	)
+}

handlers/DetachSandboxNetwork.go

@@ -0,0 +1,28 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+
+	mcputils "github.com/NodeOps-app/createos-mcp/helpers"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+type DetachSandboxNetworkParams struct {
+	ID      string `json:"id"`
+	Network string `json:"network"`
+}
+
+func DetachSandboxNetworkHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	authInfo, args, err := handleRequest(ctx, request)
+	if err != nil {
+		return nil, err
+	}
+
+	params, err := mcputils.ParamsParser[DetachSandboxNetworkParams](args)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse parameters: %w", err)
+	}
+
+	return makeSandboxDeleteRequest(fmt.Sprintf("/v1/sandboxes/%s/networks/%s", params.ID, params.Network), authInfo)
+}

handlers/GetDisk.go

@@ -0,0 +1,27 @@
+package handler
+
+import (
+	"context"
+	"fmt"
+
+	mcputils "github.com/NodeOps-app/createos-mcp/helpers"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+type GetDiskParams struct {
+	IDOrName string `json:"id_or_name"`
+}
+
+func GetDiskHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	authInfo, args, err := handleRequest(ctx, request)
+	if err != nil {
+		return nil, err
+	}
+
+	params, err := mcputils.ParamsParser[GetDiskParams](args)
+	if err != nil {
+		return nil, fmt.Errorf("failed to parse parameters: %w", err)
+	}
+
+	return makeSandboxGetRequest(fmt.Sprintf("/v1/disks/%s", params.IDOrName), nil, authInfo)
+}