agent

type AvailabilityError struct {
	IDE         string
	DisplayName string
	Command     []string
	DocsURL     string
	InstallHint string
	Output      string
	Cause       error
}

type Client interface {
	// CreateSession starts a new ACP session with the given prompt.
	CreateSession(ctx context.Context, req SessionRequest) (Session, error)
	// ResumeSession attaches to an existing ACP session and sends a new prompt into it.
	ResumeSession(ctx context.Context, req ResumeSessionRequest) (Session, error)
	// SupportsLoadSession reports whether the connected ACP agent advertised session/load support.
	SupportsLoadSession() bool
	// Close terminates the agent subprocess.
	Close() error
	// Kill force-terminates the agent subprocess immediately.
	Kill() error
}

type ClientConfig struct {
	IDE             string
	Model           string
	AddDirs         []string
	ReasoningEffort string
	AccessMode      string
	Logger          *slog.Logger
	ShutdownTimeout time.Duration
}

type DriverCatalogEntry struct {
	IDE                string
	DisplayName        string
	CanonicalCommand   []string
	CanonicalProbe     []string
	FallbackLaunchers  []DriverCatalogLauncher
	SupportsAddDirs    bool
	UsesBootstrapModel bool
	DocsURL            string
	InstallHint        string
}

type DriverCatalogLauncher struct {
	Command []string
	Probe   []string
}

type Launcher struct {
	Command   string
	FixedArgs []string
	ProbeArgs []string
}

type OverlayBootstrap struct {
	ModelFlag             string
	ReasoningEffortFlag   string
	AddDirFlag            string
	DefaultAccessModeArgs []string
	FullAccessModeArgs    []string
}

type OverlayEntry struct {
	Name               string
	DisplayName        string
	Command            string
	SetupAgentName     string
	DefaultModel       string
	SupportsAddDirs    *bool
	UsesBootstrapModel *bool
	DocsURL            string
	InstallHint        string
	FullAccessModeID   string
	FixedArgs          []string
	ProbeArgs          []string
	EnvVars            map[string]string
	Fallbacks          []Launcher
	Bootstrap          OverlayBootstrap
	Metadata           map[string]string
}

type Registry struct{}

type ResumeSessionRequest struct {
	SessionID  string               `json:"session_id,omitempty"`
	Prompt     []byte               `json:"prompt,omitempty"`
	WorkingDir string               `json:"working_dir,omitempty"`
	Model      string               `json:"model,omitempty"`
	MCPServers []model.MCPServer    `json:"mcp_servers,omitempty"`
	ExtraEnv   map[string]string    `json:"extra_env,omitempty"`
	Context    context.Context      `json:"-"`
	RunID      string               `json:"-"`
	JobID      string               `json:"-"`
	RuntimeMgr model.RuntimeManager `json:"-"`
}

type RuntimeRegistry interface {
	ValidateRuntimeConfig(cfg *model.RuntimeConfig) error
	EnsureAvailable(ctx context.Context, cfg *model.RuntimeConfig) error
}

type Session interface {
	// ID returns the ACP session identifier.
	ID() string
	// Identity returns the canonical session identifiers known for this session.
	Identity() SessionIdentity
	// Updates returns the streamed session updates in arrival order.
	Updates() <-chan model.SessionUpdate
	// Done closes when the session has fully completed.
	Done() <-chan struct{}
	// Err returns the terminal session error, if any.
	Err() error
	// SlowPublishes returns the number of publishes that waited for backpressure to clear.
	SlowPublishes() uint64
	// DroppedUpdates returns the number of updates dropped after backpressure timed out.
	DroppedUpdates() uint64
}

type SessionError struct {
	Code    int
	Message string
	Data    json.RawMessage
}

type SessionIdentity struct {
	ACPSessionID   string `json:"acp_session_id"`
	AgentSessionID string `json:"agent_session_id,omitempty"`
	Resumed        bool   `json:"resumed,omitempty"`
}

type SessionOutcome struct {
	Status model.SessionStatus `json:"status"`
	Error  string              `json:"error,omitempty"`
}

type SessionRequest struct {
	Prompt     []byte               `json:"prompt,omitempty"`
	WorkingDir string               `json:"working_dir,omitempty"`
	Model      string               `json:"model,omitempty"`
	MCPServers []model.MCPServer    `json:"mcp_servers,omitempty"`
	ExtraEnv   map[string]string    `json:"extra_env,omitempty"`
	Context    context.Context      `json:"-"`
	RunID      string               `json:"-"`
	JobID      string               `json:"-"`
	RuntimeMgr model.RuntimeManager `json:"-"`
}

type SessionSetupError struct {
	Stage SessionSetupStage
	Err   error
}

type SessionSetupStage string

type Spec struct {
	ID                 string
	DisplayName        string
	SetupAgentName     string
	DefaultModel       string
	Command            string
	FixedArgs          []string
	ProbeArgs          []string
	Fallbacks          []Launcher
	SupportsAddDirs    bool
	UsesBootstrapModel bool
	EnvVars            map[string]string
	DocsURL            string
	InstallHint        string
	FullAccessModeID   string
	BootstrapArgs      func(modelName, reasoningEffort string, addDirs []string, accessMode string) []string
}