@zenstackhq/sdk 3.7.0 → 3.7.1

package/dist/index.cjs

@@ -418,6 +418,12 @@ var EnumField$1 = class extends DeclarationBase {
 //#region src/prisma/prisma-schema-generator.ts
 const IDENTIFIER_NAME_MAX_LENGTH = 38;
 const NON_PRISMA_DATASOURCE_FIELDS = ["defaultSchema"];
+const ID_FUNCTIONS = [
+	"uuid",
+	"ulid",
+	"cuid",
+	"nanoid"
+];
 /**
 * Generates Prisma schema file
 */
@@ -568,7 +574,7 @@ var PrismaSchemaGenerator = class {
 		return new _zenstackhq_language.ZModelCodeGenerator({ quote: "double" }).generate(expr);
 	}
 	makeFunctionCall(node) {
-		return new FunctionCall(node.function.ref.name, node.args.map((arg) => {
+		return new FunctionCall(node.function.ref.name, node.args.filter((_, i) => !(ID_FUNCTIONS.includes(node.function.ref.name) && (node.function.ref.name === "ulid" && i === 0 || i === 1))).map((arg) => {
 			return new FunctionCallArg((0, ts_pattern.match)(arg.value).when(_zenstackhq_language_ast.isStringLiteral, (v) => `"${v.value}"`).when(_zenstackhq_language_ast.isLiteralExpr, (v) => v.value.toString()).when(_zenstackhq_language_ast.isNullExpr, () => "null").otherwise(() => {
 				throw new Error("Function call argument must be literal or null");
 			}));
@@ -742,6 +748,11 @@ var TsSchemaGenerator = class {
 			typescript.factory.createPropertyAssignment("idFields", typescript.factory.createArrayLiteralExpression(getIdFields(dm).map((idField) => typescript.factory.createStringLiteral(idField)))),
 			typescript.factory.createPropertyAssignment("uniqueFields", this.createUniqueFieldsObject(dm)),
 			...isDelegateModel$1(dm) ? [typescript.factory.createPropertyAssignment("isDelegate", typescript.factory.createTrue())] : [],
+			...(() => {
+				const delegateMapValue = this.getDelegateMapValue(dm);
+				if (delegateMapValue === void 0) return [];
+				return [typescript.factory.createPropertyAssignment("delegateMap", this.createLiteralNode(delegateMapValue))];
+			})(),
 			...subModels.length > 0 ? [typescript.factory.createPropertyAssignment("subModels", typescript.factory.createArrayLiteralExpression(subModels.map((subModel) => typescript.factory.createStringLiteral(subModel))))] : [],
 			...dm.isView ? [typescript.factory.createPropertyAssignment("isView", typescript.factory.createTrue())] : []
 		];
@@ -749,6 +760,17 @@ var TsSchemaGenerator = class {
 		if (computedFields.length > 0) fields.push(typescript.factory.createPropertyAssignment("computedFields", this.createComputedFieldsObject(computedFields)));
 		return typescript.factory.createObjectLiteralExpression(fields, true);
 	}
+	getDelegateMapValue(dm) {
+		const delegateMapAttr = getAttribute(dm, "@@delegateMap");
+		if (!delegateMapAttr) return;
+		const valueExpr = delegateMapAttr.args[0]?.value;
+		if (!valueExpr) return;
+		if ((0, _zenstackhq_language_ast.isLiteralExpr)(valueExpr)) {
+			const literal = this.getLiteral(valueExpr);
+			return typeof literal === "string" ? literal : void 0;
+		}
+		if ((0, _zenstackhq_language_ast.isReferenceExpr)(valueExpr) && (0, _zenstackhq_language_ast.isEnumField)(valueExpr.target.ref)) return valueExpr.target.ref.name;
+	}
 	getSubModels(dm) {
 		return dm.$container.declarations.filter(_zenstackhq_language_ast.isDataModel).filter((d) => d.baseModel?.ref === dm).map((d) => d.name);
 	}