Disallow multiple save modals for the same pane item (#11931)

Fixes https://github.com/zed-industries/zed/issues/10192 Release Notes: - Fixed multiple save modals appearing for the same file being closed ([10192](https://github.com/zed-industries/zed/issues/10192))

Kirill Bulatov committed 2 years ago

Commit decfbc61 parent 65138861 file · +33 −14Browse files

Unified · Split

crates/workspace/src/pane.rs+33−14

     IconSize, Indicator, Label, Tab, TabBar, TabPosition, Tooltip,
 };
 use ui::{v_flex, ContextMenu};
-use util::{maybe, truncate_and_remove_front, ResultExt};
+use util::{debug_panic, maybe, truncate_and_remove_front, ResultExt};
 #[derive(PartialEq, Clone, Copy, Deserialize, Debug)]
 #[serde(rename_all = "camelCase")]
     /// Otherwise, when `display_nav_history_buttons` is Some, it determines whether nav buttons should be displayed.
     display_nav_history_buttons: Option<bool>,
     double_click_dispatch_action: Box<dyn Action>,
+    save_modals_spawned: HashSet<EntityId>,
 }
 pub struct ActivationHistoryEntry {
             ),
             _subscriptions: subscriptions,
             double_click_dispatch_action,
+            save_modals_spawned: HashSet::default(),
         }
     }
                     ) && Self::can_autosave_item(item, cx)
                 })?;
                 if !will_autosave {
-                    let answer = pane.update(cx, |pane, cx| {
+                    let item_id = item.item_id();
-                        pane.activate_item(item_ix, true, true, cx);
+                    let answer_task = pane.update(cx, |pane, cx| {
-                        let prompt = dirty_message_for(item.project_path(cx));
+                        if pane.save_modals_spawned.insert(item_id) {
-                        cx.prompt(
+                            pane.activate_item(item_ix, true, true, cx);
-                            PromptLevel::Warning,
+                            let prompt = dirty_message_for(item.project_path(cx));
-                            &prompt,
+                            Some(cx.prompt(
-                            None,
+                                PromptLevel::Warning,
-                            &["Save", "Don't Save", "Cancel"],
+                                &prompt,
-                        )
+                                None,
+                                &["Save", "Don't Save", "Cancel"],
+                            ))
+                        } else {
+                            None
+                        }
                     })?;
-                    match answer.await {
+                    if let Some(answer_task) = answer_task {
-                        Ok(0) => {}
+                        let answer = answer_task.await;
-                        Ok(1) => return Ok(true), // Don't save this file
+                        pane.update(cx, |pane, _| {
-                        _ => return Ok(false),    // Cancel
+                            if !pane.save_modals_spawned.remove(&item_id) {
+                                debug_panic!(
+                                    "save modal was not present in spawned modals after awaiting for its answer"
+                                )
+                            }
+                        })?;
+                        match answer {
+                            Ok(0) => {}
+                            Ok(1) => return Ok(true), // Don't save this file
+                            _ => return Ok(false),    // Cancel
+                        }
+                    } else {
+                        return Ok(false);
                     }
                 }
             }