/*
 * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
 */

import { LivepeerCore } from "../core.js";
import { readableStreamToArrayBuffer } from "../lib/files.js";
import * as M from "../lib/matchers.js";
import { safeParse } from "../lib/schemas.js";
import { RequestOptions } from "../lib/sdks.js";
import { extractSecurity, resolveGlobalSecurity } from "../lib/security.js";
import { pathToFunc } from "../lib/url.js";
import * as components from "../models/components/index.js";
import {
  ConnectionError,
  InvalidRequestError,
  RequestAbortedError,
  RequestTimeoutError,
  UnexpectedClientError,
} from "../models/errors/httpclienterrors.js";
import * as errors from "../models/errors/index.js";
import { SDKError } from "../models/errors/sdkerror.js";
import { SDKValidationError } from "../models/errors/sdkvalidationerror.js";
import * as operations from "../models/operations/index.js";
import { isBlobLike } from "../types/blobs.js";
import { Result } from "../types/fp.js";
import { isReadableStream } from "../types/streams.js";

/**
 * Image To Image
 *
 * @remarks
 * Apply image transformations to a provided image.
 */
export async function generateImageToImage(
  client: LivepeerCore,
  request: components.BodyGenImageToImage,
  options?: RequestOptions,
): Promise<
  Result<
    operations.GenImageToImageResponse,
    | errors.GenImageToImageResponseBody
    | errors.GenImageToImageGenerateResponseBody
    | errors.GenImageToImageGenerateResponseResponseBody
    | errors.GenImageToImageGenerateResponse500ResponseBody
    | SDKError
    | SDKValidationError
    | UnexpectedClientError
    | InvalidRequestError
    | RequestAbortedError
    | RequestTimeoutError
    | ConnectionError
  >
> {
  const parsed = safeParse(
    request,
    (value) => components.BodyGenImageToImage$outboundSchema.parse(value),
    "Input validation failed",
  );
  if (!parsed.ok) {
    return parsed;
  }
  const payload = parsed.value;
  const body = new FormData();

  if (isBlobLike(payload.image)) {
    body.append("image", payload.image);
  } else if (isReadableStream(payload.image.content)) {
    const buffer = await readableStreamToArrayBuffer(payload.image.content);
    const blob = new Blob([buffer], { type: "application/octet-stream" });
    body.append("image", blob);
  } else {
    body.append(
      "image",
      new Blob([payload.image.content], { type: "application/octet-stream" }),
      payload.image.fileName,
    );
  }
  body.append("prompt", payload.prompt);
  if (payload.guidance_scale !== undefined) {
    body.append("guidance_scale", String(payload.guidance_scale));
  }
  if (payload.image_guidance_scale !== undefined) {
    body.append("image_guidance_scale", String(payload.image_guidance_scale));
  }
  if (payload.loras !== undefined) {
    body.append("loras", payload.loras);
  }
  if (payload.model_id !== undefined) {
    body.append("model_id", payload.model_id);
  }
  if (payload.negative_prompt !== undefined) {
    body.append("negative_prompt", payload.negative_prompt);
  }
  if (payload.num_images_per_prompt !== undefined) {
    body.append("num_images_per_prompt", String(payload.num_images_per_prompt));
  }
  if (payload.num_inference_steps !== undefined) {
    body.append("num_inference_steps", String(payload.num_inference_steps));
  }
  if (payload.safety_check !== undefined) {
    body.append("safety_check", String(payload.safety_check));
  }
  if (payload.seed !== undefined) {
    body.append("seed", String(payload.seed));
  }
  if (payload.strength !== undefined) {
    body.append("strength", String(payload.strength));
  }

  const path = pathToFunc("/api/generate/image-to-image")();

  const headers = new Headers({
    Accept: "application/json",
  });

  const secConfig = await extractSecurity(client._options.apiKey);
  const securityInput = secConfig == null ? {} : { apiKey: secConfig };
  const requestSecurity = resolveGlobalSecurity(securityInput);

  const context = {
    operationID: "genImageToImage",
    oAuth2Scopes: [],

    resolvedSecurity: requestSecurity,

    securitySource: client._options.apiKey,
    retryConfig: options?.retries
      || client._options.retryConfig
      || { strategy: "none" },
    retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
  };

  const requestRes = client._createRequest(context, {
    security: requestSecurity,
    method: "POST",
    path: path,
    headers: headers,
    body: body,
    timeoutMs: options?.timeoutMs || client._options.timeoutMs || -1,
  }, options);
  if (!requestRes.ok) {
    return requestRes;
  }
  const req = requestRes.value;

  const doResult = await client._do(req, {
    context,
    errorCodes: ["400", "401", "422", "4XX", "500", "5XX"],
    retryConfig: context.retryConfig,
    retryCodes: context.retryCodes,
  });
  if (!doResult.ok) {
    return doResult;
  }
  const response = doResult.value;

  const responseFields = {
    ContentType: response.headers.get("content-type")
      ?? "application/octet-stream",
    StatusCode: response.status,
    RawResponse: response,
    Headers: {},
  };

  const [result] = await M.match<
    operations.GenImageToImageResponse,
    | errors.GenImageToImageResponseBody
    | errors.GenImageToImageGenerateResponseBody
    | errors.GenImageToImageGenerateResponseResponseBody
    | errors.GenImageToImageGenerateResponse500ResponseBody
    | SDKError
    | SDKValidationError
    | UnexpectedClientError
    | InvalidRequestError
    | RequestAbortedError
    | RequestTimeoutError
    | ConnectionError
  >(
    M.json(200, operations.GenImageToImageResponse$inboundSchema, {
      key: "ImageResponse",
    }),
    M.jsonErr(400, errors.GenImageToImageResponseBody$inboundSchema),
    M.jsonErr(401, errors.GenImageToImageGenerateResponseBody$inboundSchema),
    M.jsonErr(
      422,
      errors.GenImageToImageGenerateResponseResponseBody$inboundSchema,
    ),
    M.jsonErr(
      500,
      errors.GenImageToImageGenerateResponse500ResponseBody$inboundSchema,
    ),
    M.fail(["4XX", "5XX"]),
    M.json("default", operations.GenImageToImageResponse$inboundSchema, {
      key: "studio-api-error",
    }),
  )(response, { extraFields: responseFields });
  if (!result.ok) {
    return result;
  }

  return result;
}