paulbsd/zabbixlaunch
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

WIP: added hot reload of config using inotify

Browse Source
This commit is contained in:
Paul 2021-10-10 22:08:21 +02:00
parent 0976a71057
commit b7a90a81ba
7 changed files with 133 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@ -1,2 +1,4 @@
/target
/config.json
*.swp

35
Cargo.lock generated
View File

@ -11,7 +11,7 @@ dependencies = [
"alsa-sys",
"bitflags",
"libc",
"nix",
"nix 0.15.0",
]
[[package]]
@ -64,9 +64,9 @@ checksum = "904dfeac50f3cdaba28fc6f57fdcddb75f49ed61346676a78c4ffe55877802fd"
[[package]]
name = "bitflags"
version = "1.2.1"
version = "1.3.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "cf1de2fe8c75bc145a2f577add951f8134889b4795d47466a54a5c846d691693"
checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
[[package]]
name = "bumpalo"
@ -470,9 +470,9 @@ checksum = "e2abad23fbc42b3700f2f279844dc832adb2b2eb069b2df918f455c4e18cc646"
[[package]]
name = "libc"
version = "0.2.101"
version = "0.2.103"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3cb00336871be5ed2c8ed44b60ae9959dc5b9f08539422ed43f09e34ecaeba21"
checksum = "dd8f7255a17a627354f321ef0055d63b898c6fb27eff628af4d1b66b7331edf6"
[[package]]
name = "log"
@ -501,6 +501,15 @@ version = "2.4.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "308cc39be01b73d0d18f82a0e7b2a3df85245f84af96fdddc5d202d27e47b86a"
[[package]]
name = "memoffset"
version = "0.6.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "59accc507f1338036a0477ef61afdae33cde60840f4dfe481319ce3ad116ddf9"
dependencies = [
"autocfg",
]
[[package]]
name = "micromath"
version = "1.1.1"
@ -519,7 +528,7 @@ dependencies = [
"js-sys",
"libc",
"memalloc",
"nix",
"nix 0.15.0",
"wasm-bindgen",
"web-sys",
"winapi",
@ -584,6 +593,19 @@ dependencies = [
"void",
]
[[package]]
name = "nix"
version = "0.23.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f305c2c2e4c39a82f7bf0bf65fb557f9070ce06781d4f2454295cc34b1c43188"
dependencies = [
"bitflags",
"cc",
"cfg-if 1.0.0",
"libc",
"memoffset",
]
[[package]]
name = "ntapi"
version = "0.3.6"
@ -1217,6 +1239,7 @@ dependencies = [
"clap",
"embedded-graphics",
"launchy",
"nix 0.23.0",
"reqwest",
"serde",
"serde_json",

1
Cargo.toml
View File

@ -12,3 +12,4 @@ launchy = { git = "https://github.com/kangalioo/launchy", branch = "master" }
reqwest = { version = "0.11.4", features = ["blocking", "json"] }
serde = { version = "1.0.130", features = ["derive"] }
serde_json = "1.0.68"
nix = "0.23.0"

72
src/config/mod.rs
View File

@ -1,10 +1,58 @@
use clap::{App, Arg, ArgMatches};
use nix::sys::inotify::{AddWatchFlags, InitFlags, Inotify, InotifyEvent};
use serde::{Deserialize, Serialize};
use serde_json::Error;
use std::string::String;
use std::{fs::File, io::Read};
#[derive(Debug, Deserialize, Serialize)]
pub struct Context {
pub cfg: Config,
pub configfile: String,
pub events: Option<Vec<InotifyEvent>>,
pub inotify: Inotify,
}
impl Context {
pub fn new() -> Self {
let configfile = Context::argparse()
.value_of("config")
.unwrap_or("config.json")
.to_string();
let mut ctx = Context {
cfg: Config::new(),
configfile: configfile,
events: None,
inotify: Inotify::init(InitFlags::empty()).unwrap(),
};
println!("Loading {} file ...", ctx.configfile);
ctx.cfg.load(&ctx.configfile);
ctx.inotify
.add_watch(ctx.configfile.as_str(), AddWatchFlags::IN_MODIFY)
.unwrap();
ctx
}
pub fn argparse() -> ArgMatches<'static> {
App::new("Zabbix Launch")
.version("1.0")
.author("PaulBSD <paul@paulbsd.com>")
.about("Lights up Launchpad mini using Zabbix data")
.arg(
Arg::with_name("config")
.short("c")
.long("config")
.value_name("FILE")
.help("Sets a custom config file")
.takes_value(true),
)
.get_matches()
}
}
#[derive(Debug, Deserialize, Serialize, Clone)]
pub struct Config {
pub server: String,
pub username: String,
@ -15,7 +63,7 @@ pub struct Config {
pub limit: Option<u64>,
}
impl Config {
impl<'a> Config {
pub fn new() -> Self {
Self {
server: String::from("https://zabbix.acme.com/api_jsonrpc.php"),
@ -43,7 +91,7 @@ impl Config {
}
}
pub fn load<'a>(&mut self, configfile: &'a str) {
pub fn load(&'a mut self, configfile: &str) {
let fileopen: Result<File, std::io::Error>;
let filemeta = std::fs::metadata(configfile);
let fileexists = match filemeta {
@ -74,24 +122,8 @@ impl Config {
self.save(&configfile);
}
pub fn save<'a>(&self, configfile: &'a str) {
pub fn save(&self, configfile: &str) {
let file = File::create(configfile).unwrap();
serde_json::to_writer_pretty(file, &self).unwrap();
}
}
pub fn argparse<'a>() -> ArgMatches<'a> {
App::new("Zabbix Launch")
.version("1.0")
.author("PaulBSD <paul@paulbsd.com>")
.about("Lights up Launchpad mini using Zabbix data")
.arg(
Arg::with_name("config")
.short("c")
.long("config")
.value_name("FILE")
.help("Sets a custom config file")
.takes_value(true),
)
.get_matches()
}

18
src/main.rs
View File

@ -2,24 +2,16 @@ pub mod config;
mod padcontrol;
mod zabbix;
use config::Config;
use config::Context;
use zabbix::problems::ZabbixLayout;
fn main() {
let mut datamatrix = ZabbixLayout { layout: Vec::new() };
// parse arguments
let matches = config::argparse();
// load configuration
let configfile = matches.value_of("config").unwrap_or("config.json");
let mut cfg = Config::new();
cfg.load(&configfile);
zabbix::api::get_zabbix_authtoken(&mut cfg);
cfg.save(&configfile);
// init/connect to launchpad and clear it
let mut context = Context::new();
zabbix::api::get_zabbix_authtoken(&mut context.cfg);
let (mut canvas, mut _poller) = padcontrol::initpad();
//padcontrol::input(&mut canvas, &mut _poller);
padcontrol::draw(&mut canvas, &mut datamatrix, &mut cfg);
padcontrol::draw(&mut canvas, &mut datamatrix, &mut context);
//padcontrol::test(&mut context);
}

50
src/padcontrol/mod.rs
View File

@ -1,16 +1,14 @@
use std::time::Duration;
use std::thread::sleep;
use crate::config::Config;
use crate::config::Context;
use crate::zabbix::api::get_zabbix_problems;
use crate::zabbix::problems::ZabbixLayout;
use launchy::Color;
use launchy::{self, Canvas, CanvasLayout, CanvasLayoutPoller, MsgPollingWrapper, Pad};
use std::thread::sleep;
use std::time::Duration;
pub const MATRIX_WIDTH: i32 = 8;
pub const MATRIX_SIZE: i32 = MATRIX_WIDTH * MATRIX_WIDTH;
pub fn initpad() -> (CanvasLayout<'static>, CanvasLayoutPoller) {
let (mut canvas, poller) = launchy::CanvasLayout::new_polling();
match canvas.add_by_guess_rotated::<launchy::mini::Canvas>(0, 0, launchy::Rotation::None) {
@ -35,10 +33,37 @@ pub fn _input(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller) {
}
}
pub fn draw(canvas: &mut CanvasLayout, datamatrix: &mut ZabbixLayout, cfg: &mut Config) {
println!("Refresh rate is {} seconds", cfg.refresh.unwrap());
pub fn test<'a>(ctx: &mut Context) {
println!("Refresh rate is {} seconds", ctx.cfg.refresh.unwrap());
loop {
let zabbix_data_result = get_zabbix_problems(cfg);
println!("test");
let events = match ctx.inotify.read_events() {
Ok(ok) => ok,
Err(_) => continue,
};
println!("nombre d'events: {}", events.len());
for i in events {
println!("test event");
println!("{:?}", i);
}
let zabbix_data_result = get_zabbix_problems(&ctx.cfg);
let zabbix_data = match zabbix_data_result {
Ok(z) => z,
Err(_) => {
println!("Error requesting zabbix service");
sleep(Duration::from_secs(10));
continue;
}
};
println!("{:?}", zabbix_data);
sleep(Duration::from_secs(ctx.cfg.refresh.unwrap_or(5)));
}
}
pub fn draw(canvas: &mut CanvasLayout, datamatrix: &mut ZabbixLayout, ctx: &mut Context) {
println!("Refresh rate is {} seconds", ctx.cfg.refresh.unwrap());
loop {
let zabbix_data_result = get_zabbix_problems(&ctx.cfg);
let zabbix_data = match zabbix_data_result {
Ok(zabbix_data) => zabbix_data,
Err(_) => {
@ -51,7 +76,10 @@ pub fn draw(canvas: &mut CanvasLayout, datamatrix: &mut ZabbixLayout, cfg: &mut
clear(canvas);
let mut max = 0;
for (i, j) in datamatrix.layout.iter().enumerate() {
let p = Pad { x: ((i as i32) % MATRIX_WIDTH ), y: ((i as i32) / MATRIX_WIDTH )+1 };
let p = Pad {
x: ((i as i32) % MATRIX_WIDTH),
y: ((i as i32) / MATRIX_WIDTH) + 1,
};
let c = match j.status {
5 => Color::RED,
4 => Color::from_hue(0.05),
@ -64,12 +92,12 @@ pub fn draw(canvas: &mut CanvasLayout, datamatrix: &mut ZabbixLayout, cfg: &mut
if max >= MATRIX_SIZE {
break;
}
if cfg.sloweffect.unwrap() {
if ctx.cfg.sloweffect.unwrap() {
sleep(Duration::from_millis(15));
}
canvas.flush().unwrap();
}
sleep(Duration::from_secs(cfg.refresh.unwrap_or(5)));
sleep(Duration::from_secs(ctx.cfg.refresh.unwrap_or(5)));
}
}

11
src/zabbix/api.rs
View File

@ -16,19 +16,13 @@ pub fn get_zabbix_authtoken(cfg: &mut Config) {
cfg.authtoken = Some(values["result"].as_str().unwrap().to_string());
}
pub fn get_zabbix_problems(cfg: &mut Config) -> Result<Value, reqwest::Error> {
let body = query_triggers(
&cfg.authtoken.as_ref().unwrap_or(&String::from("")),
);
pub fn get_zabbix_problems(cfg: &Config) -> Result<Value, reqwest::Error> {
let body = query_triggers(&cfg.authtoken.as_ref().unwrap_or(&String::from("")));
let resp = reqwest::blocking::Client::new()
.post(&cfg.server)
.json(&body)
.send();
//let res = match resp {
// Ok(res) => res.json()
//};
Ok(resp.unwrap().json().unwrap())
}
@ -67,7 +61,6 @@ fn _query_problems(zabbix_token: &String, zabbix_limit: i64) -> Value {
})
}
pub fn query_triggers(zabbix_token: &String) -> Value {
let zabbix_api_function = "trigger.get";
json!({