Relationship::Root treated like Relationship::Reply

src/globals.rs

@@ -176,7 +176,7 @@ impl Globals {
         let mut output: Vec<Id> = Vec::new();
         if let Some(vec) = GLOBALS.relationships.blocking_read().get(&id) {
             for (id, relationship) in vec.iter() {
-                if *relationship == Relationship::Reply {
+                if *relationship == Relationship::Reply || *relationship == Relationship::Root {
                     output.push(*id);
                 }
             }

src/process.rs

@@ -156,6 +156,22 @@ pub async fn process_new_event(
             Globals::add_relationship(id, event.id, Relationship::Reply).await;
         }
 
+        // replies to root
+        if let Some((id, _)) = event.replies_to_root() {
+            if from_relay {
+                let db_event_relationship = DbEventRelationship {
+                    original: event.id.as_hex_string(),
+                    refers_to: id.as_hex_string(),
+                    relationship: "root".to_string(),
+                    content: None,
+                };
+                db_event_relationship.insert().await?;
+            }
+
+            // Insert into relationships
+            Globals::add_relationship(id, event.id, Relationship::Root).await;
+        }
+
         // mentions
         for (id, _) in event.mentions() {
             if from_relay {

src/relationship.rs

@@ -1,6 +1,7 @@
 /// A relationship between events
 #[derive(Clone, Debug, PartialEq, Eq)]
 pub enum Relationship {
+    Root,
     Reply,
     Mention,
     Reaction(String),