import {z} from 'zod'

export type GetApiApiV2EntitiesEntityIdKycOptions = {baseUrl?: string | undefined, ApiKeyId?: string | undefined, ApiSecretKey?: string | undefined, fetch?: typeof fetch | undefined};

export const kycStatus = z.enum(["PASS", "FAIL", "PENDING", "INCOMPLETE", "NEEDS_REVIEW"]);

export const baselineKycCheckDataV0 = z.object({country_code: z.string().min(2).max(2), last_name: z.string().min(1), address_country_code: z.string().min(2).max(2), first_name: z.string().min(1).nullable().optional(), middle_name: z.string().min(1).nullable().optional(), email: z.string().nullable().optional(), tax_id_number: z.string().min(1).nullable().optional(), birth_date: z.string().nullable().optional(), address_street_1: z.string().min(1).nullable().optional(), address_street_2: z.string().nullable().optional(), address_city: z.string().min(1).nullable().optional(), address_subdivision: z.string().min(1).nullable().optional(), address_postal_code: z.string().min(1).nullable().optional()});

export const baselineKyc = z.object({id: z.string(), status: kycStatus, checked_dt: z.string().nullable().optional(), data: baselineKycCheckDataV0.nullable().optional(), jurisdiction: z.literal("BASELINE").optional()});

export const usKycIdentityDataV1 = z.object({given_name: z.string().min(1), family_name: z.string().min(2), email_address: z.string(), phone_number: z.string(), street_address: z.string().min(2), city: z.string().min(2), country_of_tax_residence: z.literal("US"), postal_code: z.string().min(1), date_of_birth: z.string(), tax_id: z.string(), country_of_citizenship: z.string().min(2).max(2), middle_name: z.string().min(1).nullable().optional(), unit: z.string().min(1).nullable().optional(), state: z.string().min(1).nullable().optional()});

export const fundingSource = z.enum(["EMPLOYMENT_INCOME", "INVESTMENTS", "INHERITANCE", "BUSINESS_INCOME", "SAVINGS", "FAMILY"]);

export const usEntityKycFinancialProfileInfoV1 = z.object({funding_sources: z.array(fundingSource), liquid_net_worth_min: z.number().int().gte(0), liquid_net_worth_max: z.number().int().gte(0)});

export const employmentStatus = z.enum(["UNEMPLOYED", "EMPLOYED", "STUDENT", "RETIRED"]);

export const usEntityKycEmploymentInfoV1 = z.object({employment_status: employmentStatus, employer_name: z.string().min(1).nullable().optional(), employer_address: z.string().min(1).nullable().optional(), employment_position: z.string().min(1).nullable().optional()});

export const usEntityKycRiskDisclosureInfoV1 = z.object({is_control_person: z.boolean(), is_affiliated_exchange_or_finra: z.boolean(), is_politically_exposed: z.boolean(), immediate_family_exposed: z.boolean()});

export const usEntityTrustedContactInfoV1 = z.object({given_name: z.string(), family_name: z.string(), email_address: z.string().nullable().optional(), phone_number: z.string().nullable().optional()});

export const alpacaCustomerAgreementInfoV1 = z.object({signed_at: z.string(), ip_address: z.string()});

export const usEntityKycMetadataV1 = z.object({ref_id: z.string().min(1), check_initiated_at: z.string(), check_completed_at: z.string(), ip_address: z.string()});

export const usEntityKycCitationInfoV1 = z.object({address_sources: z.array(z.string()), date_of_birth_sources: z.array(z.string()), tax_id_sources: z.array(z.string())});

export const usEntityAmlCheckInfoV1 = z.object({ref_id: z.string().min(1), check_created_at: z.string(), is_politically_exposed_person_match: z.boolean(), is_adverse_media_match: z.boolean(), is_sanctions_match: z.boolean(), is_monitored_lists_match: z.boolean(), records: z.array(z.string())});

export const usNonProfessionalTraderAttestationV1 = z.object({is_non_professional_trader: z.boolean(), attestation_dt: z.string()});

export const usVisaType = z.enum(["B1", "B2", "DACA", "E1", "E2", "E3", "F1", "G4", "H1B", "J1", "L1", "Other", "O1", "TN1"]);

export const usEntityKycUsImmigrationInfoV1 = z.object({is_permanent_resident: z.boolean(), country_of_birth: z.string().min(2).max(2), visa_type: usVisaType.optional(), visa_expiration_date: z.string().nullable().optional(), departure_from_us_date: z.string().nullable().optional()});

export const usKycCheckDataV1 = z.object({identity: usKycIdentityDataV1, financial_profile: usEntityKycFinancialProfileInfoV1, employment: usEntityKycEmploymentInfoV1, risk_disclosure: usEntityKycRiskDisclosureInfoV1, trusted_contact: usEntityTrustedContactInfoV1, alpaca_customer_agreement: alpacaCustomerAgreementInfoV1, kyc_metadata: usEntityKycMetadataV1, data_citation: usEntityKycCitationInfoV1, aml_check: usEntityAmlCheckInfoV1, non_professional_trader_attestation: usNonProfessionalTraderAttestationV1, us_immigration_info: usEntityKycUsImmigrationInfoV1.nullable().optional()});

export const usKyc = z.object({id: z.string(), status: kycStatus, checked_dt: z.string().nullable().optional(), data: usKycCheckDataV1.nullable().optional(), jurisdiction: z.literal("US").optional()});

export const kyc = z.discriminatedUnion("jurisdiction", [baselineKyc, usKyc]);

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

/**
 * Get KYC Data
 * 
 * GET /api/v2/entities/{entity_id}/kyc
 */
export const getApiApiV2EntitiesEntityIdKyc = async (options?: GetApiApiV2EntitiesEntityIdKycOptions) => {
const {baseUrl = "https://api-enterprise.sbt.dinari.com", fetch: fetchFn = fetch, ApiKeyId, ApiSecretKey} = options ?? {}

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

if (ApiKeyId !== undefined) headers.set("X-API-Key-Id", ApiKeyId)
if (ApiSecretKey !== undefined) headers.set("X-API-Secret-Key", ApiSecretKey)

const url = `${baseUrl.replace(/\/+$/, '')}/api/v2/entities/${encodeURIComponent(String(args.entity_id))}/kyc${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(kyc, await res.json())
};