make field selected

Author: paula-stacho

packages/compass-data-modeling/src/store/diagram.ts

@@ -27,10 +27,7 @@ import {
 } from '../services/open-and-download-diagram';
 import type { MongoDBJSONSchema } from 'mongodb-schema';
 import { getCoordinatesForNewNode } from '@mongodb-js/diagramming';
-import {
-  collectionToBaseNodeForLayout,
-  collectionToDiagramNode,
-} from '../utils/nodes-and-edges';
+import { collectionToBaseNodeForLayout } from '../utils/nodes-and-edges';
 import toNS from 'mongodb-ns';
 import { traverseSchema } from '../utils/schema-traversal';
 import { addFieldToJSONSchema, getNewUnusedFieldName } from '../utils/schema';
@@ -387,6 +384,13 @@ const updateSelectedItemsFromAppliedEdit = (
         id: edit.ns,
       };
     }
+    case 'AddField': {
+      return {
+        type: 'field',
+        namespace: edit.ns,
+        fieldPath: edit.field,
+      };
+    }
   }
 
   return currentSelection;