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

import * as z from "zod";
import { FireworksAICore } from "../core.js";
import { encodeSimple } from "../lib/encodings.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 {
  ConnectionError,
  InvalidRequestError,
  RequestAbortedError,
  RequestTimeoutError,
  UnexpectedClientError,
} from "../models/httpclienterrors.js";
import * as models from "../models/index.js";
import { SDKError } from "../models/sdkerror.js";
import { SDKValidationError } from "../models/sdkvalidationerror.js";
import { isBlobLike } from "../types/blobs.js";
import { Result } from "../types/fp.js";
import { isReadableStream } from "../types/streams.js";

/**
 * Generate a new image using ControlNet with provided image as a guidance
 */
export async function imagesGenerateFromControlNet(
  client: FireworksAICore,
  request: models.GenerateFromControlNetRequest,
  options?: RequestOptions,
): Promise<
  Result<
    ReadableStream<Uint8Array>,
    | models.BadRequest
    | models.Unauthorized
    | models.Forbidden
    | models.NotFound
    | models.TooManyRequests
    | models.InternalServerError
    | SDKError
    | SDKValidationError
    | UnexpectedClientError
    | InvalidRequestError
    | RequestAbortedError
    | RequestTimeoutError
    | ConnectionError
  >
> {
  const parsed = safeParse(
    request,
    (value) => models.GenerateFromControlNetRequest$outboundSchema.parse(value),
    "Input validation failed",
  );
  if (!parsed.ok) {
    return parsed;
  }
  const payload = parsed.value;
  const body = new FormData();

  if (isBlobLike(payload.Body_ControlNet_gen.control_image)) {
    body.append("control_image", payload.Body_ControlNet_gen.control_image);
  } else if (
    isReadableStream(payload.Body_ControlNet_gen.control_image.content)
  ) {
    const buffer = await readableStreamToArrayBuffer(
      payload.Body_ControlNet_gen.control_image.content,
    );
    const blob = new Blob([buffer], { type: "application/octet-stream" });
    body.append("control_image", blob);
  } else {
    body.append(
      "control_image",
      new Blob([payload.Body_ControlNet_gen.control_image.content], {
        type: "application/octet-stream",
      }),
      payload.Body_ControlNet_gen.control_image.fileName,
    );
  }
  body.append("prompt", payload.Body_ControlNet_gen.prompt);
  if (payload.Body_ControlNet_gen.cfg_scale !== undefined) {
    body.append("cfg_scale", String(payload.Body_ControlNet_gen.cfg_scale));
  }
  if (payload.Body_ControlNet_gen.conditioning_scale !== undefined) {
    body.append(
      "conditioning_scale",
      String(payload.Body_ControlNet_gen.conditioning_scale),
    );
  }
  if (payload.Body_ControlNet_gen.control_net_name !== undefined) {
    body.append(
      "control_net_name",
      payload.Body_ControlNet_gen.control_net_name,
    );
  }
  if (payload.Body_ControlNet_gen.height !== undefined) {
    body.append("height", String(payload.Body_ControlNet_gen.height));
  }
  if (payload.Body_ControlNet_gen.negative_prompt !== undefined) {
    body.append("negative_prompt", payload.Body_ControlNet_gen.negative_prompt);
  }
  if (payload.Body_ControlNet_gen.safety_check !== undefined) {
    body.append(
      "safety_check",
      String(payload.Body_ControlNet_gen.safety_check),
    );
  }
  if (payload.Body_ControlNet_gen.sampler !== undefined) {
    body.append("sampler", payload.Body_ControlNet_gen.sampler);
  }
  if (payload.Body_ControlNet_gen.samples !== undefined) {
    body.append("samples", String(payload.Body_ControlNet_gen.samples));
  }
  if (payload.Body_ControlNet_gen.seed !== undefined) {
    body.append("seed", String(payload.Body_ControlNet_gen.seed));
  }
  if (payload.Body_ControlNet_gen.step_schedule_end !== undefined) {
    body.append(
      "step_schedule_end",
      String(payload.Body_ControlNet_gen.step_schedule_end),
    );
  }
  if (payload.Body_ControlNet_gen.step_schedule_start !== undefined) {
    body.append(
      "step_schedule_start",
      String(payload.Body_ControlNet_gen.step_schedule_start),
    );
  }
  if (payload.Body_ControlNet_gen.steps !== undefined) {
    body.append("steps", String(payload.Body_ControlNet_gen.steps));
  }
  if (payload.Body_ControlNet_gen.width !== undefined) {
    body.append("width", String(payload.Body_ControlNet_gen.width));
  }

  const pathParams = {
    accountId: encodeSimple("accountId", payload.accountId, {
      explode: false,
      charEncoding: "percent",
    }),
    modelId: encodeSimple("modelId", payload.modelId, {
      explode: false,
      charEncoding: "percent",
    }),
  };

  const path = pathToFunc(
    "/image_generation/accounts/{accountId}/models/{modelId}/control_net",
  )(pathParams);

  const headers = new Headers({
    Accept: "image/jpeg",
  });

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

  const context = {
    operationID: "generateFromControlNet",
    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",
    baseURL: options?.serverURL,
    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", "403", "404", "429", "4XX", "500", "503", "5XX"],
    retryConfig: context.retryConfig,
    retryCodes: context.retryCodes,
  });
  if (!doResult.ok) {
    return doResult;
  }
  const response = doResult.value;

  const responseFields = {
    HttpMeta: { Response: response, Request: req },
  };

  const [result] = await M.match<
    ReadableStream<Uint8Array>,
    | models.BadRequest
    | models.Unauthorized
    | models.Forbidden
    | models.NotFound
    | models.TooManyRequests
    | models.InternalServerError
    | SDKError
    | SDKValidationError
    | UnexpectedClientError
    | InvalidRequestError
    | RequestAbortedError
    | RequestTimeoutError
    | ConnectionError
  >(
    M.stream(200, z.instanceof(ReadableStream<Uint8Array>), {
      ctype: "image/jpeg",
    }),
    M.jsonErr(400, models.BadRequest$inboundSchema),
    M.jsonErr(401, models.Unauthorized$inboundSchema),
    M.jsonErr(403, models.Forbidden$inboundSchema),
    M.jsonErr(404, models.NotFound$inboundSchema),
    M.jsonErr(429, models.TooManyRequests$inboundSchema),
    M.jsonErr(500, models.InternalServerError$inboundSchema),
    M.fail(["4XX", 503, "5XX"]),
  )(response, { extraFields: responseFields });
  if (!result.ok) {
    return result;
  }

  return result;
}