summaryrefslogtreecommitdiffstats
path: root/src/command/client/sync/login.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/command/client/sync/login.rs')
-rw-r--r--src/command/client/sync/login.rs75
1 files changed, 75 insertions, 0 deletions
diff --git a/src/command/client/sync/login.rs b/src/command/client/sync/login.rs
new file mode 100644
index 00000000..efc9c590
--- /dev/null
+++ b/src/command/client/sync/login.rs
@@ -0,0 +1,75 @@
+use std::io;
+
+use atuin_common::api::LoginRequest;
+use clap::AppSettings;
+use clap::Parser;
+use eyre::Result;
+use tokio::{fs::File, io::AsyncWriteExt};
+
+use atuin_client::api_client;
+use atuin_client::settings::Settings;
+
+#[derive(Parser)]
+#[clap(setting(AppSettings::DeriveDisplayOrder))]
+pub struct Cmd {
+ #[clap(long, short)]
+ pub username: Option<String>,
+
+ #[clap(long, short)]
+ pub password: Option<String>,
+
+ /// The encryption key for your account
+ #[clap(long, short)]
+ pub key: Option<String>,
+}
+
+fn get_input() -> Result<String> {
+ let mut input = String::new();
+ io::stdin().read_line(&mut input)?;
+ Ok(input.trim_end_matches(&['\r', '\n'][..]).to_string())
+}
+
+impl Cmd {
+ pub async fn run(&self, settings: &Settings) -> Result<()> {
+ let session_path = atuin_common::utils::data_dir().join("session");
+
+ if session_path.exists() {
+ println!(
+ "You are already logged in! Please run 'atuin logout' if you wish to login again"
+ );
+
+ return Ok(());
+ }
+
+ let username = or_user_input(&self.username, "username");
+ let password = or_user_input(&self.password, "password");
+ let key = or_user_input(&self.key, "encryption key");
+
+ let session = api_client::login(
+ settings.sync_address.as_str(),
+ LoginRequest { username, password },
+ )
+ .await?;
+
+ let session_path = settings.session_path.as_str();
+ let mut file = File::create(session_path).await?;
+ file.write_all(session.session.as_bytes()).await?;
+
+ let key_path = settings.key_path.as_str();
+ let mut file = File::create(key_path).await?;
+ file.write_all(key.as_bytes()).await?;
+
+ println!("Logged in!");
+
+ Ok(())
+ }
+}
+
+pub(super) fn or_user_input(value: &'_ Option<String>, name: &'static str) -> String {
+ value.clone().unwrap_or_else(|| read_user_input(name))
+}
+
+fn read_user_input(name: &'static str) -> String {
+ eprint!("Please enter {}: ", name);
+ get_input().expect("Failed to read from input")
+}