From 2ccbf2eafeaacf727d223a226789f3776dca18d9 Mon Sep 17 00:00:00 2001
From: Sergey Isaev <isvforall@gmail.com>
Date: Thu, 6 Aug 2020 22:36:38 +0300
Subject: [PATCH] Add initial version of the login command

---
 leo/commands/login.rs | 154 ++++++++++++++++++++++++++++++++++++++++++
 leo/commands/mod.rs   |   3 +
 leo/main.rs           |   2 +
 3 files changed, 159 insertions(+)
 create mode 100644 leo/commands/login.rs

diff --git a/leo/commands/login.rs b/leo/commands/login.rs
new file mode 100644
index 0000000000..60f43aba2f
--- /dev/null
+++ b/leo/commands/login.rs
@@ -0,0 +1,154 @@
+use crate::{cli::CLI, cli_types::*};
+use std::{
+    fs::{create_dir, File},
+    io::prelude::*,
+    path::Path,
+};
+
+const LEO_CREDENTIALS_DIR: &str = ".leo";
+const LEO_CREDENTIALS_FILE: &str = "credentials";
+
+#[derive(Debug)]
+pub struct LoginCommand;
+
+impl CLI for LoginCommand {
+    type Options = Option<String>;
+    type Output = ();
+
+    const ABOUT: AboutType = "Login to the package manager (*)";
+    const ARGUMENTS: &'static [ArgumentType] = &[
+        // (name, description, required, index)
+        ("NAME", "Sets token for login to the package manager", false, 1u64),
+    ];
+    const FLAGS: &'static [FlagType] = &[];
+    const NAME: NameType = "login";
+    const OPTIONS: &'static [OptionType] = &[];
+    const SUBCOMMANDS: &'static [SubCommandType] = &[];
+
+    fn parse(arguments: &clap::ArgMatches) -> Result<Self::Options, crate::errors::CLIError> {
+        match arguments.value_of("NAME") {
+            Some(name) => Ok(Some(name.to_string())),
+            None => Ok(None),
+        }
+    }
+
+    fn output(options: Self::Options) -> Result<Self::Output, crate::errors::CLIError> {
+        let token = match options {
+            Some(token) => token,
+            None => {
+                // TODO JWT or make a request to login
+                unimplemented!()
+            }
+        };
+
+        match Path::new(LEO_CREDENTIALS_DIR).exists() {
+            true => {}
+            false => {
+                create_dir(LEO_CREDENTIALS_DIR)?;
+            }
+        }
+
+        let mut credentials = File::create(&format!("{}/{}", LEO_CREDENTIALS_DIR, LEO_CREDENTIALS_FILE))?;
+        let mut buf = String::new();
+        buf.push_str(token.as_str());
+        credentials.write_all(&buf.as_bytes())?;
+        Ok(())
+    }
+}
+
+// Run tests in single thread
+#[cfg(test)]
+mod tests {
+
+    use super::*;
+
+    use std::{
+        env,
+        fs::{remove_dir, remove_file},
+        io,
+    };
+
+    const TEST_DIR: &str = ".test";
+
+    fn setup() -> Result<(), io::Error> {
+        create_dir(TEST_DIR)?;
+        env::set_current_dir(TEST_DIR)?;
+        Ok(())
+    }
+
+    fn clean() -> Result<(), io::Error> {
+        remove_file(&format!("{}/{}", LEO_CREDENTIALS_DIR, LEO_CREDENTIALS_FILE))?;
+        remove_dir(LEO_CREDENTIALS_DIR)?;
+        env::set_current_dir("../")?;
+        remove_dir(TEST_DIR)?;
+        Ok(())
+    }
+
+    fn get_token() -> Result<String, io::Error> {
+        let mut credentials = File::open(&format!("{}/{}", LEO_CREDENTIALS_DIR, LEO_CREDENTIALS_FILE))?;
+        let mut buf = String::new();
+        credentials.read_to_string(&mut buf)?;
+        Ok(buf)
+    }
+
+    fn create_token(token: &str) -> Result<(), io::Error> {
+        let mut f = File::create(&format!("{}/{}", LEO_CREDENTIALS_DIR, LEO_CREDENTIALS_FILE))?;
+        f.write_all(token.as_bytes())?;
+        Ok(())
+    }
+
+    #[test]
+    fn test_credential_dir_exists() -> Result<(), io::Error> {
+        setup()?;
+        create_dir(LEO_CREDENTIALS_DIR)?;
+
+        let token = "SOME_TOKEN".to_string();
+        let options = Some(token.clone());
+        LoginCommand::output(options).unwrap();
+        assert_eq!(token, get_token()?);
+        clean()?;
+        Ok(())
+    }
+
+    #[test]
+    fn test_credential_file_exists() -> Result<(), io::Error> {
+        setup()?;
+        create_dir(LEO_CREDENTIALS_DIR)?;
+        create_token("OLD_TOKEN")?;
+
+        let token = "NEW_TOKEN".to_string();
+        let options = Some(token.clone());
+        LoginCommand::output(options).unwrap();
+
+        assert_eq!(token, get_token()?);
+        clean()?;
+        Ok(())
+    }
+
+    #[test]
+    fn test_credential_dir_does_not_exist() -> Result<(), io::Error> {
+        setup()?;
+
+        let token = "SOME_TOKEN".to_string();
+        let options = Some(token.clone());
+        LoginCommand::output(options).unwrap();
+
+        assert_eq!(token, get_token()?);
+        clean()?;
+        Ok(())
+    }
+
+    #[test]
+    fn test_credential_file_does_not_exist() -> Result<(), io::Error> {
+        setup()?;
+        create_dir(LEO_CREDENTIALS_DIR)?;
+
+        let token = "SOME_TOKEN".to_string();
+        let options = Some(token.clone());
+        LoginCommand::output(options).unwrap();
+
+        assert_eq!(token, get_token()?);
+        clean()?;
+        Ok(())
+    }
+}
diff --git a/leo/commands/mod.rs b/leo/commands/mod.rs
index 37021057fc..d6181e024e 100644
--- a/leo/commands/mod.rs
+++ b/leo/commands/mod.rs
@@ -16,6 +16,9 @@ pub use self::lint::*;
 pub mod load;
 pub use self::load::*;
 
+pub mod login;
+pub use self::login::*;
+
 pub mod new;
 pub use self::new::*;
 
diff --git a/leo/main.rs b/leo/main.rs
index 2af0c6973a..d67a1af4b4 100644
--- a/leo/main.rs
+++ b/leo/main.rs
@@ -33,6 +33,7 @@ fn main() -> Result<(), CLIError> {
             PublishCommand::new().display_order(10),
             DeployCommand::new().display_order(11),
             CleanCommand::new().display_order(12),
+            LoginCommand::new().display_order(13),
         ])
         .set_term_width(0)
         .get_matches();
@@ -51,6 +52,7 @@ fn main() -> Result<(), CLIError> {
         ("publish", Some(arguments)) => PublishCommand::process(arguments),
         ("deploy", Some(arguments)) => DeployCommand::process(arguments),
         ("clean", Some(arguments)) => CleanCommand::process(arguments),
+        ("login", Some(arguments)) => LoginCommand::process(arguments),
         _ => unreachable!(),
     }
 }