From 0ca5165b7329b67d71b7218c87fcfd72993322c2 Mon Sep 17 00:00:00 2001
From: alemi <me@alemi.dev>
Date: Thu, 8 Aug 2024 21:55:52 +0200
Subject: [PATCH] feat: improve InternallyMutable

---
 src/buffer/tools.rs | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/src/buffer/tools.rs b/src/buffer/tools.rs
index f4a8fd8..17249f0 100644
--- a/src/buffer/tools.rs
+++ b/src/buffer/tools.rs
@@ -51,26 +51,38 @@ pub async fn select_buffer(
 
 /// wraps sender and receiver to allow mutable field with immutable ref
 #[derive(Debug)]
-pub struct InternallyMutable<T: Clone> {
+pub struct InternallyMutable<T> {
 	getter: tokio::sync::watch::Receiver<T>,
 	setter: tokio::sync::watch::Sender<T>,
 }
 
-impl<T: Clone + Default> Default for InternallyMutable<T> {
+impl<T: Default> Default for InternallyMutable<T> {
 	fn default() -> Self {
-		let (tx, rx) = tokio::sync::watch::channel(T::default());
+		Self::new(T::default())
+	}
+}
+
+impl<T> InternallyMutable<T> {
+	pub fn new(init: T) -> Self {
+		let (tx, rx) = tokio::sync::watch::channel(init);
 		InternallyMutable {
 			getter: rx,
 			setter: tx,
 		}
 	}
-}
 
-impl<T: Clone> InternallyMutable<T> {
 	pub fn set(&self, state: T) -> T {
 		self.setter.send_replace(state)
 	}
 
+	pub async fn wait(&self) {
+		let mut new = self.getter.clone();
+		new.changed().await; // first time unlocks immediately
+		new.changed().await;
+	}
+}
+
+impl<T: Clone> InternallyMutable<T> {
 	pub fn get(&self) -> T {
 		self.getter.borrow().clone()
 	}