fix: lowercase addresses

WalletConnect · Feb 9, 2024 · 21da085 · 21da085
1 parent cf06689
commit 21da085
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 5 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -12,7 +12,7 @@ build = "build.rs"
 
 [dependencies]
 wc = { git = "https://github.com/WalletConnect/utils-rs.git", tag = "v0.7.0", features = ["geoip", "geoblock"] }
-relay_rpc = { git = "https://github.com/WalletConnect/WalletConnectRust.git", tag = "v0.26.0", features = ["cacao"] }
+relay_rpc = { git = "https://github.com/WalletConnect/WalletConnectRust.git", tag = "v0.26.1", features = ["cacao"] }
 
 aws-config = "0.56"
 aws-sdk-s3 = "0.31"

diff --git a/src/stores/keys.rs b/src/stores/keys.rs
@@ -180,10 +180,38 @@ impl KeysPersistentStorage for MongoPersistentStorage {
 
         match MongoKeys::find_one_and_update(&self.db, filter, update, None).await? {
             Some(_) => Ok(()),
-            None => Err(StoreError::NotFound(
-                "Account".to_string(),
-                account.to_string(),
-            )),
+            None => {
+
+                // TODO: This is a temporary fix to handle the case where the account is not found becuase we lowercased it
+                return if(!account.starts_with("eip155")){
+                    Err(StoreError::NotFound(
+                        "Account".to_string(),
+                        account.to_string(),
+                    ))
+                } else {
+
+                    let lowercase_account = account.to_lowercase();
+                    let filter = doc! {
+                        "account": &lowercase_account,
+                    };
+
+                    let update = doc! {
+                        "$pull": {
+                            "identities" : {
+                                "identity_key": &identity_key,
+                            }
+                        }
+                    };
+
+                    match MongoKeys::find_one_and_update(&self.db, filter, update, None).await? {
+                        Some(_) => Ok(()),
+                        None => Err(StoreError::NotFound(
+                            "Account".to_string(),
+                            lowercase_account.to_string(),
+                        )),
+                    }
+                }
+            }
         }
     }