shecret 0.2.0

A CLI SSH/SFTP client
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

use colored::*;
use home::home_dir;
use rusqlite::{Connection, Result};
use shecret::*;
use std::{env, io::stdin};

fn main() -> Result<()> {
    let sql_file = format!(
        "{}/{}.db3",
        home_dir().unwrap().display(),
        env!("CARGO_PKG_NAME")
    );

    let conn = Connection::open(&sql_file)?;
    create_database(&conn)?;

    let mut input = String::new();

    let title = format!(
        "\n{} - {}\nAuthors: {}\nVersion: {}\nLicense: {}\nDatabase path: {}\nDocumentation: {}\nCrafted with ❤️ using Rust language\n",
        env!("CARGO_PKG_NAME").to_uppercase(),
        env!("CARGO_PKG_DESCRIPTION"),
        env!("CARGO_PKG_AUTHORS"),
        env!("CARGO_PKG_VERSION"),
        env!("CARGO_PKG_LICENSE"),
        &sql_file,
        env!("CARGO_PKG_HOMEPAGE")
    )
    .color("yellow");
    println!("{title}");

    let options_menu = "Available options (case insensitive):
    CC: Create a server connection
    LC: List all server connections
    SC: Start a connection
    DC: Delete a server connection
    PD: Purge database (delete all server connections)
    CK: Create SSH key
    IC: Issue SSH command to multiple servers
    Q:  Quit";

    loop {
        println!("{options_menu}");
        stdin().read_line(&mut input).unwrap();

        match input.to_uppercase().as_str().trim() {
            "CC" => create_server_connection(&conn)?,
            "LC" => display_connections(&get_connections(&conn).unwrap()),
            "SC" => start_connection(&get_connections(&conn).unwrap())?,
            "DC" => delete_record(&conn, &get_input("ID to delete:", "0"))?,
            "PD" => purge_database(&conn)?,
            "CK" => create_key()?,
            "IC" => issue_command(&get_connections(&conn).unwrap())?,
            "Q" => break,
            _ => {
                display_message("ERROR", "Invalid option", "red");
            }
        }
        input.clear();
    }
    Ok(())
}