import {z} from 'zod'

export type GetApiPotsPotIdConstitutionWithEdgesArgs = {pot_id: string};

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

export const potScoreLevel = z.enum(["CONSTITUTION", "VERIFIED", "SUPPORTED", "INFERRED", "HYPOTHESIS", "SPECULATION"]);

export const axiomKind = z.enum(["fact", "rule", "scope"]);

export const constitutionAxiom = z.object({id: z.string(), pot_id: z.string(), workspace_id: z.string(), content: z.string().min(1), pot_score: z.number().optional(), level: potScoreLevel.optional(), created_by: z.string(), approved_by: z.string(), rationale: z.string().nullable().optional(), kind: axiomKind.optional(), domain: z.string().nullable().optional(), tags: z.array(z.string()).optional(), created_at: z.string().optional(), updated_at: z.string().optional()}).and(z.record(z.string(), z.unknown()));

export const edgeType = z.enum(["supports", "contradicts", "resolved_contradiction", "refines", "derives_from", "is_part_of", "extends", "cites", "related_to", "supersedes", "restates"]);

export const edge = z.object({id: z.string(), pot_id: z.string(), workspace_id: z.string(), from_id: z.string(), to_id: z.string(), type: edgeType, weight: z.number().gte(0).lte(1), rationale: z.string().optional(), detected_by: z.string().optional(), companion_of: z.string().nullable().optional(), episode_id: z.string().nullable().optional(), confidence: z.number().gte(0).lte(1).optional(), judge: z.string().max(200).nullable().optional(), partial: z.boolean().nullable().optional(), shared_figures: z.number().int().gte(0).nullable().optional(), similarity: z.number().gte(0).lte(1).nullable().optional(), validation_status: z.string().optional(), decided_by: z.string().nullable().optional(), decided_at: z.string().nullable().optional(), resolution_type: z.string().nullable().optional(), resolved_by_name: z.string().nullable().optional(), resolved_by_role: z.string().nullable().optional(), resolved_by_contributor_id: z.string().nullable().optional(), resolved_at: z.string().nullable().optional(), resolution_notes: z.string().nullable().optional(), resolution_score_shifts: z.record(z.string(), z.number()).nullable().optional(), kind: z.enum(["express", "tacit", "partial", "hierarchy", "suspension", "extension", "manual"]).nullable().optional(), basis: z.string().nullable().optional(), scope_section: z.string().max(300).nullable().optional(), effective_at: z.string().nullable().optional(), resume_at: z.string().nullable().optional(), applied_at: z.string().nullable().optional(), axiom_conflict: z.array(z.string()).nullable().optional(), axiom_hold: z.enum(["conflict", "unverified"]).nullable().optional(), created_at: z.string().nullable().optional(), updated_at: z.string().nullable().optional()}).and(z.record(z.string(), z.unknown()));

export const provenance = z.object({type: z.string(), sources: z.array(z.record(z.string(), z.unknown())).optional(), derivation: z.record(z.string(), z.unknown()).nullable().optional(), profile_id: z.string().nullable().optional(), profile_version: z.number().int().gte(1).nullable().optional()});

export const classification = z.object({domain: z.string().optional(), subdomain: z.string().nullable().optional(), tags: z.array(z.string()).optional(), entity_mentions: z.array(z.string()).optional()});

export const documentLocation = z.object({page_number: z.number().int().gte(1).nullable().optional(), section_path: z.string().nullable().optional(), source_text: z.string().max(500).nullable().optional(), sheet_name: z.string().nullable().optional(), row_number: z.number().int().gte(1).nullable().optional(), cell_ref: z.string().nullable().optional()});

export const factPii = z.object({applied: z.boolean(), mode: z.enum(["off", "anonymize", "block"]), entities: z.record(z.string(), z.number().int()).optional(), coverage: z.enum(["full", "partial", "none"]).optional(), language: z.string().nullable().optional(), unmasked_content: z.string().nullable().optional()});

export const factWarning = z.object({pattern: z.string(), message: z.string(), detail: z.string().nullable().optional(), probability: z.number().nullable().optional()});

export const edgeTypeCounts = z.object({in: z.number().int().gte(0).optional(), out: z.number().int().gte(0).optional()});

export const edgeSummary = z.object({total_in: z.number().int().gte(0).optional(), total_out: z.number().int().gte(0).optional(), by_type: z.record(z.string(), edgeTypeCounts).optional()});

export const factExtended = z.object({id: z.string(), pot_id: z.string(), workspace_id: z.string().nullable().optional(), content: z.string().min(1), title: z.string().max(80).nullable().optional(), content_hash: z.string().nullable().optional(), pot_score: z.number().gte(0).lte(1), original_pot_score: z.number().gte(0).lte(1).nullable().optional(), level: potScoreLevel, embedding_model: z.string().nullable().optional(), provenance: provenance, classification: classification.optional(), status: z.string().optional(), validation_status: z.string().optional(), created_at: z.string().optional(), updated_at: z.string().optional(), valid_from: z.string().nullable().optional(), valid_from_source: z.enum(["document_effective", "document_date", "contributor", "imported", "curator"]).nullable().optional(), document_id: z.string().nullable().optional(), language: z.string().nullable().optional(), location: documentLocation.nullable().optional(), scope_inferred: z.enum(["own", "third_party", "mixed_context", "unknown"]).nullable().optional(), scope_confidence: z.number().gte(0).lte(1).nullable().optional(), scope_reason: z.string().max(500).nullable().optional(), context_attachment_ids_used: z.array(z.string()).nullable().optional(), pii: factPii.nullable().optional(), version: z.number().int().optional(), flags: z.array(z.string()).optional(), notes: z.string().nullable().optional(), warnings: z.array(factWarning).optional(), edge_summary: edgeSummary.optional()}).and(z.record(z.string(), z.unknown()));

export const axiomWithEdges = z.object({axiom: constitutionAxiom, edges_out: z.array(edge).optional(), edges_in: z.array(edge).optional(), connected_facts: z.array(factExtended).optional(), total_edges: z.number().int().optional()});

export const constitutionWithEdgesResponse = z.object({pot_id: z.string(), axioms: z.array(axiomWithEdges), total_axioms: z.number().int(), total_edges: z.number().int(), total_edges_in_pot: z.number().int().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
};

/**
 * Get constitution axioms with their edges and connected facts
 * 
 * GET /pots/{pot_id}/constitution/with-edges
 */
export const getApiPotsPotIdConstitutionWithEdges = async (args: GetApiPotsPotIdConstitutionWithEdgesArgs, options: GetApiPotsPotIdConstitutionWithEdgesOptions) => {
const {baseUrl, fetch: fetchFn = fetch} = options ?? {}

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

const url = `${baseUrl.replace(/\/+$/, '')}/pots/${encodeURIComponent(String(args.pot_id))}/constitution/with-edges${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(constitutionWithEdgesResponse, await res.json())
};