domain

type ConversationEntry struct {
	// Core message fields
	Message          sdk.Message                   `json:"message"`
	Model            string                        `json:"model,omitempty"`
	Time             time.Time                     `json:"time"`
	Hidden           bool                          `json:"hidden,omitempty"`
	Images           []agentdomain.ImageAttachment `json:"images,omitempty"`
	ReasoningContent string                        `json:"reasoning_content,omitempty"`

	// Tool-related fields
	ToolExecution      *agentdomain.ToolExecutionResult   `json:"tool_execution,omitempty"`
	PendingToolCall    *sdk.ChatCompletionMessageToolCall `json:"pending_tool_call,omitempty"`
	ToolApprovalStatus ToolApprovalStatus                 `json:"tool_approval_status,omitempty"`

	// Plan mode fields
	Rejected           bool               `json:"rejected,omitempty"`
	IsPlan             bool               `json:"is_plan,omitempty"`
	PlanApprovalStatus PlanApprovalStatus `json:"plan_approval_status,omitempty"`
}

type ConversationMetadata struct {
	ID                  string            `json:"id"`
	Title               string            `json:"title"`
	CreatedAt           time.Time         `json:"created_at"`
	UpdatedAt           time.Time         `json:"updated_at"`
	MessageCount        int               `json:"message_count"`
	TokenStats          SessionTokenStats `json:"token_stats"`
	CostStats           SessionCostStats  `json:"cost_stats,omitempty"`
	Model               string            `json:"model,omitempty"`
	Tags                []string          `json:"tags,omitempty"`
	TitleGenerated      bool              `json:"title_generated,omitempty"`
	TitleInvalidated    bool              `json:"title_invalidated,omitempty"`
	TitleGenerationTime *time.Time        `json:"title_generation_time,omitempty"`
	ContextID           string            `json:"context_id,omitempty"`

	// ParentSessionID is the orchestrator session that spawned this one
	// (e.g. a subagent's parent). Empty for top-level sessions.
	ParentSessionID string `json:"parent_session_id"`

	// InvokedBy indicates who started the session: "human" or "agent".
	// Defaults to "human" for sessions with no explicit invoker.
	InvokedBy string `json:"invoked_by"`
}

type ConversationOptimizer interface {
	OptimizeMessages(messages []sdk.Message, model string, force bool) []sdk.Message
}

type ConversationRepository interface {
	AddMessage(msg ConversationEntry) error
	GetMessages() []ConversationEntry
	Clear() error
	ClearExceptFirstUserMessage() error
	GetMessageCount() int
	UpdateLastMessage(content string) error
	UpdateLastMessageToolCalls(toolCalls *[]sdk.ChatCompletionMessageToolCall) error
	DeleteMessagesAfterIndex(index int) error

	AddTokenUsage(model string, inputTokens, outputTokens, totalTokens, cachedTokens, cacheWriteTokens int) error
	AddCachedTokens(tokens int)
	GetSessionTokens() SessionTokenStats
	GetSessionCostStats() SessionCostStats

	FormatToolResultForLLM(result *agentdomain.ToolExecutionResult) string
	FormatToolResultForUI(result *agentdomain.ToolExecutionResult, terminalWidth int) string
	FormatToolResultExpanded(result *agentdomain.ToolExecutionResult, terminalWidth int) string

	RemovePendingToolCallByID(toolCallID string)

	StartNewConversation(title string) error
	LoadConversation(ctx context.Context, conversationID string) error
	GetCurrentConversationTitle() string
	GetCurrentConversationID() string

	Export(format ExportFormat) ([]byte, error)
}

type ConversationSummary struct {
	ID                  string            `json:"id"`
	Title               string            `json:"title"`
	CreatedAt           time.Time         `json:"created_at"`
	UpdatedAt           time.Time         `json:"updated_at"`
	MessageCount        int               `json:"message_count"`
	TokenStats          SessionTokenStats `json:"token_stats"`
	CostStats           SessionCostStats  `json:"cost_stats,omitempty"`
	Model               string            `json:"model,omitempty"`
	Tags                []string          `json:"tags,omitempty"`
	Summary             string            `json:"summary,omitempty"`
	TitleGenerated      bool              `json:"title_generated,omitempty"`
	TitleInvalidated    bool              `json:"title_invalidated,omitempty"`
	TitleGenerationTime *time.Time        `json:"title_generation_time,omitempty"`

	// ParentSessionID is the orchestrator session that spawned this one.
	ParentSessionID string `json:"parent_session_id"`

	// InvokedBy indicates who started the session: "human" or "agent".
	InvokedBy string `json:"invoked_by"`
}

type ExportFormat string

type MessageQueue interface {
	// Enqueue adds a message to the queue
	Enqueue(message sdk.Message, requestID string)

	// Dequeue removes and returns the next message from the queue
	// Returns nil if the queue is empty
	Dequeue() *QueuedMessage

	// Peek returns the next message without removing it
	// Returns nil if the queue is empty
	Peek() *QueuedMessage

	// Size returns the number of messages in the queue
	Size() int

	// IsEmpty returns true if the queue has no messages
	IsEmpty() bool

	// Clear removes all messages from the queue
	Clear()

	// GetAll returns all messages in the queue without removing them
	GetAll() []QueuedMessage
}

type ModelCostStats struct {
	Model        string
	InputTokens  int
	OutputTokens int
	InputCost    float64
	OutputCost   float64
	TotalCost    float64
	RequestCount int
}

type ModelService interface {
	ListModels(ctx context.Context) ([]string, error)
	SelectModel(modelID string) error
	GetCurrentModel() string
	IsModelAvailable(modelID string) bool
	ValidateModel(modelID string) error
}

type PlanApprovalStatus int

type PricingService interface {
	// IsEnabled returns whether pricing is enabled in the configuration.
	IsEnabled() bool

	// GetInputPrice retrieves the input price per million tokens for a specific model.
	// Returns 0.0 for unknown models (e.g., Ollama, custom models).
	GetInputPrice(model string) float64

	// GetOutputPrice retrieves the output price per million tokens for a specific model.
	// Returns 0.0 for unknown models (e.g., Ollama, custom models).
	GetOutputPrice(model string) float64

	// CalculateCost computes the total cost for a given number of input and
	// output tokens. cachedTokens and cacheWriteTokens are the cache-read and
	// cache-creation subsets of inputTokens, billed at the gateway's
	// cache-read/cache-write rates when known (full input rate otherwise).
	CalculateCost(model string, inputTokens, outputTokens, cachedTokens, cacheWriteTokens int) (inputCost, outputCost, totalCost float64)

	// RequiresPro reports whether the model is gated behind a paid Pro
	// subscription (e.g. some Ollama Cloud models). Resolves custom prices
	// first, then defaults. Returns false when pricing is disabled or the
	// model has no entry.
	RequiresPro(model string) bool

	// FormatModelPricing returns a formatted string describing the model's pricing.
	// Returns empty string if pricing is disabled or the model has no pricing entry.
	// Returns "free" only when an explicit pricing entry sets both prices to 0.0.
	// Returns "$X.XX/$Y.YY per MTok" for paid models.
	FormatModelPricing(model string) string
}

type QueuedMessage struct {
	Message   sdk.Message
	QueuedAt  time.Time
	RequestID string
}

type SessionCostStats struct {
	TotalCost       float64
	TotalInputCost  float64
	TotalOutputCost float64
	PerModelStats   map[string]*ModelCostStats
	Currency        string
}

type SessionID string

type SessionTokenStats struct {
	TotalInputTokens      int `json:"total_input_tokens"`
	TotalOutputTokens     int `json:"total_output_tokens"`
	TotalTokens           int `json:"total_tokens"`
	RequestCount          int `json:"request_count"`
	LastInputTokens       int `json:"last_input_tokens"`
	TotalCachedTokens     int `json:"total_cached_tokens"`
	TotalCacheWriteTokens int `json:"total_cache_write_tokens"`
}

type TokenEstimator interface {
	// GetToolStats returns token count and tool count for a given agent mode
	GetToolStats(toolService agentdomain.ToolService, agentMode agentdomain.AgentMode) (tokens int, count int)

	// EstimateMessagesTokens estimates the total tokens for a slice of messages
	EstimateMessagesTokens(messages []sdk.Message) int

	// EffectiveContextTokens estimates what the *next* request will carry: the
	// larger of the gateway-reported last-request size and a fresh estimate of
	// the current buffer. The max catches a single-turn tool-output spike that a
	// stale lastInputTokens alone would miss.
	EffectiveContextTokens(lastInputTokens int, messages []sdk.Message) int
}

type ToolApprovalStatus int