import {z} from 'zod'

export type AdditionalInfoType = 'KBA' | 'KENYA_KRA_PIN' | 'BANK_STATEMENT_DOCUMENT' | 'BUSINESS_REGISTRATION_DOCUMENT' | 'OPERATING_LICENSE_DOCUMENT' | 'SOURCE_OF_FUNDS_DOCUMENT' | 'VISA_DOCUMENT' | 'US_AAMVA' | 'AADHAAR_DOCUMENT_FRONT' | 'AADHAAR_DOCUMENT_BACK' | 'USER_SELFIE' | 'GENDER' | 'SEON_SESSION_PAYLOAD' | 'ADDRESS_PROOF_DOCUMENT' | 'NIGERIA_BVN_NUMBER' | 'TAX_IDENTIFICATION_NUMBER' | 'PREVIOUS_VERIFICATION_SESSION_ID' | 'FRAUD_CHECK_DOCUMENT' | 'OPERATION_CARD' | 'TRANSPORTATION_PERMIT' | 'VEHICLE_OWNERSHIP_CARD' | 'TAX_CERTIFICATE' | 'OTHER';

export type AddressProofFileSubType = 'NATIONAL_ID_FRONT' | 'NATIONAL_ID_BACK' | 'UTILITY_SERVICE_BILLS' | 'FINANCIAL_STATEMENTS' | 'GOVERNMENT_CORRESPONDENCE' | 'RESIDENTIAL_DOCUMENT' | 'EMPLOYMENT_EDUCATION_DOCUMENT' | 'LEGAL_DOCUMENT' | 'ADDRESS_DOCUMENT_OTHERS';

export type IndividualAdditionalInfo = {additional_info_type?: AdditionalInfoType | undefined, additional_info_data?: (string | {file_data?: string | undefined, file_uuid?: string | undefined, document_input_title?: (string | null) | undefined} | Array<{file: string, file_sub_type?: AddressProofFileSubType | undefined}> | Array<{type: string, answer: string}>) | undefined};

export type AdditionalInfo = Array<IndividualAdditionalInfo>;

export type CustomAMLRequest = {entity_type: 'person' | 'business', entity_data: {first_name?: string | undefined, last_name?: string | undefined, date_of_birth?: string | undefined, country_code?: string | undefined, business_name?: string | undefined, registration_number?: string | undefined} | {[key: string]: unknown}, aml_data: {risks?: Array<{[key: string]: unknown}> | undefined} | {[key: string]: unknown} | Array<{[key: string]: unknown}>, template_id: string, search_metadata?: {[key: string]: unknown} | undefined, customer_reference_id?: string | undefined, trigger_analysis?: boolean | undefined, client_reference_id?: string | undefined, client_reference_data?: {[key: string]: unknown} | undefined, additional_info?: Array<AdditionalInfo> | undefined, verification_options?: {[key: string]: unknown} | undefined, callback_url?: string | undefined, events_callback_url?: string | undefined};

export type CreateApiVerifyAnalyzeMeshAmlDataArgs = {body: CustomAMLRequest};

export type CreateApiVerifyAnalyzeMeshAmlDataOptions = {baseUrl?: string | undefined, apiKey?: string | undefined, fetch?: typeof fetch | undefined};

export const customAMLResponse = z.object({status: z.enum(["success", "error"]).optional(), data: z.object({verification_session_id: z.string().optional(), entity_hits_count: z.number().int().optional(), analysis_tasks: z.array(z.string()).optional(), message: z.string().optional()}).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
};

/**
 * Analyze Custom AML Data (Mesh Format), Its a BETA feature
 * 
 * POST /verify/analyze_mesh_aml_data
 */
export const createApiVerifyAnalyzeMeshAmlData = async (args: CreateApiVerifyAnalyzeMeshAmlDataArgs, options?: CreateApiVerifyAnalyzeMeshAmlDataOptions) => {
const {baseUrl = "https://api-sandbox.aiprise.com/api/v1", fetch: fetchFn = fetch, apiKey} = options ?? {}

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

if (apiKey !== undefined) headers.set("X-API-KEY", apiKey)

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