extern crate serde_json; #[macro_use] extern crate failure; use clap; use clap::Arg; use futures::future::{done, ok, Either, Future}; use http::StatusCode; use hyper::service::service_fn; use hyper::{Body, Request, Response, Server}; use log::{error, info, trace}; use std::env; mod options; use options::Options; mod exporter_error; use exporter_error::ExporterError; mod render_to_prometheus; use render_to_prometheus::RenderToPrometheus; mod wireguard; use std::convert::TryFrom; use std::process::Command; use std::string::String; use wireguard::WireGuard; fn check_compliance(req: &Request) -> Result<(), Response> { if req.uri() != "/metrics" { trace!("uri not allowed {}", req.uri()); Err(Response::builder() .status(StatusCode::NOT_FOUND) .body(hyper::Body::empty()) .unwrap()) } else if req.method() != "GET" { trace!("method not allowed {}", req.method()); Err(Response::builder() .status(StatusCode::METHOD_NOT_ALLOWED) .body(hyper::Body::empty()) .unwrap()) } else { Ok(()) } } fn handle_request( req: Request, options: Options, ) -> impl Future, Error = failure::Error> { trace!("{:?}", req); done(check_compliance(&req)).then(move |res| match res { Ok(_) => Either::A(perform_request(req, &options).then(|res| match res { Ok(body) => ok(body), Err(err) => { error!("internal server error: {:?}", err); ok(Response::builder() .status(StatusCode::INTERNAL_SERVER_ERROR) .body(hyper::Body::empty()) .unwrap()) } })), Err(err) => Either::B(ok(err)), }) } fn perform_request( _req: Request, _options: &Options, ) -> impl Future, Error = ExporterError> { trace!("perform_request"); done( Command::new("wg") .arg("show") .arg("all") .arg("dump") .output(), ) .from_err() .and_then(|output| { done(String::from_utf8(output.stdout)) .from_err() .and_then(|output_str| { trace!("{}", output_str); done(WireGuard::try_from(&output_str as &str)) .from_err() .and_then(|wg| ok(Response::new(Body::from(wg.render())))) }) }) } fn main() { let matches = clap::App::new("prometheus_wireguard_exporter") .version("0.1") .author("Francesco Cogno ") .arg( Arg::with_name("port") .short("p") .help("exporter port (default 9576)") .default_value("9576") .takes_value(true), ) .arg( Arg::with_name("verbose") .short("v") .help("verbose logging") .takes_value(false), ) .get_matches(); let options = Options::from_claps(&matches); if options.verbose { env::set_var("RUST_LOG", "prometheus_wireguard_exporter=trace"); } else { env::set_var("RUST_LOG", "prometheus_wireguard_exporter=info"); } env_logger::init(); info!("using options: {:?}", options); let bind = matches.value_of("port").unwrap(); let bind = u16::from_str_radix(&bind, 10).expect("port must be a valid number"); let addr = ([0, 0, 0, 0], bind).into(); info!("starting exporter on {}", addr); let new_svc = move || { let options = options.clone(); service_fn(move |req| handle_request(req, options.clone())) }; let server = Server::bind(&addr) .serve(new_svc) .map_err(|e| eprintln!("server error: {}", e)); hyper::rt::run(server); }