implement refresh token handling

1 files changed, 23 insertions, 4 deletions

diff --git a/src/bin/rbw-agent/actions.rs b/src/bin/rbw-agent/actions.rs
index 81e2597..4662164 100644
--- a/src/bin/rbw-agent/actions.rs
+++ b/src/bin/rbw-agent/actions.rs
@@ -133,13 +133,32 @@ pub async fn sync(sock: &mut crate::sock::Sock) -> anyhow::Result<()> {
         .context("failed to load local database")?;
 
     let access_token = if let Some(access_token) = &db.access_token {
-        access_token
+        access_token.clone()
     } else {
         return Err(anyhow::anyhow!("failed to find access token in db"));
     };
-    let (protected_key, ciphers) = rbw::actions::sync(&access_token)
-        .await
-        .context("failed to sync database from server")?;
+    let res = rbw::actions::sync(&access_token).await;
+    let res = if let Err(e) = &res {
+        if let rbw::error::Error::RequestUnauthorized = e {
+            if let Some(refresh_token) = &db.refresh_token {
+                let access_token =
+                    rbw::actions::exchange_refresh_token(refresh_token)
+                        .await?;
+                db.access_token = Some(access_token.clone());
+                rbw::actions::sync(&access_token).await
+            } else {
+                return Err(anyhow::anyhow!(
+                    "failed to find refresh token in db"
+                ));
+            }
+        } else {
+            res
+        }
+    } else {
+        res
+    };
+    let (protected_key, ciphers) =
+        res.context("failed to sync database from server")?;
     db.protected_key = Some(protected_key);
     db.ciphers = ciphers;
     db.save_async(&email)