media

type EnrichmentConfig struct {
	AutoEnrichImages    bool `yaml:"auto_enrich_images" json:"auto_enrich_images"`
	AutoEnrichAudio     bool `yaml:"auto_enrich_audio" json:"auto_enrich_audio"`
	AutoEnrichDocuments bool `yaml:"auto_enrich_documents" json:"auto_enrich_documents"`
}

type EnrichmentResult struct {
	MediaID     string
	Type        MediaType
	Description string // Vision description
	Transcript  string // Audio transcript
	TextContent string // Document text
	Metadata    map[string]any
}

type ExtractDocFunc func(ctx context.Context, data []byte, mimeType string) (string, error)

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

type LLMMediaPrep struct {
	MediaID    string
	Type       MediaType
	Base64Data string
	MimeType   string
	Caption    string
	FileSize   int64
}

type ListFilter struct {
	Channel   string
	SessionID string
	Type      MediaType
	Temporary *bool
	Limit     int
	Offset    int
}

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

type MediaServiceOption func(*MediaService)

type MediaStore interface {
	// Save stores media data and returns metadata.
	Save(ctx context.Context, req SaveRequest) (*StoredMedia, error)

	// Get retrieves media data by ID.
	Get(ctx context.Context, id string) (io.ReadCloser, *StoredMedia, error)

	// GetBytes retrieves media data as bytes.
	GetBytes(ctx context.Context, id string) ([]byte, *StoredMedia, error)

	// Delete removes media by ID.
	Delete(ctx context.Context, id string) error

	// List returns media matching the filter.
	List(ctx context.Context, filter ListFilter) ([]*StoredMedia, error)

	// DeleteExpired removes all temporary media past their expiration.
	DeleteExpired(ctx context.Context) (int, error)

	// URL returns a URL for accessing the media.
	URL(id string) string
}

type MediaType string

type SaveRequest struct {
	Data      []byte
	Filename  string
	MimeType  string
	Type      MediaType
	Channel   string
	SessionID string
	Temporary bool
	TTL       time.Duration
	Metadata  map[string]any
}

type ServiceConfig struct {
	Enabled         bool   `yaml:"enabled" json:"enabled"`
	MaxImageSize    int64  `yaml:"max_image_size" json:"max_image_size"`
	MaxAudioSize    int64  `yaml:"max_audio_size" json:"max_audio_size"`
	MaxDocSize      int64  `yaml:"max_doc_size" json:"max_doc_size"`
	MaxUploadSize   int64  `yaml:"max_upload_size" json:"max_upload_size"`
	TempTTL         string `yaml:"temp_ttl" json:"temp_ttl"`
	CleanupEnabled  bool   `yaml:"cleanup_enabled" json:"cleanup_enabled"`
	CleanupInterval string `yaml:"cleanup_interval" json:"cleanup_interval"`
}

type StoreConfig struct {
	BaseDir     string `yaml:"base_dir" json:"base_dir"`
	TempDir     string `yaml:"temp_dir" json:"temp_dir"`
	BaseURL     string `yaml:"base_url" json:"base_url"`
	MaxFileSize int64  `yaml:"max_file_size" json:"max_file_size"`
}

type StoredMedia struct {
	ID        string         `json:"id"`
	Filename  string         `json:"filename"`
	MimeType  string         `json:"mime_type"`
	Type      MediaType      `json:"type"`
	Size      int64          `json:"size"`
	Channel   string         `json:"channel"`
	SessionID string         `json:"session_id,omitempty"`
	Temporary bool           `json:"temporary"`
	CreatedAt time.Time      `json:"created_at"`
	ExpiresAt *time.Time     `json:"expires_at,omitempty"`
	Metadata  map[string]any `json:"metadata,omitempty"`
}

type TranscribeFunc func(ctx context.Context, audioData []byte, filename string) (string, error)

type URLValidator interface {
	IsAllowed(url string) error
}

type UploadRequest struct {
	Data      []byte
	Filename  string
	MimeType  string
	Channel   string
	SessionID string
	Temporary bool
}

type ValidationResult struct {
	MimeType string
	Type     MediaType
	Size     int64
	Valid    bool
	Error    string
}

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

type VisionFunc func(ctx context.Context, imageData []byte, mimeType string) (string, error)