types

type FileMetadataExtractor struct {
	DateLayouts []string
	LogfileName string
}

type GenerationParameters interface {

	// The version of the software that generated the metadata.
	Version() string

	// The prompt used for the generation.
	PositivePrompt() string
	// The negative prompt used for the generation.
	NegativePrompt() string
	// The model used for the generation.
	Model() string
	// The LoRAs used for the generation.
	LoRAs() []Lora
	// The seed used for the generation.
	Seed() string

	// Raw returns the underlying metadata struct (e.g. fooocus.Metadata).
	// The caller can type-assert it if needed.
	Raw() interface{}
}

type ImageMetadataContext struct {

	// Original filepath of the image, used to locate any sidecar files
	// if supported. If the image was read from a stream, this may be empty.
	//
	// Note: This is the original path of the image, not the path of any
	// sidecar file. Sidecar files must be located by the extractor if
	// supported.
	Filepath string

	// MIME type of the image, e.g. "image/jpeg", "image/png"
	MIME string

	// All embedded metadata extracted from the image, usually
	// from EXIF blocks or PNG tEXt chunks.
	EmbeddedMetadata map[string]imagemeta.TagInfo
}

type Lora struct {
	Name   string
	Weight float32
}

type PngMetadataWriter struct {
	Template io.Reader
}

type Reader[T any] interface {
	// Decode reads software-specific metadata from the image.
	Decode(ImageMetadataContext) (T, error)
	// Extract reads structured metadata for the image.
	// This could come from embedded metadata or an external source
	// such as a sidecar file.
	Extract(ImageMetadataContext) (StructuredMetadata, error)
}

type StructuredMetadata struct {
	// Source is the identifier for the tool that generated the metadata,
	// e.g., "Fooocus", "FooocusPlus" or "RuinedFooocus".
	Source string

	Created time.Time
	Params  GenerationParameters
}

type Writer[M any] interface {
	Write(target io.Writer, metadata M) error
	CopyWrite(source io.Reader, target io.Writer, metadata M) error
}