import {z} from 'zod'

export type SimulatedBinaryCrossoverOptions = {name?: 'SimulatedBinaryCrossover' | undefined, xover_probability?: number | undefined, xover_distribution?: number | undefined};

export type SinglePointBinaryCrossoverOptions = {name?: 'SinglePointBinaryCrossover' | undefined};

export type UniformIntegerCrossoverOptions = {name?: 'UniformIntegerCrossover' | undefined};

export type UniformMixedIntegerCrossoverOptions = {name?: 'UniformMixedIntegerCrossover' | undefined};

export type BlendAlphaCrossoverOptions = {name?: 'BlendAlphaCrossover' | undefined, alpha?: number | undefined, xover_probability?: number | undefined};

export type SingleArithmeticCrossoverOptions = {name?: 'SingleArithmeticCrossover' | undefined, xover_probability?: number | undefined};

export type LocalCrossoverOptions = {name?: 'LocalCrossover' | undefined, xover_probability?: number | undefined};

export type BoundedExponentialCrossoverOptions = {name?: 'BoundedExponentialCrossover' | undefined, xover_probability?: number | undefined, lambda_?: number | undefined};

export type BoundedPolynomialMutationOptions = {name?: 'BoundedPolynomialMutation' | undefined, mutation_probability?: (number | null) | undefined, distribution_index?: number | undefined};

export type BinaryFlipMutationOptions = {name?: 'BinaryFlipMutation' | undefined, mutation_probability?: (number | null) | undefined};

export type IntegerRandomMutationOptions = {name?: 'IntegerRandomMutation' | undefined, mutation_probability?: (number | null) | undefined};

export type MixedIntegerRandomMutationOptions = {name?: 'MixedIntegerRandomMutation' | undefined, mutation_probability?: (number | null) | undefined};

export type MPTMutationOptions = {name?: 'MPTMutation' | undefined, mutation_probability?: (number | null) | undefined, mutation_exponent?: number | undefined};

export type NonUniformMutationOptions = {name?: 'NonUniformMutation' | undefined, mutation_probability?: (number | null) | undefined, max_generations: number, b?: number | undefined};

export type SelfAdaptiveGaussianMutationOptions = {name?: 'SelfAdaptiveGaussianMutation' | undefined, mutation_probability?: (number | null) | undefined};

export type PowerMutationOptions = {name?: 'PowerMutation' | undefined, mutation_probability?: (number | null) | undefined, p?: number | undefined};

export type ReferenceVectorOptions = {adaptation_frequency?: number | undefined, creation_type?: ('simplex' | 's_energy') | undefined, vector_type?: ('spherical' | 'planar') | undefined, lattice_resolution?: (number | null) | undefined, number_of_vectors?: number | undefined, adaptation_distance?: number | undefined, reference_point?: ({[key: string]: number} | null) | undefined, preferred_solutions?: ({[key: string]: Array<number>} | null) | undefined, non_preferred_solutions?: ({[key: string]: Array<number>} | null) | undefined, preferred_ranges?: ({[key: string]: Array<number>} | null) | undefined};

export type ParameterAdaptationStrategy = 'GENERATION_BASED' | 'FUNCTION_EVALUATION_BASED' | 'OTHER';

export type RVEASelectorOptions = {name?: 'RVEASelector' | undefined, reference_vector_options?: ReferenceVectorOptions | undefined, parameter_adaptation_strategy?: ParameterAdaptationStrategy | undefined, alpha?: number | undefined};

export type NSGA3SelectorOptions = {name?: 'NSGA3Selector' | undefined, reference_vector_options?: ReferenceVectorOptions | undefined, invert_reference_vectors?: boolean | undefined};

export type IBEASelectorOptions = {name?: 'IBEASelector' | undefined, population_size: number, kappa?: number | undefined, binary_indicator?: ('eps' | 'hv') | undefined};

export type MaxGenerationsTerminatorOptions = {name?: 'MaxGenerationsTerminator' | undefined, max_generations?: number | undefined};

export type MaxEvaluationsTerminatorOptions = {name?: 'MaxEvaluationsTerminator' | undefined, max_evaluations?: number | undefined};

export type MaxTimeTerminatorOptions = {name?: 'MaxTimeTerminator' | undefined, max_time?: number | undefined};

export type ExternalCheckTerminatorOptions = {name?: 'ExternalCheckTerminator' | undefined};

export type CompositeTerminatorOptions = {name?: 'CompositeTerminator' | undefined, terminators?: (Array<MaxEvaluationsTerminatorOptions | MaxGenerationsTerminatorOptions | MaxTimeTerminatorOptions | ExternalCheckTerminatorOptions>) | undefined, mode?: ('all' | 'any') | undefined};

export type LHSGeneratorOptions = {n_points: number, name?: 'LHSGenerator' | undefined};

export type RandomBinaryGeneratorOptions = {n_points: number, name?: 'RandomBinaryGenerator' | undefined};

export type RandomGeneratorOptions = {n_points: number, name?: 'RandomGenerator' | undefined};

export type RandomIntegerGeneratorOptions = {n_points: number, name?: 'RandomIntegerGenerator' | undefined};

export type RandomMixedIntegerGeneratorOptions = {n_points: number, name?: 'RandomMixedIntegerGenerator' | undefined};

export type ClipRepairOptions = {name?: 'ClipRepair' | undefined, lower_bounds?: ({[key: string]: number} | null) | undefined, upper_bounds?: ({[key: string]: number} | null) | undefined};

export type NoRepairOptions = {name?: 'NoRepair' | undefined};

export type Template1Options = {crossover: SimulatedBinaryCrossoverOptions | SinglePointBinaryCrossoverOptions | UniformIntegerCrossoverOptions | UniformMixedIntegerCrossoverOptions | BlendAlphaCrossoverOptions | SingleArithmeticCrossoverOptions | LocalCrossoverOptions | BoundedExponentialCrossoverOptions, mutation: BoundedPolynomialMutationOptions | BinaryFlipMutationOptions | IntegerRandomMutationOptions | MixedIntegerRandomMutationOptions | MPTMutationOptions | NonUniformMutationOptions | SelfAdaptiveGaussianMutationOptions | PowerMutationOptions, selection: RVEASelectorOptions | NSGA3SelectorOptions | IBEASelectorOptions, termination: MaxGenerationsTerminatorOptions | MaxEvaluationsTerminatorOptions | MaxTimeTerminatorOptions | ExternalCheckTerminatorOptions | CompositeTerminatorOptions, generator: LHSGeneratorOptions | RandomBinaryGeneratorOptions | RandomGeneratorOptions | RandomIntegerGeneratorOptions | RandomMixedIntegerGeneratorOptions, repair?: (ClipRepairOptions | NoRepairOptions) | undefined, use_archive?: boolean | undefined, seed?: number | undefined, verbosity?: number | undefined, algorithm_name: string, name?: 'Template1' | undefined};

export type TournamentSelectionOptions = {name?: 'TournamentSelection' | undefined, tournament_size?: number | undefined, winner_size: number};

export type RouletteWheelSelectionOptions = {name?: 'RouletteWheelSelection' | undefined, tournament_size?: number | undefined, winner_size: number};

export type Template2Options = {crossover: SimulatedBinaryCrossoverOptions | SinglePointBinaryCrossoverOptions | UniformIntegerCrossoverOptions | UniformMixedIntegerCrossoverOptions | BlendAlphaCrossoverOptions | SingleArithmeticCrossoverOptions | LocalCrossoverOptions | BoundedExponentialCrossoverOptions, mutation: BoundedPolynomialMutationOptions | BinaryFlipMutationOptions | IntegerRandomMutationOptions | MixedIntegerRandomMutationOptions | MPTMutationOptions | NonUniformMutationOptions | SelfAdaptiveGaussianMutationOptions | PowerMutationOptions, selection: RVEASelectorOptions | NSGA3SelectorOptions | IBEASelectorOptions, termination: MaxGenerationsTerminatorOptions | MaxEvaluationsTerminatorOptions | MaxTimeTerminatorOptions | ExternalCheckTerminatorOptions | CompositeTerminatorOptions, generator: LHSGeneratorOptions | RandomBinaryGeneratorOptions | RandomGeneratorOptions | RandomIntegerGeneratorOptions | RandomMixedIntegerGeneratorOptions, repair?: (ClipRepairOptions | NoRepairOptions) | undefined, use_archive?: boolean | undefined, seed?: number | undefined, verbosity?: number | undefined, algorithm_name: string, name?: 'Template2' | undefined, mate_selection: TournamentSelectionOptions | RouletteWheelSelectionOptions};

export type ReferencePointOptions = {name?: 'reference_point' | undefined, preference: {[key: string]: number}, method?: ('Hakanen' | 'IOPIS') | undefined};

export type DesirableRangesOptions = {name?: 'preferred_ranges' | undefined, aspiration_levels: {[key: string]: number}, reservation_levels: {[key: string]: number}, method?: ('Hakanen' | 'DF transformation') | undefined, desirability_levels?: Array<unknown> | undefined};

export type PreferredSolutionsOptions = {name?: 'preferred_solutions' | undefined, preference: {[key: string]: Array<number>}, method?: 'Hakanen' | undefined};

export type NonPreferredSolutionsOptions = {name?: 'non_preferred_solutions' | undefined, preference: {[key: string]: Array<number>}, method?: 'Hakanen' | undefined};

export type EMOIterateRequest = {problem_id: number, session_id?: (number | null) | undefined, parent_state_id?: (number | null) | undefined, template_options?: (Array<Template1Options | Template2Options> | null) | undefined, preference_options?: (ReferencePointOptions | DesirableRangesOptions | PreferredSolutionsOptions | NonPreferredSolutionsOptions | null) | undefined};

export type CreateApiMethodEmoIterateArgs = {body: EMOIterateRequest};

export type CreateApiMethodEmoIterateOptions = {baseUrl: string, OAuth2PasswordBearer?: string | undefined, fetch?: typeof fetch | undefined};

export const eMOIterateResponse = z.object({method_ids: z.array(z.string()), client_id: z.string(), state_id: z.number().int()});

export const validateResponse = async <Output>(
  schema: {
    readonly '~standard': {
      readonly validate: (value: unknown) => ({ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray<PropertyKey | { readonly key: PropertyKey }> | undefined }> }) | Promise<{ readonly value: Output; readonly issues?: undefined } | { readonly issues: ReadonlyArray<{ readonly message: string; readonly path?: ReadonlyArray<PropertyKey | { readonly key: PropertyKey }> | undefined }> }>
    }
  },
  value: unknown
): Promise<Output> => {
  const result = await schema['~standard'].validate(value)

  if (result.issues !== undefined) {
    const detail = result.issues
      .map(({ message, path }) =>
        path?.length
          ? `${path.map((segment) => String(typeof segment === 'object' ? segment.key : segment)).join('.')}: ${message}`
          : message
      )
      .join('; ')

    throw new Error(`Response does not match the schema: ${detail}`, { cause: { issues: result.issues } })
  }

  return result.value
};

/**
 * Iterate
 * 
 * POST /method/emo/iterate
 * 
 * Also accepts auth: APIKeyCookie
 */
export const createApiMethodEmoIterate = async (args: CreateApiMethodEmoIterateArgs, options: CreateApiMethodEmoIterateOptions) => {
const {baseUrl, fetch: fetchFn = fetch, OAuth2PasswordBearer} = options ?? {}

const query = new URLSearchParams()
const headers = new Headers()
headers.set('content-type', "application/json")
headers.set('accept', "application/json")

if (OAuth2PasswordBearer !== undefined) headers.set('authorization', `Bearer ${OAuth2PasswordBearer}`)

const url = `${baseUrl.replace(/\/+$/, '')}/method/emo/iterate${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "POST", headers, body: JSON.stringify(args.body)})

if (!res.ok) throw new Error(`Request failed with status ${res.status}`, {cause: {status: res.status, body: await res.text()}})

return validateResponse(eMOIterateResponse, await res.json())
};