read and query from config file works

master
Edward Shen 2019-04-30 01:44:26 -04:00
commit e93d2136dd
Signed by: edward
GPG Key ID: F350507060ED6C90
4 changed files with 1586 additions and 0 deletions

3
.gitignore vendored Normal file
View File

@ -0,0 +1,3 @@
/target
**/*.rs.bk
*.ron

1470
Cargo.lock generated Normal file

File diff suppressed because it is too large Load Diff

10
Cargo.toml Normal file
View File

@ -0,0 +1,10 @@
[package]
name = "endstat"
version = "0.1.0"
authors = ["Edward Shen <code@eddie.sh>"]
edition = "2018"
[dependencies]
reqwest = "0.9.15"
serde = { version = "1.0.90", features = ["derive"] }
ron = "0.5"

103
src/main.rs Normal file
View File

@ -0,0 +1,103 @@
extern crate reqwest;
extern crate ron;
extern crate serde;
use reqwest::{Client, Url};
use ron::de::from_str;
use serde::Deserialize;
use std::error::Error;
use std::fs::read_to_string;
#[derive(Deserialize, Debug)]
struct EndpointConf {
label: Option<String>,
endpoint: Option<String>,
port: Option<u16>,
code: Option<u16>,
body: Option<String>,
}
#[derive(Deserialize, Debug)]
struct WebsiteConf {
label: String,
base: String,
endpoints: Vec<EndpointConf>,
}
#[derive(Deserialize, Debug)]
struct Config {
websites: Vec<WebsiteConf>,
}
#[derive(Debug)]
struct Status {
status: u8,
location: String,
domain: String,
endpoint: String,
error: Option<String>,
}
fn main() -> Result<(), Box<Error>> {
let client = Client::new();
let config = from_str::<Config>(&read_to_string("./endstat_conf.ron")?)?;
let mut results: Vec<Status> = vec![];
for website_conf in config.websites {
for endpoint in website_conf.endpoints {
let (label, path, port, code, body) = get_endpoint_info(endpoint);
let mut url = Url::parse(&website_conf.base)?.join(&path)?;
if let Err(e) = url.set_port(port) {
println!("{:?}", e);
}
let url = url.into_string();
let mut res = client.get(&url).send()?;
let does_code_match = res.status() == code;
let does_body_match = body.is_empty() || res.text()? == body;
results.push(if !does_code_match {
Status {
status: 1,
location: url,
domain: website_conf.label.clone(),
endpoint: label,
error: Some(format!(
"Status code mismatch! {} != {}",
res.status().as_u16(),
code
)),
}
} else if !does_body_match {
Status {
status: 2,
location: url,
domain: website_conf.label.clone(),
endpoint: label,
error: Some(format!("Body mismatch! {} != {}", res.text()?, body)),
}
} else {
Status {
status: 0,
location: url,
domain: website_conf.label.clone(),
endpoint: label,
error: None,
}
});
}
}
dbg!(results);
Ok(())
}
fn get_endpoint_info(endpoint: EndpointConf) -> (String, String, Option<u16>, u16, String) {
let path = endpoint.endpoint.unwrap_or_default();
let label = endpoint.label.unwrap_or_else(|| path.clone());
let code = endpoint.code.unwrap_or_else(|| 200);
let body = endpoint.body.unwrap_or_default();
(label, path, endpoint.port, code, body)
}