3 files changed, 140 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")
+}
diff --git a/src/command/client/sync/logout.rs b/src/command/client/sync/logout.rs
new file mode 100644
index 00000000..a7e9541d
--- /dev/null
+++ b/src/command/client/sync/logout.rs
@@ -0,0 +1,15 @@
+use eyre::{Context, Result};
+use fs_err::remove_file;
+
+pub fn run() -> Result<()> {
+    let session_path = atuin_common::utils::data_dir().join("session");
+
+    if session_path.exists() {
+        remove_file(session_path.as_path()).context("Failed to remove session file")?;
+        println!("You have logged out!");
+    } else {
+        println!("You are not logged in");
+    }
+
+    Ok(())
+}
diff --git a/src/command/client/sync/register.rs b/src/command/client/sync/register.rs
new file mode 100644
index 00000000..2c60a2e9
--- /dev/null
+++ b/src/command/client/sync/register.rs
@@ -0,0 +1,50 @@
+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 email: Option<String>,
+
+    #[clap(long, short)]
+    pub password: Option<String>,
+}
+
+impl Cmd {
+    pub async fn run(self, settings: &Settings) -> Result<()> {
+        run(settings, &self.username, &self.email, &self.password).await
+    }
+}
+
+pub async fn run(
+    settings: &Settings,
+    username: &Option<String>,
+    email: &Option<String>,
+    password: &Option<String>,
+) -> Result<()> {
+    use super::login::or_user_input;
+    let username = or_user_input(username, "username");
+    let email = or_user_input(email, "email");
+    let password = or_user_input(password, "password");
+
+    let session =
+        api_client::register(settings.sync_address.as_str(), &username, &email, &password).await?;
+
+    let path = settings.session_path.as_str();
+    let mut file = File::create(path).await?;
+    file.write_all(session.session.as_bytes()).await?;
+
+    // Create a new key, and save it to disk
+    let _key = atuin_client::encryption::new_key(settings)?;
+
+    Ok(())
+}