diff options
| author | Conrad Ludgate <conrad.ludgate@truelayer.com> | 2022-04-13 18:08:49 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-04-13 18:08:49 +0100 |
| commit | 3c5fbc573443a7a4e59a0b898e1586d219f105eb (patch) | |
| tree | 20a9498f7b4a81cb44810c84c2067d4eee367d25 /atuin-client/src | |
| parent | remove default db uri (#299) (diff) | |
| download | atuin-3c5fbc573443a7a4e59a0b898e1586d219f105eb.zip | |
provide better error messages (#300)
Diffstat (limited to 'atuin-client/src')
| -rw-r--r-- | atuin-client/src/database.rs | 3 | ||||
| -rw-r--r-- | atuin-client/src/encryption.rs | 19 | ||||
| -rw-r--r-- | atuin-client/src/settings.rs | 8 |
3 files changed, 16 insertions, 14 deletions
diff --git a/atuin-client/src/database.rs b/atuin-client/src/database.rs index 3ca61717..8b950de9 100644 --- a/atuin-client/src/database.rs +++ b/atuin-client/src/database.rs @@ -9,6 +9,7 @@ use eyre::Result; use itertools::Itertools; use regex::Regex; +use fs_err as fs; use sqlx::sqlite::{ SqliteConnectOptions, SqliteJournalMode, SqlitePool, SqlitePoolOptions, SqliteRow, }; @@ -62,7 +63,7 @@ impl Sqlite { let create = !path.exists(); if create { if let Some(dir) = path.parent() { - std::fs::create_dir_all(dir)?; + fs::create_dir_all(dir)?; } } diff --git a/atuin-client/src/encryption.rs b/atuin-client/src/encryption.rs index 5721c2f9..d91ad076 100644 --- a/atuin-client/src/encryption.rs +++ b/atuin-client/src/encryption.rs @@ -8,11 +8,11 @@ // clients must share the secret in order to be able to sync, as it is needed // to decrypt -use std::fs::File; +use fs_err as fs; use std::io::prelude::*; use std::path::PathBuf; -use eyre::{eyre, Result}; +use eyre::{eyre, Context, Result}; use sodiumoxide::crypto::secretbox; use crate::history::History; @@ -30,7 +30,7 @@ pub fn new_key(settings: &Settings) -> Result<secretbox::Key> { let key = secretbox::gen_key(); let encoded = encode_key(key.clone())?; - let mut file = File::create(path)?; + let mut file = fs::File::create(path)?; file.write_all(encoded.as_bytes())?; Ok(key) @@ -41,7 +41,7 @@ pub fn load_key(settings: &Settings) -> Result<secretbox::Key> { let path = settings.key_path.as_str(); let key = if PathBuf::from(path).exists() { - let key = std::fs::read_to_string(path)?; + let key = fs_err::read_to_string(path)?; decode_key(key)? } else { new_key(settings)? @@ -54,13 +54,13 @@ pub fn load_encoded_key(settings: &Settings) -> Result<String> { let path = settings.key_path.as_str(); if PathBuf::from(path).exists() { - let key = std::fs::read_to_string(path)?; + let key = fs::read_to_string(path)?; Ok(key) } else { let key = secretbox::gen_key(); let encoded = encode_key(key)?; - let mut file = File::create(path)?; + let mut file = fs::File::create(path)?; file.write_all(encoded.as_bytes())?; Ok(encoded) @@ -68,15 +68,16 @@ pub fn load_encoded_key(settings: &Settings) -> Result<String> { } pub fn encode_key(key: secretbox::Key) -> Result<String> { - let buf = rmp_serde::to_vec(&key)?; + let buf = rmp_serde::to_vec(&key).wrap_err("could not encode key to message pack")?; let buf = base64::encode(buf); Ok(buf) } pub fn decode_key(key: String) -> Result<secretbox::Key> { - let buf = base64::decode(key)?; - let buf: secretbox::Key = rmp_serde::from_read_ref(&buf)?; + let buf = base64::decode(key).wrap_err("encryption key is not a valid base64 encoding")?; + let buf: secretbox::Key = rmp_serde::from_read_ref(&buf) + .wrap_err("encryption key is not a valid message pack encoding")?; Ok(buf) } diff --git a/atuin-client/src/settings.rs b/atuin-client/src/settings.rs index 0e115909..537b1ca3 100644 --- a/atuin-client/src/settings.rs +++ b/atuin-client/src/settings.rs @@ -1,4 +1,4 @@ -use std::fs::{create_dir_all, File}; +use fs_err::{create_dir_all, File}; use std::io::prelude::*; use std::path::{Path, PathBuf}; @@ -76,7 +76,7 @@ impl Settings { let sync_time_path = data_dir.join("last_sync_time"); - std::fs::write(sync_time_path, Utc::now().to_rfc3339())?; + fs_err::write(sync_time_path, Utc::now().to_rfc3339())?; Ok(()) } @@ -91,7 +91,7 @@ impl Settings { return Ok(Utc.ymd(1970, 1, 1).and_hms(0, 0, 0)); } - let time = std::fs::read_to_string(sync_time_path)?; + let time = fs_err::read_to_string(sync_time_path)?; let time = chrono::DateTime::parse_from_rfc3339(time.as_str())?; Ok(time.with_timezone(&Utc)) @@ -188,7 +188,7 @@ impl Settings { // Finally, set the auth token if Path::new(session_path.to_string().as_str()).exists() { - let token = std::fs::read_to_string(session_path.to_string())?; + 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"); |
