@slicemachine/manager 0.24.14-alpha.jp-update-cr-links-unit.10 → 0.24.14-alpha.jp-update-cr-links-remove-recursion.11

package/src/managers/customTypes/CustomTypesManager.ts

@@ -188,6 +188,141 @@ export class CustomTypesManager extends BaseManager {
 		};
 	}
 
+	private updateCRCustomType(
+		args: { customType: CrCustomType } & CustomTypeFieldIdChangedMeta,
+	): CrCustomType {
+		const [previousCustomTypeId, previousFieldId] = args.previousPath;
+		const [newCustomTypeId, newFieldId] = args.newPath;
+
+		if (!previousCustomTypeId || !newCustomTypeId) {
+			throw new Error(
+				"Could not find a customtype id in previousPath and/or newPath, which should not be possible.",
+			);
+		}
+
+		if (!previousFieldId || !newFieldId) {
+			throw new Error(
+				"Could not find a field id in previousPath and/or newPath, which should not be possible.",
+			);
+		}
+
+		const customType = shallowCloneIfObject(args.customType);
+
+		if (typeof customType === "string" || !customType.fields) {
+			return customType;
+		}
+
+		const matchedCustomTypeId = customType.id === previousCustomTypeId;
+
+		const newFields = customType.fields.map((fieldArg) => {
+			const nestedField = shallowCloneIfObject(fieldArg);
+
+			if (typeof nestedField === "string") {
+				if (
+					matchedCustomTypeId &&
+					nestedField === previousFieldId &&
+					nestedField !== newFieldId
+				) {
+					// We have reached a field id that matches the id that was renamed,
+					// so we update it new one. The field is a string, so return the new
+					// id.
+					return newFieldId;
+				}
+
+				return nestedField;
+			}
+
+			if (
+				matchedCustomTypeId &&
+				nestedField.id === previousFieldId &&
+				nestedField.id !== newFieldId
+			) {
+				// We have reached a field id that matches the id that was renamed,
+				// so we update it new one.
+				// Since field is not a string, we don't exit, as we might have
+				// something to update further down in customtypes.
+				nestedField.id = newFieldId;
+			}
+
+			return {
+				...nestedField,
+				customtypes: nestedField.customtypes.map((customTypeArg) => {
+					const customTypeField = shallowCloneIfObject(customTypeArg);
+
+					if (typeof customTypeField === "string" || !customTypeField.fields) {
+						return customTypeField;
+					}
+
+					const matchedNestedCustomTypeId =
+						customTypeField.id === previousCustomTypeId;
+
+					return {
+						...customTypeField,
+						fields: customTypeField.fields.map((fieldArg) => {
+							const nestedCustomTypeField = shallowCloneIfObject(fieldArg);
+
+							if (
+								matchedNestedCustomTypeId &&
+								nestedCustomTypeField === previousFieldId &&
+								nestedCustomTypeField !== newFieldId
+							) {
+								// Matches the previous id, so we update it and return because
+								// it's the last level.
+								return newFieldId;
+							}
+
+							return nestedCustomTypeField;
+						}),
+					};
+				}),
+			};
+		});
+
+		return { ...customType, fields: newFields };
+	}
+
+	/**
+	 * Map over the custom types of a Content Relationship Link and update the API
+	 * IDs that were changed during the custom type update.
+	 */
+	private updateCRCustomTypes(
+		args: { customTypes: CrCustomTypes } & CustomTypeFieldIdChangedMeta,
+	): CrCustomTypes {
+		const { customTypes, ...updateMeta } = args;
+
+		return customTypes.map((customType) => {
+			return this.updateCRCustomType({ customType, ...updateMeta });
+		});
+	}
+
+	/**
+	 * Update the Content Relationship API IDs of a single field. The change is
+	 * determined by the `previousPath` and `newPath` properties.
+	 */
+	private updateFieldContentRelationships<
+		T extends UID | NestableWidget | Group | NestedGroup,
+	>(args: { field: T } & CustomTypeFieldIdChangedMeta): T {
+		const { field, ...updateMeta } = args;
+		if (
+			field.type !== "Link" ||
+			field.config?.select !== "document" ||
+			!field.config?.customtypes
+		) {
+			// not a content relationship field
+			return field;
+		}
+
+		const newCustomTypes = this.updateCRCustomTypes({
+			...updateMeta,
+			customTypes: field.config.customtypes,
+		});
+
+		return {
+			...field,
+			config: { ...field.config, customtypes: newCustomTypes },
+		};
+	}
+
 	/**
 	 * Update the Content Relationship API IDs for all existing custom types and
 	 * slices. The change is determined by properties inside the `updateMeta`
@@ -213,19 +348,24 @@ export class CustomTypesManager extends BaseManager {
 				// any custom type and update them to use the new one.
 				const customTypes = await this.readAllCustomTypes();
 
-				updateCustomTypeContentRelationships({
-					models: customTypes.models,
-					onUpdate: (model) => {
-						pushIfDefined(
-							crUpdates,
-							this.sliceMachinePluginRunner?.callHook("custom-type:update", {
-								model,
-							}),
-						);
-					},
-					previousPath,
-					newPath,
-				});
+				for (const customType of customTypes.models) {
+					const updatedCustomTypeModel = traverseCustomType({
+						customType: customType.model,
+						onField: ({ field }) => {
+							return this.updateFieldContentRelationships({
+								field,
+								previousPath,
+								newPath,
+							});
+						},
+					});
+
+					crUpdates.push(
+						this.sliceMachinePluginRunner.callHook("custom-type:update", {
+							model: updatedCustomTypeModel,
+						}),
+					);
+				}
 
 				// Find existing slice with content relationships that link to the renamed
 				// field id in all libraries and update them to use the new one.
@@ -236,20 +376,26 @@ export class CustomTypesManager extends BaseManager {
 						libraryID: library.libraryID,
 					});
 
-					updateSharedSliceContentRelationships({
-						models: slices.models,
-						onUpdate: (model) => {
-							pushIfDefined(
-								crUpdates,
-								this.sliceMachinePluginRunner?.callHook("slice:update", {
-									libraryID: library.libraryID,
-									model,
-								}),
-							);
-						},
-						previousPath,
-						newPath,
-					});
+					for (const slice of slices.models) {
+						const updatedSliceModel = traverseSharedSlice({
+							path: ["."],
+							slice: slice.model,
+							onField: ({ field }) => {
+								return this.updateFieldContentRelationships({
+									field,
+									previousPath,
+									newPath,
+								});
+							},
+						});
+
+						crUpdates.push(
+							this.sliceMachinePluginRunner.callHook("slice:update", {
+								libraryID: library.libraryID,
+								model: updatedSliceModel,
+							}),
+						);
+					}
 				}
 
 				// Process all the Content Relationship updates at once.
@@ -495,196 +641,6 @@ const InferSliceResponse = z.object({
 	langSmithUrl: z.string().url().optional(),
 });
 
-function updateCRCustomType(
-	args: { customType: CrCustomType } & CustomTypeFieldIdChangedMeta,
-): CrCustomType {
-	const { previousPath, newPath } = args;
-
-	const customType = shallowCloneIfObject(args.customType);
-
-	const [previousCustomTypeId, previousFieldId] = previousPath;
-	const [newCustomTypeId, newFieldId] = newPath;
-
-	if (!previousCustomTypeId || !newCustomTypeId) {
-		throw new Error(
-			"Didn't find any customtype id, which should not be possible.",
-		);
-	}
-
-	if (!previousFieldId || !newFieldId) {
-		throw new Error("Didn't find any field id, which should not be possible.");
-	}
-
-	if (typeof customType === "string") {
-		return customType; // we don't support custom type id renaming
-	}
-
-	const matchedCustomTypeId = customType.id === previousCustomTypeId;
-
-	if (customType.fields) {
-		const newFields = customType.fields.map((fieldArg) => {
-			const nestedField = shallowCloneIfObject(fieldArg);
-
-			if (typeof nestedField === "string") {
-				if (
-					matchedCustomTypeId &&
-					nestedField === previousFieldId &&
-					nestedField !== newFieldId
-				) {
-					// We have reached a field id that matches the id that was renamed,
-					// so we update it new one. The field is a string, so return the new
-					// id.
-					return newFieldId;
-				}
-
-				return nestedField;
-			}
-
-			if (nestedField.id === previousFieldId) {
-				if (matchedCustomTypeId && nestedField.id !== newFieldId) {
-					// We have reached a field id that matches the id that was renamed,
-					// so we update it new one.
-					// Since field is not a string, we don't exit, as we might have
-					// something to update further down in customtypes.
-					nestedField.id = newFieldId;
-				}
-			}
-
-			return {
-				...nestedField,
-				customtypes: nestedField.customtypes.map((customTypeArg) => {
-					const customTypeField = shallowCloneIfObject(customTypeArg);
-
-					if (typeof customTypeField === "string") {
-						return customTypeField; // we don't support custom type id renaming
-					}
-
-					const matchedNestedCustomTypeId =
-						customTypeField.id === previousCustomTypeId;
-
-					if (customTypeField.fields) {
-						return {
-							...customTypeField,
-							fields: customTypeField.fields.map((fieldArg) => {
-								const nestedCustomTypeField = shallowCloneIfObject(fieldArg);
-
-								if (
-									matchedNestedCustomTypeId &&
-									nestedCustomTypeField === previousFieldId &&
-									nestedCustomTypeField !== newFieldId
-								) {
-									// Matches the previous id, so we update it and return because
-									// it's the last level.
-									return newFieldId;
-								}
-
-								return nestedCustomTypeField;
-							}),
-						};
-					}
-
-					return customTypeField;
-				}),
-			};
-		});
-
-		return { ...customType, fields: newFields };
-	}
-
-	return customType;
-}
-
-/**
- * Map over the custom types of a Content Relationship Link and update the API
- * IDs that were changed during the custom type update.
- */
-function updateCRCustomTypes(
-	args: { customTypes: CrCustomTypes } & CustomTypeFieldIdChangedMeta,
-): CrCustomTypes {
-	const { customTypes, ...updateMeta } = args;
-
-	return customTypes.map((customType) => {
-		return updateCRCustomType({ customType, ...updateMeta });
-	});
-}
-
-/**
- * Update the Content Relationship API IDs of a single field. The change is
- * determined by the `previousPath` and `newPath` properties.
- */
-function updateFieldContentRelationships<
-	T extends UID | NestableWidget | Group | NestedGroup,
->(args: { field: T } & CustomTypeFieldIdChangedMeta): T {
-	const { field, ...updateMeta } = args;
-	if (
-		field.type !== "Link" ||
-		field.config?.select !== "document" ||
-		!field.config?.customtypes
-	) {
-		// not a content relationship field
-		return field;
-	}
-
-	const newCustomTypes = updateCRCustomTypes({
-		...updateMeta,
-		customTypes: field.config.customtypes,
-	});
-
-	return {
-		...field,
-		config: { ...field.config, customtypes: newCustomTypes },
-	};
-}
-
-export function updateCustomTypeContentRelationships(
-	args: {
-		models: { model: CustomType }[];
-		onUpdate: (model: CustomType) => void;
-	} & CustomTypeFieldIdChangedMeta,
-): void {
-	const { models, previousPath, newPath, onUpdate } = args;
-
-	for (const customType of models) {
-		const updatedCustomTypeModel = traverseCustomType({
-			customType: customType.model,
-			onField: ({ field }) => {
-				return updateFieldContentRelationships({
-					field,
-					previousPath,
-					newPath,
-				});
-			},
-		});
-
-		onUpdate(updatedCustomTypeModel);
-	}
-}
-
-export function updateSharedSliceContentRelationships(
-	args: {
-		models: { model: SharedSlice }[];
-		onUpdate: (model: SharedSlice) => void;
-	} & CustomTypeFieldIdChangedMeta,
-): void {
-	const { models, previousPath, newPath, onUpdate } = args;
-
-	for (const slice of models) {
-		const updatedSliceModel = traverseSharedSlice({
-			path: ["."],
-			slice: slice.model,
-			onField: ({ field }) => {
-				return updateFieldContentRelationships({
-					field,
-					previousPath,
-					newPath,
-				});
-			},
-		});
-
-		onUpdate(updatedSliceModel);
-	}
-}
-
 function shallowCloneIfObject<T>(value: T): T {
 	if (typeof value === "object") {
 		return { ...value };
@@ -692,9 +648,3 @@ function shallowCloneIfObject<T>(value: T): T {
 
 	return value;
 }
-
-function pushIfDefined<T>(array: T[], value: T | undefined) {
-	if (value) {
-		array.push(value);
-	}
-}