@zenstackhq/sdk 3.9.0-beta.1 → 3.9.0

package/dist/index.cjs

@@ -527,6 +527,9 @@ var PrismaSchemaGenerator = class {
 		if (!attr.decl.ref) return false;
 		return attr.decl.ref.attributes.some((a) => a.decl.ref?.name === "@@@prisma");
 	}
+	isNativeTypeMappingAttribute(attr) {
+		return attr.decl.$refText.startsWith("@db.") && this.isPrismaAttribute(attr);
+	}
 	getUnsupportedFieldType(fieldType) {
 		if (fieldType.unsupported) {
 			const value = (0, _zenstackhq_language_utils.getStringLiteral)(fieldType.unsupported.value);
@@ -557,7 +560,11 @@ var PrismaSchemaGenerator = class {
 		return langium.AstUtils.streamAst(expr).some(_zenstackhq_language_utils.isAuthInvocation);
 	}
 	makeFieldAttribute(attr) {
-		const attrName = attr.decl.ref.name;
+		let attrName = attr.decl.ref.name;
+		if (this.isNativeTypeMappingAttribute(attr)) {
+			const dataSource = this.zmodel.declarations.find(_zenstackhq_language_ast.isDataSource);
+			if (dataSource) attrName = attrName.replace("@db", `@${dataSource.name}`);
+		}
 		return new FieldAttribute(attrName, attr.args.map((arg) => this.makeAttributeArg(arg)));
 	}
 	makeAttributeArg(arg) {
@@ -785,7 +792,24 @@ var TsSchemaGenerator = class {
 		return typescript.factory.createObjectLiteralExpression(fields, true);
 	}
 	createComputedFieldsObject(fields) {
-		return typescript.factory.createObjectLiteralExpression(fields.map((field) => typescript.factory.createMethodDeclaration(void 0, void 0, field.name, void 0, void 0, [typescript.factory.createParameterDeclaration(void 0, void 0, "_context", void 0, typescript.factory.createTypeLiteralNode([typescript.factory.createPropertySignature(void 0, "modelAlias", void 0, typescript.factory.createKeywordTypeNode(typescript.SyntaxKind.StringKeyword))]), void 0)], typescript.factory.createTypeReferenceNode(this.mapFieldTypeToTSType(field.type)), typescript.factory.createBlock([typescript.factory.createThrowStatement(typescript.factory.createNewExpression(typescript.factory.createIdentifier("Error"), void 0, [typescript.factory.createStringLiteral("This is a stub for computed field")]))], true))), true);
+		return typescript.factory.createObjectLiteralExpression(fields.map((field) => {
+			const params = [typescript.factory.createParameterDeclaration(void 0, void 0, "_context", void 0, typescript.factory.createTypeLiteralNode([typescript.factory.createPropertySignature(void 0, "modelAlias", void 0, typescript.factory.createKeywordTypeNode(typescript.SyntaxKind.StringKeyword))]), void 0)];
+			if (field.params.length > 0) params.push(typescript.factory.createParameterDeclaration(void 0, void 0, "args", void 0, typescript.factory.createTypeLiteralNode(field.params.map((param) => typescript.factory.createPropertySignature(void 0, param.name, param.optional ? typescript.factory.createToken(typescript.SyntaxKind.QuestionToken) : void 0, typescript.factory.createTypeReferenceNode(this.mapFunctionParamTypeToTSType(param.type))))), void 0));
+			return typescript.factory.createMethodDeclaration(void 0, void 0, field.name, void 0, void 0, params, typescript.factory.createTypeReferenceNode(this.mapFieldTypeToTSType(field.type)), typescript.factory.createBlock([typescript.factory.createThrowStatement(typescript.factory.createNewExpression(typescript.factory.createIdentifier("Error"), void 0, [typescript.factory.createStringLiteral("This is a stub for computed field")]))], true));
+		}), true);
+	}
+	createFieldParamsObject(params) {
+		return typescript.factory.createObjectLiteralExpression(params.map((param) => typescript.factory.createPropertyAssignment(param.name, typescript.factory.createObjectLiteralExpression([
+			typescript.factory.createPropertyAssignment("name", typescript.factory.createStringLiteral(param.name)),
+			...param.optional ? [typescript.factory.createPropertyAssignment("optional", typescript.factory.createTrue())] : [],
+			...param.type.array ? [typescript.factory.createPropertyAssignment("array", typescript.factory.createTrue())] : [],
+			typescript.factory.createPropertyAssignment("type", typescript.factory.createStringLiteral(param.type.type ?? param.type.reference.$refText))
+		]))), true);
+	}
+	mapFunctionParamTypeToTSType(type) {
+		let result = (0, ts_pattern.match)(type.type).with("String", () => "string").with("Boolean", () => "boolean").with("Int", () => "number").with("Float", () => "number").with("BigInt", () => "bigint").with("Decimal", () => "number").with("DateTime", () => "Date").otherwise(() => "unknown");
+		if (type.array) result = `${result}[]`;
+		return result;
 	}
 	createUpdatedAtObject(ignoreArg) {
 		return typescript.factory.createObjectLiteralExpression([typescript.factory.createPropertyAssignment("ignore", typescript.factory.createArrayLiteralExpression(ignoreArg.value.items.map((item) => typescript.factory.createStringLiteral(item.target.$refText))))]);
@@ -828,6 +852,7 @@ var TsSchemaGenerator = class {
 			objectFields.push(typescript.factory.createPropertyAssignment("default", this.createDefaultTypeAssertion(defaultExpr)));
 		}
 		if (hasAttribute(field, "@computed")) objectFields.push(typescript.factory.createPropertyAssignment("computed", typescript.factory.createTrue()));
+		if (field.params.length > 0) objectFields.push(typescript.factory.createPropertyAssignment("params", this.createFieldParamsObject(field.params)));
 		if ((0, _zenstackhq_language_ast.isDataModel)(field.type.reference?.ref)) objectFields.push(typescript.factory.createPropertyAssignment("relation", this.createRelationObject(field, contextModel)));
 		const fkFor = this.getForeignKeyFor(field, contextModel);
 		if (fkFor && fkFor.length > 0) objectFields.push(typescript.factory.createPropertyAssignment("foreignKeyFor", typescript.factory.createAsExpression(typescript.factory.createArrayLiteralExpression(fkFor.map((fk) => typescript.factory.createStringLiteral(fk)), true), typescript.factory.createTypeOperatorNode(typescript.SyntaxKind.ReadonlyKeyword, typescript.factory.createArrayTypeNode(typescript.factory.createKeywordTypeNode(typescript.SyntaxKind.StringKeyword))))));
@@ -940,9 +965,9 @@ var TsSchemaGenerator = class {
 		const relationName = this.getRelationName(field);
 		for (const otherField of (0, _zenstackhq_language_utils.getAllFields)(targetModel)) {
 			if (otherField === field) continue;
-			if (otherField.type.reference?.ref === sourceModel) if (relationName) {
+			if (otherField.type.reference?.ref === sourceModel) {
 				if (this.getRelationName(otherField) === relationName) return otherField;
-			} else return otherField;
+			}
 		}
 	}
 	getRelationName(field) {