Refactor things a tiny little bit

src/streaming.rs

@@ -13,32 +13,29 @@ pub async fn start_stream() -> Result<()> {
         connect_async("wss://bsky.social/xrpc/com.atproto.sync.subscribeRepos").await?;
 
     while let Some(Ok(tungstenite::Message::Binary(message))) = stream.next().await {
-        let commit = match parse_commit_message(&message) {
+        if let Err(e) = handle_message(&message).await {
-            Ok(Some(commit)) => commit,
+            println!("Error handling a message: {:?}", e);
-            Ok(None) => continue,
-            Err(e) => {
-                println!("Couldn't parse commit: {:?}", e);
-                continue;
-            }
-        };
-
-        let post_messages = extract_post_messages(&commit).await;
-        match post_messages {
-            Ok(post_messages) => {
-                if !post_messages.is_empty() {
-                    println!("{:?}", post_messages);
-                }
-            }
-            Err(e) => {
-                println!("Coudln't extract post messages: {:?}", e);
-            }
         }
     }
 
     Ok(())
 }
 
-fn parse_commit_message(message: &[u8]) -> Result<Option<Commit>> {
+async fn handle_message(message: &[u8]) -> Result<()> {
+    let commit = match parse_commit_from_message(&message)? {
+        Some(commit) => commit,
+        None => return Ok(()),
+    };
+
+    let post_operations = extract_post_operations(&commit).await?;
+    for operation in &post_operations {
+        println!("{:?}", operation);
+    }
+
+    Ok(())
+}
+
+fn parse_commit_from_message(message: &[u8]) -> Result<Option<Commit>> {
     match Frame::try_from(message)? {
         Frame::Message(message) => match message.body {
             Message::Commit(commit) => Ok(Some(*commit)),
@@ -49,23 +46,21 @@ fn parse_commit_message(message: &[u8]) -> Result<Option<Commit>> {
 }
 
 #[derive(Debug)]
-enum Action {
+enum PostOperation {
-    Create,
+    Create {
-    Delete,
+        author_did: String,
+        cid: String,
+        uri: String,
+        languages: Vec<String>,
+        text: String,
+    },
+    Delete {
+        cid: String,
+    },
 }
 
-#[derive(Debug)]
+async fn extract_post_operations(commit: &Commit) -> Result<Vec<PostOperation>> {
-struct PostMessage {
+    let mut operations = Vec::new();
-    action: Action,
-    author_did: String,
-    cid: String,
-    uri: String,
-    languages: Vec<String>,
-    text: String,
-}
-
-async fn extract_post_messages(commit: &Commit) -> Result<Vec<PostMessage>> {
-    let mut posts = Vec::new();
 
     let (items, _) = rs_car::car_read_all(&mut commit.blocks.as_slice(), true).await?;
     for op in &commit.ops {
@@ -74,23 +69,24 @@ async fn extract_post_messages(commit: &Commit) -> Result<Vec<PostMessage>> {
             continue;
         }
 
+        let cid = op.cid.expect("cid is not there, what").to_string();
+
         if let Some((_, item)) = items.iter().find(|(cid, _)| Some(*cid) == op.cid) {
             let record: Record = ciborium::from_reader(&mut item.as_slice())?;
 
-            posts.push(PostMessage {
+            operations.push(match op.action.as_str() {
-                action: if op.action == "create" {
+                "create" => PostOperation::Create {
-                    Action::Create
+                    languages: record.langs.unwrap_or_else(Vec::new),
-                } else {
+                    text: record.text,
-                    Action::Delete
+                    author_did: commit.repo.clone(),
+                    cid,
+                    uri: format!("at://{}/{}", commit.repo, op.path),
                 },
-                languages: record.langs.unwrap_or_else(Vec::new),
+                "delete" => PostOperation::Delete { cid },
-                text: record.text,
+                _ => unreachable!(),
-                author_did: commit.repo.clone(),
+            });
-                cid: op.cid.expect("cid is not there, what").to_string(),
-                uri: format!("at://{}/{}", commit.repo, op.path),
-            })
         }
     }
 
-    Ok(posts)
+    Ok(operations)
 }