import {z} from 'zod'

export type KnowledgeGraphConfig = {auto_build?: boolean | undefined, exclude_path_prefixes?: Array<string> | undefined, exclude_categories?: Array<string> | undefined};

export type CurrencyPolicy = {default?: (string | null) | undefined};

export type LedgerPackageScope = {path_prefix?: (string | null) | undefined, paths?: Array<string> | undefined, selection_id?: (string | null) | undefined};

export type PreparedTableSelectionConfig = {path: string, component?: (string | null) | undefined};

export type PreparedIdentityNormalizationConfig = {kind?: 'identity' | undefined};

export type PreparedLeftPadNormalizationConfig = {kind?: 'left_pad' | undefined, width: number, fill: string};

export type PreparedKeyConfig = {left_column: string, right_column: string, left_normalization?: (PreparedIdentityNormalizationConfig | PreparedLeftPadNormalizationConfig) | undefined, right_normalization?: (PreparedIdentityNormalizationConfig | PreparedLeftPadNormalizationConfig) | undefined};

export type PreparedRelationshipConfig = {name: string, left: PreparedTableSelectionConfig, right: PreparedTableSelectionConfig, keys: Array<PreparedKeyConfig>, bucket_count: number};

export type LedgerUnderstandingConfig = {engine?: ('measured' | 'llm') | undefined, package_confirmation?: ('required' | 'automatic') | undefined, package_scope?: LedgerPackageScope | undefined, trigger?: ('auto' | 'always' | 'off') | undefined, gl_parsing?: boolean | undefined, prepared_relationships?: Array<PreparedRelationshipConfig> | undefined};

export type UpdateWorkspaceRequest = {name?: string | undefined, knowledge_graph?: (KnowledgeGraphConfig | null) | undefined, currency?: (CurrencyPolicy | null) | undefined, ledger_understanding?: (LedgerUnderstandingConfig | null) | undefined, global_context?: (string | null) | undefined};

export type PatchApiV1WorkspacesWorkspaceIdArgs = {workspace_id: string, body: UpdateWorkspaceRequest};

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

export const managedWorkspaceStorage = z.object({type: z.literal("managed").optional()});

export const tenantBucketWorkspaceStorage = z.object({type: z.literal("tenant_bucket").optional(), provider: z.literal("s3").optional(), location: z.string()});

export const externalWorkspaceStorage = z.object({type: z.literal("external").optional(), connection_id: z.string().nullable().optional(), provider: z.enum(["s3", "azure_blob"]).nullable().optional(), location: z.string().nullable().optional(), health: z.enum(["draft", "ready", "degraded", "revoked"]).nullable().optional()});

export const accessLabel = z.object({name: z.string().max(64), description: z.string()});

export const accessConfig = z.object({labels: z.array(accessLabel).optional(), default_label: z.string().nullable().optional()});

export const knowledgeGraphConfig = z.object({auto_build: z.boolean().optional(), exclude_path_prefixes: z.array(z.string().max(1024)).optional(), exclude_categories: z.array(z.string().max(1024)).optional()});

export const currencyPolicy = z.object({default: z.string().nullable().optional()});

export const ledgerPackageScope = z.object({path_prefix: z.string().nullable().optional(), paths: z.array(z.string()).optional(), selection_id: z.string().nullable().optional()});

export const preparedTableSelectionConfig = z.object({path: z.string().min(1), component: z.string().nullable().optional()});

export const preparedIdentityNormalizationConfig = z.object({kind: z.literal("identity").optional()});

export const preparedLeftPadNormalizationConfig = z.object({kind: z.literal("left_pad").optional(), width: z.number().int().gt(0).lte(1024), fill: z.string().min(1).max(1)});

export const preparedKeyConfig = z.object({left_column: z.string().min(1), right_column: z.string().min(1), left_normalization: z.discriminatedUnion("kind", [preparedIdentityNormalizationConfig, preparedLeftPadNormalizationConfig]).optional(), right_normalization: z.discriminatedUnion("kind", [preparedIdentityNormalizationConfig, preparedLeftPadNormalizationConfig]).optional()});

export const preparedRelationshipConfig = z.object({name: z.string().min(1).max(100), left: preparedTableSelectionConfig, right: preparedTableSelectionConfig, keys: z.array(preparedKeyConfig), bucket_count: z.number().int().gt(0).lte(1024)});

export const ledgerUnderstandingConfig = z.object({engine: z.enum(["measured", "llm"]).optional(), package_confirmation: z.enum(["required", "automatic"]).optional(), package_scope: ledgerPackageScope.optional(), trigger: z.enum(["auto", "always", "off"]).optional(), gl_parsing: z.boolean().optional(), prepared_relationships: z.array(preparedRelationshipConfig).optional()});

export const workspaceStatus = z.enum(["active", "deleting"]);

export const workspacePurpose = z.literal("perception_playground");

export const workspace = z.object({storage: z.discriminatedUnion("type", [managedWorkspaceStorage, tenantBucketWorkspaceStorage, externalWorkspaceStorage]).optional(), workspace_id: z.string(), owner_user_id: z.string().nullable(), share_count: z.number().int().gte(0), name: z.string(), domain_id: z.string(), domain_slug: z.string(), domain_version: z.number().int(), access: accessConfig, knowledge_graph: knowledgeGraphConfig, currency: currencyPolicy, ledger_understanding: ledgerUnderstandingConfig.optional(), global_context: z.string().optional(), config_version: z.number().int(), status: workspaceStatus, purpose: workspacePurpose.nullable().optional(), created_at: z.string(), updated_at: z.string()});

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

/**
 * update_workspace
 * 
 * PATCH /v1/workspaces/{workspace_id}
 */
export const patchApiV1WorkspacesWorkspaceId = async (args: PatchApiV1WorkspacesWorkspaceIdArgs, options: PatchApiV1WorkspacesWorkspaceIdOptions) => {
const {baseUrl, 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 (APIKeyHeader !== undefined) headers.set("X-API-Key", APIKeyHeader)

const url = `${baseUrl.replace(/\/+$/, '')}/v1/workspaces/${encodeURIComponent(String(args.workspace_id))}${query.size > 0 ? `?${query}` : ''}`

const res = await fetchFn(url, {method: "PATCH", 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(workspace, await res.json())
};