From dcf8b00656eab32299338c2a63b69dc128fff19d Mon Sep 17 00:00:00 2001
From: Nathan Sobo <nathan@zed.dev>
Date: Sat, 5 Aug 2023 18:00:44 -0600
Subject: [PATCH] WIP

---
 crates/gpui/src/app.rs        | 59 ++++++++++++++++++++---------------
 crates/gpui/src/app/window.rs |  2 +-
 2 files changed, 35 insertions(+), 26 deletions(-)

diff --git a/crates/gpui/src/app.rs b/crates/gpui/src/app.rs
index d98033820b..62d7d1d48a 100644
--- a/crates/gpui/src/app.rs
+++ b/crates/gpui/src/app.rs
@@ -1442,7 +1442,7 @@ impl AppContext {
         window
     }
 
-    pub fn replace_root_view<V, F>(
+    pub(crate) fn replace_root_view<V, F>(
         &mut self,
         window_id: usize,
         build_root_view: F,
@@ -3900,28 +3900,6 @@ impl<V: View> WindowHandle<V> {
         cx.update_window(self.window_id(), update)
     }
 
-    // pub fn update_root<C, F, R>(&self, cx: &mut C, update: F) -> C::Result<Option<R>>
-    // where
-    //     C: BorrowWindowContext,
-    //     F: FnOnce(&mut V, &mut ViewContext<V>) -> R,
-    // {
-    //     cx.update_window(self.window_id, |cx| {
-    //         cx.root_view()
-    //             .clone()
-    //             .downcast::<V>()
-    //             .map(|v| v.update(cx, update))
-    //     })
-    // }
-
-    pub fn read_root<'a>(&self, cx: &'a AppContext) -> &'a V {
-        let root_view = cx
-            .read_window(self.window_id(), |cx| {
-                cx.root_view().clone().downcast().unwrap()
-            })
-            .unwrap();
-        root_view.read(cx)
-    }
-
     pub fn read_root_with<C, F, R>(&self, cx: &C, read: F) -> C::Result<R>
     where
         C: BorrowWindowContext,
@@ -3935,6 +3913,20 @@ impl<V: View> WindowHandle<V> {
         })
     }
 
+    pub fn update_root<C, F, R>(&self, cx: &mut C, update: F) -> C::Result<R>
+    where
+        C: BorrowWindowContext,
+        F: FnOnce(&mut V, &mut ViewContext<V>) -> R,
+    {
+        cx.update_window(self.window_id, |cx| {
+            cx.root_view()
+                .clone()
+                .downcast::<V>()
+                .unwrap()
+                .update(cx, update)
+        })
+    }
+
     pub fn add_view<C, U, F>(&self, cx: &mut C, build_view: F) -> C::Result<ViewHandle<U>>
     where
         C: BorrowWindowContext,
@@ -3943,6 +3935,23 @@ impl<V: View> WindowHandle<V> {
     {
         self.update(cx, |cx| cx.add_view(build_view))
     }
+
+    pub(crate) fn replace_root_view<C, F>(
+        &self,
+        cx: &mut C,
+        build_root_view: F,
+    ) -> C::Result<ViewHandle<V>>
+    where
+        C: BorrowWindowContext,
+        F: FnOnce(&mut ViewContext<V>) -> V,
+    {
+        cx.update_window(self.window_id, |cx| {
+            let root_view = self.add_view(cx, |cx| build_root_view(cx));
+            cx.window.root_view = Some(root_view.clone().into_any());
+            cx.window.focused_view_id = Some(root_view.id());
+            root_view
+        })
+    }
 }
 
 #[repr(transparent)]
@@ -5329,7 +5338,7 @@ mod tests {
             TestView { events: Vec::new() }
         });
 
-        assert_eq!(window.read_root(cx).events, ["before emit"]);
+        window.read_root_with(cx, |view, _| assert_eq!(view.events, ["before emit"]));
     }
 
     #[crate::test(self)]
@@ -5381,7 +5390,7 @@ mod tests {
             TestView { events: Vec::new() }
         });
 
-        assert_eq!(window.read_root(cx).events, ["before notify"]);
+        assert_eq!(window.read_root_with(cx, |view, _| assert_eq!(view.events, ["before notify"])));
     }
 
     #[crate::test(self)]
diff --git a/crates/gpui/src/app/window.rs b/crates/gpui/src/app/window.rs
index 789341d46f..20992bfbaa 100644
--- a/crates/gpui/src/app/window.rs
+++ b/crates/gpui/src/app/window.rs
@@ -1157,7 +1157,7 @@ impl<'a> WindowContext<'a> {
         self.window.platform_window.prompt(level, msg, answers)
     }
 
-    pub fn replace_root_view<V, F>(&mut self, build_root_view: F) -> WindowHandle<V>
+    pub(crate) fn replace_root_view<V, F>(&mut self, build_root_view: F) -> WindowHandle<V>
     where
         V: View,
         F: FnOnce(&mut ViewContext<V>) -> V,