This commit is contained in:
Nathan Sobo 2023-04-21 15:36:52 -06:00
parent 55db28e074
commit 83bf3d071d
2 changed files with 73 additions and 16 deletions

View file

@ -121,6 +121,11 @@ pub trait View: Entity + Sized {
} }
} }
pub trait AccessAppContext {
fn read<T, F: FnOnce(&AppContext) -> T>(&self, f: F) -> T;
fn update<T, F: FnOnce(&mut AppContext) -> T>(&mut self, f: F) -> T;
}
pub trait ReadModelWith { pub trait ReadModelWith {
fn read_model_with<E: Entity, T>( fn read_model_with<E: Entity, T>(
&self, &self,
@ -148,11 +153,11 @@ pub trait UpgradeModelHandle {
fn upgrade_any_model_handle(&self, handle: &AnyWeakModelHandle) -> Option<AnyModelHandle>; fn upgrade_any_model_handle(&self, handle: &AnyWeakModelHandle) -> Option<AnyModelHandle>;
} }
pub trait UpgradeViewHandle { // pub trait UpgradeViewHandle {
fn upgrade_view_handle<T: View>(&self, handle: &WeakViewHandle<T>) -> Option<ViewHandle<T>>; // fn upgrade_view_handle<T: View>(&self, handle: &WeakViewHandle<T>) -> Option<ViewHandle<T>>;
fn upgrade_any_view_handle(&self, handle: &AnyWeakViewHandle) -> Option<AnyViewHandle>; // fn upgrade_any_view_handle(&self, handle: &AnyWeakViewHandle) -> Option<AnyViewHandle>;
} // }
pub trait ReadViewWith { pub trait ReadViewWith {
fn read_view_with<V, T>( fn read_view_with<V, T>(
@ -179,9 +184,6 @@ pub trait UpdateView {
#[derive(Clone)] #[derive(Clone)]
pub struct App(Rc<RefCell<AppContext>>); pub struct App(Rc<RefCell<AppContext>>);
#[derive(Clone)]
pub struct AsyncAppContext(Rc<RefCell<AppContext>>);
impl App { impl App {
pub fn new(asset_source: impl AssetSource) -> Result<Self> { pub fn new(asset_source: impl AssetSource) -> Result<Self> {
let platform = platform::current::platform(); let platform = platform::current::platform();
@ -328,6 +330,9 @@ impl App {
} }
} }
#[derive(Clone)]
pub struct AsyncAppContext(Rc<RefCell<AppContext>>);
impl AsyncAppContext { impl AsyncAppContext {
pub fn spawn<F, Fut, T>(&self, f: F) -> Task<T> pub fn spawn<F, Fut, T>(&self, f: F) -> Task<T>
where where
@ -406,6 +411,16 @@ impl AsyncAppContext {
} }
} }
impl AccessAppContext for AsyncAppContext {
fn read<T, F: FnOnce(&AppContext) -> T>(&self, f: F) -> T {
self.0.borrow().read(f)
}
fn update<T, F: FnOnce(&mut AppContext) -> T>(&mut self, f: F) -> T {
self.0.borrow_mut().update(f)
}
}
impl UpdateModel for AsyncAppContext { impl UpdateModel for AsyncAppContext {
fn update_model<E: Entity, O>( fn update_model<E: Entity, O>(
&mut self, &mut self,
@ -2089,6 +2104,16 @@ impl AppContext {
} }
} }
impl AccessAppContext for AppContext {
fn read<T, F: FnOnce(&AppContext) -> T>(&self, f: F) -> T {
f(self)
}
fn update<T, F: FnOnce(&mut AppContext) -> T>(&mut self, f: F) -> T {
f(self)
}
}
impl UpdateModel for AppContext { impl UpdateModel for AppContext {
fn update_model<T: Entity, V>( fn update_model<T: Entity, V>(
&mut self, &mut self,
@ -3427,6 +3452,16 @@ impl<'a, 'b, V: View> ViewContext<'a, 'b, V> {
} }
} }
impl<V> AccessAppContext for ViewContext<'_, '_, V> {
fn read<T, F: FnOnce(&AppContext) -> T>(&self, f: F) -> T {
self.window_context.read(f)
}
fn update<T, F: FnOnce(&mut AppContext) -> T>(&mut self, f: F) -> T {
self.window_context.update(f)
}
}
impl<V> UpgradeModelHandle for ViewContext<'_, '_, V> { impl<V> UpgradeModelHandle for ViewContext<'_, '_, V> {
fn upgrade_model_handle<T: Entity>( fn upgrade_model_handle<T: Entity>(
&self, &self,
@ -4223,13 +4258,13 @@ impl<T> WeakHandle for WeakViewHandle<T> {
} }
} }
impl<T: View> WeakViewHandle<T> { impl<V: View> WeakViewHandle<V> {
fn new(window_id: usize, view_id: usize) -> Self { fn new(window_id: usize, view_id: usize) -> Self {
Self { Self {
any_handle: AnyWeakViewHandle { any_handle: AnyWeakViewHandle {
window_id, window_id,
view_id, view_id,
view_type: TypeId::of::<T>(), view_type: TypeId::of::<V>(),
}, },
view_type: PhantomData, view_type: PhantomData,
} }
@ -4247,8 +4282,20 @@ impl<T: View> WeakViewHandle<T> {
self.any_handle self.any_handle
} }
pub fn upgrade(&self, cx: &impl UpgradeViewHandle) -> Option<ViewHandle<T>> { pub fn upgrade(&self, cx: &impl AccessAppContext) -> Option<ViewHandle<V>> {
cx.upgrade_view_handle(self) cx.read(|cx| cx.upgrade_view_handle(self))
}
pub fn update<T>(
&self,
cx: &mut impl AccessAppContext,
update: impl FnOnce(&mut V, &mut ViewContext<V>) -> T,
) -> Option<T> {
cx.update(|cx| {
let handle = cx.upgrade_view_handle(self)?;
cx.update_window(self.window_id, |cx| handle.update(cx, update))
})
} }
} }

View file

@ -13,11 +13,11 @@ use crate::{
}, },
text_layout::TextLayoutCache, text_layout::TextLayoutCache,
util::post_inc, util::post_inc,
Action, AnyModelHandle, AnyView, AnyViewHandle, AnyWeakModelHandle, AnyWeakViewHandle, AccessAppContext, Action, AnyModelHandle, AnyView, AnyViewHandle, AnyWeakModelHandle,
AppContext, Effect, Element, Entity, Handle, ModelContext, ModelHandle, MouseRegion, AnyWeakViewHandle, AppContext, Effect, Element, Entity, Handle, ModelContext, ModelHandle,
MouseRegionId, ParentId, SceneBuilder, Subscription, UpdateModel, UpdateView, MouseRegion, MouseRegionId, ParentId, SceneBuilder, Subscription, UpdateModel, UpdateView,
UpgradeModelHandle, UpgradeViewHandle, View, ViewContext, ViewHandle, WeakModelHandle, UpgradeModelHandle, View, ViewContext, ViewHandle, WeakModelHandle, WeakViewHandle,
WeakViewHandle, WindowInvalidation, WindowInvalidation,
}; };
use anyhow::{anyhow, bail, Result}; use anyhow::{anyhow, bail, Result};
use collections::{HashMap, HashSet}; use collections::{HashMap, HashSet};
@ -133,6 +133,16 @@ impl DerefMut for WindowContext<'_> {
} }
} }
impl AccessAppContext for WindowContext<'_> {
fn read<T, F: FnOnce(&AppContext) -> T>(&self, f: F) -> T {
self.app_context.read(f)
}
fn update<T, F: FnOnce(&mut AppContext) -> T>(&mut self, f: F) -> T {
self.app_context.update(f)
}
}
impl UpdateModel for WindowContext<'_> { impl UpdateModel for WindowContext<'_> {
fn update_model<T: Entity, R>( fn update_model<T: Entity, R>(
&mut self, &mut self,