import {z} from 'zod'

export type UserMonitoringTypeEnum = 'COMPANY' | 'PERSON';

export type CountryEnum = 'AF' | 'AX' | 'AL' | 'DZ' | 'AS' | 'AD' | 'AO' | 'AI' | 'AQ' | 'AG' | 'AR' | 'AM' | 'AW' | 'AU' | 'AT' | 'AZ' | 'BS' | 'BH' | 'BD' | 'BB' | 'BY' | 'BE' | 'BZ' | 'BJ' | 'BM' | 'BT' | 'BO' | 'BQ' | 'BA' | 'BW' | 'BV' | 'BR' | 'IO' | 'BN' | 'BG' | 'BF' | 'BI' | 'CV' | 'KH' | 'CM' | 'CA' | 'KY' | 'CF' | 'TD' | 'CL' | 'CN' | 'CX' | 'CC' | 'CO' | 'KM' | 'CG' | 'CK' | 'CR' | 'CI' | 'HR' | 'CU' | 'CW' | 'CY' | 'CZ' | 'CD' | 'DK' | 'DJ' | 'DM' | 'DO' | 'EC' | 'EG' | 'SV' | 'GQ' | 'ER' | 'EE' | 'SZ' | 'ET' | 'FK' | 'FO' | 'FJ' | 'FI' | 'FR' | 'GF' | 'PF' | 'TF' | 'GA' | 'GM' | 'GE' | 'DE' | 'GH' | 'GI' | 'GR' | 'GL' | 'GD' | 'GP' | 'GU' | 'GT' | 'GG' | 'GN' | 'GW' | 'GY' | 'HT' | 'HM' | 'HN' | 'HK' | 'HU' | 'IS' | 'IN' | 'ID' | 'IR' | 'IQ' | 'IE' | 'IM' | 'IL' | 'IT' | 'JM' | 'JP' | 'JE' | 'JO' | 'KZ' | 'KE' | 'KI' | 'XK' | 'KW' | 'KG' | 'LA' | 'LV' | 'LB' | 'LS' | 'LR' | 'LY' | 'LI' | 'LT' | 'LU' | 'MO' | 'MG' | 'MW' | 'MY' | 'MV' | 'ML' | 'MT' | 'MH' | 'MQ' | 'MR' | 'MU' | 'YT' | 'MX' | 'FM' | 'MD' | 'MC' | 'MN' | 'ME' | 'MS' | 'MA' | 'MZ' | 'MM' | 'NA' | 'NR' | 'NP' | 'NL' | 'NC' | 'NZ' | 'NI' | 'NE' | 'NG' | 'NU' | 'NF' | 'KP' | 'MK' | 'MP' | 'NO' | 'OM' | 'PK' | 'PW' | 'PS' | 'PA' | 'PG' | 'PY' | 'PE' | 'PH' | 'PN' | 'PL' | 'PT' | 'PR' | 'QA' | 'RE' | 'RO' | 'RU' | 'RW' | 'BL' | 'SH' | 'KN' | 'LC' | 'MF' | 'PM' | 'VC' | 'WS' | 'SM' | 'ST' | 'SA' | 'SN' | 'RS' | 'SC' | 'SL' | 'SG' | 'SX' | 'SK' | 'SI' | 'SB' | 'SO' | 'ZA' | 'GS' | 'KR' | 'SS' | 'ES' | 'LK' | 'SD' | 'SR' | 'SJ' | 'SE' | 'CH' | 'SY' | 'TW' | 'TJ' | 'TZ' | 'TH' | 'TL' | 'TG' | 'TK' | 'TO' | 'TT' | 'TN' | 'TR' | 'TM' | 'TC' | 'TV' | 'UG' | 'UA' | 'AE' | 'GB' | 'UM' | 'US' | 'UY' | 'UZ' | 'VU' | 'VA' | 'VE' | 'VN' | 'VG' | 'VI' | 'WF' | 'EH' | 'YE' | 'ZM' | 'ZW';

export type MainSexEnum = 'MALE' | 'FEMALE';

export type PersonAmlCheckInputRequest = {fullName: string, nationality?: (CountryEnum | null) | undefined, dateOfBirth?: (string | null) | undefined, sex?: (MainSexEnum | null) | undefined};

export type PersonAmlCheckFilterDatasetsEnum = 'PEP_CURRENT' | 'PEP_FORMER' | 'PEP_LINKED' | 'SAN_CURRENT' | 'SAN_FORMER' | 'INS' | 'RRE' | 'DD' | 'POI' | 'REL';

export type DateOfBirthMatchingEnum = 'EXACT' | 'SAME_YEAR' | 'WITHIN_ONE_YEAR' | 'WITHIN_TWO_YEARS' | 'WITHIN_THREE_YEARS' | 'WITHIN_FOUR_YEARS' | 'WITHIN_FIVE_YEARS';

export type PepTierEnum = 'PEP_TIER_1' | 'PEP_TIER_2' | 'PEP_TIER_3';

export type PersonAmlCheckFilterRequest = {threshold?: (number | null) | undefined, datasets?: Array<PersonAmlCheckFilterDatasetsEnum> | undefined, sanctionBodies?: (Array<number> | null) | undefined, dateOfBirthMatching?: (DateOfBirthMatchingEnum | null) | undefined, pepTiers?: (Array<PepTierEnum> | null) | undefined};

export type PersonAmlCheckRequest = {clientId: string, userType: UserMonitoringTypeEnum, input: PersonAmlCheckInputRequest, filter: PersonAmlCheckFilterRequest};

export type CompanyAmlCheckInputRequest = {companyName: string, country?: (CountryEnum | null) | undefined};

export type CompanyAmlCheckFilterDatasetsEnum = 'PEP_LINKED' | 'SAN_CURRENT' | 'SAN_FORMER' | 'INS' | 'RRE' | 'POI' | 'REL' | 'SOE_CURRENT' | 'SOE_FORMER';

export type CompanyAmlCheckFilterRequest = {threshold?: (number | null) | undefined, datasets?: Array<CompanyAmlCheckFilterDatasetsEnum> | undefined, sanctionBodies?: (Array<number> | null) | undefined};

export type CompanyAmlCheckRequest = {clientId: string, userType: UserMonitoringTypeEnum, input: CompanyAmlCheckInputRequest, filter: CompanyAmlCheckFilterRequest};

export type AmlCheckRequest = PersonAmlCheckRequest | CompanyAmlCheckRequest;

export type CreateApiAmlChecksArgs = {body: AmlCheckRequest};

export type CreateApiAmlChecksOptions = {baseUrl?: string | undefined, apiKeyPair?: {username: string, password: string} | undefined, fetch?: typeof fetch | undefined};

export const userMonitoringTypeEnum = z.enum(["COMPANY", "PERSON"]);

export const countryEnum = z.enum(["AF", "AX", "AL", "DZ", "AS", "AD", "AO", "AI", "AQ", "AG", "AR", "AM", "AW", "AU", "AT", "AZ", "BS", "BH", "BD", "BB", "BY", "BE", "BZ", "BJ", "BM", "BT", "BO", "BQ", "BA", "BW", "BV", "BR", "IO", "BN", "BG", "BF", "BI", "CV", "KH", "CM", "CA", "KY", "CF", "TD", "CL", "CN", "CX", "CC", "CO", "KM", "CG", "CK", "CR", "CI", "HR", "CU", "CW", "CY", "CZ", "CD", "DK", "DJ", "DM", "DO", "EC", "EG", "SV", "GQ", "ER", "EE", "SZ", "ET", "FK", "FO", "FJ", "FI", "FR", "GF", "PF", "TF", "GA", "GM", "GE", "DE", "GH", "GI", "GR", "GL", "GD", "GP", "GU", "GT", "GG", "GN", "GW", "GY", "HT", "HM", "HN", "HK", "HU", "IS", "IN", "ID", "IR", "IQ", "IE", "IM", "IL", "IT", "JM", "JP", "JE", "JO", "KZ", "KE", "KI", "XK", "KW", "KG", "LA", "LV", "LB", "LS", "LR", "LY", "LI", "LT", "LU", "MO", "MG", "MW", "MY", "MV", "ML", "MT", "MH", "MQ", "MR", "MU", "YT", "MX", "FM", "MD", "MC", "MN", "ME", "MS", "MA", "MZ", "MM", "NA", "NR", "NP", "NL", "NC", "NZ", "NI", "NE", "NG", "NU", "NF", "KP", "MK", "MP", "NO", "OM", "PK", "PW", "PS", "PA", "PG", "PY", "PE", "PH", "PN", "PL", "PT", "PR", "QA", "RE", "RO", "RU", "RW", "BL", "SH", "KN", "LC", "MF", "PM", "VC", "WS", "SM", "ST", "SA", "SN", "RS", "SC", "SL", "SG", "SX", "SK", "SI", "SB", "SO", "ZA", "GS", "KR", "SS", "ES", "LK", "SD", "SR", "SJ", "SE", "CH", "SY", "TW", "TJ", "TZ", "TH", "TL", "TG", "TK", "TO", "TT", "TN", "TR", "TM", "TC", "TV", "UG", "UA", "AE", "GB", "UM", "US", "UY", "UZ", "VU", "VA", "VE", "VN", "VG", "VI", "WF", "EH", "YE", "ZM", "ZW"]);

export const mainSexEnum = z.enum(["MALE", "FEMALE"]);

export const personAmlCheckInput = z.object({fullName: z.string().max(100), nationality: countryEnum.nullable().optional(), dateOfBirth: z.string().nullable().optional(), sex: mainSexEnum.nullable().optional()});

export const personAmlCheckFilterDatasetsEnum = z.enum(["PEP_CURRENT", "PEP_FORMER", "PEP_LINKED", "SAN_CURRENT", "SAN_FORMER", "INS", "RRE", "DD", "POI", "REL"]);

export const dateOfBirthMatchingEnum = z.enum(["EXACT", "SAME_YEAR", "WITHIN_ONE_YEAR", "WITHIN_TWO_YEARS", "WITHIN_THREE_YEARS", "WITHIN_FOUR_YEARS", "WITHIN_FIVE_YEARS"]);

export const pepTierEnum = z.enum(["PEP_TIER_1", "PEP_TIER_2", "PEP_TIER_3"]);

export const personAmlCheckFilter = z.object({threshold: z.number().int().gte(75).lte(100).nullable().optional(), datasets: z.array(personAmlCheckFilterDatasetsEnum).nullable().optional(), sanctionBodies: z.array(z.number().int().gte(1).lte(80)).nullable().optional(), dateOfBirthMatching: dateOfBirthMatchingEnum.nullable().optional(), pepTiers: z.array(pepTierEnum).nullable().optional()});

export const searchAddress = z.object({geography: z.string(), city: z.string().optional()});

export const individualDatasetEnum = z.enum(["PEP_CURRENT", "PEP_FORMER", "PEP_LINKED", "SAN_CURRENT", "SAN_FORMER", "INS", "RRE", "DD", "POI", "REL", "GRI"]);

export const individualSearchResult = z.object({version: z.number().int().optional(), resourceId: z.string(), score: z.number().int(), match: z.string(), name: z.string(), addresses: z.array(searchAddress), nationalities: z.array(countryEnum), datesOfBirth: z.array(z.string()), gender: mainSexEnum.optional(), profileImage: z.string().optional(), datasets: z.array(individualDatasetEnum), currentSanBodyIds: z.array(z.number().int()), formerSanBodyIds: z.array(z.number().int()), pepTier: pepTierEnum.optional()});

export const amlArticleTypeEnum = z.literal("NEGATIVE");

export const adverseMediaSearchResult = z.object({title: z.string().nullable(), url: z.string(), text: z.string().nullable(), score: z.number().int().nullable(), type: amlArticleTypeEnum.nullable(), publishDate: z.string().nullable()});

export const amlCheckStatusEnum = z.enum(["NOT_CHECKED", "NO_FLAGS", "FALSE_POSITIVE", "TRUE_POSITIVE", "FLAGS_FOUND"]);

export const monitoringStatusEnum = z.enum(["ACTIVE", "STOPPED", "PENDING", "ALERT"]);

export const amlCheckMonitoring = z.object({id: z.string(), status: monitoringStatusEnum.optional(), isSubscribed: z.boolean().optional(), expiration: z.string().optional(), tags: z.array(z.string().max(32)).optional(), created: z.string()});

export const personAmlCheck = z.object({id: z.string(), clientId: z.string().max(100), checkedAt: z.string().nullable(), created: z.string(), userType: userMonitoringTypeEnum, input: personAmlCheckInput, filter: personAmlCheckFilter, amlCheck: z.array(individualSearchResult).nullable(), adverseMedia: z.array(adverseMediaSearchResult).nullable(), overallStatus: amlCheckStatusEnum, monitoring: amlCheckMonitoring.nullable()});

export const companyAmlCheckInput = z.object({companyName: z.string().max(100), country: countryEnum.nullable().optional()});

export const companyAmlCheckFilterDatasetsEnum = z.enum(["PEP_LINKED", "SAN_CURRENT", "SAN_FORMER", "INS", "RRE", "POI", "REL", "SOE_CURRENT", "SOE_FORMER"]);

export const companyAmlCheckFilter = z.object({threshold: z.number().int().gte(75).lte(100).nullable().optional(), datasets: z.array(companyAmlCheckFilterDatasetsEnum).nullable().optional(), sanctionBodies: z.array(z.number().int().gte(1).lte(80)).nullable().optional()});

export const businessSearchDatasetEnum = z.enum(["PEP", "PEP_LINKED", "SAN", "SAN_CURRENT", "SAN_FORMER", "INS", "RRE", "POI", "REL", "SOE", "SOE_CURRENT", "SOE_FORMER"]);

export const businessSearchResult = z.object({version: z.number().int().optional(), resourceId: z.string(), score: z.number().int(), match: z.string(), name: z.string(), addresses: z.array(searchAddress), datasets: z.array(businessSearchDatasetEnum), currentSanBodyIds: z.array(z.number().int()), formerSanBodyIds: z.array(z.number().int())});

export const companyAmlCheck = z.object({id: z.string(), clientId: z.string().max(100), checkedAt: z.string().nullable(), created: z.string(), userType: userMonitoringTypeEnum, input: companyAmlCheckInput, filter: companyAmlCheckFilter, amlCheck: z.array(businessSearchResult).nullable(), adverseMedia: z.array(adverseMediaSearchResult).nullable(), overallStatus: amlCheckStatusEnum, monitoring: amlCheckMonitoring.nullable()});

export const amlCheck = z.discriminatedUnion("userType", [personAmlCheck, companyAmlCheck]);

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 AML check
 * 
 * POST /aml/checks/
 */
export const createApiAmlChecks = async (args: CreateApiAmlChecksArgs, options?: CreateApiAmlChecksOptions) => {
const {baseUrl = "https://ivs.idenfy.com", fetch: fetchFn = fetch, apiKeyPair} = options ?? {}

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

if (apiKeyPair !== undefined) headers.set('authorization', `Basic ${btoa(`${apiKeyPair.username}:${apiKeyPair.password}`)}`)

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