|
|
|
use clap::{App, Arg, SubCommand};
|
|
|
|
|
|
|
|
pub struct CLI {
|
|
|
|
pub url: String,
|
|
|
|
pub key: String,
|
|
|
|
pub datasource_url: String,
|
|
|
|
pub query: String,
|
|
|
|
pub from: u64,
|
|
|
|
pub to: u64,
|
|
|
|
pub step: u64,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl CLI {
|
|
|
|
pub fn new() -> CLI {
|
|
|
|
let matches = App::new("subbeat")
|
|
|
|
.version("0.0.2")
|
|
|
|
.about("Timeseries toolkit")
|
|
|
|
.subcommand(
|
|
|
|
SubCommand::with_name("grafana")
|
|
|
|
.about("Use Grafana as datasource")
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("GRAFANA_URL")
|
|
|
|
.help("URL to your Grafana instance")
|
|
|
|
.required(true)
|
|
|
|
.index(1),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("GRAFANA_API_KEY")
|
|
|
|
.help(
|
|
|
|
"Grafna API Key. Go to http://<grafana-url>/org/apikeys to get one",
|
|
|
|
)
|
|
|
|
.required(true)
|
|
|
|
.index(2),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("datasource_url")
|
|
|
|
.help("relative path to datasource")
|
|
|
|
.required(true)
|
|
|
|
.index(3),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("query")
|
|
|
|
.help("your query to datasource")
|
|
|
|
.required(true)
|
|
|
|
.index(4),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("from")
|
|
|
|
.help("timestamp")
|
|
|
|
.required(true)
|
|
|
|
.index(5),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("to")
|
|
|
|
.help("timestampt")
|
|
|
|
.required(true)
|
|
|
|
.index(6),
|
|
|
|
)
|
|
|
|
.arg(
|
|
|
|
Arg::with_name("step")
|
|
|
|
.help("aggregation step")
|
|
|
|
.required(true)
|
|
|
|
.index(7),
|
|
|
|
),
|
|
|
|
)
|
|
|
|
.get_matches();
|
|
|
|
|
|
|
|
if let Some(matches) = matches.subcommand_matches("grafana") {
|
|
|
|
let url = matches.value_of("GRAFANA_URL").unwrap();
|
|
|
|
let key = matches.value_of("GRAFANA_API_KEY").unwrap();
|
|
|
|
let datasource_url = matches.value_of("datasource_url").unwrap();
|
|
|
|
let query = matches.value_of("query").unwrap();
|
|
|
|
let from = matches.value_of("from").unwrap().parse().unwrap();
|
|
|
|
let to = matches.value_of("to").unwrap().parse().unwrap();
|
|
|
|
let step = matches.value_of("step").unwrap().parse().unwrap();
|
|
|
|
return CLI {
|
|
|
|
url: url.to_owned(),
|
|
|
|
key: key.to_owned(),
|
|
|
|
datasource_url: datasource_url.to_owned(),
|
|
|
|
query: query.to_owned(),
|
|
|
|
from,
|
|
|
|
to,
|
|
|
|
step,
|
|
|
|
};
|
|
|
|
} else {
|
|
|
|
return CLI {
|
|
|
|
url: "url.to_owned()".to_string(),
|
|
|
|
key: "key.to_owned()".to_string(),
|
|
|
|
datasource_url: "datasource_url.to_owned()".to_string(),
|
|
|
|
query: "query.to_owned()".to_string(),
|
|
|
|
from: 0,
|
|
|
|
to: 0,
|
|
|
|
step: 0,
|
|
|
|
};
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|