feat: add MCP file support for ACP server configuration

Author: 35C4n0r

cmd/server/server.go

@@ -136,6 +136,11 @@ func runServer(ctx context.Context, logger *slog.Logger, argsToPass []string) er
 	}
 
 	experimentalACP := viper.GetBool(FlagExperimentalACP)
+	acpMCPFile := viper.GetString(FlagMCPFile)
+
+	if acpMCPFile != "" && !experimentalACP {
+		return xerrors.Errorf("--mcp-file requires --experimental-acp requires to be set")
+	}
 
 	if experimentalACP && (saveState || loadState) {
 		return xerrors.Errorf("ACP mode doesn't support state persistence")
@@ -169,6 +174,7 @@ func runServer(ctx context.Context, logger *slog.Logger, argsToPass []string) er
 		acpResult, err = httpapi.SetupACP(ctx, httpapi.SetupACPConfig{
 			Program:     agent,
 			ProgramArgs: argsToPass[1:],
+			MCPFilePath: acpMCPFile,
 		})
 		if err != nil {
 			return xerrors.Errorf("failed to setup ACP: %w", err)
@@ -382,6 +388,7 @@ const (
 	FlagSaveState       = "save-state"
 	FlagPidFile         = "pid-file"
 	FlagExperimentalACP = "experimental-acp"
+	FlagMCPFile         = "mcp-file"
 )
 
 func CreateServerCmd() *cobra.Command {
@@ -425,6 +432,7 @@ func CreateServerCmd() *cobra.Command {
 		{FlagSaveState, "", false, "Save state to state-file on shutdown (defaults to true when state-file is set)", "bool"},
 		{FlagPidFile, "", "", "Path to file where the server process ID will be written for shutdown scripts", "string"},
 		{FlagExperimentalACP, "", false, "Use experimental ACP transport instead of PTY", "bool"},
+		{FlagMCPFile, "", "", "MCP file for the ACP server", "string"},
 	}
 
 	for _, spec := range flagSpecs {

lib/httpapi/setup.go

@@ -53,6 +53,7 @@ func SetupProcess(ctx context.Context, config SetupProcessConfig) (*termexec.Pro
 type SetupACPConfig struct {
 	Program     string
 	ProgramArgs []string
+	MCPFilePath string
 	Clock       quartz.Clock
 }
 
@@ -88,7 +89,7 @@ func SetupACP(ctx context.Context, config SetupACPConfig) (*SetupACPResult, erro
 		return nil, fmt.Errorf("failed to start process: %w", err)
 	}
 
-	agentIO, err := acpio.NewWithPipes(ctx, stdin, stdout, logger, os.Getwd)
+	agentIO, err := acpio.NewWithPipes(ctx, stdin, stdout, logger, os.Getwd, config.MCPFilePath)
 	if err != nil {
 		_ = cmd.Process.Kill()
 		return nil, fmt.Errorf("failed to initialize ACP connection: %w", err)

x/acpio/acpio.go

@@ -2,14 +2,17 @@ package acpio
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 	"io"
 	"log/slog"
+	"os"
 	"strings"
 	"sync"
 
 	acp "github.com/coder/acp-go-sdk"
 	st "github.com/coder/agentapi/lib/screentracker"
+	"golang.org/x/xerrors"
 )
 
 // Compile-time assertion that ACPAgentIO implements st.AgentIO
@@ -31,6 +34,10 @@ type acpClient struct {
 	agentIO *ACPAgentIO
 }
 
+type McpConfig struct {
+	McpServers []acp.McpServer `json:"mcpServers"`
+}
+
 var _ acp.Client = (*acpClient)(nil)
 
 func (c *acpClient) SessionUpdate(ctx context.Context, params acp.SessionNotification) error {
@@ -131,7 +138,7 @@ func (a *ACPAgentIO) SetOnChunk(fn func(chunk string)) {
 }
 
 // NewWithPipes creates an ACPAgentIO connected via the provided pipes
-func NewWithPipes(ctx context.Context, toAgent io.Writer, fromAgent io.Reader, logger *slog.Logger, getwd func() (string, error)) (*ACPAgentIO, error) {
+func NewWithPipes(ctx context.Context, toAgent io.Writer, fromAgent io.Reader, logger *slog.Logger, getwd func() (string, error), mcpFilePath string) (*ACPAgentIO, error) {
 	if logger == nil {
 		logger = slog.Default()
 	}
@@ -154,6 +161,12 @@ func NewWithPipes(ctx context.Context, toAgent io.Writer, fromAgent io.Reader, l
 	}
 	logger.Debug("ACP initialized", "protocolVersion", initResp.ProtocolVersion)
 
+	// Prepare the MCPs for the session
+	supportedMCPList, err := getSupportedMCPConfig(mcpFilePath, logger, &initResp)
+	if err != nil {
+		return nil, err
+	}
+
 	// Create a session
 	cwd, err := getwd()
 	if err != nil {
@@ -162,7 +175,7 @@ func NewWithPipes(ctx context.Context, toAgent io.Writer, fromAgent io.Reader, l
 	}
 	sessResp, err := conn.NewSession(ctx, acp.NewSessionRequest{
 		Cwd:        cwd,
-		McpServers: []acp.McpServer{},
+		McpServers: supportedMCPList,
 	})
 	if err != nil {
 		logger.Error("Failed to create ACP session", "error", err)
@@ -174,6 +187,36 @@ func NewWithPipes(ctx context.Context, toAgent io.Writer, fromAgent io.Reader, l
 	return agentIO, nil
 }
 
+func getSupportedMCPConfig(mcpFilePath string, logger *slog.Logger, initResp *acp.InitializeResponse) ([]acp.McpServer, error) {
+	mcpFile, err := os.Open(mcpFilePath)
+	if err != nil {
+		return nil, xerrors.Errorf("Failed to open mcp file: %v", err)
+	}
+
+	defer func() {
+		if closeErr := mcpFile.Close(); closeErr != nil {
+			logger.Error("Failed to close mcp file", "error", err)
+		}
+	}()
+
+	var allMcpList McpConfig
+	decoder := json.NewDecoder(mcpFile)
+
+	if err = decoder.Decode(&allMcpList); err != nil {
+		return nil, xerrors.Errorf("Failed to decode mcp file: %v", err)
+	}
+
+	// Only send the MCPs that are supported by the agents
+	var supportedMCPList []acp.McpServer
+	for _, mcp := range allMcpList.McpServers {
+		if (mcp.Http != nil && !initResp.AgentCapabilities.McpCapabilities.Http) || (mcp.Sse != nil && !initResp.AgentCapabilities.McpCapabilities.Sse) {
+			continue
+		}
+		supportedMCPList = append(supportedMCPList, mcp)
+	}
+	return supportedMCPList, err
+}
+
 // Write sends a message to the agent via ACP prompt
 func (a *ACPAgentIO) Write(data []byte) (int, error) {
 	text := string(data)