agscheduler-cli 0.5.0

Command line interface for AGScheduler
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
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

#![cfg_attr(coverage_nightly, feature(coverage_attribute))]

use std::env;

use clap::Parser;
use dialoguer::{theme::ColorfulTheme, Select};
use sha2::{Digest, Sha256};

use agscheduler_cli::api_client::AGScheduler;
use agscheduler_cli::http;
use agscheduler_cli::interaction::Interaction;

/// Command line interface for AGScheduler
#[derive(Parser, Debug)]
#[command(version, about, long_about = None)]
struct Args {
    /// AGScheduler HTTP endpoint
    #[arg(short, long, default_value = "http://127.0.0.1:36370")]
    endpoint: String,
    /// AGScheduler password
    /// You can also use the AGSCHEDULERCLI_AUTH environment variable to pass this password more safely
    #[arg(short, long, default_value = "", verbatim_doc_comment)]
    password: String,
}

#[tokio::main]
#[cfg_attr(coverage_nightly, coverage(off))]
async fn main() {
    let args = Args::parse();

    let mut auth = env::var("AGSCHEDULERCLI_AUTH").unwrap_or("".to_string());
    if !args.password.is_empty() {
        auth = args.password;
    }
    if !auth.is_empty() {
        unsafe {
            let hash = Sha256::digest(auth);
            http::PASSWORD_SHA2 = hex::encode(hash);
        }
    }

    println!("Connecting to `{}`...", args.endpoint);

    let ags = AGScheduler {
        endpoint: args.endpoint,
    };

    loop {
        let selections = &[
            "Add Job",
            "Get Job",
            "Get All Jobs",
            "Update Job",
            "Delete Job",
            "Delete All Jobs",
            "Pause Job",
            "Resume Job",
            "Run Job",
            "Schedule Job",
            "Start",
            "Stop",
            "Get Records",
            "Get All Records",
            "Delete Records",
            "Delete All Records",
            "Get Info",
            "Get Funcs",
            "Get Queues",
            "Get Cluster Nodes",
        ];

        let interaction = Interaction {};

        let selection = Select::with_theme(&ColorfulTheme::default())
            .with_prompt("Select your operation")
            .default(0)
            .max_length(8)
            .items(&selections[..])
            .interact()
            .unwrap();

        match selection {
            0 => ags.add_job(&interaction).await,
            1 => ags.get_job(&interaction).await,
            2 => ags.get_all_jobs().await,
            3 => ags.update_job(&interaction).await,
            4 => ags.delete_job(&interaction).await,
            5 => ags.delete_all_jobs(&interaction).await,
            6 => ags.pause_or_resume_job("pause", &interaction).await,
            7 => ags.pause_or_resume_job("resume", &interaction).await,
            8 => ags.run_or_schedule_job("run", &interaction).await,
            9 => ags.run_or_schedule_job("schedule", &interaction).await,
            10 => ags.start_or_stop("start").await,
            11 => ags.start_or_stop("stop").await,
            12 => ags.get_records(&interaction).await,
            13 => ags.get_all_records(&interaction).await,
            14 => ags.delete_records(&interaction).await,
            15 => ags.delete_all_records(&interaction).await,
            16 => ags.get_info().await,
            17 => ags.get_funcs().await,
            18 => ags.get_queues().await,
            19 => ags.get_cluster_nodes().await,
            _ => panic!("Error"),
        };
    }
}