refactor

2 files changed, 26 insertions, 54 deletions

diff --git a/src/bin/rbw-agent/actions.rs b/src/bin/rbw-agent/actions.rs
index 9559f73..75c1c25 100644
--- a/src/bin/rbw-agent/actions.rs
+++ b/src/bin/rbw-agent/actions.rs
@@ -132,28 +132,18 @@ pub async fn sync(sock: &mut crate::sock::Sock) -> anyhow::Result<()> {
     } else {
         return Err(anyhow::anyhow!("failed to find access token in db"));
     };
-    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_async(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
-        }
+    let refresh_token = if let Some(refresh_token) = &db.refresh_token {
+        refresh_token.clone()
     } else {
-        res
+        return Err(anyhow::anyhow!("failed to find refresh token in db"));
     };
-    let (protected_key, ciphers) =
-        res.context("failed to sync database from server")?;
+    let (access_token, protected_key, ciphers) =
+        rbw::actions::sync(&access_token, &refresh_token)
+            .await
+            .context("failed to sync database from server")?;
+    if let Some(access_token) = access_token {
+        db.access_token = Some(access_token);
+    }
     db.protected_key = Some(protected_key);
     db.ciphers = ciphers;
     db.save_async(&email)
diff --git a/src/bin/rbw/commands.rs b/src/bin/rbw/commands.rs
index acf3f32..253ce5d 100644
--- a/src/bin/rbw/commands.rs
+++ b/src/bin/rbw/commands.rs
@@ -114,8 +114,9 @@ pub fn add(name: &str, username: Option<&str>) -> anyhow::Result<()> {
     let email = config_email()?;
     let mut db = rbw::db::Db::load(&email)?;
     // unwrap is safe here because the call to unlock above is guaranteed to
-    // populate it or error
-    let access_token = db.access_token.unwrap();
+    // populate these or error
+    let access_token = db.access_token.as_ref().unwrap();
+    let refresh_token = db.refresh_token.as_ref().unwrap();
 
     let name = crate::actions::encrypt(name)?;
 
@@ -153,21 +154,11 @@ pub fn add(name: &str, username: Option<&str>) -> anyhow::Result<()> {
         notes,
     };
 
-    let res = rbw::actions::add(&access_token, &cipher);
-    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)?;
-                db.access_token = Some(access_token.clone());
-                db.save(&email).context("failed to save database")?;
-                rbw::actions::add(&access_token, &cipher)?;
-            } else {
-                return Err(anyhow::anyhow!(
-                    "failed to find refresh token in db"
-                ));
-            }
-        }
+    if let Some(access_token) =
+        rbw::actions::add(&access_token, &refresh_token, &cipher)?
+    {
+        db.access_token = Some(access_token);
+        db.save(&email).context("failed to save database")?;
     }
 
     crate::actions::sync()?;
@@ -190,8 +181,9 @@ pub fn generate(
         let email = config_email()?;
         let mut db = rbw::db::Db::load(&email)?;
         // unwrap is safe here because the call to unlock above is guaranteed
-        // to populate it or error
-        let access_token = db.access_token.unwrap();
+        // to populate these or error
+        let access_token = db.access_token.as_ref().unwrap();
+        let refresh_token = db.refresh_token.as_ref().unwrap();
 
         let name = crate::actions::encrypt(name)?;
         let username = username
@@ -208,21 +200,11 @@ pub fn generate(
             notes: None,
         };
 
-        let res = rbw::actions::add(&access_token, &cipher);
-        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)?;
-                    db.access_token = Some(access_token.clone());
-                    db.save(&email).context("failed to save database")?;
-                    rbw::actions::add(&access_token, &cipher)?;
-                } else {
-                    return Err(anyhow::anyhow!(
-                        "failed to find refresh token in db"
-                    ));
-                }
-            }
+        if let Some(access_token) =
+            rbw::actions::add(&access_token, &refresh_token, &cipher)?
+        {
+            db.access_token = Some(access_token);
+            db.save(&email).context("failed to save database")?;
         }
 
         crate::actions::sync()?;