package llm

import "context"

// Provider is the interface for LLM backends.
type Provider interface {
	Summarize(ctx context.Context, input SummaryInput) (string, error)
	RewriteAction(ctx context.Context, action ActionInput) (string, error)
	GenerateHeatPlan(ctx context.Context, input HeatPlanInput) (string, error)
	GenerateActions(ctx context.Context, input ActionsInput) (string, error)
	Name() string
}

// HeatSource represents a ranked heat source for summary.
type HeatSource struct {
	Name  string
	Watts float64
}

// RiskWindowSummary is a simplified risk window for LLM input.
type RiskWindowSummary struct {
	StartHour int
	EndHour   int
	PeakTempC float64
	Level     string
}

// SummaryInput holds computed data for the 3-bullet summary.
type SummaryInput struct {
	Date           string
	PeakTempC      float64
	MinNightTempC  float64
	RiskLevel      string
	TopHeatSources []HeatSource
	ACHeadroomBTUH float64
	BudgetStatus   string
	ActiveWarnings []string
	RiskWindows    []RiskWindowSummary
	Language       string
}

// ActionInput holds data for rewriting a technical action.
type ActionInput struct {
	ActionName  string
	Description string
	TempC       float64
	Hour        int
}

// TimelineSlotSummary is a simplified timeline slot for LLM input.
type TimelineSlotSummary struct {
	Hour         int
	TempC        float64
	RiskLevel    string
	BudgetStatus string
	Actions      []string
}

// ActionSummary is a simplified action for LLM input.
type ActionSummary struct {
	Name     string
	Category string
	Impact   string
	Hour     int
}

// HeatPlanInput holds full day data for the 1-page plan.
type HeatPlanInput struct {
	Summary       SummaryInput
	Timeline      []TimelineSlotSummary
	Actions       []ActionSummary
	CareChecklist []string
}

// ActionsInput holds context for AI-generated actions.
type ActionsInput struct {
	Date          string
	Language      string
	IndoorTempC   float64
	PeakTempC     float64
	MinNightTempC float64
	PoorNightCool bool
	RiskLevel     string
	RiskWindows   []RiskWindowSummary
	Timeline      []ActionsTimelineSlot
	Rooms         []ActionsRoom
}

// ActionsTimelineSlot is one hour's data for AI action generation.
type ActionsTimelineSlot struct {
	Hour         int
	TempC        float64
	HumidityPct  float64
	BudgetStatus string
	CoolMode     string
	GainsW       float64
}

// ActionsRoom describes a room for AI action generation.
type ActionsRoom struct {
	Name        string
	Orientation string
	ShadingType string
	HasAC       bool
}