diff options
| author | Ellie Huxtable <ellie@elliehuxtable.com> | 2024-05-21 10:43:08 +0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-21 10:43:08 +0700 |
| commit | 8dc8448de0b9acaf9c3c27556d3c6c4342d8d493 (patch) | |
| tree | a0fc95ebb1bf2fd2f923ac54634c8c021d93e476 /crates/atuin-client/src | |
| parent | chore(deps): bump tokio from 1.36.0 to 1.37.0 (#2033) (diff) | |
| download | atuin-8dc8448de0b9acaf9c3c27556d3c6c4342d8d493.zip | |
fix(daemon): do not try to sync if logged out (#2037)
* fix(daemon): do not try to sync if logged out
I've also added Settings::logged_in, as there are a few places where we
switch on login state.
* make session_token a function
Diffstat (limited to 'crates/atuin-client/src')
| -rw-r--r-- | crates/atuin-client/src/record/sync.rs | 10 | ||||
| -rw-r--r-- | crates/atuin-client/src/settings.rs | 36 | ||||
| -rw-r--r-- | crates/atuin-client/src/sync.rs | 2 |
3 files changed, 25 insertions, 23 deletions
diff --git a/crates/atuin-client/src/record/sync.rs b/crates/atuin-client/src/record/sync.rs index 234c6442..c3794b59 100644 --- a/crates/atuin-client/src/record/sync.rs +++ b/crates/atuin-client/src/record/sync.rs @@ -55,7 +55,10 @@ pub async fn diff( ) -> Result<(Vec<Diff>, RecordStatus), SyncError> { let client = Client::new( &settings.sync_address, - &settings.session_token, + settings + .session_token() + .map_err(|e| SyncError::RemoteRequestError { msg: e.to_string() })? + .as_str(), settings.network_connect_timeout, settings.network_timeout, ) @@ -270,7 +273,10 @@ pub async fn sync_remote( ) -> Result<(i64, Vec<RecordId>), SyncError> { let client = Client::new( &settings.sync_address, - &settings.session_token, + settings + .session_token() + .map_err(|e| SyncError::RemoteRequestError { msg: e.to_string() })? + .as_str(), settings.network_connect_timeout, settings.network_timeout, ) diff --git a/crates/atuin-client/src/settings.rs b/crates/atuin-client/src/settings.rs index 98a40496..a97d05eb 100644 --- a/crates/atuin-client/src/settings.rs +++ b/crates/atuin-client/src/settings.rs @@ -1,10 +1,5 @@ use std::{ - collections::HashMap, - convert::TryFrom, - fmt, - io::prelude::*, - path::{Path, PathBuf}, - str::FromStr, + collections::HashMap, convert::TryFrom, fmt, io::prelude::*, path::PathBuf, str::FromStr, }; use atuin_common::record::HostId; @@ -458,11 +453,6 @@ pub struct Settings { #[serde(default)] pub daemon: Daemon, - - // This is automatically loaded when settings is created. Do not set in - // config! Keep secrets and settings apart. - #[serde(skip)] - pub session_token: String, } impl Settings { @@ -568,6 +558,21 @@ impl Settings { } } + pub fn logged_in(&self) -> bool { + let session_path = self.session_path.as_str(); + + PathBuf::from(session_path).exists() + } + + pub fn session_token(&self) -> Result<String> { + if !self.logged_in() { + return Err(eyre!("Tried to load session; not logged in")); + } + + let session_path = self.session_path.as_str(); + Ok(fs_err::read_to_string(session_path)?) + } + #[cfg(feature = "check-update")] fn needs_update_check(&self) -> Result<bool> { let last_check = Settings::last_version_check()?; @@ -688,7 +693,6 @@ impl Settings { )? .set_default("scroll_context_lines", 1)? .set_default("shell_up_key_binding", false)? - .set_default("session_token", "")? .set_default("workspaces", false)? .set_default("ctrl_n_shortcuts", false)? .set_default("secrets_filter", true)? @@ -778,14 +782,6 @@ impl Settings { let session_path = shellexpand::full(&session_path)?; settings.session_path = session_path.to_string(); - // Finally, set the auth token - if Path::new(session_path.to_string().as_str()).exists() { - let token = fs_err::read_to_string(session_path.to_string())?; - settings.session_token = token.trim().to_string(); - } else { - settings.session_token = String::from("not logged in"); - } - Ok(settings) } diff --git a/crates/atuin-client/src/sync.rs b/crates/atuin-client/src/sync.rs index 0a5021cb..c2377baa 100644 --- a/crates/atuin-client/src/sync.rs +++ b/crates/atuin-client/src/sync.rs @@ -191,7 +191,7 @@ async fn sync_upload( pub async fn sync(settings: &Settings, force: bool, db: &impl Database) -> Result<()> { let client = api_client::Client::new( &settings.sync_address, - &settings.session_token, + settings.session_token()?.as_str(), settings.network_connect_timeout, settings.network_timeout, )?; |
