iterm2

type AgentResolveFixture struct {
	Kind      string // grok | codex | none
	SessionID string
	Title     string // optional GrokTitle
	Tree      []AgentTreeNode
}

type AgentTreeNode struct {
	PID  int    `json:"pid"`
	PPID int    `json:"ppid"`
	Role string `json:"role,omitempty"` // input | agent-run | … | grok | codex | other
	Cmd  string `json:"cmd"`
}

type CaptureOpts struct {
	// NoEnrich skips procresolve agent attach (CLI --no-enrich).
	NoEnrich bool
	// SpaceAllow, when non-empty, enables space-first filtering in the lib.
	SpaceAllow []int
	// SpaceSkipped, when non-nil, receives skipped window-header count.
	SpaceSkipped *int
}

type CommandDecision struct {
	Key       string   `json:"key"`
	Cwd       string   `json:"cwd"`
	Argv      []string `json:"argv"`
	Decision  string   `json:"decision"` // allow | deny
	DecidedAt string   `json:"decided_at"`
}

type CurrentSpaceIndexFunc func() (int, error)

type DecisionsDocument struct {
	Version   int               `json:"version"`
	Decisions []CommandDecision `json:"decisions"`
}

type FocusCandidate struct {
	WindowID      string
	WindowTitle   string
	TabIndex      int
	SessionID     string
	Path          string
	KoolTargetDir string
}

type FocusFake struct {
	Candidates    []FocusCandidate
	Focused       []string
	DiscoverCalls int
}

type ForegroundCommand struct {
	PID      int      `json:"pid"`
	Cwd      string   `json:"cwd,omitempty"`
	Argv     []string `json:"argv,omitempty"`
	Observed string   `json:"observed,omitempty"`
	Reason   string   `json:"reason,omitempty"`
	Complex  bool     `json:"complex,omitempty"`
}

type ForkPlan struct {
	SourceItermID  string
	Cwd            string
	GrokSessionID  string
	FollowUp       string
	AgentRunBinary string
}

type FormatFlags struct {
	JSON     bool
	Markdown bool
	HTML     bool
}

type MultiAppPreflight struct {
	// AsApp is the canonical app that bare `tell application "iTerm2"` targets.
	AsApp string
	// RunningApps is the set of canonical installs with a live process
	// (home and/or system). Order is not significant.
	RunningApps []string
}

type MultiAppPreflightFn func() (MultiAppPreflight, error)

type OutputFormat string

type PhasedFixtureOpts struct {
	Windows           []SnapshotWindow
	ITermRunning      bool
	OnListWindows     func()
	OnListTabs        func(windowIndex int)
	IdleTTYs          []string
	BusyTTYs          []string
	BusyLeafByTTY     map[string]string
	CwdByTTY          map[string]string
	Now               time.Time
	Hostname          string
	AgentResolveByTTY map[string]AgentResolveFixture
}

type RenderOptions struct {
	Format  OutputFormat
	NoColor bool
	// NoTree omits FormatTree process-tree lines while keeping agent session id.
	NoTree bool
	// ForceColor forces ANSI even when stdout is not a TTY (tests).
	ForceColor bool
	// ColorWriter is checked for TTY (defaults to os.Stdout).
	ColorWriter *os.File
}

type RestoreAppTarget struct {
	// Canonical is ~/Applications/iTerm.app or /Applications/iTerm.app, or empty when Bare.
	Canonical string
	// Bare means fall back to tell application "iTerm2" (no path).
	Bare bool
	// Warning is optional user message without the "warning:" prefix.
	Warning string
}

type SaveDocument struct {
	Version    int          `json:"version"`
	SavedAt    string       `json:"saved_at"`
	RestoredAt *string      `json:"restored_at"` // null until restore succeeds
	Host       string       `json:"host"`
	Source     string       `json:"source"`
	Filter     *SaveFilter  `json:"filter,omitempty"` // present when save used --spaces
	Summary    SaveSummary  `json:"summary"`
	Windows    []SaveWindow `json:"windows"`
}

type SaveFilter struct {
	// Spaces is the sorted, deduped 0-based allowlist from --spaces.
	Spaces []int `json:"spaces,omitempty"`
}

type SaveSummary struct {
	Windows  int            `json:"windows"`
	Tabs     int            `json:"tabs"`
	Sessions int            `json:"sessions"`
	ByKind   map[string]int `json:"by_kind"`
}

type SaveTab struct {
	SourceTabIndex  int           `json:"source_tab_index,omitempty"`
	SourcePaneIndex int           `json:"source_pane_index,omitempty"`
	Name            string        `json:"name,omitempty"`
	Cwd             string        `json:"cwd"`
	Kind            string        `json:"kind"` // grok | codex | mark | command
	Command         *SavedCommand `json:"command,omitempty"`
	SessionID       string        `json:"session_id,omitempty"`
	Message         string        `json:"message,omitempty"` // mark only
	Title           string        `json:"title,omitempty"`
	ResumeCmd       string        `json:"resume_cmd"`
	ItermSessionID  string        `json:"iterm_session_id,omitempty"`
	SourceCmdLine   string        `json:"source_command_line,omitempty"`
	// ResolvedAllow / ResolvedDeny are in-memory restore-time decisions (from
	// the decision store or the confirmation prompt). Never serialized.
	ResolvedAllow bool `json:"-"`
	ResolvedDeny  bool `json:"-"`
}

type SaveWindow struct {
	SourceIndex   int       `json:"source_index"`
	Name          string    `json:"name,omitempty"`
	App           string    `json:"app,omitempty"`             // canonical install; restore prefer-home or --same-app
	Space         int       `json:"space"`                     // 0-based Desktop; always emitted when not ignore
	ItermWindowID int64     `json:"iterm_window_id,omitempty"` // info only at save; restore never uses it
	Tabs          []SaveTab `json:"tabs"`
	// contains filtered or unexported fields
}

type SavedCommand struct {
	Argv          []string `json:"argv,omitempty"`
	Source        string   `json:"source"`
	Category      string   `json:"category,omitempty"`
	RestorePolicy string   `json:"restore_policy"`
	Reason        string   `json:"reason,omitempty"`
}

type SessionAgent struct {
	Kind      string          `json:"kind"`
	SessionID string          `json:"session_id"`
	Title     string          `json:"title,omitempty"`
	Tree      []AgentTreeNode `json:"tree,omitempty"`
}

type SessionListRow struct {
	ItermID      string `json:"iterm_id"`
	Status       string `json:"status"` // idle | busy | unknown
	Name         string `json:"name,omitempty"`
	Cwd          string `json:"cwd,omitempty"`
	Kind         string `json:"kind,omitempty"` // grok | codex | …
	AgentSession string `json:"agent_session,omitempty"`
	Title        string `json:"title,omitempty"`
	WindowIndex  int    `json:"window_index,omitempty"`
	TabIndex     int    `json:"tab_index,omitempty"`
	PaneIndex    int    `json:"pane_index,omitempty"`
}

type Snapshot struct {
	CapturedAt string           `json:"captured_at"`
	Host       string           `json:"host"`
	Source     string           `json:"source"`
	Summary    SnapshotSummary  `json:"summary"`
	Windows    []SnapshotWindow `json:"windows"`
}

type SnapshotCollector struct {

	// OnListWindows is an optional test hook invoked at the start of ListWindows.
	OnListWindows func()
	// OnListTabs is an optional test hook invoked at the start of ListTabsAndSessions.
	OnListTabs func(windowIndex int)

	// ResolveFromPID optionally overrides live procresolve (production default).
	ResolveFromPID func(pid int) (*procresolve.Result, error)

	// AppTell is the AppleScript application target (forwarded to lib).
	AppTell string
	// AppTag is stamped on windows when App is empty (forwarded to lib).
	AppTag string

	// RunAppleScript / ListProcs / ListTTYProcs / ListCwds / ITermRunning are
	// injectable for live_scan and tests; synced onto lib before capture.
	RunAppleScript func(script string) (string, error)
	ListProcs      func(ttyShort string) ([]snapshot.ProcRow, error)
	ListTTYProcs   func(ttyShorts []string) ([]liveTTYProc, error)
	ListCwds       func(pids []int) (map[int]string, error)
	ITermRunning   func() bool
	Now            func() time.Time
	Hostname       func() (string, error)
	// contains filtered or unexported fields
}

type SnapshotProc struct {
	PID             int     `json:"pid"`
	PPID            int     `json:"ppid"`
	Stat            string  `json:"stat"`
	Etime           string  `json:"etime"`
	DurationSeconds int64   `json:"duration_seconds"`
	Duration        string  `json:"duration"`
	StartTime       *string `json:"start_time"`
	StartTimeUnix   *int64  `json:"start_time_unix"`
	RSSKB           int64   `json:"rss_kb"`
	Command         string  `json:"command"`
}

type SnapshotSession struct {
	Index             int            `json:"index"`
	ID                string         `json:"id"`
	Name              string         `json:"name"`
	TTY               string         `json:"tty"`
	Profile           string         `json:"profile"`
	ItermIsProcessing bool           `json:"iterm_is_processing"`
	Idle              *bool          `json:"idle"` // nil = unknown
	Cwd               *string        `json:"cwd"`
	ShellPID          *int           `json:"shell_pid"`
	PID               *int           `json:"pid"`
	PPID              *int           `json:"ppid"`
	Stat              *string        `json:"stat"`
	Command           *string        `json:"command"`
	CommandLine       *string        `json:"command_line"`
	StartTime         *string        `json:"start_time"`
	StartTimeUnix     *int64         `json:"start_time_unix"`
	DurationSeconds   *int64         `json:"duration_seconds"`
	Duration          *string        `json:"duration"`
	Etime             *string        `json:"etime"`
	RSSKB             *int64         `json:"rss_kb"`
	Processes         []SnapshotProc `json:"processes"`
	// Agent is set when procresolve finds a grok/codex session on a busy pane.
	Agent *SessionAgent `json:"agent,omitempty"`
	// Foreground contains live launcher evidence, separate from the chosen leaf.
	Foreground *ForegroundCommand `json:"foreground,omitempty"`
	// Layout hints (not required for id resolution).
	WindowIndex int `json:"window_index,omitempty"`
	TabIndex    int `json:"tab_index,omitempty"`
}

type SnapshotSummary struct {
	Windows  int `json:"windows"`
	Tabs     int `json:"tabs"`
	Sessions int `json:"sessions"`
	Idle     int `json:"idle"`
	Busy     int `json:"busy"`
	Unknown  int `json:"unknown"`
}

type SnapshotTab struct {
	Index    int               `json:"index"`
	Name     string            `json:"name"`
	Sessions []SnapshotSession `json:"sessions"`
}

type SnapshotWindow struct {
	Index int    `json:"index"`
	Name  string `json:"name"`
	// WindowID is the iTerm/AppleScript window id (CG window number when available).
	// Used to resolve macOS Space on sessions save; zero means unknown.
	WindowID uint64 `json:"window_id,omitempty"`
	// FixedSpace, when non-nil, is the resolved 0-based Space for tests/fixtures.
	// Prefer this over SpaceIndexForWindow so parallel tests need no global hook.
	FixedSpace *int `json:"-"`
	// App is the canonical iTerm install that owns this window (fixture tag or
	// multi-app save source). Only "~/Applications/iTerm.app" or
	// "/Applications/iTerm.app". Not emitted on sessions snapshot JSON.
	App  string        `json:"-"`
	Tabs []SnapshotTab `json:"tabs"`
}

type SpaceBackend = space.Backend

type SpaceIndexResolver func(windowID uint64) (int, error)

type TestRun struct {
	TabSetDir string
	GOOS      string
	Installed func() bool
	Osascript func(script string) error
	// Contents overrides lib.Contents (tests).
	Contents func(sessionID string, cfg *lib.ContentsConfig) (lib.ContentsResult, error)
	// SendText overrides lib.SendText (tests).
	SendText func(sessionID, text string, opts lib.SendTextOptions, cfg *lib.SendTextConfig) error
	// ListSessions overrides lib.ListSessions for send resolve (tests).
	ListSessions func() ([]lib.SessionRef, error)
	// CurrentStatus overrides resolve probes for window/tab status (tests).
	CurrentStatus *lib.CurrentStatusConfig
}