refactor: reorganise code

1 files changed, 39 insertions, 0 deletions

diff --git a/src/message.rs b/src/message.rs
new file mode 100644
index 0000000..e5caed1
--- /dev/null
+++ b/src/message.rs
@@ -0,0 +1,39 @@
+use std::ops::{Deref, DerefMut};
+
+use filamento::{chat::Message, user::User};
+use reactive_stores::ArcStore;
+use uuid::Uuid;
+use leptos::prelude::*;
+
+use crate::{state_store::{StateListener, StateStore}, user::MacawUser};
+
+#[derive(Clone)]
+pub struct MacawMessage {
+    pub message: StateListener<Uuid, ArcStore<Message>>,
+    pub user: MacawUser,
+}
+
+impl MacawMessage {
+    pub fn got_message_and_user(message: Message, user: User) -> Self {
+        let message_state_store: StateStore<Uuid, ArcStore<Message>> =
+            use_context().expect("no message state store");
+        let message = message_state_store.store(message.id, ArcStore::new(message));
+        let user = MacawUser::got_user(user);
+        Self { message, user }
+    }
+}
+
+impl Deref for MacawMessage {
+    type Target = StateListener<Uuid, ArcStore<Message>>;
+
+    fn deref(&self) -> &Self::Target {
+        &self.message
+    }
+}
+
+impl DerefMut for MacawMessage {
+    fn deref_mut(&mut self) -> &mut Self::Target {
+        &mut self.message
+    }
+}
+