import type { UntrustedValue } from "./untrusted-value.js";

export interface BetterWrightArtifactLike {
  kind?: UntrustedValue;
  media?: UntrustedValue;
  path?: UntrustedValue;
  [key: string]: UntrustedValue;
}

export interface BetterWrightResultLike {
  artifacts?: readonly BetterWrightArtifactLike[];
  [key: string]: UntrustedValue;
}

export interface PiImageContentOptions {
  maxImageBytes?: number;
}

export interface PiImageContentBlock {
  type: "image";
  data: string;
  mimeType: "image/png" | "image/jpeg" | "image/webp" | "image/gif";
}

export interface PiImageArtifact {
  path: string;
  mimeType: "image/png" | "image/jpeg" | "image/webp" | "image/gif";
}

export function piImageArtifacts(
  result: BetterWrightResultLike | null | undefined,
): PiImageArtifact[];

export function piImageContent(
  result: BetterWrightResultLike | null | undefined,
  options?: PiImageContentOptions,
): Promise<PiImageContentBlock[]>;

export function piPrimaryImageArtifact(
  result: BetterWrightResultLike | null | undefined,
): PiImageArtifact | null;