fetchlib 0.0.1

A lib for remote server connection through ssh
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

use std::fs;
use std::path::Path;
use std::path::PathBuf;

use rand::Rng;
use rand::RngExt;
use rand::distr::Alphabetic;
use rand::distr::Alphanumeric;
use rand::rngs::ThreadRng;
use serde::Deserialize;
use serde::Serialize;
use ssh_key::PrivateKey;
use ssh_key::rand_core::OsRng;

pub mod credentials;

pub struct Secrets {
    pub pass: Option<String>,
    key: PrivateKey,
    pub fname: String,
}

impl Secrets {
    pub fn gen_key(password: Option<String>) -> PrivateKey {
        let generated = PrivateKey::random(&mut OsRng, ssh_key::Algorithm::Ed25519).unwrap();
        let pass;
        if password.is_none() {
            pass = String::new();
        } else {
            pass = password.unwrap();
        }
        let encrypted = generated.encrypt(&mut OsRng, pass).unwrap();
        encrypted
    }

    pub fn gen_name() -> String {
        let length = 20;
        rand::rng()
            .sample_iter(&Alphanumeric)
            .take(length)
            .map(char::from)
            .collect()
    }

    pub fn generate(pass: Option<String>) -> Self {
        let encyrpted_key = Secrets::gen_key(pass.clone());
        let fname = Secrets::gen_name();
        Self {
            pass: pass,
            key: encyrpted_key,
            fname: fname,
        }
    }

    pub fn write_openssh_to_file(&self, priv_path: PathBuf) {
        let path = priv_path.join(self.fname.clone());
        self.key
            .write_openssh_file(path.as_path(), ssh_key::LineEnding::LF)
            .unwrap();
    }

    pub fn get_pass(pass: String) -> Option<String> {
        if pass.is_empty() {
            return None;
        } else {
            return Some(pass);
        }
    }
}