From adda436b41dd4e48e1d4ed917a773541cbfb8a3c Mon Sep 17 00:00:00 2001
From: JesusPerez <jpl@jesusperez.pro>
Date: Wed, 1 Sep 2021 20:12:41 +0100
Subject: [PATCH] chore: add ssh_utils

---
 ssh_utils/.gitignore | 10 ++++++
 ssh_utils/Cargo.toml | 12 +++++++
 ssh_utils/README.md  |  3 ++
 ssh_utils/TODO.md    | 11 +++++++
 ssh_utils/src/lib.rs | 74 ++++++++++++++++++++++++++++++++++++++++++++
 5 files changed, 110 insertions(+)
 create mode 100644 ssh_utils/.gitignore
 create mode 100644 ssh_utils/Cargo.toml
 create mode 100644 ssh_utils/README.md
 create mode 100644 ssh_utils/TODO.md
 create mode 100644 ssh_utils/src/lib.rs

diff --git a/ssh_utils/.gitignore b/ssh_utils/.gitignore
new file mode 100644
index 0000000..c3a29c2
--- /dev/null
+++ b/ssh_utils/.gitignore
@@ -0,0 +1,10 @@
+/target
+target
+Cargo.lock
+.cache
+.temp
+.env
+*.log
+.DS_Store
+logs
+tmp
diff --git a/ssh_utils/Cargo.toml b/ssh_utils/Cargo.toml
new file mode 100644
index 0000000..610384c
--- /dev/null
+++ b/ssh_utils/Cargo.toml
@@ -0,0 +1,12 @@
+[package]
+name = "ssh_utils"
+version = "0.1.0"
+authors = ["JesusPerez <jpl@jesusperez.pro>"]
+edition = "2018"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+anyhow = "1.0.40"
+openssh = "0.8.0"
+tokio = { version = "1.5.0", features = ["full"] }
\ No newline at end of file
diff --git a/ssh_utils/README.md b/ssh_utils/README.md
new file mode 100644
index 0000000..56aa081
--- /dev/null
+++ b/ssh_utils/README.md
@@ -0,0 +1,3 @@
+### SSH utils library
+
+SSH utils library
\ No newline at end of file
diff --git a/ssh_utils/TODO.md b/ssh_utils/TODO.md
new file mode 100644
index 0000000..efb642e
--- /dev/null
+++ b/ssh_utils/TODO.md
@@ -0,0 +1,11 @@
+### SSH utils library
+
+SSH utils library
+
+- [ ] Test, test
+
+- [ ] Connect witj Extend **config** to support other **data stores**
+
+- [ ] Add encryption to **config**
+
+- [ ] Complete implementation for this types by moving some code here.
diff --git a/ssh_utils/src/lib.rs b/ssh_utils/src/lib.rs
new file mode 100644
index 0000000..cc26fd2
--- /dev/null
+++ b/ssh_utils/src/lib.rs
@@ -0,0 +1,74 @@
+// use openssh::{Session, KnownHosts};
+use openssh::*;
+// use std::io;
+// use std::process::Stdio;
+use tokio::io::{AsyncWriteExt,AsyncReadExt};
+
+/// ssh 
+/// ````rust
+/// // Prepare address 
+/// let host= String::from("hostname-or-ip");
+/// let user= String::from("root");
+/// let port: u16 = 22;
+/// let addr = format!("ssh://{}@{}:{}",&user,&host,port);
+/// 
+/// // for scp_to_add data content into /tmp/hola
+/// let task= String::from("scp_to_add");
+/// let trgt=String::from("/tmp/hola"); 
+/// let mut data = String::from("Dime \n");
+//
+/// // for ssh ls /tmp
+/// let task= String::from("ssh");
+/// let trgt=String::from("ls"); 
+/// let mut data = String::from("/tmp");
+///
+/// // Call command and "macth" result  
+/// match  cmds::ssh(&task, &addr, &trgt, &mut data) {
+///   Ok(rs) => println!("ssh res: {:?} -> {:?}", rs, &data),
+///   Err(e) => println!("ssh error: {:?}", e),
+/// }
+/// ```
+// #[tokio::main]
+pub async fn ssh(task: &str, addr: &str, trgt: &str, data: &mut String ) -> anyhow::Result<()> {
+
+  let session = Session::connect(&addr,KnownHosts::Strict).await?;
+
+  if task == "ssh" {
+    let ls = session.command(trgt).arg(data).output().await?;
+    match  String::from_utf8(ls.stdout) {
+      Ok(res) => println!("ls : {:?}",&res),
+      Err(e) => println!("Error {:?}",e),
+    };
+   } else {
+    let mut sftp = session.sftp();
+    match task {
+    "scp_to" => {
+      let mut w = sftp.write_to(trgt).await?; 
+      let content = data.as_bytes();
+      w.write_all(content).await?;
+      w.close().await?;
+    },
+    "scp_to_add" => {
+      let mut w = sftp.append_to(trgt).await?;
+      let content = data.as_bytes();
+      w.write_all(content).await?;
+      w.close().await?;
+    },
+    "scp_from" => {
+      let mut r = sftp.read_from(trgt).await?;
+      r.read_to_string(data).await?;
+      // println!("source: {:?}",&data);
+      r.close().await?;
+    },
+    _ => println!("Undefined {:?}",&task),
+    }
+  }
+
+  session.close().await?;
+  Ok(())
+}
+//      println!("SSH error no KeyPair found");
+// .map_err(|e| {
+//   debug!("e = {:?}", e);
+//   Error::SendError
+// })?;