refactor: reorganise code

1 files changed, 78 insertions, 0 deletions

diff --git a/src/user.rs b/src/user.rs
new file mode 100644
index 0000000..f55c0dd
--- /dev/null
+++ b/src/user.rs
@@ -0,0 +1,78 @@
+use std::ops::{Deref, DerefMut};
+
+use filamento::user::{User, UserStoreFields};
+use jid::BareJID;
+use reactive_stores::{ArcStore, Store};
+use leptos::prelude::*;
+
+use crate::{client::Client, roster::{Roster, RosterStoreFields}, state_store::{StateListener, StateStore}};
+
+#[derive(Clone)]
+pub struct MacawUser {
+    pub user: StateListener<BareJID, ArcStore<User>>,
+}
+
+impl MacawUser {
+    pub fn got_user(user: User) -> Self {
+        
+        let user_state_store: StateStore<BareJID, ArcStore<User>> =
+            use_context().expect("no user state store");
+        let user = user_state_store.store(user.jid.clone(), ArcStore::new(user));
+        Self { user }
+    }
+}
+
+impl Deref for MacawUser {
+    type Target = StateListener<BareJID, ArcStore<User>>;
+
+    fn deref(&self) -> &Self::Target {
+        &self.user
+    }
+}
+
+impl DerefMut for MacawUser {
+    fn deref_mut(&mut self) -> &mut Self::Target {
+        &mut self.user
+    }
+}
+
+pub const NO_AVATAR: &str = "/assets/no-avatar.png";
+
+pub async fn get_avatar(user: Store<User>) -> String {
+    if let Some(avatar) = &user.read().avatar {
+        let client = use_context::<Client>().expect("client not in context");
+        if let Some(data) = client.file_store.get_src(avatar).await {
+            data
+        } else {
+            NO_AVATAR.to_string()
+        }
+        // TODO: enable avatar fetching
+        // format!("/files/{}", avatar)
+    } else {
+        NO_AVATAR.to_string()
+    }
+}
+
+pub fn get_name(user: Store<User>, note_to_self: bool) -> String {
+    let roster: Store<Roster> = use_context().expect("no roster in context");
+    if note_to_self {
+        let client: Client = use_context().expect("no client in context");
+        if *client.jid == *user.jid().read() {
+            return "Note to self".to_string()
+        }
+    }
+    if let Some(name) = roster
+        .contacts()
+        .read()
+        .get(&user.read().jid)
+        .map(|contact| contact.read().name.clone())
+        .unwrap_or_default()
+    {
+        name.to_string()
+    } else if let Some(nick) = &user.read().nick {
+        nick.to_string()
+    } else {
+        user.read().jid.to_string()
+    }
+}
+