import {z} from 'zod'

export type GetApiApiProvidersOptions = {baseUrl?: string | undefined, ManagementBearerAuth?: string | undefined, fetch?: typeof fetch | undefined};

export const modelProvider = z.enum(["openai", "azure", "anthropic", "bedrock", "cohere", "vertex", "vllm", "mistral", "ollama", "groq", "sgl", "parasail", "perplexity", "replicate", "cerebras", "deepseek", "gemini", "openrouter", "elevenlabs", "huggingface", "nebius", "xai", "runway", "fireworks"]);

export const networkConfig = z.object({base_url: z.string().optional(), extra_headers: z.record(z.string(), z.string()).optional(), default_request_timeout_in_seconds: z.number().int().optional(), max_retries: z.number().int().optional(), retry_backoff_initial: z.number().int().optional(), retry_backoff_max: z.number().int().optional(), insecure_skip_verify: z.boolean().optional(), ca_cert_pem: z.string().optional(), allow_private_network: z.boolean().optional()});

export const concurrencyAndBufferSize = z.object({concurrency: z.number().int().optional(), buffer_size: z.number().int().optional()});

export const providerResponse = z.object({name: modelProvider.optional(), network_config: networkConfig.optional(), concurrency_and_buffer_size: concurrencyAndBufferSize.optional(), proxy_config: z.object({type: z.enum(["none", "http", "socks5", "environment"]).optional(), url: z.string().optional(), username: z.string().optional(), password: z.string().optional(), ca_cert_pem: z.string().optional()}).optional(), send_back_raw_request: z.boolean().optional(), send_back_raw_response: z.boolean().optional(), store_raw_request_response: z.boolean().optional(), custom_provider_config: z.object({is_key_less: z.boolean().optional(), base_provider_type: modelProvider.optional(), allowed_requests: z.object({list_models: z.boolean().optional(), text_completion: z.boolean().optional(), text_completion_stream: z.boolean().optional(), chat_completion: z.boolean().optional(), chat_completion_stream: z.boolean().optional(), responses: z.boolean().optional(), responses_stream: z.boolean().optional(), count_tokens: z.boolean().optional(), embedding: z.boolean().optional(), speech: z.boolean().optional(), speech_stream: z.boolean().optional(), transcription: z.boolean().optional(), transcription_stream: z.boolean().optional(), image_generation: z.boolean().optional(), image_generation_stream: z.boolean().optional(), batch_create: z.boolean().optional(), batch_list: z.boolean().optional(), batch_retrieve: z.boolean().optional(), batch_cancel: z.boolean().optional(), batch_results: z.boolean().optional(), file_upload: z.boolean().optional(), file_list: z.boolean().optional(), file_retrieve: z.boolean().optional(), file_delete: z.boolean().optional(), file_content: z.boolean().optional()}).optional(), request_path_overrides: z.record(z.string(), z.string()).optional()}).optional(), provider_status: z.enum(["active", "error", "deleted"]).optional(), status: z.string().optional(), description: z.string().optional(), config_hash: z.string().optional()});

export const listProvidersResponse = z.object({providers: z.array(providerResponse).optional(), total: z.number().int().optional()});

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

/**
 * List all providers
 * 
 * GET /api/providers
 */
export const getApiApiProviders = async (options?: GetApiApiProvidersOptions) => {
const {baseUrl = "http://localhost:8080", fetch: fetchFn = fetch, ManagementBearerAuth} = options ?? {}

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

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

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

const res = await fetchFn(url, {method: "GET", headers})

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

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