@danielhritcu/zenstack-custom 1.1.0 → 1.2.1

package/dist/cli.cjs

@@ -940,6 +940,32 @@ __name(sortTypedJsonFields, "sortTypedJsonFields");
 
 // src/codegen/emit.ts
 var DB_ATTR_RE = /@db\.(\w+)(?:\((\d+)\))?/;
+function isUuidField(t, fieldName) {
+  const col = t.table[fieldName];
+  if (!col) return false;
+  return col.columnType === "PgUUID";
+}
+__name(isUuidField, "isUuidField");
+function isPkField(t, fieldName) {
+  const col = t.table[fieldName];
+  if (!col) return false;
+  if (col.primary) return true;
+  const sqlName = col.name;
+  return t.cfg.primaryKeys.some((pk) => pk.columns.some((c) => c.name === sqlName));
+}
+__name(isPkField, "isPkField");
+function findFkTarget(t, fieldName, fkRelations) {
+  const modelFks = fkRelations[t.modelName];
+  if (!modelFks) return null;
+  for (const fkInfo of Object.values(modelFks)) {
+    const idx = fkInfo.fields.indexOf(fieldName);
+    if (idx >= 0) {
+      return fkInfo.targetExportName;
+    }
+  }
+  return null;
+}
+__name(findFkTarget, "findFkTarget");
 function buildDerivedEnumName(modelName, fieldName) {
   return `${toPascalCase(modelName)}${toPascalCase(fieldName)}`;
 }
@@ -1435,8 +1461,9 @@ function emitSchemaTs({ tables, enumMap, modelRelations, inverseRelations, searc
   return out.join("\n");
 }
 __name(emitSchemaTs, "emitSchemaTs");
-function emitModelsTs(tables, typedJsonFields = []) {
+function emitModelsTs(tables, typedJsonFields = [], branding) {
   const jsonFieldAliases = createJsonFieldAliasMap(typedJsonFields);
+  const hasBranding = branding && (Object.keys(branding.brandedFields).length > 0 || tables.some((t) => getEntityFieldNames(t).some((f) => isUuidField(t, f) && isPkField(t, f))));
   const out = [
     ...emitGeneratedFileBanner()
   ];
@@ -1444,6 +1471,10 @@ function emitModelsTs(tables, typedJsonFields = []) {
   out.push('import type * as $ from "@zenstackhq/orm";');
   out.push('import type * as Json from "./json";');
   out.push('import { type SchemaType as Schema } from "./schema";');
+  if (hasBranding) {
+    out.push("");
+    out.push("declare const __brand: unique symbol;");
+  }
   out.push("");
   for (const t of tables) {
     out.push(`export interface ${t.typeName} extends $.ModelResult<Schema, ${esc(t.modelName)}> {}`);
@@ -1454,7 +1485,12 @@ function emitModelsTs(tables, typedJsonFields = []) {
       if (jsonAlias) {
         out.push(`  export interface ${exportName} extends Json.${jsonAlias} {}`);
       } else {
-        out.push(`  export type ${exportName} = ${t.typeName}[${esc(fieldName)}];`);
+        const brandedLine = branding ? emitBrandedFieldLine(t, fieldName, exportName, branding, "  ") : null;
+        if (brandedLine) {
+          out.push(...brandedLine);
+        } else {
+          out.push(`  export type ${exportName} = ${t.typeName}[${esc(fieldName)}];`);
+        }
       }
     }
     out.push("}");
@@ -1463,6 +1499,33 @@ function emitModelsTs(tables, typedJsonFields = []) {
   return out.join("\n");
 }
 __name(emitModelsTs, "emitModelsTs");
+function emitBrandedFieldLine(t, fieldName, exportName, branding, indent) {
+  const customBranded = branding.brandedFields[t.modelName]?.[fieldName];
+  if (isUuidField(t, fieldName) && isPkField(t, fieldName)) {
+    const brand = `${t.modelName}.${fieldName}`;
+    return [
+      `${indent}export type ${exportName} = string & { [__brand]: '${brand}' };`,
+      `${indent}export const ${exportName} = (id: string): ${exportName} => id as ${exportName};`
+    ];
+  }
+  if (isUuidField(t, fieldName)) {
+    const targetExportName = findFkTarget(t, fieldName, branding.fkRelations);
+    if (targetExportName) {
+      return [
+        `${indent}export type ${exportName} = ${targetExportName}.Id;`
+      ];
+    }
+  }
+  if (customBranded) {
+    const brand = `${t.modelName}.${fieldName}`;
+    return [
+      `${indent}export type ${exportName} = string & { [__brand]: '${brand}' };`,
+      `${indent}export const ${exportName} = (id: string): ${exportName} => id as ${exportName};`
+    ];
+  }
+  return null;
+}
+__name(emitBrandedFieldLine, "emitBrandedFieldLine");
 function emitViewsTs(views, typedJsonFields = []) {
   const jsonFieldAliases = createJsonFieldAliasMap(typedJsonFields);
   const out = [
@@ -1528,12 +1591,13 @@ function emitInputTs(tables) {
   return out.join("\n");
 }
 __name(emitInputTs, "emitInputTs");
-function emitIndexTs({ tables, views, enumMap, typeDefs, typedJsonFields }) {
+function emitIndexTs({ tables, views, enumMap, typeDefs, typedJsonFields, branding }) {
   const allEnums = new Map([
     ...enumMap,
     ...collectDerivedEnums(tables, enumMap)
   ]);
   const jsonFieldAliases = createJsonFieldAliasMap(typedJsonFields);
+  const hasBranding = branding && (Object.keys(branding.brandedFields).length > 0 || tables.some((t) => getEntityFieldNames(t).some((f) => isUuidField(t, f) && isPkField(t, f))));
   const out = [
     ...emitGeneratedFileBanner()
   ];
@@ -1541,6 +1605,10 @@ function emitIndexTs({ tables, views, enumMap, typeDefs, typedJsonFields }) {
   out.push('import type * as $ from "@zenstackhq/orm";');
   out.push('import type * as RawJson from "../../supabase/schema/types";');
   out.push('import { type SchemaType as Schema } from "./schema";');
+  if (hasBranding) {
+    out.push("");
+    out.push("declare const __brand: unique symbol;");
+  }
   out.push("");
   out.push("export namespace Json {");
   out.push(...emitJsonNamespaceBlock(typeDefs, typedJsonFields, "  "));
@@ -1555,7 +1623,7 @@ function emitIndexTs({ tables, views, enumMap, typeDefs, typedJsonFields }) {
   out.push("");
   out.push("export namespace Table {");
   for (const table of tables) {
-    out.push(...emitEntityNamespaceBlock(table, table.modelName, jsonFieldAliases, "  "));
+    out.push(...emitEntityNamespaceBlock(table, table.modelName, jsonFieldAliases, "  ", branding));
     out.push("");
   }
   out.push("}");
@@ -1714,7 +1782,7 @@ function createJsonFieldAliasMap(typedJsonFields) {
   ]));
 }
 __name(createJsonFieldAliasMap, "createJsonFieldAliasMap");
-function emitEntityNamespaceBlock(entity, modelName, jsonFieldAliases, indent = "") {
+function emitEntityNamespaceBlock(entity, modelName, jsonFieldAliases, indent = "", branding) {
   const out = [];
   out.push(`${indent}export interface ${entity.typeName} extends $.ModelResult<Schema, ${esc(modelName)}> {}`);
   out.push(`${indent}export namespace ${entity.typeName} {`);
@@ -1724,7 +1792,13 @@ function emitEntityNamespaceBlock(entity, modelName, jsonFieldAliases, indent =
     if (jsonAlias) {
       out.push(`${indent}  export interface ${exportName} extends Json.${jsonAlias} {}`);
     } else {
-      out.push(`${indent}  export type ${exportName} = ${entity.typeName}[${esc(fieldName)}];`);
+      const isTable = "table" in entity;
+      const brandedLine = isTable && branding ? emitBrandedFieldLine(entity, fieldName, exportName, branding, `${indent}  `) : null;
+      if (brandedLine) {
+        out.push(...brandedLine);
+      } else {
+        out.push(`${indent}  export type ${exportName} = ${entity.typeName}[${esc(fieldName)}];`);
+      }
     }
   }
   out.push(`${indent}}`);
@@ -2273,6 +2347,7 @@ function resolveConfig(config) {
   const derivedFieldsMap = {};
   const validationSchemas = {};
   const fkRelations = {};
+  const brandedFields = {};
   for (const [derivedName, desc] of Object.entries(derived)) {
     const baseExportName = sqlNameToExportName(tables, desc.baseName);
     if (!derivedModels[baseExportName]) {
@@ -2405,6 +2480,9 @@ function resolveConfig(config) {
     if (modelConfig.validate) {
       validationSchemas[modelName] = modelConfig.validate;
     }
+    if (modelConfig.branded) {
+      brandedFields[modelName] = modelConfig.branded;
+    }
     if (modelConfig.search) {
       const modelSearchDefaults = {};
       for (const [profileName, profile] of Object.entries(modelConfig.search)) {
@@ -2424,7 +2502,8 @@ function resolveConfig(config) {
     modelScopes,
     derivedFields: derivedFieldsMap,
     validationSchemas,
-    fkRelations
+    fkRelations,
+    brandedFields
   };
 }
 __name(resolveConfig, "resolveConfig");
@@ -2458,12 +2537,17 @@ async function runCodegen(config) {
     fieldToTypeDef,
     views
   });
+  const branding = {
+    fkRelations: resolved.fkRelations,
+    brandedFields: resolved.brandedFields
+  };
   const indexContent = emitIndexTs({
     tables,
     views,
     enumMap,
     typeDefs,
-    typedJsonFields
+    typedJsonFields,
+    branding
   });
   const compatContent = emitCompatTs({
     tables,
@@ -2472,7 +2556,7 @@ async function runCodegen(config) {
     typedJsonFields
   });
   const ormTypesContent = emitOrmTypesTs(tables);
-  const modelsContent = emitModelsTs(tables, typedJsonFields);
+  const modelsContent = emitModelsTs(tables, typedJsonFields, branding);
   const inputContent = emitInputTs(tables);
   const enumsContent = emitEnumsTs(enumMap);
   const jsonContent = emitJsonTs(typeDefs, typedJsonFields);