@kurotako/gen-angular 0.1.0 → 0.2.0

package/dist/index.js

@@ -99,6 +99,18 @@ function zodSymbol(zod, namespace, entity, role) {
 function zodModule(zod, namespace, entity) {
   return zodEntity(zod, namespace, entity).module;
 }
+function zodRefType(zod, namespace, ref) {
+  const key = entityKey(namespace, ref);
+  const entry = zod.entities[key];
+  if (entry === void 0) {
+    throw new MissingZodSymbolError(key, "type");
+  }
+  const id = entry.symbols.type;
+  if (id === void 0) {
+    throw new MissingZodSymbolError(key, "type");
+  }
+  return { typeName: id, module: entry.module };
+}
 function zodExtra(zod) {
   return zod.extra;
 }
@@ -225,6 +237,36 @@ function importStmt(spec, names, typeOnly) {
 
 // src/render/controls.ts
 import { resolveEnum } from "@kurotako/ir";
+
+// src/render/unions.ts
+import { flattenUnion } from "@kurotako/ir";
+function discriminatedUnion(field, source) {
+  const type = field.type;
+  if (type.kind !== "union" || type.discriminator?.mapping === void 0) {
+    return void 0;
+  }
+  if (field.list || field.nullable) {
+    return void 0;
+  }
+  const variants = [];
+  for (const [value, ref] of Object.entries(type.discriminator.mapping)) {
+    if (source.entities[ref] === void 0) {
+      return void 0;
+    }
+    variants.push({ value, entity: ref });
+  }
+  return variants.length >= 2 ? { discriminator: type.discriminator.propertyName, variants } : void 0;
+}
+function refNames(type, into = /* @__PURE__ */ new Set()) {
+  if (type.kind === "ref") {
+    into.add(type.ref);
+  } else if (type.kind === "union") {
+    for (const variant of type.variants) {
+      refNames(variant, into);
+    }
+  }
+  return into;
+}
 var SCALAR_BASE = {
   string: "string",
   uuid: "string",
@@ -238,18 +280,71 @@ var SCALAR_BASE = {
   datetime: "Date",
   json: "unknown"
 };
-function baseType(field, zodEnumTypeName) {
+function variantType(type, refTypeName, enumTypeName, cyclicRefs) {
+  switch (type.kind) {
+    case "scalar":
+      return SCALAR_BASE[type.scalar] ?? "unknown";
+    case "enum":
+      return enumTypeName(type.ref);
+    case "ref":
+      return refTypeName(type.ref);
+    case "unknown":
+      return "unknown";
+    case "union":
+      return unionType(type, refTypeName, enumTypeName, cyclicRefs).text;
+  }
+}
+function unionType(type, refTypeName, enumTypeName, cyclicRefs = /* @__PURE__ */ new Set()) {
+  const variants = flattenUnion(type);
+  if (variants.length === 0) {
+    return { text: "unknown", recursive: false };
+  }
+  const recursive = [...refNames(type)].some((ref) => cyclicRefs.has(ref));
+  if (recursive) {
+    return { text: "unknown", recursive: true };
+  }
+  const text = variants.map((variant) => {
+    const inner = variantType(variant, refTypeName, enumTypeName, cyclicRefs);
+    return variant.kind === "union" ? `(${inner})` : inner;
+  }).join(" | ");
+  return { text, recursive: false };
+}
+
+// src/render/controls.ts
+var SCALAR_BASE2 = {
+  string: "string",
+  uuid: "string",
+  decimal: "string",
+  bytes: "string",
+  int: "number",
+  float: "number",
+  bigint: "bigint",
+  boolean: "boolean",
+  date: "Date",
+  datetime: "Date",
+  json: "unknown"
+};
+function baseType(field, resolvers) {
   switch (field.type.kind) {
     case "scalar":
-      return SCALAR_BASE[field.type.scalar];
+      return SCALAR_BASE2[field.type.scalar];
     case "enum":
-      return zodEnumTypeName(field.type.ref);
+      return resolvers.enumTypeName(field.type.ref);
     case "unknown":
       return "unknown";
+    case "ref":
+      return resolvers.refTypeName(field.type.ref);
+    case "union":
+      return unionType(
+        field.type,
+        resolvers.refTypeName,
+        resolvers.enumTypeName,
+        resolvers.cyclicRefs
+      ).text;
   }
 }
-function controlType(field, zodEnumTypeName) {
-  let t = baseType(field, zodEnumTypeName);
+function controlType(field, resolvers) {
+  let t = baseType(field, resolvers);
   if (field.list) {
     t = `${t}[]`;
   }
@@ -305,12 +400,18 @@ function controlExpr(field, typeArg, sourceExpr) {
   if (field.nullable) {
     return `new FormControl<${typeArg}>(${sourceExpr})`;
   }
+  if (field.type.kind === "union") {
+    return `new FormControl<${typeArg}>((${sourceExpr}) as ${typeArg}) /* union: validated by zodValidator(schema) */`;
+  }
+  if (field.type.kind === "ref") {
+    return `new FormControl<${typeArg}>((${sourceExpr}) as ${typeArg})`;
+  }
   return `new FormControl(${sourceExpr}, { nonNullable: true })`;
 }
-function fieldControlEntry(field, zodEnumTypeName) {
+function fieldControlEntry(field, resolvers) {
   return {
     name: field.name,
-    fullType: `FormControl<${controlType(field, zodEnumTypeName)}>`
+    fullType: `FormControl<${controlType(field, resolvers)}>`
   };
 }
 function controlsInterface(interfaceName, entries) {
@@ -365,10 +466,42 @@ function variantFields(entity, variant) {
 }
 
 // src/render/reactive.ts
+function discriminatedControlEntry(union, fieldName, variant, family) {
+  const lit = union.variants.map((v2) => JSON.stringify(v2.value)).join(" | ");
+  const subs = union.variants.map(
+    (v2) => `${JSON.stringify(v2.value)}: FormGroup<${controlsTypeName(v2.entity, variant, family)}>`
+  ).join("; ");
+  return {
+    name: fieldName,
+    fullType: `FormGroup<{ ${JSON.stringify(union.discriminator)}: FormControl<${lit}>; ${subs} }>`
+  };
+}
+function discriminatedFactoryExpr(union, fieldName, variant, deep, zod, namespace, injectedFactories) {
+  const lit = union.variants.map((v2) => JSON.stringify(v2.value)).join(" | ");
+  const first = JSON.stringify(union.variants[0]?.value ?? "");
+  const method = factoryMethod(variant);
+  const subLines = union.variants.map((v2) => {
+    const param = injectedFactories.get(v2.entity) ?? `${lowerFirst2(v2.entity)}FormFactory`;
+    if (variant === "Create") {
+      return `    ${JSON.stringify(v2.value)}: this.${param}.${method}(),`;
+    }
+    const role = deep ? "updateDeepType" : "updateType";
+    const dto = zodSymbol(zod, namespace, v2.entity, role);
+    return `    ${JSON.stringify(v2.value)}: this.${param}.${method}(value.${fieldName} as unknown as ${dto}),`;
+  }).join("\n");
+  return `(() => {
+  const g = new FormGroup({
+    ${JSON.stringify(union.discriminator)}: new FormControl<${lit}>(${first}, { nonNullable: true }),
+${subLines}
+  });
+  switchDiscriminatedGroup(g, ${JSON.stringify(union.discriminator)});
+  return g;
+})()`;
+}
 function lowerFirst2(s) {
   return s.length === 0 ? s : s.charAt(0).toLowerCase() + s.slice(1);
 }
-function reactiveEntity(entity, namespace, source, options, zod, imports, logger) {
+function reactiveEntity(entity, namespace, source, options, zod, imports, cyclicRefs, logger) {
   const deep = options.relations === "deep";
   const enumZero = enumZeroFromSource(source, entity);
   imports.value("@angular/core", "Injectable");
@@ -381,6 +514,7 @@ function reactiveEntity(entity, namespace, source, options, zod, imports, logger
   };
   const blocks = [];
   const injectedFactories = /* @__PURE__ */ new Map();
+  const warnedUnionFields = /* @__PURE__ */ new Set();
   for (const variant of ["Create", "Update"]) {
     const family = deep ? "Deep" : "";
     const schemaRole = deep ? variant === "Create" ? "createDeepSchema" : "updateDeepSchema" : variant === "Create" ? "createSchema" : "updateSchema";
@@ -393,8 +527,61 @@ function reactiveEntity(entity, namespace, source, options, zod, imports, logger
     imports.value(`${namespace}/angular/zod-forms.runtime`, "zodValidator");
     const interfaceName = controlsTypeName(entity.name, variant, family);
     const formType = formTypeName(entity.name, variant, family);
+    const resolvers = {
+      enumTypeName: zodEnumTypeName,
+      refTypeName: (ref) => {
+        const r = zodRefType(zod, namespace, ref);
+        imports.type(r.module, r.typeName);
+        return r.typeName;
+      },
+      cyclicRefs
+    };
     const fieldEntries = variantFields(entity, variant).map(
-      (field) => fieldControlEntry(field, zodEnumTypeName)
+      (field) => {
+        const union = field.type.kind === "union" ? discriminatedUnion(field, source) : void 0;
+        if (union !== void 0) {
+          imports.value(
+            `${namespace}/angular/zod-forms.runtime`,
+            "switchDiscriminatedGroup"
+          );
+          for (const v2 of union.variants) {
+            const targetModule = `${namespace}/angular/${v2.entity}.form`;
+            imports.type(
+              targetModule,
+              controlsTypeName(v2.entity, variant, family)
+            );
+            if (!injectedFactories.has(v2.entity)) {
+              injectedFactories.set(
+                v2.entity,
+                `${lowerFirst2(v2.entity)}FormFactory`
+              );
+              imports.value(targetModule, factoryName(v2.entity));
+            }
+            if (variant === "Update") {
+              const role = deep ? "updateDeepType" : "updateType";
+              imports.type(
+                zodModule(zod, namespace, v2.entity),
+                zodSymbol(zod, namespace, v2.entity, role)
+              );
+            }
+          }
+          return discriminatedControlEntry(union, field.name, variant, family);
+        }
+        if (field.type.kind === "union" && !warnedUnionFields.has(field.name)) {
+          warnedUnionFields.add(field.name);
+          const recursive = unionType(
+            field.type,
+            resolvers.refTypeName,
+            resolvers.enumTypeName,
+            cyclicRefs
+          ).recursive;
+          const cause = recursive ? "a recursive ref branch (chains into a cycle)" : "no discriminator mapping, an alias target, or a list/nullable union";
+          logger?.warn(
+            `gen-angular: union field '${entity.name}.${field.name}' has no usable discriminated sub-form (${cause}); emitting a free FormControl${recursive ? "<unknown>" : ""} validated by zodValidator`
+          );
+        }
+        return fieldControlEntry(field, resolvers);
+      }
     );
     const relations = deep ? deepRelations(entity, variant, namespace, logger) : [];
     const manyRelations = relations.filter((r) => r.many);
@@ -441,15 +628,17 @@ function reactiveEntity(entity, namespace, source, options, zod, imports, logger
     renderFactoryClass(
       entity,
       namespace,
+      source,
       options,
       zod,
       injectedFactories,
-      enumZero
+      enumZero,
+      cyclicRefs
     )
   );
   return blocks.join("\n\n");
 }
-function renderFactoryClass(entity, namespace, options, zod, injectedFactories, enumZero) {
+function renderFactoryClass(entity, namespace, source, options, zod, injectedFactories, enumZero, cyclicRefs) {
   const deep = options.relations === "deep";
   const className = factoryName(entity.name);
   const ctorParams = [...injectedFactories.entries()].map(
@@ -464,11 +653,13 @@ function renderFactoryClass(entity, namespace, options, zod, injectedFactories,
       renderFactoryMethod(
         entity,
         namespace,
+        source,
         variant,
         options,
         zod,
         injectedFactories,
-        enumZero
+        enumZero,
+        cyclicRefs
       )
     );
   }
@@ -506,7 +697,7 @@ export class ${className} {
 ${body}
 }`;
 }
-function renderFactoryMethod(entity, namespace, variant, options, zod, injectedFactories, enumZero) {
+function renderFactoryMethod(entity, namespace, source, variant, options, zod, injectedFactories, enumZero, cyclicRefs) {
   const deep = options.relations === "deep";
   const family = deep ? "Deep" : "";
   const typeRole = deep && variant === "Create" ? "createDeepType" : deep && variant === "Update" ? "updateDeepType" : variant === "Create" ? "createType" : "updateType";
@@ -516,13 +707,29 @@ function renderFactoryMethod(entity, namespace, variant, options, zod, injectedF
   const interfaceName = controlsTypeName(entity.name, variant, family);
   const formType = formTypeName(entity.name, variant, family);
   const methodName = factoryMethod(variant);
-  const zodEnumTypeName = (ref) => ref;
+  const resolvers = {
+    enumTypeName: (ref) => ref,
+    refTypeName: (ref) => zodRefType(zod, namespace, ref).typeName,
+    cyclicRefs
+  };
   const fields = variantFields(entity, variant);
   const lines = fields.map((field) => {
-    const typeArg = controlType(field, zodEnumTypeName);
+    const union = field.type.kind === "union" ? discriminatedUnion(field, source) : void 0;
+    if (union !== void 0) {
+      return `    ${field.name}: ${discriminatedFactoryExpr(
+        union,
+        field.name,
+        variant,
+        deep,
+        zod,
+        namespace,
+        injectedFactories
+      )},`;
+    }
+    const typeArg = controlType(field, resolvers);
     const accessor = variant === "Create" ? `init?.${field.name}` : `value.${field.name}`;
-    const source = `${accessor} ?? ${initExpr(field, enumZero)}`;
-    return `    ${field.name}: ${controlExpr(field, typeArg, source)},`;
+    const seed = `${accessor} ?? ${initExpr(field, enumZero)}`;
+    return `    ${field.name}: ${controlExpr(field, typeArg, seed)},`;
   });
   const relations = deep ? deepRelations(entity, variant, namespace, void 0) : [];
   const relationLines = relations.map((r) => {
@@ -583,9 +790,12 @@ function signalEntity(entity, namespace, source, options, zod, imports, logger)
     imports.value(module, schemaId);
     imports.value(`${namespace}/angular/zod-forms.runtime`, "zodTreeValidate");
     const fields = variantFields(entity, variant);
-    const fieldLines = fields.map(
-      (field) => `    ${field.name}: init?.${field.name} ?? ${initExpr(field, enumZero)},`
-    );
+    const fieldLines = fields.map((field) => {
+      if (field.type.kind === "ref" || field.type.kind === "union") {
+        return `    ${field.name}: (init?.${field.name} ?? undefined) as ${typeId}[${JSON.stringify(field.name)}],`;
+      }
+      return `    ${field.name}: init?.${field.name} ?? ${initExpr(field, enumZero)},`;
+    });
     const relations = deep ? deepRelations(entity, variant, namespace, logger) : [];
     const relationLines = relations.map((rel) => {
       if (rel.many) {
@@ -625,12 +835,21 @@ ${modelBody}
 }
 
 // src/emit/entity.ts
-function emitEntity(entity, namespace, source, options, zod, logger) {
+function emitEntity(entity, namespace, source, options, zod, cyclicRefs, logger) {
   const imports = new ImportsRecorder();
   const blocks = [];
   if (options.forms.includes("reactive")) {
     blocks.push(
-      reactiveEntity(entity, namespace, source, options, zod, imports, logger)
+      reactiveEntity(
+        entity,
+        namespace,
+        source,
+        options,
+        zod,
+        imports,
+        cyclicRefs,
+        logger
+      )
     );
   }
   if (options.forms.includes("signal")) {
@@ -724,6 +943,53 @@ function collectControls(control: AbstractControl): AbstractControl[] {
   }
   return out;
 }`;
+var SWITCH_DISCRIMINATED_GROUP = `/**
+ * Wire a discriminated-union sub-FormGroup: enable only the sub-group named by
+ * the discriminator control's value (disable the rest), keep it in sync on
+ * valueChanges, and rewrite the group's getRawValue() to the flat
+ * active-variant shape a z.discriminatedUnion schema can parse.
+ */
+export function switchDiscriminatedGroup(
+  group: AbstractControl,
+  discriminator: string,
+): void {
+  const controls = (group as unknown as {
+    controls: Record<string, AbstractControl>;
+  }).controls;
+  const selector = controls[discriminator];
+  if (selector === undefined) {
+    return;
+  }
+  const variantKeys = Object.keys(controls).filter(
+    (key) => key !== discriminator,
+  );
+
+  const apply = (active: unknown): void => {
+    for (const key of variantKeys) {
+      const sub = controls[key];
+      if (sub === undefined) {
+        continue;
+      }
+      if (key === active) {
+        sub.enable({ emitEvent: false });
+      } else {
+        sub.disable({ emitEvent: false });
+      }
+    }
+  };
+
+  apply(selector.value);
+  selector.valueChanges.subscribe(apply);
+
+  (group as { getRawValue: () => unknown }).getRawValue = () => {
+    const active = controls[String(selector.value)];
+    const variant =
+      active === undefined || active === selector
+        ? {}
+        : (active.getRawValue() as Record<string, unknown>);
+    return { ...variant, [discriminator]: selector.value };
+  };
+}`;
 var ZOD_TREE_VALIDATE = `export function zodTreeValidate<T>(
   path: SchemaPath<T>,
   schema: ZodType<T>,
@@ -773,7 +1039,7 @@ function emitRuntime(_source, options) {
   imports.push("import type { ZodType } from 'zod';");
   const blocks = [];
   if (reactive) {
-    blocks.push(ZOD_VALIDATOR);
+    blocks.push(ZOD_VALIDATOR, SWITCH_DISCRIMINATED_GROUP);
   }
   if (signal) {
     blocks.push(ZOD_TREE_VALIDATE);
@@ -810,6 +1076,13 @@ var angularGenerator = defineGenerator({
     for (const [namespace, source] of Object.entries(ctx.ir.sources)) {
       const prefix = `${namespace}/angular`;
       const entities = Object.values(source.entities);
+      const cyclicRefs = /* @__PURE__ */ new Set();
+      for (const key of ctx.cycles) {
+        const dot = key.indexOf(".");
+        if (dot > 0 && key.slice(0, dot) === namespace) {
+          cyclicRefs.add(key.slice(dot + 1));
+        }
+      }
       if (entities.length > 0 && options.forms.length > 0) {
         files.push({
           path: `${prefix}/zod-forms.runtime.ts`,
@@ -825,6 +1098,7 @@ var angularGenerator = defineGenerator({
             source,
             options,
             zod,
+            cyclicRefs,
             ctx.logger
           )
         });