package contracts

import (
	"context"
	"encoding/json"

	"gitea.maximumdirect.net/eric/notarius/internal/core/artifacts"
	"gitea.maximumdirect.net/eric/notarius/internal/core/source"
)

type LLMMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type StructuredCompletionRequest struct {
	StageName          string          `json:"stage_name"`
	Messages           []LLMMessage    `json:"messages"`
	Model              string          `json:"model,omitempty"`
	ResponseSchemaName string          `json:"response_schema_name,omitempty"`
	ResponseSchema     json.RawMessage `json:"response_schema,omitempty"`
}

type StructuredCompletionResponse struct {
	Content          json.RawMessage `json:"content"`
	Provider         string          `json:"provider,omitempty"`
	Model            string          `json:"model,omitempty"`
	PromptTokens     int             `json:"prompt_tokens,omitempty"`
	CompletionTokens int             `json:"completion_tokens,omitempty"`
	TotalTokens      int             `json:"total_tokens,omitempty"`
}

type StructuredLLMClient interface {
	CompleteStructured(ctx context.Context, req StructuredCompletionRequest, out any) (StructuredCompletionResponse, error)
}

type ParseRequest struct {
	SourceID   string         `json:"source_id,omitempty"`
	Path       string         `json:"path,omitempty"`
	Raw        []byte         `json:"-"`
	LLMProfile string         `json:"llm_profile,omitempty"`
	Options    map[string]any `json:"options,omitempty"`
	Metadata   map[string]any `json:"metadata,omitempty"`
}

type InputAdapter interface {
	Key() string
	Parse(ctx context.Context, req ParseRequest) (*source.SourceDocument, error)
}

type SourceChunk struct {
	ID       string              `json:"id"`
	SourceID string              `json:"source_id"`
	Index    int                 `json:"index"`
	Units    []source.SourceUnit `json:"units"`
	Metadata map[string]any      `json:"metadata,omitempty"`
}

type ChunkRequest struct {
	Source     *source.SourceDocument `json:"-"`
	LLMClient  StructuredLLMClient    `json:"-"`
	LLMProfile string                 `json:"llm_profile,omitempty"`
	Options    map[string]any         `json:"options,omitempty"`
	Metadata   map[string]any         `json:"metadata,omitempty"`
}

type ChunkResult struct {
	Chunks   []SourceChunk `json:"chunks"`
	Warnings []Warning     `json:"warnings,omitempty"`
}

type Chunker interface {
	Key() string
	Chunk(ctx context.Context, req ChunkRequest) (ChunkResult, error)
}

const (
	ReferenceBindingSourceConfig = "config"
	ReferenceBindingSourceCLI    = "cli"
)

type ReferenceSlot struct {
	Name               string   `json:"name"`
	Description        string   `json:"description,omitempty"`
	Required           bool     `json:"required,omitempty"`
	AcceptedMediaTypes []string `json:"accepted_media_types,omitempty"`
	Multiple           bool     `json:"multiple,omitempty"`
	MaxBytes           int64    `json:"max_bytes,omitempty"`
}

type ReferenceOrigin struct {
	Type string `json:"type"`
	URI  string `json:"uri,omitempty"`
}

type ReferenceItem struct {
	SlotName      string          `json:"slot_name"`
	MediaType     string          `json:"media_type,omitempty"`
	Content       []byte          `json:"-"`
	Digest        string          `json:"digest,omitempty"`
	Origin        ReferenceOrigin `json:"origin"`
	SizeBytes     int64           `json:"size_bytes,omitempty"`
	BindingSource string          `json:"binding_source,omitempty"`
}

type ResolvedReferenceSlot struct {
	Slot  ReferenceSlot   `json:"slot"`
	Items []ReferenceItem `json:"items,omitempty"`
}

type ReferenceSet struct {
	Slots map[string]ResolvedReferenceSlot `json:"slots,omitempty"`
}

type ExtractionRequest struct {
	Source         *source.SourceDocument `json:"-"`
	Chunk          *SourceChunk           `json:"chunk,omitempty"`
	AmbientContext map[string]any         `json:"ambient_context,omitempty"`
	References     ReferenceSet           `json:"references,omitempty"`
	LLMClient      StructuredLLMClient    `json:"-"`
	LLMProfile     string                 `json:"llm_profile,omitempty"`
	Options        map[string]any         `json:"options,omitempty"`
	Metadata       map[string]any         `json:"metadata,omitempty"`
}

type ExtractionResult struct {
	Candidates []artifacts.ArtifactCandidate `json:"candidates,omitempty"`
	Warnings   []Warning                     `json:"warnings,omitempty"`
}

type Extractor interface {
	Key() string
	ArtifactType() string
	SchemaVersion() string
	ReferenceSlots() []ReferenceSlot
	Validators() []Validator
	Extract(ctx context.Context, req ExtractionRequest) (ExtractionResult, error)
}

type ChunkArtifacts struct {
	Chunk      SourceChunk                   `json:"chunk"`
	Candidates []artifacts.ArtifactCandidate `json:"candidates"`
}

type MergeRequest struct {
	Source         *source.SourceDocument `json:"-"`
	LaneID         string                 `json:"lane_id"`
	ChunkArtifacts []ChunkArtifacts       `json:"chunk_artifacts"`
	LLMProfile     string                 `json:"llm_profile,omitempty"`
	Options        map[string]any         `json:"options,omitempty"`
	Metadata       map[string]any         `json:"metadata,omitempty"`
}

type MergeResult struct {
	Candidates []artifacts.ArtifactCandidate `json:"candidates"`
	Warnings   []Warning                     `json:"warnings,omitempty"`
}

type Merger interface {
	Key() string
	Merge(ctx context.Context, req MergeRequest) (MergeResult, error)
}

type NormalizeRequest struct {
	Source     *source.SourceDocument        `json:"-"`
	LaneID     string                        `json:"lane_id"`
	Candidates []artifacts.ArtifactCandidate `json:"candidates"`
	LLMProfile string                        `json:"llm_profile,omitempty"`
	Options    map[string]any                `json:"options,omitempty"`
	Metadata   map[string]any                `json:"metadata,omitempty"`
}

type NormalizeResult struct {
	Candidates []artifacts.ArtifactCandidate `json:"candidates"`
	Warnings   []Warning                     `json:"warnings,omitempty"`
}

type Normalizer interface {
	Key() string
	Normalize(ctx context.Context, req NormalizeRequest) (NormalizeResult, error)
}

type ValidationRequest struct {
	Source     *source.SourceDocument        `json:"-"`
	Candidates []artifacts.ArtifactCandidate `json:"candidates"`
	LLMProfile string                        `json:"llm_profile,omitempty"`
	Options    map[string]any                `json:"options,omitempty"`
	Metadata   map[string]any                `json:"metadata,omitempty"`
}

type ValidationDecision struct {
	CandidateIndex         int    `json:"candidate_index"`
	Approved               bool   `json:"approved"`
	ReasonCode             string `json:"reason_code"`
	Message                string `json:"message"`
	DiagnosticArtifactPath string `json:"diagnostic_artifact_path,omitempty"`
}

type ValidationResult struct {
	ValidatorName string               `json:"validator_name"`
	Decisions     []ValidationDecision `json:"decisions"`
	Warnings      []Warning            `json:"warnings,omitempty"`
}

type Validator interface {
	Name() string
	Validate(ctx context.Context, req ValidationRequest) (ValidationResult, error)
}

type Warning struct {
	Scope      string `json:"scope,omitempty"`
	ReasonCode string `json:"reason_code"`
	Message    string `json:"message"`
}

type OutputRequest struct {
	Manifest   artifacts.RunManifest        `json:"manifest"`
	Approved   []artifacts.Artifact         `json:"approved,omitempty"`
	Rejected   []artifacts.RejectedArtifact `json:"rejected,omitempty"`
	Warnings   []Warning                    `json:"warnings,omitempty"`
	LLMProfile string                       `json:"llm_profile,omitempty"`
	Options    map[string]any               `json:"options,omitempty"`
	Metadata   map[string]any               `json:"metadata,omitempty"`
}

type OutputFile struct {
	Name        string `json:"name"`
	ContentType string `json:"content_type,omitempty"`
	Bytes       []byte `json:"-"`
}

type OutputResult struct {
	Files    []OutputFile `json:"files,omitempty"`
	Warnings []Warning    `json:"warnings,omitempty"`
}

type OutputEncoder interface {
	Key() string
	Encode(ctx context.Context, req OutputRequest) (OutputResult, error)
}

type ManifestMetadataProvider interface {
	ManifestMetadata() map[string]any
}