feature-flags (#328)

* use feature flags

* fmt

* fix features

* update ci

* fmt

Co-authored-by: Ellie Huxtable <ellie@elliehuxtable.com>

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")
+}