Rotate KeyPackage after processing Welcomes

xmtp · Mar 11, 2024 · 8345016 · 8345016
1 parent 114d132
commit 8345016
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/xmtp_mls/src/client.rs b/xmtp_mls/src/client.rs
@@ -439,6 +439,12 @@ where
             })
             .collect();
 
+        if groups.len() > 0 {
+            if let Err(err) = self.rotate_key_package().await {
+                log::error!("Error rotating key package: {:?}", err);
+            }
+        }
+
         Ok(groups)
     }
 

diff --git a/xmtp_mls/src/subscriptions.rs b/xmtp_mls/src/subscriptions.rs
@@ -88,7 +88,12 @@ where
             })
             .filter_map(|res| async {
                 match res.await {
-                    Ok(group) => Some(group),
+                    Ok(group) => {
+                        if let Err(err) = self.rotate_key_package().await {
+                            log::error!("Error rotating key package: {:?}", err);
+                        }
+                        Some(group)
+                    },
                     Err(err) => {
                         log::error!("Error processing stream entry: {:?}", err);
                         None