ensureCollectionExists

Author: paula-stacho

packages/compass-data-modeling/src/store/apply-edit.ts

@@ -32,6 +32,16 @@ function renameFieldInRelationshipSide(
   };
 }
 
+function ensureCollectionExists(
+  collections: DataModelCollection[],
+  ns: string
+): void {
+  const collection = collections.find((c) => c.ns === ns);
+  if (!collection) {
+    throw new Error(`Collection '${ns}' not found`);
+  }
+}
+
 export function applyEdit(edit: Edit, model?: StaticModel): StaticModel {
   if (edit.type === 'SetModel') {
     return edit.model;
@@ -81,6 +91,7 @@ export function applyEdit(edit: Edit, model?: StaticModel): StaticModel {
       };
     }
     case 'MoveCollection': {
+      ensureCollectionExists(model.collections, edit.ns);
       return {
         ...model,
         collections: model.collections.map((collection) => {
@@ -95,6 +106,7 @@ export function applyEdit(edit: Edit, model?: StaticModel): StaticModel {
       };
     }
     case 'RemoveCollection': {
+      ensureCollectionExists(model.collections, edit.ns);
       return {
         ...model,
         // Remove any relationships involving the collection being removed.
@@ -109,6 +121,7 @@ export function applyEdit(edit: Edit, model?: StaticModel): StaticModel {
       };
     }
     case 'RenameCollection': {
+      ensureCollectionExists(model.collections, edit.fromNS);
       return {
         ...model,
         // Update relationships to point to the renamed namespace.
@@ -137,6 +150,7 @@ export function applyEdit(edit: Edit, model?: StaticModel): StaticModel {
       };
     }
     case 'UpdateCollectionNote': {
+      ensureCollectionExists(model.collections, edit.ns);
       return {
         ...model,
         collections: model.collections.map((collection) => {
@@ -151,6 +165,7 @@ export function applyEdit(edit: Edit, model?: StaticModel): StaticModel {
       };
     }
     case 'AddField': {
+      ensureCollectionExists(model.collections, edit.ns);
       return {
         ...model,
         collections: model.collections.map((collection) => {
@@ -169,6 +184,7 @@ export function applyEdit(edit: Edit, model?: StaticModel): StaticModel {
       };
     }
     case 'RemoveField': {
+      ensureCollectionExists(model.collections, edit.ns);
       return {
         ...model,
         // Remove any relationships involving the field being removed.
@@ -193,6 +209,7 @@ export function applyEdit(edit: Edit, model?: StaticModel): StaticModel {
       };
     }
     case 'RenameField': {
+      ensureCollectionExists(model.collections, edit.ns);
       return {
         ...model,
         // Update any relationships involving the field being renamed.
@@ -227,6 +244,7 @@ export function applyEdit(edit: Edit, model?: StaticModel): StaticModel {
       };
     }
     case 'ChangeFieldType': {
+      ensureCollectionExists(model.collections, edit.ns);
       return {
         ...model,
         collections: model.collections.map((collection) => {