aboutsummaryrefslogtreecommitdiffstats
path: root/crates/atuin-client/src
diff options
context:
space:
mode:
authorEllie Huxtable <ellie@elliehuxtable.com>2024-05-21 10:43:08 +0700
committerGitHub <noreply@github.com>2024-05-21 10:43:08 +0700
commit8dc8448de0b9acaf9c3c27556d3c6c4342d8d493 (patch)
treea0fc95ebb1bf2fd2f923ac54634c8c021d93e476 /crates/atuin-client/src
parentchore(deps): bump tokio from 1.36.0 to 1.37.0 (#2033) (diff)
downloadatuin-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.rs10
-rw-r--r--crates/atuin-client/src/settings.rs36
-rw-r--r--crates/atuin-client/src/sync.rs2
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,
)?;