handle: actually sweep PR numbers from database

Cuprate · Jul 10, 2024 · 961ee1b · 961ee1b
1 parent 23ba6ec
commit 961ee1b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/command/handle.rs b/src/command/handle.rs
@@ -260,7 +260,7 @@ impl Command {
     /// TODO
     #[instrument]
     pub async fn handle_sweep(db: Arc<Database>) -> RoomMessageEventContent {
-        let db = db.write().await;
+        let mut db = db.write().await;
         return_if_empty!(db);
 
         let mut keep = vec![];
@@ -269,12 +269,14 @@ impl Command {
         for pr in db.keys() {
             // Check if open PR.
             match pr_is_open(*pr).await {
-                Ok(true) => keep.push(pr),
-                Ok(false) => sweep.push(pr),
+                Ok(true) => keep.push(*pr),
+                Ok(false) => sweep.push(*pr),
                 Err(e) => return RoomMessageEventContent::text_plain(e.to_string()),
             }
         }
 
+        db.retain(|k, _| keep.contains(k));
+
         let msg = format!("Keeping: {keep:?}, sweeping: {sweep:?}");
         trace!(msg);
         RoomMessageEventContent::text_plain(msg)