Merge branch 'release/v0.4.0' into stable

Author: peeeuzin

Cargo.lock

@@ -1,14 +1,14 @@
 [package]
 name = "rustbase-cli"
-version = "0.2.0"
+version = "0.4.0"
 edition = "2021"
 authors = ["Rustbase"]
 license = "MIT"
 
 [dependencies]
-tokio = { version = "1.20.1", features = ["macros", "rt-multi-thread", "net", "io-std", "io-util"] }
+tokio = { version = "1.23.1", features = ["macros", "rt-multi-thread", "net", "io-std", "io-util"] }
 bson = "2.4.0"
-rustyline = "10.0.0"
+rustyline = "10.1.1"
 colored = "2.0.0"
 clap = "3.2.19"
 clap_derive = "3.2.18"
@@ -17,4 +17,7 @@ serde = { version = "1.0.147", features = ["derive"] }
 tokio-rustls = "0.23.4"
 rustls-pemfile = "1.0.1"
 rustls = "0.20.7"
-webpki = "0.22.0"
+webpki = "0.22.0"
+scram = "0.6.0"
+unicode-width = "0.1.10"
+rustyline-derive = "0.7.0"

Cargo.toml

@@ -0,0 +1,26 @@
+use scram::ScramClient;
+use tokio::io::{AsyncRead, AsyncWrite};
+
+use super::AuthConfig;
+use super::Rustbase;
+
+pub async fn auth<IO>(auth_config: AuthConfig, client: &mut IO)
+where
+    IO: AsyncRead + AsyncWrite + Unpin,
+{
+    let scram = ScramClient::new(&auth_config.username, &auth_config.password, None);
+    let (scram, client_first) = scram.client_first();
+
+    let server_first = Rustbase::send_and_receive(client, client_first.as_bytes().to_vec()).await;
+    let server_first = String::from_utf8(server_first).unwrap();
+
+    let scram = scram
+        .handle_server_first(&server_first)
+        .expect("Invalid server first message, maybe server is not using scram? or maybe the username is wrong?");
+    let (scram, client_final) = scram.client_final();
+
+    let server_final = Rustbase::send_and_receive(client, client_final.as_bytes().to_vec()).await;
+    let server_final = String::from_utf8(server_final).unwrap();
+
+    scram.handle_server_final(&server_final).unwrap();
+}