resm 0.1.2

Remote Server Management tools
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123

//! Utility functions.

use crate::ssh_config::SshConfig;

use std::collections::BTreeMap;
use std::path::PathBuf;

use glob::glob;
use colored::Colorize;
use openssh::{ Session, SessionBuilder };
use openssh_sftp_client::{ Sftp, SftpOptions };
use chrono::Local;

//------------------------------------------------------------------------------
/// Loads JSON files.
//------------------------------------------------------------------------------
pub fn load_json( path: &str ) -> BTreeMap<String, SshConfig>
{
    //  Gets the path to JSON files.
    let json_path = path.trim_end_matches("/").to_string() + "/**/*.json";

    //  Loads JSON files.
    let mut config_entries: Vec<SshConfig> = Vec::new();
    for entry in glob(&json_path).unwrap()
    {
        match entry
        {
            Ok(path) =>
            {
                let json_data = std::fs::read_to_string(path).unwrap();
                match serde_json::from_str::<Vec<SshConfig>>(&json_data)
                {
                    Ok(config_entry) => config_entries.extend(config_entry),
                    Err(e) => println!("{:?}", e),
                }
            },
            Err(e) => println!("{:?}", e),
        }
    }

    config_entries
        .into_iter()
        .map(|entry|
        {
            let mut key = entry.project();
            if entry.environment().len() > 0
            {
                key += &("_".to_string() + entry.environment().as_str());
            }
            (key, entry)
        })
        .collect()
}

//------------------------------------------------------------------------------
/// Gets SSH channel for the specified project.
//------------------------------------------------------------------------------
pub async fn get_session( project: &str ) -> Session
{
    SessionBuilder::default().connect(project).await.unwrap()
}

//------------------------------------------------------------------------------
/// Gets SFTP channel for the specified project.
//------------------------------------------------------------------------------
pub async fn get_sftp_session( project: &str ) -> Sftp
{
    let session = get_session(project).await;
    Sftp::from_session(session, SftpOptions::default()).await.unwrap()
}

//------------------------------------------------------------------------------
/// Gets file paths in the specified directory.
//------------------------------------------------------------------------------
pub fn get_file_paths( dir: &str ) -> Vec<PathBuf>
{
    let mut file_paths: Vec<PathBuf> = Vec::new();
    println!("{}", dir);
    for entry in glob(&(dir.to_string() + "/**/*")).unwrap()
    {
        match entry
        {
            Ok(path) =>
            {
                if path.is_dir()
                {
                    continue;
                }
                file_paths.push(path);
            },
            Err(e) => println!("{:?}", e),
        }
    }
    file_paths
}

//------------------------------------------------------------------------------
/// Prints separator.
//------------------------------------------------------------------------------
pub fn print_sep()
{
    println!("{}", "=".to_string().repeat(80));
}

//------------------------------------------------------------------------------
/// Confirm.
//------------------------------------------------------------------------------
pub fn confirm( message: &str ) -> bool
{
    println!("{} [y/N]", message.yellow());
    let mut input = String::new();
    std::io::stdin().read_line(&mut input).unwrap();
    input.trim() == "y"
}

//------------------------------------------------------------------------------
/// Gets current time for filename.
//------------------------------------------------------------------------------
pub fn get_current_time_for_filename() -> String
{
    let now = Local::now();
    now.format("%Y%m%d_%H%M%S").to_string()
}