agent

type Agent struct {
	// contains filtered or unexported fields
}

type AgentEnd struct {
	Type         string       `json:"type"`
	Turns        int          `json:"turns"`
	FinishReason string       `json:"finish_reason"`
	Error        string       `json:"error,omitempty"`
	Message      ai.Assistant `json:"message,omitempty"`
}

type AgentOptions struct {
	MaxTurns      int
	MaxToolOutput int
	PromptCustom  string
	PromptAppend  []string
	PromptContext string
	CWD           string
}

type AgentStart struct {
	Type      string `json:"type"`
	ModelName string `json:"model_name"`
}

type AssistantMessageEvent struct {
	Type    string       `json:"type"`
	Message ai.Assistant `json:"message"`
}

type Channel interface {
	// Start runs the channel until the context is cancelled or Stop
	// is called. Blocks.
	Start(ctx context.Context, deps ChannelDeps) error
	// Stop shuts the channel down. Called by the host after the
	// context is cancelled.
	Stop() error
}

type ChannelDeps struct {
	Ctx   *Context
	Store StoreProvider
}

type CmdAction struct {
	Type  string   // "set_model", "refresh_title", "fetch_model_info", "run_compact", "set_model_list", "set_tool_display", "set_thinking", "new_session", "load_session", "branch_session", "set_label"
	Value string   // primary value (model name, session ID, label text)
	List  []string // list values (model names)
}

type CommandResult struct {
	Text    string      // what to display (may be empty)
	Actions []CmdAction // optional TUI actions
}

type CompactionConfig struct {
	Enabled       bool    `yaml:"enabled"`
	Threshold     float64 `yaml:"threshold"`
	ContextWindow int     `yaml:"context_window"`
	KeepFirst     int     `yaml:"keep_first"`
	KeepLast      int     `yaml:"keep_last"`
	ReserveTokens int     `yaml:"reserve_tokens"`
	MaxToolOutput int     `yaml:"max_tool_output"`
}

type CompactionProvider interface {
	// Compact compacts the message history. Returns the compacted
	// history, or the original if no compaction is needed. A nil
	// CompactionProvider means compaction is disabled.
	Compact(ctx context.Context, messages []ai.Message) ([]ai.Message, error)
}

type Context struct {
	// Seam slots (exclusive — one plugin per slot).
	Provider      ai.Provider
	Compactor     CompactionProvider
	Store         StoreProvider
	Skills        SkillsProvider
	Loop          LoopProvider
	Logger        LoggerProvider
	Memory        MemoryProvider
	PromptBuilder PromptBuilder

	// Tool providers (additive — multiple plugins contribute).
	ToolProviders []ToolProvider

	// Frontend (exclusive — one frontend at a time). The host calls
	// Run after mounting all extensions.
	Frontend Frontend
	Trust    TrustProvider

	// Channels (additive — multiple plugins contribute delivery
	// transports). The host starts all mounted channels after MountAll.
	Channels []Channel

	// Cross-cutting (set by specific plugins).
	Commands           []ExtensionCommand
	CommandHandler     func(ctx context.Context, name, args string) (CommandResult, error)
	InjectedMessages   <-chan InjectedMessage
	PendingDelegations func() int

	// Shared state.
	CWD     string
	Configs map[string]any // per-plugin typed config, keyed by plugin Name()

	// Metadata (built during MountAll).
	Infos []ExtensionInfo
}

type DayStat struct {
	Day   string // "Mon", "Tue", etc.
	Count int
	Bar   string // visual bar string
}

type DefaultToolProvider struct {
	ToolsMap map[string]Tool
}

type Event interface {
	// contains filtered or unexported methods
}

type ExtensionCommand struct {
	Name        string `json:"name"` // includes leading slash, e.g. "/mycmd"
	Description string `json:"description"`
}

type ExtensionInfo struct {
	Name     string
	Tools    int
	Commands int
	Seams    []string   // declared seam types ("prompt_builder", "compactor", etc.)
	ToolList []ToolInfo // tools provided by this extension (name + description)
}

type Frontend interface {
	Run(ctx context.Context, pctx *Context, opts FrontendOptions) error
}

type FrontendOptions struct {
	ModelName     string
	ProviderType  string
	Compaction    *CompactionConfig
	PromptCustom  string
	PromptAppend  []string
	PromptContext string
	Skills        []Skill
	ThemeName     string
	TrustState    string
	Notifications string
	CWD           string
	Version       string
}

type InjectedMessage struct {
	Text   string // the message content
	Source string // "delegate:<task_id>" — for display
}

type Insights struct {
	Period         string
	PeriodStart    string
	PeriodEnd      string
	Days           int
	Sessions       int
	Messages       int
	UserMessages   int
	ToolCalls      int
	TotalTokens    int
	AvgSessionMsgs float64
	Tools          []ToolStat
	Daily          [7]DayStat
	NotableMsgs    NotableStat
	NotableTokens  NotableStat
	NotableTools   NotableStat
}

type LoggerProvider interface {
	// Printf writes an info-level log entry.
	Printf(format string, args ...any)
	// Errorf writes an error-level log entry.
	Errorf(format string, args ...any)
	// Close flushes and closes the log output.
	Close() error
}

type LoopOptions struct {
	Provider           ai.Provider
	Messages           []ai.Message
	Tools              map[string]Tool
	ToolProviders      []ToolProvider // additive tool sources (extensions)
	CompactionProvider CompactionProvider
	PromptBuilder      PromptBuilder   // builds system prompt + guidelines
	ExtensionInfos     []ExtensionInfo // for prompt building context
	MaxTurns           int
	MaxToolOutput      int
	CWD                string
	PromptCustom       string   // user-supplied prompt from config, for extension-built prompts
	PromptAppend       []string // extra prompts from --append-system-prompt
	PromptContext      string   // trust-gated AGENTS.md project context
	Events             chan<- Event
	InjectedMessages   <-chan InjectedMessage // subagent results (nil if no delegate extension)
	UserInput          <-chan string          // user messages while running (nil for one-shot mode)
	PendingDelegations func() int             // returns count of running subagent tasks
	Logger             LoggerProvider         // optional logger; nil = no logging
}

type LoopProvider interface {
	Run(ctx context.Context, opts LoopOptions) error
}

type MemoryProvider interface {
	// Snapshot returns the formatted prompt block for system prompt
	// injection. Reads files fresh on each call.
	Snapshot() string

	// Add appends a new entry to the target store ("memory" or "user").
	// Returns the new usage string and an error if the entry would
	// exceed the char limit or is an exact duplicate.
	Add(target, content string) (usage string, err error)

	// Replace finds the entry matching oldText (unique substring) and
	// replaces it with content. Returns usage string and error.
	Replace(target, oldText, content string) (usage string, err error)

	// Remove finds the entry matching oldText (unique substring) and
	// deletes it. Returns usage string and error.
	Remove(target, oldText string) (usage string, err error)

	// List returns all entries for the target.
	List(target string) ([]string, error)
}

type NotableStat struct {
	Value  int
	Detail string // date or session label
}

type Plugin interface {
	// Name is shown in the /extensions listing.
	Name() string
	// Provides lists seam names this plugin mounts (e.g. "provider",
	// "store", "tools"). Exclusive seams conflict if two plugins
	// provide the same one. The "tools" seam is additive.
	Provides() []string
	// Requires lists seam names that must be mounted before this
	// plugin. MountAll uses these for topological ordering.
	Requires() []string
	// Mount populates Context slots. Called once, in dependency order.
	Mount(ctx *Context) error
}

type PromptBuildOptions struct {
	CWD            string
	Messages       []ai.Message
	Extensions     []ExtensionInfo
	ProjectContext string   // AGENTS.md contents, already trust-gated by Go
	Custom         string   // user-supplied prompt from config, may be empty
	Append         []string // extra prompts from --append-system-prompt, may be nil
}

type PromptBuilder interface {
	// BuildPrompt assembles the full system prompt. Returns ok=false
	// if the builder does not want to provide a prompt; the agent gets
	// no system prompt in that case.
	BuildPrompt(ctx context.Context, opts PromptBuildOptions) (string, bool)
	// Guidelines returns extra lines appended under
	// "## Extension Guidelines" in the system prompt.
	Guidelines() []string
}

type SearchResult struct {
	SessionID string `json:"session_id"`
	Snippet   string `json:"snippet"`
}

type Session struct {
	ID        string `json:"id"`
	ParentID  string `json:"parent_id,omitempty"`
	Label     string `json:"label,omitempty"`
	CreatedAt string `json:"created_at"`
	UpdatedAt string `json:"updated_at"`
}

type SessionNode struct {
	Session
	Children []*SessionNode
}

type Skill struct {
	Name        string
	Description string
	Content     string
	Dir         string
}

type SkillsProvider interface {
	LoadSkills(dir string) ([]Skill, error)
}

type StoreProvider interface {
	Open(dsn string) error
	Close() error
	CreateSession(ctx context.Context, id, parentID, label string) error
	GetSession(ctx context.Context, id string) (Session, error)
	ListSessions(ctx context.Context) ([]Session, error)
	DeleteSession(ctx context.Context, id string) error
	UpdateSession(ctx context.Context, id, label string) error
	AppendMessage(ctx context.Context, sessionID string, msg ai.Message) error
	GetMessages(ctx context.Context, sessionID string) ([]ai.Message, error)
	GetSessionTree(ctx context.Context) ([]*SessionNode, error)
	GetAncestorMessages(ctx context.Context, sessionID string) ([]ai.Message, error)
	SearchMessages(ctx context.Context, query string) ([]SearchResult, error)
	ComputeInsights(ctx context.Context, days int) (*Insights, error)
	CountMessages(ctx context.Context, sessionID string) (int, error)
}

type StreamEvent struct {
	Event ai.StreamEvent `json:"-"`
}

type Tool struct {
	Description string
	Parameters  map[string]any
	Run         func(ctx context.Context, args map[string]any) (string, error)
}

type ToolInfo struct {
	Name        string
	Description string
}

type ToolProvider interface {
	Tools() map[string]Tool
}

type ToolResultEvent struct {
	Type    string        `json:"type"`
	Message ai.ToolResult `json:"message"`
}

type ToolStat struct {
	Name  string
	Count int
}

type TrustProvider interface {
	ResolveContext(cwd string, approve, noApprove, interactive bool) string
	State(cwd string, approve, noApprove bool) string
}

type TurnEnd struct {
	Type      string `json:"type"`
	Turn      int    `json:"turn"`
	ToolCalls int    `json:"tool_calls"`
}

type TurnStart struct {
	Type string `json:"type"`
	Turn int    `json:"turn"`
}