import {z} from 'zod'

export type BaseFetchMode = 'full' | 'preview';

export type ExistsFilter = {negate?: boolean | undefined, value: boolean};

export type JobSeniorityLevel = 'owner' | 'c-suite' | 'vice president' | 'director' | 'senior non-managerial' | 'manager' | 'partner' | 'non-managerial' | 'junior' | 'president' | 'senior manager' | 'advisor' | 'freelancer' | 'board member' | 'founder';

export type IncludesFilterJobSeniorityLevel = {negate?: boolean | undefined, values: Array<JobSeniorityLevel>};

export type JobDepartment = 'administration' | 'real estate' | 'healthcare' | 'partnerships' | 'c-suite' | 'design' | 'human resources' | 'engineering' | 'education' | 'strategy' | 'product' | 'sales' | 'r&d' | 'retail' | 'customer success' | 'security' | 'public service' | 'creative' | 'it' | 'support' | 'marketing' | 'trade' | 'legal' | 'operations' | 'procurement' | 'data' | 'manufacturing' | 'logistics' | 'finance';

export type IncludesFilterJobDepartment = {negate?: boolean | undefined, values: Array<JobDepartment>};

export type IncludesFilterBusinessId = {negate?: boolean | undefined, values: Array<string>};

export type RangeFilterInt = {negate?: boolean | undefined, gte?: number | undefined, lte?: number | undefined};

export type IncludesFilterCountryCodeAlpha2 = {negate?: boolean | undefined, values: Array<string>};

export type IncludesFilterRegionCode = {negate?: boolean | undefined, values: Array<string>};

export type NumberOfEmployeesRange = '1-10' | '11-50' | '51-200' | '201-500' | '501-1000' | '1001-5000' | '5001-10000' | '10001+';

export type IncludesFilterNumberOfEmployeesRange = {negate?: boolean | undefined, values: Array<NumberOfEmployeesRange>};

export type RevenueRange = '0-500K' | '500K-1M' | '1M-5M' | '5M-10M' | '10M-25M' | '25M-75M' | '75M-200M' | '200M-500M' | '500M-1B' | '1B-10B' | '10B-100B' | '100B-1T' | '1T-10T' | '10T+';

export type IncludesFilterRevenueRange = {negate?: boolean | undefined, values: Array<RevenueRange>};

export type IncludesFilterStandardizedText = {negate?: boolean | undefined, values: Array<string>};

export type IncludesFilterNAICS = {negate?: boolean | undefined, values: Array<string>};

export type IncludesFilterText = {negate?: boolean | undefined, values: Array<string>};

export type JobTitleFilter = {negate?: boolean | undefined, values: Array<string>, include_related_job_titles?: (boolean | null) | undefined};

export type AnyMatchFilterOrganizationName = {negate?: boolean | undefined, values: Array<string>};

export type ProspectsFetchFilters = {has_email?: ExistsFilter | undefined, has_phone_number?: ExistsFilter | undefined, job_level?: IncludesFilterJobSeniorityLevel | undefined, job_department?: IncludesFilterJobDepartment | undefined, business_id?: IncludesFilterBusinessId | undefined, total_experience_months?: RangeFilterInt | undefined, country_code?: IncludesFilterCountryCodeAlpha2 | undefined, region_country_code?: IncludesFilterRegionCode | undefined, current_role_months?: RangeFilterInt | undefined, company_size?: IncludesFilterNumberOfEmployeesRange | undefined, company_revenue?: IncludesFilterRevenueRange | undefined, google_category?: IncludesFilterStandardizedText | undefined, naics_category?: IncludesFilterNAICS | undefined, linkedin_category?: IncludesFilterText | undefined, job_title?: JobTitleFilter | undefined, company_country_code?: IncludesFilterCountryCodeAlpha2 | undefined, company_region_country_code?: IncludesFilterRegionCode | undefined, city_region_country?: IncludesFilterText | undefined, company_name?: AnyMatchFilterOrganizationName | undefined, has_website?: (ExistsFilter | null) | undefined};

export type ProspectsFetchRequest = {request_context?: (Record<string, never> | null) | undefined, mode: BaseFetchMode, size: number, page_size: number, page?: (number | null) | undefined, exclude?: (Array<string> | null) | undefined, filters?: (ProspectsFetchFilters | null) | undefined, next_cursor?: (string | null) | undefined};

export type CreateApiV1ProspectsArgs = {tenant?: string | undefined, body: ProspectsFetchRequest};

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

export const requestStatus = z.enum(["success", "miss", "failure"]);

export const responseContext = z.object({correlation_id: z.string(), request_status: requestStatus, time_took_in_seconds: z.number()});

export const prospect = z.object({prospect_id: z.string().optional(), professional_email_hashed: z.string().optional(), first_name: z.string().optional(), last_name: z.string().optional(), full_name: z.string().optional(), country_name: z.string().optional(), region_name: z.string().optional(), city: z.string().max(256).optional(), linkedin: z.string().optional(), experience: z.array(z.string()).optional(), skills: z.array(z.string()).optional(), interests: z.array(z.string()).optional(), company_name: z.string().max(256).optional(), company_website: z.string().optional(), company_linkedin: z.string().optional(), job_department: z.string().optional(), job_department_array: z.array(z.string()).optional(), job_department_main: z.string().optional(), job_seniority_level: z.array(z.string()).optional(), job_level_array: z.array(z.string()).optional(), job_level_main: z.string().optional(), job_title: z.string().optional(), business_id: z.string().optional(), linkedin_url_array: z.array(z.string()).optional()});

export const prospectsFetchResponse = z.object({response_context: responseContext, data: z.array(prospect).optional(), total_results: z.number().int().gte(0).optional(), page: z.number().int().gte(0).optional(), total_pages: z.number().int().gte(0)});

export const partnerServiceModelsProspectsRequestsPageInfo = z.object({size: z.number().int(), next_cursor: z.string().optional()});

export const prospectsFetchResponseV2 = z.object({response_context: responseContext, data: z.array(prospect).optional(), total_results: z.number().int().gte(0).optional(), page: partnerServiceModelsProspectsRequestsPageInfo});

export const createApiV1ProspectsResponse = z.union([prospectsFetchResponse, prospectsFetchResponseV2]);

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

/**
 * Fetch Prospects
 * 
 * POST /v1/prospects
 * 
 * Also accepts auth: APIKeyHeader
 */
export const createApiV1Prospects = async (args: CreateApiV1ProspectsArgs, options?: CreateApiV1ProspectsOptions) => {
const {baseUrl = "https://api.explorium.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 (args.tenant != null) headers.set("tenant", String(args.tenant))
if (APIKeyHeader !== undefined) headers.set("api_key", APIKeyHeader)

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