import {z} from 'zod'

export type GetApiApiV1UsStatutesCoverageArgs = {state?: ('federal' | 'al' | 'ak' | 'as' | 'az' | 'ar' | 'ca' | 'co' | 'ct' | 'de' | 'dc' | 'fl' | 'ga' | 'gu' | 'hi' | 'id' | 'il' | 'in' | 'ia' | 'ks' | 'ky' | 'la' | 'me' | 'md' | 'ma' | 'mi' | 'mn' | 'ms' | 'mo' | 'mt' | 'ne' | 'nv' | 'nh' | 'nj' | 'nm' | 'ny' | 'nc' | 'nd' | 'mp' | 'oh' | 'ok' | 'or' | 'pa' | 'pr' | 'ri' | 'sc' | 'sd' | 'tn' | 'tx' | 'ut' | 'vt' | 'vi' | 'va' | 'wa' | 'wv' | 'wi' | 'wy' | null) | undefined};

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

export const usCorpusTypeInfo = z.object({token: z.string(), description: z.string(), scope: z.enum(["federal", "state"]), browsable: z.boolean().optional(), scopeNote: z.string().nullable().optional(), sourceScopeNotes: z.record(z.string(), z.string()).optional()});

export const sessionLawDenominator = z.object({kind: z.enum(["stated_count", "complete_listing", "max_number", "feed_total", "our_recount", "registry_union", "none"]), value: z.number().int().gte(0).nullable().optional(), derivation: z.string()});

export const sessionLawSessionCoverage = z.object({code: z.string(), label: z.string().nullable().optional(), type: z.string().nullable().optional(), year: z.number().int().nullable().optional(), status: z.enum(["complete", "partial"]), servedCount: z.number().int().gte(0), withheldCount: z.number().int().gte(0).optional(), notHeldCount: z.number().int().gte(0).optional(), registryCount: z.number().int().gte(0).optional(), collectedCount: z.number().int().gte(0).optional(), denominator: sessionLawDenominator.nullable().optional(), openGaps: z.number().int().gte(0).optional(), numberingHoles: z.number().int().gte(0).nullable().optional(), denominatorStale: z.boolean().nullable().optional(), measuredAt: z.string().nullable().optional()});

export const sessionLawJurisdictionCoverage = z.object({sessions: z.array(sessionLawSessionCoverage).optional(), sessionCount: z.number().int().gte(0), completeSessionCount: z.number().int().gte(0), servedCount: z.number().int().gte(0), series: z.array(z.string()).nullable().optional(), instrumentTypes: z.array(z.string()).nullable().optional(), earliestApprovedDate: z.string().nullable().optional(), latestApprovedDate: z.string().nullable().optional()});

export const usJurisdictionCoverage = z.object({code: z.string(), name: z.string(), kind: z.enum(["federal", "state", "territory"]), sectionCount: z.number().int().gte(0), hasData: z.boolean(), statusBreakdown: z.record(z.string(), z.record(z.string(), z.number().int())).optional(), corpora: z.record(z.string(), z.number().int()).optional(), sessionLaws: sessionLawJurisdictionCoverage.nullable().optional()});

export const usProposedCorpusCoverage = z.object({corpusType: z.string(), name: z.string(), sectionCount: z.number().int().optional(), passageCount: z.number().int().optional(), held: z.boolean(), optInOnly: z.boolean().optional(), note: z.string()});

export const usStatutesCoverageResponse = z.object({corpusTypes: z.array(usCorpusTypeInfo), jurisdictions: z.array(usJurisdictionCoverage), totalSections: z.number().int(), totalPassages: z.number().int().optional(), stateCountWithData: z.number().int(), proposedCorpora: z.array(usProposedCorpusCoverage).optional(), measuredAt: z.string().nullable().optional()});

export const usFreeCallMeta = z.object({processingTimeMs: z.number(), creditsConsumed: z.number().optional()});

export const usStatutesCoverageEnvelope = z.object({data: usStatutesCoverageResponse, meta: usFreeCallMeta});

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

/**
 * Coverage matrix by jurisdiction and corpusType
 * 
 * GET /api/v1/us/statutes/coverage
 * 
 * Also accepts auth: ApiKeyHeader
 */
export const getApiApiV1UsStatutesCoverage = async (args: GetApiApiV1UsStatutesCoverageArgs, options?: GetApiApiV1UsStatutesCoverageOptions) => {
const {baseUrl = "https://api.vaquill.ai", fetch: fetchFn = fetch, ApiKeyAuth} = options ?? {}

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

if (args.state != null) query.append("state", String(args.state))
if (ApiKeyAuth !== undefined) headers.set('authorization', `Bearer ${ApiKeyAuth}`)

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