import type { APIPromise } from '../core/api-promise';
import type { RequestOptions } from '../internal/request-options';
import {
  Generate,
  type GenerateCaptionParams,
  type GenerateCaptionResponse,
  type GenerateDescribeStyleParams,
  type GenerateDescribeStyleResponse,
  type GenerateDetectParams,
  type GenerateDetectResponse,
  type GenerateEmbedParams,
  type GenerateEmbedResponse,
  type GeneratePatchParams,
  type GeneratePatchResponse,
  type GenerateRunModelParams,
  type GenerateRunModelResponse,
  type GenerateTranslateParams,
  type GenerateTranslateResponse,
} from '../resources/generate';
import { type WithJob, enhanceJob } from './job';

/**
 * Enhanced Generate resource.
 * All original methods are inherited. Job-producing methods return
 * responses where `job` has `.wait()`.
 */
export class EnhancedGenerate extends Generate {
  override caption(
    params: GenerateCaptionParams,
    options?: RequestOptions,
  ): APIPromise<WithJob<GenerateCaptionResponse>> {
    return enhanceJob(this._client, super.caption(params, options), options);
  }

  override describeStyle(
    params: GenerateDescribeStyleParams,
    options?: RequestOptions,
  ): APIPromise<WithJob<GenerateDescribeStyleResponse>> {
    return enhanceJob(this._client, super.describeStyle(params, options), options);
  }

  override detect(
    params: GenerateDetectParams,
    options?: RequestOptions,
  ): APIPromise<WithJob<GenerateDetectResponse>> {
    return enhanceJob(this._client, super.detect(params, options), options);
  }

  override embed(
    params: GenerateEmbedParams,
    options?: RequestOptions,
  ): APIPromise<WithJob<GenerateEmbedResponse>> {
    return enhanceJob(this._client, super.embed(params, options), options);
  }

  override patch(
    params: GeneratePatchParams,
    options?: RequestOptions,
  ): APIPromise<WithJob<GeneratePatchResponse>> {
    return enhanceJob(this._client, super.patch(params, options), options);
  }

  override runModel(
    modelID: string,
    params: GenerateRunModelParams,
    options?: RequestOptions,
  ): APIPromise<WithJob<GenerateRunModelResponse>> {
    return enhanceJob(this._client, super.runModel(modelID, params, options), options);
  }

  override translate(
    params: GenerateTranslateParams,
    options?: RequestOptions,
  ): APIPromise<WithJob<GenerateTranslateResponse>> {
    return enhanceJob(this._client, super.translate(params, options), options);
  }
}