import {z} from 'zod'

export type Name = string;

export type Identifier = string;

export type AutoValidationThreshold = number;

export type EnableAutoValidationThreshold = boolean;

export type AnnotationContentType = 'text' | 'integer' | 'float' | 'decimal' | 'date' | 'datetime' | 'daterange' | 'boolean' | 'enum' | 'location' | 'phonenumber' | 'json' | 'table' | 'expectedremuneration' | 'jobtitle' | 'language' | 'skill' | 'yearsexperience' | 'group' | 'table_deprecated' | 'url' | 'image' | 'docclf';

export type FieldDeprecated = {label: string, slug?: (string | null) | undefined, fieldType: AnnotationContentType, dataSource?: (string | null) | undefined, mapping?: (string | null) | undefined, dataPoint: string, mandatory?: boolean | undefined, disabled?: boolean | undefined, autoValidationThreshold?: (number | null) | undefined, showDropdown?: boolean | undefined, displayRawText?: boolean | undefined, dropNull?: boolean | undefined, displayEnumValue?: boolean | undefined, fields?: Array<FieldDeprecated> | undefined} | {[key: string]: unknown};

export type FieldGroup = {label: string, fields: Array<FieldDeprecated>};

export type Fields = Array<FieldGroup> | null;

export type Field = {label: string, dataPoint: string, fieldType: AnnotationContentType, dataSource?: (string | null) | undefined, mapping?: (string | null) | undefined, mandatory?: boolean | undefined, autoValidationThreshold?: (number | null) | undefined, enableAutoValidationThreshold?: (boolean | null) | undefined, showDropdown?: boolean | undefined, displayEnumValue?: boolean | undefined, hideEnumDetail?: boolean | undefined, dropNull?: boolean | undefined, enabledChildFields?: Array<Field> | undefined, disabledChildFields?: Array<Field> | undefined, slug?: (string | null) | undefined, displayRawText?: boolean | undefined, fields?: (Array<Record<string, never>> | null) | undefined} | {[key: string]: unknown};

export type FieldCategory = {label: string, enabledFields: Array<Field>, disabledFields: Array<Field>};

export type FieldsLayout = {defaultCategory: FieldCategory, categories: Array<FieldCategory>};

export type DateFormatPreference = 'DMY' | 'MDY' | 'YMD' | null;

export type DateFormatFromDocument = boolean | null;

export type RedactConfig = {redactHeadshot?: boolean | undefined, redactPersonalDetails?: boolean | undefined, redactWorkDetails?: boolean | undefined, redactReferees?: boolean | undefined, redactEducationDetails?: boolean | undefined, redactLocations?: boolean | undefined, redactDates?: boolean | undefined, redactGender?: boolean | undefined};

export type ExtractorConfig = {resumeRedact?: RedactConfig | undefined} | null;

export type AllowOpenai = boolean;

export type TrainsExtractor = boolean;

export type DisableConfirmationIfValidationRulesFail = boolean;

export type CollectionCreate = {name: Name, workspace: Identifier, extractor?: string | undefined, baseExtractor?: string | undefined, autoValidationThreshold?: AutoValidationThreshold | undefined, enableAutoValidationThreshold?: EnableAutoValidationThreshold | undefined, fields?: Fields | undefined, fieldsLayout?: FieldsLayout | undefined, dateFormatPreference?: DateFormatPreference | undefined, dateFormatFromDocument?: DateFormatFromDocument | undefined, extractorConfig?: ExtractorConfig | undefined, allowOpenai?: AllowOpenai | undefined, trainsExtractor?: TrainsExtractor | undefined, disableConfirmationIfValidationRulesFail?: DisableConfirmationIfValidationRulesFail | undefined};

export type CreateApiV3CollectionsArgs = {body: CollectionCreate};

export type CreateApiV3CollectionsOptions = {baseUrl?: string | undefined, ApiKeyAuth?: string | undefined, fetch?: typeof fetch | undefined};

export const identifier = z.string();

export const organizationRole = z.enum(["admin", "member"]);

export const paletteColorOptions = z.object({main: z.string(), light: z.string().optional(), dark: z.string().optional(), contrastText: z.string().optional()});

export const themeConfig = z.object({palette: z.object({mode: z.enum(["light", "dark"]).optional(), background: z.union([z.string(), z.object({default: z.string().optional(), paper: z.string().optional()})]).optional(), text: z.object({primary: z.string().optional(), secondary: z.string().optional(), disabled: z.string().optional()}).optional(), divider: z.string().optional(), primary: paletteColorOptions.optional(), secondary: paletteColorOptions.optional(), success: paletteColorOptions.optional(), annotation: paletteColorOptions.optional(), error: paletteColorOptions.optional(), info: paletteColorOptions.optional(), warning: paletteColorOptions.optional()}).optional(), typography: z.object({fontFamily: z.string().optional(), fontSize: z.union([z.string(), z.number()]).optional(), fontWeightRegular: z.string().optional(), fontWeightMedium: z.string().optional(), fontWeightBold: z.string().optional()}).optional(), borderRadius: z.number().optional(), fontUrl: z.string().optional()});

export const organization = z.object({identifier: z.string(), name: z.string(), userRole: organizationRole.nullable(), avatar: z.string().nullable(), resthookSignatureKey: z.string().nullable(), isTrial: z.boolean(), validationToolConfig: z.object({theme: themeConfig.optional(), hideActions: z.boolean().optional(), hideCollection: z.boolean().optional(), hideEditPages: z.boolean().optional(), hideExport: z.boolean().optional(), hideFilename: z.boolean().optional(), hideShowRawValues: z.boolean().optional(), hideReject: z.boolean().optional(), hideReparse: z.boolean().optional(), hideRunOcr: z.boolean().optional(), hideTags: z.boolean().optional(), hideWarnings: z.boolean().optional(), restrictDocumentSplitting: z.boolean().optional(), disableCurrencyFormatting: z.boolean().optional(), disableEditDocumentMetadata: z.boolean().optional(), disableManualAnnotationEditing: z.boolean().optional(), hideDocumentStatus: z.boolean().optional()}).nullable(), showCustomFieldCreation: z.boolean(), allowDataSourceRecordCreation: z.boolean()});

export const name = z.string();

export const extractorPropertiesIdentifier = z.string();

export const namePlural = z.string();

export const validatable = z.boolean();

export const isCustom = z.boolean();

export const createdDt = z.string();

export const annotationContentType = z.enum(["text", "integer", "float", "decimal", "date", "datetime", "daterange", "boolean", "enum", "location", "phonenumber", "json", "table", "expectedremuneration", "jobtitle", "language", "skill", "yearsexperience", "group", "table_deprecated", "url", "image", "docclf"]);

export const fieldDeprecated: z.ZodType<FieldDeprecated> = z.object({label: z.string(), slug: z.string().nullable().optional(), fieldType: annotationContentType, dataSource: z.string().nullable().optional(), mapping: z.string().nullable().optional(), dataPoint: z.string(), mandatory: z.boolean().optional(), disabled: z.boolean().optional(), autoValidationThreshold: z.number().nullable().optional(), showDropdown: z.boolean().optional(), displayRawText: z.boolean().optional(), dropNull: z.boolean().optional(), displayEnumValue: z.boolean().optional(), fields: z.array(z.lazy(() => fieldDeprecated)).optional()}).and(z.record(z.string(), z.unknown()));

export const fieldGroup = z.object({label: z.string(), fields: z.array(fieldDeprecated)});

export const extractor = z.object({identifier: z.string(), name: z.string(), namePlural: z.string(), baseExtractor: z.object({identifier: extractorPropertiesIdentifier, name: name, namePlural: namePlural, validatable: validatable, isCustom: isCustom.optional(), createdDt: createdDt.optional()}).nullable().optional(), organization: organization.optional(), category: z.string().nullable().optional(), validatable: z.boolean(), isCustom: z.boolean().optional(), fieldGroups: z.array(fieldGroup).nullable().optional(), createdDt: z.string().optional(), lastTrainedDt: z.string().nullable().optional()});

export const field: z.ZodType<Field> = z.object({label: z.string(), dataPoint: z.string(), fieldType: annotationContentType, dataSource: z.string().nullable().optional(), mapping: z.string().nullable().optional(), mandatory: z.boolean().optional(), autoValidationThreshold: z.number().gte(0).lte(1).nullable().optional(), enableAutoValidationThreshold: z.boolean().nullable().optional(), showDropdown: z.boolean().optional(), displayEnumValue: z.boolean().optional(), hideEnumDetail: z.boolean().optional(), dropNull: z.boolean().optional(), enabledChildFields: z.array(z.lazy(() => field)).optional(), disabledChildFields: z.array(z.lazy(() => field)).optional(), slug: z.string().nullable().optional(), displayRawText: z.boolean().optional(), fields: z.array(z.object({})).nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const fieldCategory = z.object({label: z.string(), enabledFields: z.array(field), disabledFields: z.array(field)});

export const fieldsLayout = z.object({defaultCategory: fieldCategory, categories: z.array(fieldCategory)});

export const redactConfig = z.object({redactHeadshot: z.boolean().optional(), redactPersonalDetails: z.boolean().optional(), redactWorkDetails: z.boolean().optional(), redactReferees: z.boolean().optional(), redactEducationDetails: z.boolean().optional(), redactLocations: z.boolean().optional(), redactDates: z.boolean().optional(), redactGender: z.boolean().optional()});

export const extractorConfig = z.object({resumeRedact: redactConfig.optional()}).nullable();

export const collection = z.object({identifier: z.string(), name: z.string().optional(), workspace: z.object({identifier: identifier.optional(), organization: organization.optional(), name: name.optional()}).optional(), extractor: extractor.optional(), autoValidationThreshold: z.number().optional(), enableAutoValidationThreshold: z.boolean().optional(), autoValidateIfValidationRulesPass: z.boolean().optional(), fields: z.array(fieldGroup).nullable().optional(), fieldsLayout: fieldsLayout.optional(), fieldsConfigured: z.boolean().optional(), dateFormatPreference: z.enum(["DMY", "MDY", "YMD"]).nullable().optional(), dateFormatFromDocument: z.boolean().nullable().optional(), extractorConfig: extractorConfig.optional(), unvalidatedDocsCount: z.number().int().gte(0).nullable().optional(), confirmedDocsCount: z.number().int().gte(0).nullable().optional(), ingestEmail: z.string().optional(), tailoredExtractorRequested: z.boolean().optional(), allowOpenai: z.boolean().optional(), trainsExtractor: z.boolean().optional(), disableConfirmationIfValidationRulesFail: z.boolean().optional(), autoRefreshValidationResults: z.boolean().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
};

/**
 * Create a collection
 * 
 * POST /v3/collections
 */
export const createApiV3Collections = async (args: CreateApiV3CollectionsArgs, options?: CreateApiV3CollectionsOptions) => {
const {baseUrl = "https://api.affinda.com", fetch: fetchFn = fetch, ApiKeyAuth} = options ?? {}

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

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

const url = `${baseUrl.replace(/\/+$/, '')}/v3/collections${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(collection, await res.json())
};