Bugfix/UI: Fix widget lookup by id

Fixes #424
NetLogo · Aug 2, 2024 · edefa41 · edefa41
1 parent 1ff0975
commit edefa41
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/app/assets/javascripts/beak/widgets/widget-controller.coffee b/app/assets/javascripts/beak/widgets/widget-controller.coffee
@@ -96,9 +96,18 @@ class WidgetController
   # (Number, Boolean, Array[Any]) => Unit
   removeWidgetById: (id, wasNew, extraNotificationArgs) ->
 
-    widgetType = @ractive.get('widgetObj')[id].type
+    widgetType = null
+    widgetObj = @ractive.get('widgetObj')
+
+    for key, w of widgetObj 
+      if w.id is id
+        widgetType = w.type
+        delete widgetObj[key]
+        break
+
+    if widgetType is null
+      throw new Error('Could not find widget to remove by id') 
 
-    delete @ractive.get('widgetObj')[id]
     @ractive.update('widgetObj')
     @ractive.fire('deselect-widgets')