tree

type BranchAnnotation struct {
	PRNumber      *int
	PRAction      string // "create", "update", "skip", ""
	CheckStatus   string // PASSING, FAILING, PENDING, NONE, ""
	ReviewStatus  string // "Approved", "In Review", "Changes Requested", "Commented", ""
	IsDraft       bool
	IsLocked      bool
	IsFrozen      bool
	NeedsRestack  bool
	CustomLabel   string // Additional text to display after branch name
	Scope         string
	ExplicitScope string

	CommitCount  int
	LinesAdded   int
	LinesDeleted int
	PRState      string // "OPEN", "MERGED", "CLOSED"

	// WorktreePath is set if this branch is the stack root of a managed worktree
	WorktreePath string

	// IsEmptyWorktree indicates this is a worktree anchor with no child branches
	IsEmptyWorktree bool

	// LocalSHA is the short commit SHA of the branch head (for debugging/diagnostics)
	LocalSHA string

	// MergedDownstack holds historical merged parents for display
	MergedDownstack []MergedParentDisplay

	// CommitMessages holds formatted commit messages (e.g., "abc123 Commit message")
	// Used when ShowCommitMessages is enabled in RenderOptions
	CommitMessages []string

	// PRURL is the GitHub PR URL for this branch
	PRURL string
}

type CachedBranchRender struct {
	Name               string   // Branch name
	LinesUnselected    []string // Lines without selection (padding prefix, no highlight)
	CursorLineSelected string   // The cursor line with selection styling (cursor prefix + highlighted name)
	CursorLineIndex    int      // Which line gets the cursor
	IsNonSelectable    bool     // If true, never show as selected
}

type CachedTreeData struct {
	Branches []CachedBranchRender
}

type Data interface {
	// CurrentBranch returns the name of the currently checked out branch
	CurrentBranch() string
	// Trunk returns the name of the trunk/main branch
	Trunk() string
	// Children returns the child branches of the given branch
	Children(branchName string) []string
	// Parent returns the parent branch of the given branch
	Parent(branchName string) string
	// IsTrunk returns whether the given branch is a trunk branch
	IsTrunk(branchName string) bool
	// IsFixed returns whether the branch is up-to-date with its parent
	IsFixed(branchName string) bool
}

type MergedParentDisplay struct {
	BranchName string
	PRNumber   *int
	PRState    string // "MERGED", "CLOSED"
}

type MockTreeData struct {
	// CurrentVal is the current branch name
	CurrentVal string
	// TrunkVal is the trunk branch name
	TrunkVal string
	// ChildrenMap maps branch name to child branch names
	ChildrenMap map[string][]string
	// ParentsMap maps branch name to parent branch name
	ParentsMap map[string]string
	// FixedMap maps branch name to whether it's fixed (up-to-date)
	FixedMap map[string]bool
}

type Model struct {
	Renderer *StackTreeRenderer
	Options  RenderOptions
	Width    int
	Height   int
}

type RenderMode int

type RenderOptions struct {
	// Mode specifies the rendering style. Defaults to RenderModeFull.
	Mode RenderMode

	// Steps limits traversal depth in each direction.
	Steps *int

	// OmitCurrentBranch hides the current branch from the tree.
	OmitCurrentBranch bool

	// NoStyleBranchName disables styling on branch names.
	NoStyleBranchName bool

	// HideStats hides commit count and line change stats.
	HideStats bool

	// HideSummary hides the entire summary line (stats, PR info, CI status).
	HideSummary bool

	// ShowSHAs shows commit SHAs next to branch names (for debugging).
	ShowSHAs bool

	// ShowCommitMessages shows commit messages below each branch.
	// Uses CommitMessages from BranchAnnotation.
	ShowCommitMessages bool

	// SelectedBranch is the name of the currently selected branch (for cursor).
	SelectedBranch string

	// Collapsed maps branch names to whether they are collapsed.
	Collapsed map[string]bool

	// SearchQuery is the current search filter text.
	SearchQuery string

	// SearchMatches maps branch names to whether they match the search.
	SearchMatches map[string]bool

	// NonSelectable marks branches that are visible but not selectable.
	NonSelectable map[string]bool

	// SkipSelectionPrefix omits the selection cursor/padding prefix.
	SkipSelectionPrefix bool
}

type RenderedBranch struct {
	Name  string
	Lines []string
	// CursorLineIndex is the index within Lines where the selection cursor should appear (typically 0)
	CursorLineIndex int
	// CursorLineSelected is the pre-computed cursor line with selection styling.
	// This is populated during render to enable fast selection updates without re-rendering.
	CursorLineSelected string
}

type StackTree struct {
	Branches       []string            // branches in the stack, in order
	CurrentBranchV string              // the currently checked out branch
	TrunkBranch    string              // the trunk/main branch name
	ParentMap      map[string]string   // branch -> parent mapping
	ChildrenMap    map[string][]string // branch -> children mapping
	FixedMap       map[string]bool     // branch -> whether it's fixed (optional)
}

type StackTreeRenderer struct {
	Annotations map[string]BranchAnnotation
	// contains filtered or unexported fields
}