chat

type Block struct {
	Source blockSource
	Block  string
}

type Client interface {
	AgentForPrompt(string) string
	Chat(context.Context, ai.Request) (ai.Response, error)
	Complete(context.Context, ai.Request) (ai.Response, error)
	ChatStream(context.Context, ai.Request) (<-chan ai.StreamEvent, error)
	SupportsTools(string) bool
	GenerateTitle(context.Context, string) (string, error)
}

type ClipboardRequested struct{ Text string }

type Completion struct {
	Items    []CompletionItem
	Matches  []CompletionItem
	Prefix   string
	Selected int
}

type CompletionItem struct {
	Label      string
	InsertText string
	Detail     string
	Kind       string // display category, e.g. "command"
}

type ConfirmationRequested struct {
	Statement  string
	Generation int64
}

type Context struct {
	ConnectionID string
	Database     sharedsql.DatabaseInfo
	Schema       []sharedsql.SchemaObject
	Query        string
	Results      sharedsql.Result
}

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

type Executor interface {
	ExecuteReadOnly(ctx context.Context, statement string) (sharedsql.Result, error)
}

type History interface {
	NewConversation(context.Context, string, string) (ai.Conversation, error)
	AppendMessage(context.Context, string, string, ai.Message) error
	Conversations(context.Context, string) ([]ai.Conversation, error)
	Messages(context.Context, string, string) ([]ai.Message, error)
	DeleteConversation(context.Context, string, string) error
	Clear(context.Context, string) error
	RenameConversation(context.Context, string, string, string) error
}

type HistoryDeletedMsg struct {
	Err            error
	ConnectionID   string
	ConversationID string // conversation the delete targeted ("" for clear-all)
	Clear          bool
}

type HistoryLoadedMsg struct {
	ConnectionID  string
	Conversations []ai.Conversation
	Err           error
}

type HistoryPickerOutcome struct {
	Picked string // selected conversation id; empty when the picker closed
	Closed bool
}

type MessagesLoadedMsg struct {
	ConnectionID   string
	ConversationID string
	Messages       []ai.Message
	Seq            int64
	Err            error
}

type Mode uint8

type Model struct {
	Input    textarea.Model
	Viewport viewport.Model
	Client   Client
	History  History
	ActiveID string // conversation shown; "" is the fresh unsent view
	Runs     map[string]*Run
	NextGen  int64 // globally unique turn ids across concurrent runs
	LoadSeq  int64 // bumps on each /history selection; drops stale loads
	Enabled  bool
	Visible  bool

	ShareResults  bool
	HistoryChoice string
	ChatMode      Mode

	Completion Completion // slash-command suggestions while typing
	// HistoryPicker is the /history conversation picker overlay.
	HistoryPicker *huh.Form
	// KeepInsert keeps the chat input in insert mode after a release
	// click inside the chat pane.
	KeepInsert bool

	// PromptHistory is the newest-first list of accepted user prompts for
	// this process; HistoryIndex == -1 means not browsing recall.
	PromptHistory []string
	HistoryIndex  int

	// YoloWrites: when true, sql_write executes without per-statement modal.
	YoloWrites bool

	// AppContext is the root application context run commands derive
	// from, so quitting cancels in-flight streams and history writes.
	AppContext context.Context

	// Session state the root keeps current (the chat-scoped mirror of the
	// connection): the executor read-only tools query through, the
	// read-only flag (gates the sql_write tool), the opened target
	// (connection info), and the last failed statement the assistant
	// context summarizes.
	Executor        Executor
	ReadOnly        bool
	Target          string
	LastFailedQuery string
	LastFailedError string
	// contains filtered or unexported fields
}

type PendingWrite struct {
	Generation int64
	Call       ai.ToolCall
	Statement  string
}

type PersistMsg struct{ Err error }

type ResponseMsg struct {
	ConversationID string
	Response       ai.Response
	Err            error
}

type Run struct {
	ConversationID string
	ConnectionID   string
	Messages       []ai.Message
	// BlockCache holds rendered viewport blocks per message so
	// RefreshView re-renders only appended/replaced messages instead of
	// the whole conversation on every stream delta.
	BlockCache  []Block
	CachedWidth int

	StreamBuffer string // accumulated streaming content
	Loading      bool
	Canceled     bool
	Cancel       context.CancelFunc
	Gen          int64 // turn id; checked on async completions
	SpinnerFrame int   // progress spinner frame while loading
	RoundState   *ToolRoundState
	PendingWrite *PendingWrite
	// contains filtered or unexported fields
}

type SQLRequested struct {
	Statement string
	ReadOnly  bool
	Source    string
}

type SchemaRequested struct{}

type SpinnerTickMsg struct{}

type StatusChanged struct{ Text string }

type StreamMsg struct {
	Ch             <-chan ai.StreamEvent
	ConversationID string
	Delta          string
	Response       ai.Response
	Done           bool
	Err            error
}

type TitleMsg struct {
	ConnectionID   string
	ConversationID string
	Title          string
	Err            error
}

type ToolContinueMsg struct{ Gen int64 }

type ToolPhaseExpiredMsg struct {
	Gen   int64
	State *ToolRoundState
}

type ToolResultMsg struct {
	Gen      int64
	CallID   string
	CallName string
	Content  string
	Err      string
}

type ToolRoundState struct {
	Gen                     int64
	Messages                []ai.Message
	AgentID                 string
	Client                  Client
	History                 History
	RootContext             context.Context
	ChatContext             context.Context
	Cancel                  context.CancelFunc
	ToolCancel              context.CancelFunc
	FinalizationCancel      context.CancelFunc
	ContextText             string
	ToolsDefs               []ai.ToolDefinition
	ConversationID          string
	ToolCalls               []ai.ToolCall
	NextCall                int
	ToolCallCount           int
	ToolDeadline            time.Time
	Finalizing              bool
	LastToolResultSignature string
	RepeatedToolResults     int
}

type ToolStartMsg struct {
	Gen   int64
	State ToolRoundState
}

type WriteRequest struct {
	Statement  string
	Generation int64
	Deadline   time.Time
}

type WriteResultMsg struct {
	Gen      int64
	CallID   string
	CallName string
	Content  string
	Err      string
	Declined bool // user declined the write; stop round
}