import {z} from 'zod'

export type ProblemGetRequest = {problem_id: number};

export type CreateApiProblemGetArgs = {body: ProblemGetRequest};

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

export const variableDomainTypeEnum = z.enum(["continuous", "binary", "integer", "mixed"]);

export const constantDB = z.object({name: z.string(), symbol: z.string(), value: z.number(), id: z.number().int().nullable().optional(), problem_id: z.number().int().nullable().optional()});

export const tensor: z.ZodType<Tensor> = z.union([z.array(z.lazy(() => tensor)), z.array(z.union([z.number(), z.number().int(), z.boolean()])), z.number(), z.number().int(), z.boolean(), z.literal("List")]).nullable();

export const tensorConstantDB = z.object({values: tensor, shape: z.array(z.number().int()), name: z.string(), symbol: z.string(), id: z.number().int().nullable().optional(), problem_id: z.number().int().nullable().optional()});

export const variableTypeEnum = z.enum(["real", "integer", "binary"]);

export const variableDB = z.object({name: z.string(), symbol: z.string(), variable_type: variableTypeEnum, lowerbound: z.number().nullable().optional(), upperbound: z.number().nullable().optional(), initial_value: z.number().nullable().optional(), id: z.number().int().nullable().optional(), problem_id: z.number().int().nullable().optional()});

export const tensorVariableDB = z.object({initial_values: tensor.nullable(), lowerbounds: tensor.nullable(), upperbounds: tensor.nullable(), shape: z.array(z.number().int()), name: z.string(), symbol: z.string(), variable_type: variableTypeEnum, id: z.number().int().nullable().optional(), problem_id: z.number().int().nullable().optional()});

export const url = z.object({url: z.string(), auth: z.array(z.unknown()).nullable().optional()});

export const objectiveTypeEnum = z.enum(["analytical", "data_based", "simulator", "surrogate"]);

export const objectiveDB = z.object({func: z.array(z.unknown()).nullable(), scenario_keys: z.array(z.string()).nullable().optional(), surrogates: z.array(z.string()).nullable().optional(), simulator_path: z.union([z.string(), url]).nullable().optional(), name: z.string(), symbol: z.string(), unit: z.string().nullable().optional(), maximize: z.boolean().optional(), ideal: z.number().nullable().optional(), nadir: z.number().nullable().optional(), objective_type: objectiveTypeEnum.optional(), is_linear: z.boolean().optional(), is_convex: z.boolean().optional(), is_twice_differentiable: z.boolean().optional(), id: z.number().int().nullable().optional(), problem_id: z.number().int().nullable().optional()});

export const constraintTypeEnum = z.enum(["=", "<="]);

export const constraintDB = z.object({func: z.array(z.unknown()), scenario_keys: z.array(z.string()).nullable().optional(), surrogates: z.array(z.string()).nullable().optional(), simulator_path: z.union([z.string(), url]).nullable().optional(), name: z.string(), symbol: z.string(), cons_type: constraintTypeEnum, is_linear: z.boolean().optional(), is_convex: z.boolean().optional(), is_twice_differentiable: z.boolean().optional(), id: z.number().int().nullable().optional(), problem_id: z.number().int().nullable().optional()});

export const scalarizationFunctionDB = z.object({func: z.array(z.unknown()), scenario_keys: z.array(z.string()), name: z.string(), symbol: z.string().nullable().optional(), is_linear: z.boolean().optional(), is_convex: z.boolean().optional(), is_twice_differentiable: z.boolean().optional(), id: z.number().int().nullable().optional(), problem_id: z.number().int().nullable().optional()});

export const extraFunctionDB = z.object({func: z.array(z.unknown()), scenario_keys: z.array(z.string()).nullable().optional(), surrogates: z.array(z.string()).nullable().optional(), simulator_path: z.union([z.string(), url]).nullable().optional(), name: z.string(), symbol: z.string(), is_linear: z.boolean().optional(), is_convex: z.boolean().optional(), is_twice_differentiable: z.boolean().optional(), id: z.number().int().nullable().optional(), problem_id: z.number().int().nullable().optional()});

export const discreteRepresentationDB = z.object({non_dominated: z.boolean().optional(), variable_values: z.record(z.string(), z.array(z.union([z.number(), z.number().int(), z.boolean()]))), objective_values: z.record(z.string(), z.array(z.number())), id: z.number().int().nullable().optional(), problem_id: z.number().int().nullable().optional()});

export const simulatorDB = z.object({file: z.string().nullable().optional(), url: url.nullable().optional(), parameter_options: z.record(z.string(), z.unknown()).nullable().optional(), name: z.string(), symbol: z.string(), id: z.number().int().nullable().optional(), problem_id: z.number().int().nullable().optional()});

export const forestProblemMetaData = z.object({id: z.number().int().nullable().optional(), metadata_id: z.number().int().nullable().optional(), metadata_type: z.string().optional(), map_json: z.string(), schedule_dict: z.record(z.string(), z.unknown()), years: z.array(z.string()), stand_id_field: z.string(), stand_descriptor: z.record(z.string(), z.unknown()).nullable().optional(), compensation: z.number().nullable().optional()});

export const representativeNonDominatedSolutions = z.object({id: z.number().int().nullable().optional(), metadata_id: z.number().int().nullable().optional(), metadata_type: z.string().optional(), name: z.string(), description: z.string().nullable().optional(), solution_data: z.record(z.string(), z.array(z.number())), ideal: z.record(z.string(), z.number()), nadir: z.record(z.string(), z.number())});

export const problemMetaDataPublic = z.object({problem_id: z.number().int(), forest_metadata: z.array(forestProblemMetaData).nullable(), representative_nd_metadata: z.array(representativeNonDominatedSolutions).nullable()});

export const problemInfo = z.object({name: z.string(), description: z.string(), is_convex: z.boolean().nullable(), is_linear: z.boolean().nullable(), is_twice_differentiable: z.boolean().nullable(), scenario_keys: z.array(z.string()).nullable(), variable_domain: variableDomainTypeEnum, id: z.number().int(), user_id: z.number().int(), constants: z.array(constantDB).nullable(), tensor_constants: z.array(tensorConstantDB).nullable(), variables: z.array(variableDB).nullable(), tensor_variables: z.array(tensorVariableDB).nullable(), objectives: z.array(objectiveDB), constraints: z.array(constraintDB).nullable(), scalarization_funcs: z.array(scalarizationFunctionDB).nullable(), extra_funcs: z.array(extraFunctionDB).nullable(), discrete_representation: discreteRepresentationDB.nullable(), simulators: z.array(simulatorDB).nullable(), problem_metadata: problemMetaDataPublic.nullable()});

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
};

/**
 * Get Problem
 * 
 * POST /problem/get
 * 
 * Also accepts auth: APIKeyCookie
 */
export const createApiProblemGet = async (args: CreateApiProblemGetArgs, options: CreateApiProblemGetOptions) => {
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(/\/+$/, '')}/problem/get${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(problemInfo, await res.json())
};