adapters

type ArenaAssertionResult struct {
	Type    string `json:"type"`
	Passed  bool   `json:"passed"`
	Message string `json:"message,omitempty"`
}

type ArenaContentPart struct {
	Type  string              `json:"type"`
	Text  *string             `json:"text,omitempty"`
	Media *types.MediaContent `json:"media,omitempty"`
}

type ArenaCost struct {
	InputTokens  int     `json:"input_tokens"`
	OutputTokens int     `json:"output_tokens"`
	TotalCost    float64 `json:"total_cost"`
}

type ArenaMessage struct {
	Content string             `json:"content"`
	Parts   []ArenaContentPart `json:"parts,omitempty"`
}

type ArenaOutputAdapter struct{}

type ArenaOutputFile struct {
	ScenarioID string              `json:"scenario_id"`
	ProviderID string              `json:"provider_id"`
	Metadata   ArenaOutputMetadata `json:"metadata"`
	Turns      []ArenaOutputTurn   `json:"turns"`
	Summary    ArenaOutputSummary  `json:"summary"`
}

type ArenaOutputMetadata struct {
	Tags   []string               `json:"tags,omitempty"`
	Extras map[string]interface{} `json:"extras,omitempty"`
}

type ArenaOutputSummary struct {
	TotalTurns  int     `json:"total_turns"`
	PassedTurns int     `json:"passed_turns"`
	FailedTurns int     `json:"failed_turns"`
	TotalCost   float64 `json:"total_cost"`
}

type ArenaOutputTurn struct {
	TurnIndex   int                    `json:"turn_index"`
	Timestamp   time.Time              `json:"timestamp"`
	UserMessage ArenaMessage           `json:"user_message"`
	Response    ArenaResponse          `json:"response"`
	ToolResults []ArenaToolResult      `json:"tool_results,omitempty"`
	Assertions  []ArenaAssertionResult `json:"assertions,omitempty"`
}

type ArenaResponse struct {
	Message types.Message `json:"message"`
	Cost    *ArenaCost    `json:"cost,omitempty"`
}

type ArenaToolResult struct {
	ToolCallID string `json:"tool_call_id"`
	Content    string `json:"content"`
}

type MediaSource struct {
	MIMEType string
	Data     string
	URI      string
	Path     string
	Size     int64
	Width    int
	Height   int
	Duration int64 // milliseconds
}

type ProviderSpec struct {
	Type  string `json:"type" yaml:"type"`
	Model string `json:"model" yaml:"model"`
	ID    string `json:"id" yaml:"id"`
}

type RecordedContentPart struct {
	Type  string             `json:"type"`
	Text  *string            `json:"text,omitempty"`
	Media *RecordedMediaPart `json:"media,omitempty"`
}

type RecordedMediaPart struct {
	MIMEType string `json:"mime_type,omitempty"`
	Data     string `json:"data,omitempty"` // Base64 encoded
	URI      string `json:"uri,omitempty"`  // URL
	Path     string `json:"path,omitempty"` // File path
	Size     int64  `json:"size,omitempty"`
	Width    int    `json:"width,omitempty"`
	Height   int    `json:"height,omitempty"`
	Duration int64  `json:"duration,omitempty"` // milliseconds
}

type RecordedMsg struct {
	Role       string                `json:"role"`
	Content    string                `json:"content"`
	Name       string                `json:"name,omitempty"`
	Parts      []RecordedContentPart `json:"parts,omitempty"`
	ToolCalls  []RecordedToolCall    `json:"tool_calls,omitempty"`
	ToolCallID string                `json:"tool_call_id,omitempty"`
}

type RecordedToolCall struct {
	ID       string                   `json:"id"`
	Type     string                   `json:"type"`
	Function RecordedToolCallFunction `json:"function"`
}

type RecordedToolCallFunction struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type RecordingAdapter interface {
	// CanHandle returns true if this adapter supports the given source/type hint.
	// The source could be a file path, glob pattern, database query, etc.
	// typeHint is an optional explicit format indicator from the eval config.
	CanHandle(source string, typeHint string) bool

	// Enumerate expands a source into individual recording references.
	// For file-based adapters, this expands glob patterns to matching files.
	// For database adapters, this could execute a query and return record IDs.
	// Returns a single-element slice for non-expandable sources.
	Enumerate(source string) ([]RecordingReference, error)

	// Load converts a recording to Arena message format.
	// The reference should have been obtained from Enumerate.
	// Returns the messages, metadata, and any error encountered.
	Load(ref RecordingReference) ([]types.Message, *RecordingMetadata, error)
}

type RecordingEvent struct {
	Type      string      `json:"type"`
	Timestamp time.Time   `json:"timestamp"`
	Message   RecordedMsg `json:"message,omitempty"`
	Event     interface{} `json:"event,omitempty"`
}

type RecordingMetadata struct {
	// JudgeTargets maps judge names to provider specifications.
	// Used by LLM judge assertions to determine which provider to use.
	JudgeTargets map[string]ProviderSpec `json:"judge_targets,omitempty" yaml:"judge_targets,omitempty"`

	// ProviderInfo contains information about the original provider(s)
	// that generated the recorded conversation.
	ProviderInfo map[string]interface{} `json:"provider_info,omitempty" yaml:"provider_info,omitempty"`

	// Tags are optional labels for categorizing/filtering recordings.
	Tags []string `json:"tags,omitempty" yaml:"tags,omitempty"`

	// Timestamps contains the timestamp for each turn in the conversation.
	// The length should match the number of messages.
	Timestamps []time.Time `json:"timestamps,omitempty" yaml:"timestamps,omitempty"`

	// SessionID is the unique identifier for the recorded session.
	SessionID string `json:"session_id,omitempty" yaml:"session_id,omitempty"`

	// Duration is the total duration of the conversation.
	Duration time.Duration `json:"duration,omitempty" yaml:"duration,omitempty"`

	// Extras holds any additional metadata from the recording.
	Extras map[string]interface{} `json:"extras,omitempty" yaml:"extras,omitempty"`
}

type RecordingMetadataFile struct {
	SessionID  string   `json:"session_id"`
	ProviderID string   `json:"provider_id,omitempty"`
	Model      string   `json:"model,omitempty"`
	Tags       []string `json:"tags,omitempty"`
}

type RecordingReference struct {
	// ID is a unique identifier for this recording reference.
	// For file-based adapters, this is typically the file path.
	// For database adapters, this could be a record ID.
	ID string `json:"id" yaml:"id"`

	// Source is the original source pattern/query that produced this reference.
	// For file-based adapters, this is the original glob pattern or path.
	Source string `json:"source" yaml:"source"`

	// TypeHint is an optional format indicator (e.g., "session", "arena_output", "transcript").
	TypeHint string `json:"type_hint,omitempty" yaml:"type_hint,omitempty"`

	// Metadata contains adapter-specific metadata about this reference.
	// This can include file size, modification time, database metadata, etc.
	Metadata map[string]interface{} `json:"metadata,omitempty" yaml:"metadata,omitempty"`
}

type Registry struct {
	// contains filtered or unexported fields
}

type SessionRecordingAdapter struct{}

type SessionRecordingFile struct {
	Metadata RecordingMetadataFile `json:"metadata"`
	Events   []RecordingEvent      `json:"events"`
}

type TranscriptAdapter struct{}

type TranscriptContentPart struct {
	Type  string               `yaml:"type"`
	Text  *string              `yaml:"text,omitempty"`
	Media *TranscriptMediaPart `yaml:"media,omitempty"`
}

type TranscriptFile struct {
	Metadata TranscriptMetadata  `yaml:"metadata"`
	Messages []TranscriptMessage `yaml:"messages"`
}

type TranscriptMediaPart struct {
	MIMEType string `yaml:"mime_type,omitempty"`
	Data     string `yaml:"data,omitempty"` // Base64 encoded
	URI      string `yaml:"uri,omitempty"`  // URL
	Path     string `yaml:"path,omitempty"` // File path
	Size     int64  `yaml:"size,omitempty"`
	Width    int    `yaml:"width,omitempty"`
	Height   int    `yaml:"height,omitempty"`
	Duration int64  `yaml:"duration,omitempty"` // milliseconds
}

type TranscriptMessage struct {
	Role       string                  `yaml:"role"`
	Content    string                  `yaml:"content"`
	Name       string                  `yaml:"name,omitempty"`
	Timestamp  string                  `yaml:"timestamp,omitempty"`
	Parts      []TranscriptContentPart `yaml:"parts,omitempty"`
	ToolCalls  []TranscriptToolCall    `yaml:"tool_calls,omitempty"`
	ToolCallID string                  `yaml:"tool_call_id,omitempty"`
}

type TranscriptMetadata struct {
	SessionID    string                            `yaml:"session_id,omitempty"`
	Provider     string                            `yaml:"provider,omitempty"`
	Model        string                            `yaml:"model,omitempty"`
	Tags         []string                          `yaml:"tags,omitempty"`
	JudgeTargets map[string]TranscriptProviderSpec `yaml:"judge_targets,omitempty"`
}

type TranscriptProviderSpec struct {
	Type  string `yaml:"type"`
	Model string `yaml:"model"`
	ID    string `yaml:"id"`
}

type TranscriptToolCall struct {
	ID       string                     `yaml:"id"`
	Type     string                     `yaml:"type"`
	Function TranscriptToolCallFunction `yaml:"function"`
}

type TranscriptToolCallFunction struct {
	Name      string `yaml:"name"`
	Arguments string `yaml:"arguments"`
}