1 files changed, 39 insertions, 3 deletions
diff --git a/filamento/src/roster.rs b/filamento/src/roster.rs
index 99682b1..6b61e10 100644
--- a/filamento/src/roster.rs
+++ b/filamento/src/roster.rs
@@ -1,4 +1,4 @@
-use std::collections::HashSet;
+use std::{collections::HashSet, fmt::Display};
 
 use jid::JID;
 use rusqlite::{
@@ -11,7 +11,8 @@ pub struct ContactUpdate {
     pub groups: HashSet<String>,
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, PartialEq, Eq)]
+#[cfg_attr(feature = "reactive_stores", derive(reactive_stores::Store))]
 pub struct Contact {
     // jid is the id used to reference everything, but not the primary key
     pub user_jid: JID,
@@ -21,19 +22,30 @@ pub struct Contact {
     // TODO: avatar, nickname
     /// nickname picked by contact
     // nickname: Option<String>,
+    #[cfg_attr(feature = "reactive_stores", store(key: String = |group| group.clone()))]
     pub groups: HashSet<String>,
 }
 
-#[derive(Debug, Clone)]
+#[derive(Debug, Clone, PartialEq, Eq)]
+/// Contact subscription state.
 pub enum Subscription {
+    /// No subscriptions.
     None,
+    /// Pending outgoing subscription request.
     PendingOut,
+    /// Pending incoming subscription request.
     PendingIn,
+    /// Pending incoming & pending outgoing subscription requests.
     PendingInPendingOut,
+    /// Subscribed to.
     OnlyOut,
+    /// Subscription from.
     OnlyIn,
+    /// Subscribed to & pending incoming subscription request.
     OutPendingIn,
+    /// Subscription from & pending outgoing subscription request.
     InPendingOut,
+    /// Buddy (subscriptions both ways).
     Buddy,
     // TODO: perhaps don't need, just emit event to remove contact
     // Remove,
@@ -79,6 +91,30 @@ impl FromSql for Subscription {
     }
 }
 
+impl Display for Subscription {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        match self {
+            Subscription::None => write!(f, "No Subscriptions"),
+            Subscription::PendingOut => write!(f, "Pending Outgoing Subscription Request"),
+            Subscription::PendingIn => write!(f, "Pending Incoming Subscription Request"),
+            Subscription::PendingInPendingOut => write!(
+                f,
+                "Pending Incoming & Pending Outgoing Subscription Requests"
+            ),
+            Subscription::OnlyOut => write!(f, "Subscribed To"),
+            Subscription::OnlyIn => write!(f, "Subscription From"),
+            Subscription::OutPendingIn => {
+                write!(f, "Subscribed To & Pending Incoming Subscription Request")
+            }
+            Subscription::InPendingOut => write!(
+                f,
+                "Subscription From & Pending Outgoing Subscription Request"
+            ),
+            Subscription::Buddy => write!(f, "Buddy (Subscriptions Both Ways)"),
+        }
+    }
+}
+
 // none
 // >
 // >>