Manage deletion of attributes

concord-consortium · Sep 13, 2024 · 9c22dc6 · 9c22dc6
1 parent 9ec6bc0
commit 9c22dc6
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/plugins/bar-graph/bar-graph-content.ts b/src/plugins/bar-graph/bar-graph-content.ts
@@ -161,9 +161,16 @@ export const BarGraphContentModel = TileContentModel
             self.setPrimaryAttribute(atts[0].id);
           }
         }
+      }
+      // Check if primary or secondary attribute has been deleted
+      if (self.primaryAttribute && !self.sharedModel?.dataSet.attrFromID(self.primaryAttribute)) {
+        self.setPrimaryAttribute(undefined); // this will also unset secondaryAttribute
+      }
+      if (self.secondaryAttribute && !self.sharedModel?.dataSet.attrFromID(self.secondaryAttribute)) {
+        self.setSecondaryAttribute(undefined);
+      }
     }
-  }
-}));
+  }));
 
 export interface BarGraphContentModelType extends Instance<typeof BarGraphContentModel> {}