#53. Add with_context

audulus · Aug 19, 2023 · f74b700 · f74b700
1 parent a9e707b
commit f74b700
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/context.rs b/src/context.rs
@@ -5,6 +5,7 @@ use std::any::TypeId;
 use std::collections::{HashMap, HashSet};
 use std::iter::FromIterator;
 use std::ops;
+use std::cell::RefCell;
 
 pub type LocalSpace = vger::defs::LocalSpace;
 pub type WorldSpace = vger::defs::WorldSpace;
@@ -122,6 +123,14 @@ pub struct Context {
     pub(crate) prev_grab_cursor: bool,
 }
 
+thread_local! {
+    pub(crate) static CONTEXT: RefCell<Context> = RefCell::new(Context::new());
+}
+
+pub(crate) fn with_context<T>(f: impl FnOnce(&mut Context) -> T) -> T {
+    CONTEXT.with(|cx| f(&mut cx.borrow_mut()))
+}
+
 impl Default for Context {
     fn default() -> Self {
         Self::new()