/**
 * Lexicon JSON generator — produces lexicon-azure.json with metadata for all resources.
 *
 * Output is a Record<string, LexiconEntry> keyed by generated class name.
 */

import type { ArmSchemaParseResult } from "../spec/parse";
import { armShortName } from "../spec/parse";
import type { NamingStrategy } from "./naming";
import type { PropertyConstraints } from "@intentius/chant/codegen/json-schema";
import {
  buildRegistry,
  serializeRegistry,
  type RegistryResource,
} from "@intentius/chant/codegen/generate-registry";

export interface LexiconEntry {
  resourceType: string;
  kind: "resource" | "property";
  lexicon: "azure";
  apiVersion?: string;
  attrs?: Record<string, string>;
  propertyConstraints?: Record<string, PropertyConstraints>;
  resourceLevelFields?: string[];
  tagging?: { taggable: boolean; tagOnCreate: boolean; tagUpdatable: boolean };
  createOnly?: string[];
  writeOnly?: string[];
  /**
   * ARM deployment scopes the resource can be declared at (#1545).
   * Absent means plain resource-group scope.
   */
  deployScopes?: string[];
}

/**
 * Curated create-only properties for common ARM resources.
 * ARM public schemas lack x-ms-mutability; this map captures
 * well-known immutable properties that cannot be changed after creation.
 */
const CURATED_CREATE_ONLY: Record<string, string[]> = {
  "Microsoft.Storage/storageAccounts": ["kind", "location"],
  "Microsoft.Compute/virtualMachines": ["location", "zones"],
  "Microsoft.Compute/virtualMachineScaleSets": ["location", "zones"],
  "Microsoft.Network/virtualNetworks": ["location"],
  "Microsoft.Network/networkSecurityGroups": ["location"],
  "Microsoft.Network/publicIPAddresses": ["location", "zones", "sku"],
  "Microsoft.Network/loadBalancers": ["location", "sku"],
  "Microsoft.Web/serverfarms": ["location"],
  "Microsoft.Web/sites": ["location"],
  "Microsoft.ContainerService/managedClusters": ["location", "dnsPrefix"],
  "Microsoft.ContainerRegistry/registries": ["location"],
  "Microsoft.Sql/servers": ["location"],
  "Microsoft.Sql/servers_databases": ["location"],
  "Microsoft.KeyVault/vaults": ["location"],
  "Microsoft.DocumentDB/databaseAccounts": ["location", "kind"],
  "Microsoft.ManagedIdentity/userAssignedIdentities": ["location"],
  "Microsoft.Network/dnsZones": ["location"],
  "Microsoft.Insights/components": ["location", "kind"],
  "Microsoft.OperationalInsights/workspaces": ["location"],
  "Microsoft.EventHub/namespaces": ["location", "sku"],
  "Microsoft.ServiceBus/namespaces": ["location", "sku"],
  "Microsoft.Cache/redis": ["location"],
  "Microsoft.Resources/deployments": ["location"],
};

/**
 * Generate the lexicon-azure.json content.
 */
export function generateLexiconJSON(
  results: ArmSchemaParseResult[],
  naming: NamingStrategy,
): string {
  // Adapt ArmSchemaParseResult[] to RegistryResource[]
  const registryResources: RegistryResource[] = results.map((r) => ({
    typeName: r.resource.typeName,
    attributes: r.resource.attributes,
    properties: r.resource.properties,
    propertyTypes: r.propertyTypes,
  }));

  const entries = buildRegistry<LexiconEntry>(registryResources, naming, {
    shortName: armShortName,
    buildEntry: (resource, _tsName, attrs, propConstraints) => {
      const armType = resource.typeName;
      const r = results.find((res) => res.resource.typeName === armType)!;

      const createOnly = CURATED_CREATE_ONLY[armType];

      return {
        resourceType: armType,
        kind: "resource" as const,
        lexicon: "azure" as const,
        apiVersion: r.resource.apiVersion,
        ...(attrs && { attrs }),
        ...(propConstraints && { propertyConstraints: propConstraints }),
        ...(r.resource.resourceLevelFields.length > 0 && {
          resourceLevelFields: r.resource.resourceLevelFields,
        }),
        ...(r.resource.tagging && { tagging: r.resource.tagging }),
        ...(createOnly && { createOnly }),
        ...(r.resource.deployScopes && { deployScopes: r.resource.deployScopes }),
      };
    },
    buildPropertyEntry: (resourceType, propertyType) => ({
      resourceType: `${resourceType}.${propertyType}`,
      kind: "property" as const,
      lexicon: "azure" as const,
    }),
  });

  return serializeRegistry(entries);
}