@zenstackhq/cli 3.7.0-beta.1 → 3.7.0

package/dist/index.cjs

@@ -430,6 +430,18 @@ function getRelationFkName(decl) {
 	return ((decl?.attributes.find((a) => a.decl.ref?.name === "@relation"))?.args.find((a) => a.name === "map")?.value)?.value;
 }
 /**
+* Gets the relation name from the @relation attribute's `name` argument.
+* e.g., @relation('myRelation', fields: [...], references: [...]) -> "myRelation"
+* e.g., @relation(name: 'myRelation', fields: [...], references: [...]) -> "myRelation"
+* e.g., @relation(fields: [...], references: [...]) -> undefined
+* e.g., @relation('backRef') -> "backRef"
+*/
+function getRelationName(decl) {
+	const relationAttr = decl?.attributes?.find((a) => a.decl?.ref?.name === "@relation");
+	if (!relationAttr) return void 0;
+	return (0, _zenstackhq_language_utils.getAttributeArgLiteral)(relationAttr, "name");
+}
+/**
 * Gets the FK field names from the @relation attribute's `fields` argument.
 * Returns a sorted, comma-separated string of field names for comparison.
 * e.g., @relation(fields: [userId], references: [id]) -> "userId"
@@ -2176,6 +2188,16 @@ const providers = {
 };
 //#endregion
 //#region src/actions/db.ts
+function hasRelationFieldsArg(field) {
+	return !!field.attributes.find((a) => a.decl.ref?.name === "@relation")?.args.some((a) => a.name === "fields");
+}
+function getReferencedModelName(field) {
+	return field.type.reference?.ref ? getDbName(field.type.reference.ref) : void 0;
+}
+function matchesRelationNameFallback(field, relationName, candidate) {
+	const referencedModelName = getReferencedModelName(field);
+	return !!referencedModelName && getRelationName(candidate) === relationName && hasRelationFieldsArg(candidate) === hasRelationFieldsArg(field) && getReferencedModelName(candidate) === referencedModelName;
+}
 /**
 * CLI action for db related commands
 */
@@ -2335,16 +2357,21 @@ async function runPull(options) {
 			}
 			newDataModel.fields.forEach((f) => {
 				let originalFields = originalDataModel.fields.filter((d) => getDbName(d) === getDbName(f));
-				const isRelationField = f.$type === "DataField" && !!f.attributes?.some((a) => a?.decl?.ref?.name === "@relation");
-				if (originalFields.length === 0 && isRelationField && !getRelationFieldsKey(f)) return;
 				if (originalFields.length === 0) {
-					const newFieldsKey = getRelationFieldsKey(f);
-					if (newFieldsKey) originalFields = originalDataModel.fields.filter((d) => getRelationFieldsKey(d) === newFieldsKey);
+					const newFieldsKey = (0, _zenstackhq_language_ast.isDataField)(f) ? getRelationFieldsKey(f) : void 0;
+					if (newFieldsKey) originalFields = originalDataModel.fields.filter((d) => (0, _zenstackhq_language_ast.isDataField)(d) && getRelationFieldsKey(d) === newFieldsKey);
 				}
-				if (originalFields.length === 0) originalFields = originalDataModel.fields.filter((d) => getRelationFkName(d) === getRelationFkName(f) && !!getRelationFkName(d) && !!getRelationFkName(f));
-				if (originalFields.length === 0) originalFields = originalDataModel.fields.filter((d) => f.$type === "DataField" && d.$type === "DataField" && f.type.reference?.ref && d.type.reference?.ref && getDbName(f.type.reference.ref) === getDbName(d.type.reference.ref));
+				if (originalFields.length === 0) {
+					const newFkName = (0, _zenstackhq_language_ast.isDataField)(f) ? getRelationFkName(f) : void 0;
+					if (newFkName) originalFields = originalDataModel.fields.filter((d) => (0, _zenstackhq_language_ast.isDataField)(d) && getRelationFkName(d) === newFkName);
+				}
+				if (originalFields.length === 0) {
+					const newRelName = (0, _zenstackhq_language_ast.isDataField)(f) ? getRelationName(f) : void 0;
+					if (newRelName) originalFields = originalDataModel.fields.filter((d) => (0, _zenstackhq_language_ast.isDataField)(d) && (0, _zenstackhq_language_ast.isDataField)(f) && matchesRelationNameFallback(f, newRelName, d));
+				}
+				if (originalFields.length === 0) originalFields = originalDataModel.fields.filter((d) => (0, _zenstackhq_language_ast.isDataField)(f) && (0, _zenstackhq_language_ast.isDataField)(d) && f.type.reference?.ref && d.type.reference?.ref && getDbName(f.type.reference.ref) === getDbName(d.type.reference.ref));
 				if (originalFields.length > 1) {
-					if (!!getRelationFieldsKey(f)) console.warn(colors.default.yellow(`Found more original fields, need to tweak the search algorithm. ${originalDataModel.name}->[${originalFields.map((of) => of.name).join(", ")}](${f.name})`));
+					if (!((0, _zenstackhq_language_ast.isDataField)(f) && !getRelationFieldsKey(f))) console.warn(colors.default.yellow(`Found more original fields, need to tweak the search algorithm. ${originalDataModel.name}->[${originalFields.map((of) => of.name).join(", ")}](${f.name})`));
 					return;
 				}
 				const originalField = originalFields.at(0);
@@ -2439,12 +2466,19 @@ async function runPull(options) {
 			});
 			originalDataModel.fields.filter((f) => {
 				if (newDataModel.fields.find((d) => getDbName(d) === getDbName(f))) return false;
-				const originalFieldsKey = getRelationFieldsKey(f);
+				const originalFieldsKey = (0, _zenstackhq_language_ast.isDataField)(f) ? getRelationFieldsKey(f) : void 0;
 				if (originalFieldsKey) {
-					if (newDataModel.fields.find((d) => getRelationFieldsKey(d) === originalFieldsKey)) return false;
+					if (newDataModel.fields.find((d) => (0, _zenstackhq_language_ast.isDataField)(d) && getRelationFieldsKey(d) === originalFieldsKey)) return false;
+				}
+				const originalFkName = (0, _zenstackhq_language_ast.isDataField)(f) ? getRelationFkName(f) : void 0;
+				if (originalFkName) {
+					if (newDataModel.fields.find((d) => (0, _zenstackhq_language_ast.isDataField)(d) && getRelationFkName(d) === originalFkName)) return false;
+				}
+				const originalRelName = (0, _zenstackhq_language_ast.isDataField)(f) ? getRelationName(f) : void 0;
+				if (originalRelName) {
+					if (newDataModel.fields.find((d) => (0, _zenstackhq_language_ast.isDataField)(d) && (0, _zenstackhq_language_ast.isDataField)(f) && matchesRelationNameFallback(f, originalRelName, d))) return false;
 				}
-				if (newDataModel.fields.find((d) => getRelationFkName(d) === getRelationFkName(f) && !!getRelationFkName(d) && !!getRelationFkName(f))) return false;
-				return !newDataModel.fields.find((d) => f.$type === "DataField" && d.$type === "DataField" && f.type.reference?.ref && d.type.reference?.ref && getDbName(f.type.reference.ref) === getDbName(d.type.reference.ref));
+				return !newDataModel.fields.find((d) => (0, _zenstackhq_language_ast.isDataField)(f) && (0, _zenstackhq_language_ast.isDataField)(d) && f.type.reference?.ref && d.type.reference?.ref && getDbName(f.type.reference.ref) === getDbName(d.type.reference.ref));
 			}).forEach((f) => {
 				const _model = f.$container;
 				const index = _model.fields.findIndex((d) => d === f);
@@ -3190,7 +3224,7 @@ function startServer(client, schema, options) {
 }
 //#endregion
 //#region src/constants.ts
-const TELEMETRY_TRACKING_TOKEN = "<TELEMETRY_TRACKING_TOKEN>";
+const TELEMETRY_TRACKING_TOKEN = "74944eb779d7d3b4ce185be843fde9fc";
 //#endregion
 //#region src/utils/is-ci.ts
 const isInCi = node_process.env["CI"] !== "0" && node_process.env["CI"] !== "false" && ("CI" in node_process.env || "CONTINUOUS_INTEGRATION" in node_process.env || Object.keys(node_process.env).some((key) => key.startsWith("CI_")));