import {z} from 'zod'

export type LogRecordsIDFilter = {column_id: string, operator?: ('eq' | 'ne' | 'one_of' | 'not_in' | 'contains') | undefined, value: unknown | Array<unknown | string> | string, type?: 'id' | undefined};

export type LogRecordsDateFilter = {column_id: string, operator: 'eq' | 'ne' | 'gt' | 'gte' | 'lt' | 'lte', value: string, type?: 'date' | undefined};

export type LogRecordsNumberFilter = {column_id: string, operator: 'eq' | 'ne' | 'gt' | 'gte' | 'lt' | 'lte' | 'between', value: number | number | Array<number> | Array<number>, type?: 'number' | undefined};

export type LogRecordsBooleanFilter = {column_id: string, operator?: ('eq' | 'ne') | undefined, value: boolean, type?: 'boolean' | undefined};

export type LogRecordsCollectionFilter = {column_id: string, operator: 'eq' | 'contains' | 'one_of' | 'not_in', value: string | Array<string>, case_sensitive?: boolean | undefined, type?: 'collection' | undefined};

export type LogRecordsTextFilter = {column_id: string, operator: 'eq' | 'ne' | 'contains' | 'one_of' | 'not_in', value: string | Array<string>, case_sensitive?: boolean | undefined, type?: 'text' | undefined};

export type LogRecordsFullyAnnotatedFilter = {column_id?: 'fully_annotated' | undefined, type?: 'fully_annotated' | undefined, user_ids?: (Array<unknown> | null) | undefined};

export type FilterLeafAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorType = {filter: LogRecordsIDFilter | LogRecordsDateFilter | LogRecordsNumberFilter | LogRecordsBooleanFilter | LogRecordsCollectionFilter | LogRecordsTextFilter | LogRecordsFullyAnnotatedFilter};

export type NotNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput = {not: FilterLeafAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorType | AndNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput | OrNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput | NotNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput};

export type OrNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput = {or: Array<FilterLeafAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorType | AndNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput | OrNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput | NotNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput>};

export type AndNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput = {and: Array<FilterLeafAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorType | AndNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput | OrNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput | NotNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput>};

export type FilterExpressionAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput = FilterLeafAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorType | AndNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput | OrNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput | NotNodeAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput;

export type MetricAggregation = 'Count' | 'Sum' | 'Average' | 'Min' | 'Max' | 'P50' | 'P90' | 'P95' | 'P99' | 'PercentageFalse' | 'PercentageTrue';

export type MetricAggregationDetail = {id: string, metric_name: string, aggregation: MetricAggregation};

export type LogRecordsCustomMetricsQueryRequest = {start_time: string, end_time: string, log_stream_id?: (unknown | null) | undefined, experiment_id?: (unknown | null) | undefined, metrics_testing_id?: (unknown | null) | undefined, access_view_id?: (unknown | null) | undefined, filter_tree?: (FilterExpressionAnnotatedUnionLogRecordsIDFilterLogRecordsDateFilterLogRecordsNumberFilterLogRecordsBooleanFilterLogRecordsCollectionFilterLogRecordsTextFilterLogRecordsFullyAnnotatedFilterFieldInfoAnnotationNoneTypeRequiredTrueDiscriminatorTypeInput | null) | undefined, interval_minutes?: number | undefined, metric_details: Array<MetricAggregationDetail>, group_by?: (string | null) | undefined};

export type CreateApiV2ProjectsProjectIdMetricsCustomSearchArgs = {project_id: unknown, body: LogRecordsCustomMetricsQueryRequest};

export type CreateApiV2ProjectsProjectIdMetricsCustomSearchOptions = {baseUrl?: string | undefined, APIKeyHeader?: string | undefined, fetch?: typeof fetch | undefined};

export const bucketedMetrics = z.object({start_bucket_time: z.string(), end_bucket_time: z.string()}).and(z.record(z.string(), z.unknown()));

export const errorType = z.enum(["llm_api_error", "credentials_error", "data_validation_error", "configuration_error", "permission_error", "not_found_error", "workflow_error", "rate_limit_error", "system_error", "timeout_error", "not_applicable_reason", "uncataloged_error"]);

export const errorSeverity = z.enum(["low", "medium", "high", "critical"]);

export const standardError = z.object({error_code: z.number().int(), error_type: errorType, error_group: z.string(), severity: errorSeverity, message: z.string(), user_action: z.string().nullable().optional(), documentation_link: z.string().min(1).max(2083).nullable().optional(), retriable: z.boolean().optional(), blocking: z.boolean().optional(), http_status_code: z.number().int().nullable().optional(), source_service: z.string().nullable().optional(), context: z.record(z.string(), z.unknown()).optional()});

export const logRecordsMetricsResponse = z.object({group_by_columns: z.array(z.string()), aggregate_metrics: z.record(z.string(), z.union([z.number(), z.number().int(), z.record(z.string(), z.number().int())])), bucketed_metrics: z.record(z.string(), z.array(bucketedMetrics)), ems_captured_error: z.boolean().optional(), standard_errors: z.record(z.string(), standardError).nullable().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
};

/**
 * Query Custom Metrics
 * 
 * POST /v2/projects/{project_id}/metrics/custom_search
 * 
 * Also accepts auth: ClassicAPIKeyHeader, OAuth2PasswordBearer, HTTPBasic
 */
export const createApiV2ProjectsProjectIdMetricsCustomSearch = async (args: CreateApiV2ProjectsProjectIdMetricsCustomSearchArgs, options?: CreateApiV2ProjectsProjectIdMetricsCustomSearchOptions) => {
const {baseUrl = "https://api.galileo.ai", fetch: fetchFn = fetch, APIKeyHeader} = options ?? {}

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

if (APIKeyHeader !== undefined) headers.set("Splunk-AO-API-Key", APIKeyHeader)

const url = `${baseUrl.replace(/\/+$/, '')}/v2/projects/${encodeURIComponent(String(args.project_id))}/metrics/custom_search${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(logRecordsMetricsResponse, await res.json())
};