ssh-utils 0.1.1

Utilities for ssh.
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

mod upload;
use upload::speedtest_upload;

mod download;
use download::speedtest_download;

mod human_readable_unit;

use super::utility::{println_on_level, Level};
use super::SshSessionBuilder;

use clap::Parser;
use clap_verbosity_flag::Verbosity;
use openssh::Error;

#[derive(Debug, Parser, Copy, Clone)]
pub struct SpeedTestArgs {
    /// Disable testing upload speed.
    #[clap(long)]
    no_upload: bool,

    /// Disable testing download speed.
    #[clap(long)]
    no_download: bool,
}

pub async fn run(
    args: SpeedTestArgs,
    verbose: Verbosity,
    builder: SshSessionBuilder<'_>,
) -> Result<(), Error> {
    let dest = builder.dest();

    println_on_level!(verbose, Level::Debug, "Attempting to connect to {dest}");
    let session = builder.connect().await?;

    if !args.no_upload {
        speedtest_upload(verbose.clone(), &session).await?;
    }

    if !args.no_download {
        speedtest_download(verbose, &session).await?;
    }

    session.close().await
}