Avoid passing cx when emitting from CallbackCollection

Co-Authored-By: Nathan Sobo <nathan@zed.dev>
This commit is contained in:
Antonio Scandurra 2023-04-13 16:53:44 +02:00
parent ff774786bf
commit 84d2605ccf
2 changed files with 33 additions and 51 deletions

View File

@ -496,7 +496,7 @@ type SubscriptionCallback = Box<dyn FnMut(&dyn Any, &mut AppContext) -> bool>;
type GlobalSubscriptionCallback = Box<dyn FnMut(&dyn Any, &mut AppContext)>; type GlobalSubscriptionCallback = Box<dyn FnMut(&dyn Any, &mut AppContext)>;
type ObservationCallback = Box<dyn FnMut(&mut AppContext) -> bool>; type ObservationCallback = Box<dyn FnMut(&mut AppContext) -> bool>;
type GlobalObservationCallback = Box<dyn FnMut(&mut AppContext)>; type GlobalObservationCallback = Box<dyn FnMut(&mut AppContext)>;
type FocusObservationCallback = Box<dyn FnMut(bool, &mut AppContext) -> bool>; type FocusObservationCallback = Box<dyn FnMut(bool, &mut WindowContext) -> bool>;
type ReleaseObservationCallback = Box<dyn FnMut(&dyn Any, &mut AppContext)>; type ReleaseObservationCallback = Box<dyn FnMut(&dyn Any, &mut AppContext)>;
type ActionObservationCallback = Box<dyn FnMut(TypeId, &mut AppContext)>; type ActionObservationCallback = Box<dyn FnMut(TypeId, &mut AppContext)>;
type WindowActivationCallback = Box<dyn FnMut(bool, &mut AppContext) -> bool>; type WindowActivationCallback = Box<dyn FnMut(bool, &mut AppContext) -> bool>;
@ -1025,7 +1025,7 @@ impl AppContext {
fn observe_focus<F, V>(&mut self, handle: &ViewHandle<V>, mut callback: F) -> Subscription fn observe_focus<F, V>(&mut self, handle: &ViewHandle<V>, mut callback: F) -> Subscription
where where
F: 'static + FnMut(ViewHandle<V>, bool, &mut AppContext) -> bool, F: 'static + FnMut(ViewHandle<V>, bool, &mut WindowContext) -> bool,
V: View, V: View,
{ {
let subscription_id = post_inc(&mut self.next_subscription_id); let subscription_id = post_inc(&mut self.next_subscription_id);
@ -1705,9 +1705,8 @@ impl AppContext {
Effect::Event { entity_id, payload } => { Effect::Event { entity_id, payload } => {
let mut subscriptions = self.subscriptions.clone(); let mut subscriptions = self.subscriptions.clone();
subscriptions.emit(entity_id, self, |callback, this| { subscriptions
callback(payload.as_ref(), this) .emit(entity_id, |callback| callback(payload.as_ref(), self))
})
} }
Effect::GlobalSubscription { Effect::GlobalSubscription {
@ -1732,7 +1731,7 @@ impl AppContext {
Effect::ModelNotification { model_id } => { Effect::ModelNotification { model_id } => {
let mut observations = self.observations.clone(); let mut observations = self.observations.clone();
observations.emit(model_id, self, |callback, this| callback(this)); observations.emit(model_id, |callback| callback(self));
} }
Effect::ViewNotification { window_id, view_id } => { Effect::ViewNotification { window_id, view_id } => {
@ -1741,8 +1740,8 @@ impl AppContext {
Effect::GlobalNotification { type_id } => { Effect::GlobalNotification { type_id } => {
let mut subscriptions = self.global_observations.clone(); let mut subscriptions = self.global_observations.clone();
subscriptions.emit(type_id, self, |callback, this| { subscriptions.emit(type_id, |callback| {
callback(this); callback(self);
true true
}); });
} }
@ -2000,8 +1999,8 @@ impl AppContext {
let type_id = (&*payload).type_id(); let type_id = (&*payload).type_id();
let mut subscriptions = self.global_subscriptions.clone(); let mut subscriptions = self.global_subscriptions.clone();
subscriptions.emit(type_id, self, |callback, this| { subscriptions.emit(type_id, |callback| {
callback(payload.as_ref(), this); callback(payload.as_ref(), self);
true //Always alive true //Always alive
}); });
} }
@ -2024,15 +2023,15 @@ impl AppContext {
} }
let mut observations = self.observations.clone(); let mut observations = self.observations.clone();
observations.emit(observed_view_id, self, |callback, this| callback(this)); observations.emit(observed_view_id, |callback| callback(self));
} }
} }
fn handle_entity_release_effect(&mut self, entity_id: usize, entity: &dyn Any) { fn handle_entity_release_effect(&mut self, entity_id: usize, entity: &dyn Any) {
self.release_observations self.release_observations
.clone() .clone()
.emit(entity_id, self, |callback, this| { .emit(entity_id, |callback| {
callback(entity, this); callback(entity, self);
// Release observations happen one time. So clear the callback by returning false // Release observations happen one time. So clear the callback by returning false
false false
}) })
@ -2043,15 +2042,12 @@ impl AppContext {
cx.window.is_fullscreen = is_fullscreen; cx.window.is_fullscreen = is_fullscreen;
let mut fullscreen_observations = cx.window_fullscreen_observations.clone(); let mut fullscreen_observations = cx.window_fullscreen_observations.clone();
fullscreen_observations.emit(window_id, cx, |callback, this| { fullscreen_observations.emit(window_id, |callback| callback(is_fullscreen, cx));
callback(is_fullscreen, this)
});
if let Some(uuid) = cx.window_display_uuid() { if let Some(uuid) = cx.window_display_uuid() {
let bounds = cx.window_bounds(); let bounds = cx.window_bounds();
let mut bounds_observations = cx.window_bounds_observations.clone(); let mut bounds_observations = cx.window_bounds_observations.clone();
bounds_observations bounds_observations.emit(window_id, |callback| callback(bounds, uuid, cx));
.emit(window_id, cx, |callback, this| callback(bounds, uuid, this));
} }
Some(()) Some(())
@ -2067,8 +2063,8 @@ impl AppContext {
) { ) {
self.update(|this| { self.update(|this| {
let mut observations = this.keystroke_observations.clone(); let mut observations = this.keystroke_observations.clone();
observations.emit(window_id, this, { observations.emit(window_id, move |callback| {
move |callback, this| callback(&keystroke, &result, handled_by.as_ref(), this) callback(&keystroke, &result, handled_by.as_ref(), this)
}); });
}); });
} }
@ -2102,7 +2098,7 @@ impl AppContext {
}); });
let mut observations = cx.window_activation_observations.clone(); let mut observations = cx.window_activation_observations.clone();
observations.emit(window_id, cx, |callback, this| callback(active, this)); observations.emit(window_id, |callback| callback(active, cx));
Some(()) Some(())
}); });
@ -2110,13 +2106,7 @@ impl AppContext {
fn handle_focus_effect(&mut self, window_id: usize, mut focused_id: Option<usize>) { fn handle_focus_effect(&mut self, window_id: usize, mut focused_id: Option<usize>) {
let mut blurred_id = None; let mut blurred_id = None;
println!("FOCUS: {:?}", focused_id);
self.update_window(window_id, |cx| { self.update_window(window_id, |cx| {
println!(
"INSIDE FOCUS: {:?}. PREVIOUS FOCUS: {:?}",
focused_id,
cx.focused_view_id()
);
if cx.window.focused_view_id == focused_id { if cx.window.focused_view_id == focused_id {
focused_id = None; focused_id = None;
return; return;
@ -2138,6 +2128,9 @@ impl AppContext {
cx.views.insert((window_id, view_id), view); cx.views.insert((window_id, view_id), view);
} }
} }
let mut subscriptions = cx.focus_observations.clone();
subscriptions.emit(blurred_id, |callback| callback(false, cx));
} }
if let Some(focused_id) = focused_id { if let Some(focused_id) = focused_id {
@ -2147,18 +2140,9 @@ impl AppContext {
cx.views.insert((window_id, view_id), view); cx.views.insert((window_id, view_id), view);
} }
} }
}
});
self.update(|cx| {
if let Some(blurred_id) = blurred_id {
let mut subscriptions = cx.focus_observations.clone(); let mut subscriptions = cx.focus_observations.clone();
subscriptions.emit(blurred_id, cx, |callback, this| callback(false, this)); subscriptions.emit(focused_id, |callback| callback(true, cx));
}
if let Some(focused_id) = focused_id {
let mut subscriptions = cx.focus_observations.clone();
subscriptions.emit(focused_id, cx, |callback, this| callback(true, this));
} }
}); });
} }
@ -2215,8 +2199,8 @@ impl AppContext {
fn handle_action_dispatch_notification_effect(&mut self, action_id: TypeId) { fn handle_action_dispatch_notification_effect(&mut self, action_id: TypeId) {
self.action_dispatch_observations self.action_dispatch_observations
.clone() .clone()
.emit((), self, |callback, this| { .emit((), |callback| {
callback(action_id, this); callback(action_id, self);
true true
}); });
} }
@ -2240,8 +2224,8 @@ impl AppContext {
let bounds = cx.window_bounds(); let bounds = cx.window_bounds();
cx.window_bounds_observations cx.window_bounds_observations
.clone() .clone()
.emit(window_id, cx, move |callback, this| { .emit(window_id, move |callback| {
callback(bounds, display, this); callback(bounds, display, cx);
true true
}); });
} }
@ -2251,8 +2235,8 @@ impl AppContext {
fn handle_active_labeled_tasks_changed_effect(&mut self) { fn handle_active_labeled_tasks_changed_effect(&mut self) {
self.active_labeled_task_observations self.active_labeled_task_observations
.clone() .clone()
.emit((), self, move |callback, this| { .emit((), move |callback| {
callback(this); callback(self);
true true
}); });
} }
@ -5789,6 +5773,7 @@ mod tests {
cx.focus(&view_1); cx.focus(&view_1);
cx.focus(&view_2); cx.focus(&view_2);
}); });
assert!(cx.is_child_focused(&view_1)); assert!(cx.is_child_focused(&view_1));
assert!(!cx.is_child_focused(&view_2)); assert!(!cx.is_child_focused(&view_2));
assert_eq!( assert_eq!(

View File

@ -1,4 +1,3 @@
use crate::AppContext;
use collections::{BTreeMap, HashMap, HashSet}; use collections::{BTreeMap, HashMap, HashSet};
use parking_lot::Mutex; use parking_lot::Mutex;
use std::sync::Arc; use std::sync::Arc;
@ -93,12 +92,10 @@ impl<K: Clone + Hash + Eq + Copy, F> CallbackCollection<K, F> {
drop(callbacks); drop(callbacks);
} }
pub fn emit<C: FnMut(&mut F, &mut AppContext) -> bool>( pub fn emit<C>(&mut self, key: K, mut call_callback: C)
&mut self, where
key: K, C: FnMut(&mut F) -> bool,
cx: &mut AppContext, {
mut call_callback: C,
) {
let callbacks = self.internal.lock().callbacks.remove(&key); let callbacks = self.internal.lock().callbacks.remove(&key);
if let Some(callbacks) = callbacks { if let Some(callbacks) = callbacks {
for (subscription_id, mut callback) in callbacks { for (subscription_id, mut callback) in callbacks {
@ -110,7 +107,7 @@ impl<K: Clone + Hash + Eq + Copy, F> CallbackCollection<K, F> {
} }
drop(this); drop(this);
let alive = call_callback(&mut callback, cx); let alive = call_callback(&mut callback);
// If this callback's subscription was dropped while invoking the callback // If this callback's subscription was dropped while invoking the callback
// itself, or if the callback returns false, then just drop the callback. // itself, or if the callback returns false, then just drop the callback.