molecule-cli/internal/backends/claudecode/claudecode.go
Hongming Wang b7a3c887c0 feat(connect): M1.3 — exec + claude-code backends
After this PR `molecule connect <id>` actually does something useful:
the default `--backend claude-code` runs `claude -p` as a subprocess
per inbound message, capturing stdout as the reply. The general
`--backend exec --backend-opt cmd="..."` lets operators bridge to any
program that reads stdin and writes stdout (Ollama, custom Python,
shell pipelines).

What's in:

- `internal/backends/exec/exec.go` — generic exec backend.
  - cmd  (required): shell command, run via /bin/sh -c (or cmd /c).
  - timeout (default 60s): per-message wall clock; subprocess is
    killed on timeout and the dispatcher keeps the message queued.
  - pass_meta (default false): when true, populates env with
    MOLECULE_WORKSPACE_ID / CALLER_ID / MESSAGE_ID / TASK_ID / METHOD.
  - Stdin = joined text parts; stdout = reply text.
  - Stderr is captured + surfaced in error messages so operators can
    see what their command printed.
  - Honours ctx cancellation — SIGTERM kills the subprocess immediately.

- `internal/backends/claudecode/claudecode.go` — thin shorthand.
  Translates {bin, args, timeout, pass_meta} into the equivalent exec
  config. Defaults: bin=claude, timeout=5m, pass_meta=true.
  Reusing exec means timeout/stdin/env handling stays in one place.

- `internal/cmd/connect.go` — registers both backends so `--help`
  shows them and the default `--backend claude-code` works without
  flag tuning.

Test plan:

- exec: cmd-required, bad-timeout, zero-timeout, echo-stdin-to-stdout,
  text-only-part-concat, non-zero-exit-surfaces-stderr, timeout-kills-
  runaway, pass_meta-injects-env, no-pass_meta-leaves-env-untouched,
  parent-env-inherited-when-pass_meta-off, ctx-cancel-kills-command.
- claude-code: registered, bin/args translation produces "echo -p hello",
  bad-timeout-surfaces, default-config-builds.
- All Unix tests gated on `requireUnix(t)` — Windows-shell semantics
  diverge; cross-platform coverage is M3 work (brew/scoop/winget).
- Full suite green under -race.

UX after this PR:

  # default — Claude Code on PATH
  molecule connect ws_01HF... --token $T

  # custom handler
  molecule connect ws_01HF... --backend exec \
      --backend-opt cmd="python myhandler.py"

  # different model via Claude Code
  molecule connect ws_01HF... \
      --backend-opt args="--model claude-sonnet-4-6"

  # CI smoke
  molecule connect ws_01HF... --backend mock --backend-opt reply=ok

Out of scope (M1.4 + later):

- M1.4: GoReleaser tag-triggered release.yml workflow
- stdio backend (one persistent subprocess + line-delimited JSON-RPC)
  — moved to M2 since exec covers the common case
- openai / mcp backends — M4 per RFC

RFC: https://github.com/Molecule-AI/molecule-cli/issues/10

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-04-30 04:23:36 -07:00

62 lines
2.1 KiB
Go

// Package claudecode implements the `claude-code` backend: a thin
// shorthand for `exec` with `cmd="claude -p"`. Each inbound A2A
// message is dispatched to a fresh `claude --print` invocation; the
// model's stdout becomes the reply.
//
// Why a separate backend instead of telling users `--backend exec
// --backend-opt cmd="claude -p"`?
//
// - The default backend is "claude-code" — copy-paste-from-canvas
// should Just Work without the operator memorising flag spelling.
// - Future versions can add Claude-Code-specific config: model
// selection, system prompt, MCP forwarding. The exec backend
// stays generic.
//
// Config keys (all optional):
//
// - bin — claude binary path. Default "claude".
// - args — extra args appended after `-p`. Default "".
// - timeout — per-message timeout. Default "5m" (Claude Code
// responses can take a while; longer than exec's 60s default).
// - pass_meta — see exec backend. Default "true" — Claude Code
// sessions benefit from knowing who sent the message.
//
// Implementation: builds the equivalent exec config under the hood.
// Reusing exec means timeout/stdin/stderr/env handling stays in one
// place; bug fixes flow to both.
package claudecode
import (
"strings"
"github.com/Molecule-AI/molecule-cli/internal/backends"
exec "github.com/Molecule-AI/molecule-cli/internal/backends/exec"
)
func init() {
backends.Register("claude-code", New)
}
// New builds a claude-code backend from cfg. Translates the
// claude-code keys to an exec backend config and delegates.
func New(cfg backends.Config) (backends.Backend, error) {
bin := cfg.Get("bin", "claude")
extra := cfg.Get("args", "")
timeout := cfg.Get("timeout", "5m")
passMeta := cfg.Get("pass_meta", "true")
// Build the underlying shell command. -p (print mode) is the
// non-interactive Claude Code mode that reads stdin and writes
// stdout once.
cmd := bin + " -p"
if strings.TrimSpace(extra) != "" {
cmd += " " + extra
}
return exec.New(backends.Config{
"cmd": cmd,
"timeout": timeout,
"pass_meta": passMeta,
})
}