package scriptorium

import "time"

// ArtifactRefType defines how an artifact is referenced.
type ArtifactRefType string

const (
	ArtifactRefInline ArtifactRefType = "inline"
	ArtifactRefFile   ArtifactRefType = "file"
)

// OutputFormat defines the desired output format.
type OutputFormat string

const (
	FormatText     OutputFormat = "text"
	FormatMarkdown OutputFormat = "markdown"
	FormatJSON     OutputFormat = "json"
)

// ValidationMode defines the output validation strategy.
type ValidationMode string

const (
	ValidationNone       ValidationMode = "none"
	ValidationBasic      ValidationMode = "basic"
	ValidationJSON       ValidationMode = "json"
	ValidationJSONSchema ValidationMode = "json_schema"
)

// ValidationStatus defines the result of a validation check.
type ValidationStatus string

const (
	ValidationPassed  ValidationStatus = "passed"
	ValidationFailed  ValidationStatus = "failed"
	ValidationSkipped ValidationStatus = "skipped"
)

// CacheControlType defines provider cache behavior for prompt content.
type CacheControlType string

const (
	CacheControlEphemeral CacheControlType = "ephemeral"
)

// StructuredOutputType identifies provider-level structured output modes.
type StructuredOutputType string

const (
	StructuredOutputJSONSchema StructuredOutputType = "json_schema"
)

// RunRequest represents a request to prepare a single prompt.
type RunRequest struct {
	PromptID      string
	PromptVersion string
	ProfileID     string
	Inputs        map[string]ArtifactRef
	Vars          map[string]string
	Execution     *ExecutionTargetOverride
	Validation    *OutputContract
	Metadata      map[string]string
}

// PreparedRun contains prepared prompt execution state. It does not include
// resolved API key values, model output, validation results, or internal target
// presence metadata.
type PreparedRun struct {
	PromptID             string                `json:"prompt_id"`
	PromptVersion        string                `json:"prompt_version,omitempty"`
	PromptHash           string                `json:"prompt_hash,omitempty"`
	SelectedProfileID    string                `json:"selected_profile_id"`
	EffectiveModelParams ExecutionTarget       `json:"effective_model_params"`
	OutputContract       OutputContract        `json:"output_contract"`
	StructuredOutput     *StructuredOutputSpec `json:"structured_output,omitempty"`
	InputHashes          map[string]string     `json:"input_hashes,omitempty"`
	SessionID            string                `json:"session_id,omitempty"`
	RenderedPromptHash   string                `json:"rendered_prompt_hash"`
	Messages             []RenderedMessage     `json:"messages"`
	StartTime            time.Time             `json:"start_time,omitempty"`
	EndTime              time.Time             `json:"end_time,omitempty"`
	DurationMS           int64                 `json:"duration_ms,omitempty"`
}

// ArtifactRef represents a reference to prompt input content.
type ArtifactRef struct {
	Type ArtifactRefType
	URI  string
	Body string
}

// Artifact represents loaded artifact content.
type Artifact struct {
	Name        string
	ContentType string
	Body        []byte
	URI         string
	Size        int64
	Hash        string
}

// ExecutionTarget represents effective model runtime settings.
type ExecutionTarget struct {
	Endpoint        string         `json:"endpoint"`
	Model           string         `json:"model"`
	Temperature     float64        `json:"temperature"`
	MaxTokens       int            `json:"max_tokens"`
	TopP            float64        `json:"top_p"`
	TimeoutSeconds  int            `json:"timeout_seconds"`
	ServiceTier     string         `json:"service_tier"`
	ReasoningEffort string         `json:"reasoning_effort"`
	APIKeyEnv       string         `json:"api_key_env"`
	ExtraParams     map[string]any `json:"extra_params"`
}

// ExecutionTargetOverride represents per-request runtime setting overrides.
type ExecutionTargetOverride struct {
	Endpoint        string
	Model           string
	Temperature     *float64
	MaxTokens       *int
	TopP            *float64
	TimeoutSeconds  *int
	ServiceTier     string
	ReasoningEffort string
	APIKeyEnv       string
	ExtraParams     map[string]any
}

// ExecutionTargetPresence tracks which numeric runtime settings were explicit
// request overrides.
type ExecutionTargetPresence struct {
	Temperature    bool
	MaxTokens      bool
	TopP           bool
	TimeoutSeconds bool
}

// OutputContract defines output and validation requirements.
type OutputContract struct {
	Format         OutputFormat   `json:"format"`
	ValidationMode ValidationMode `json:"validation_mode"`
	SchemaPath     string         `json:"schema_path"`
	RepairAttempts int            `json:"repair_attempts"`
}

// ValidationResult represents output validation state.
type ValidationResult struct {
	Status         ValidationStatus
	Mode           ValidationMode
	Errors         []string
	SchemaPath     string
	RepairAttempts int
	IsValid        bool
}

// TokenUsage tracks token consumption.
type TokenUsage struct {
	PromptTokens     int
	CompletionTokens int
	TotalTokens      int
	CachedTokens     int
	CacheWriteTokens int
}

// RenderedMessage is a rendered chat message.
type RenderedMessage struct {
	Role         string        `json:"role"`
	Content      string        `json:"content"`
	CacheControl *CacheControl `json:"cache_control,omitempty"`
}

// CacheControl describes provider cache metadata attached to prompt content.
type CacheControl struct {
	Type CacheControlType `json:"type"`
	TTL  string           `json:"ttl,omitempty"`
}

// StructuredOutputSpec describes provider-level structured output.
type StructuredOutputSpec struct {
	Type       StructuredOutputType      `json:"type"`
	JSONSchema *StructuredOutputJSONSpec `json:"json_schema,omitempty"`
}

// StructuredOutputJSONSpec contains JSON Schema output constraints.
type StructuredOutputJSONSpec struct {
	Name   string `json:"name"`
	Strict bool   `json:"strict"`
	Schema any    `json:"schema"`
}

// File returns a file-backed artifact reference.
func File(path string) ArtifactRef {
	return ArtifactRef{Type: ArtifactRefFile, URI: path}
}

// Inline returns an inline artifact reference.
func Inline(body string) ArtifactRef {
	return ArtifactRef{Type: ArtifactRefInline, Body: body}
}

// InlineWithURI returns an inline artifact reference with URI metadata.
func InlineWithURI(uri string, body string) ArtifactRef {
	return ArtifactRef{Type: ArtifactRefInline, URI: uri, Body: body}
}