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

came with various optimizations

Browse Source
This commit is contained in:
Paul 2021-09-12 20:16:11 +02:00
parent 8b22768886
commit 355643ecdd
5 changed files with 69 additions and 87 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/config/mod.rs
View File

@ -1,8 +1,8 @@
use clap::{App, Arg, ArgMatches};
use serde::{Deserialize, Serialize};
use serde_json::Error;
use std::string::String;
use std::{fs::File, io::Read};
use clap::{App, Arg, ArgMatches};
#[derive(Debug, Deserialize, Serialize)]
pub struct Config {
@ -27,7 +27,7 @@ impl Config {
}
pub fn load<'a>(&mut self, configfile: &'a str) {
let fileopen: Result<File,std::io::Error>;
let fileopen: Result<File, std::io::Error>;
let filemeta = std::fs::metadata(configfile);
let fileexists = match filemeta {
Ok(_) => true,
@ -43,7 +43,7 @@ impl Config {
Ok(f) => f,
Err(e) => {
panic!("{}", e);
},
}
};
let mut contents = String::new();
file.read_to_string(&mut contents).unwrap();
@ -63,15 +63,17 @@ impl Config {
}
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()
}
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()
}

16
src/main.rs
View File

@ -3,27 +3,23 @@ mod padcontrol;
mod zabbix;
use config::Config;
//use std::rc::Rc;
//use std::vec::Vec;
use zabbix::problems::ZabbixLayout;
fn main() {
let mut datamatrix = ZabbixLayout{layout: vec!()};
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 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 canvas, mut poller) = padcontrol::initpad();
//padcontrol::input(&mut canvas, &mut poller);
padcontrol::draw(&mut canvas, &mut poller, &mut datamatrix, &mut cfg);
}
let (mut canvas, mut _poller) = padcontrol::initpad();
//padcontrol::input(&mut canvas, &mut _poller);
padcontrol::draw(&mut canvas, &mut datamatrix, &mut cfg);
}

28
src/padcontrol/mod.rs
View File

@ -1,9 +1,8 @@
use crate::config::Config;
use crate::zabbix::api::get_zabbix_problems;
use crate::zabbix::problems::ZabbixLayout;
use launchy::Color;
use launchy::{self, Canvas, CanvasLayout, CanvasLayoutPoller, MsgPollingWrapper, Pad};
use crate::zabbix;
use crate::zabbix::problems::{ZabbixLayout};
use crate::config::Config;
pub fn initpad() -> (CanvasLayout<'static>, CanvasLayoutPoller) {
let (mut canvas, poller) = launchy::CanvasLayout::new_polling();
@ -12,14 +11,14 @@ pub fn initpad() -> (CanvasLayout<'static>, CanvasLayoutPoller) {
_ => {
eprintln!("Failed to connect to device, check USB connection");
std::process::exit(1);
},
}
};
println!("Connected to device using USB");
clear(&mut canvas);
(canvas, poller)
}
pub fn input(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller) {
pub fn _input(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller) {
for color in (0u64..).map(|f| Color::red_green_color(f as f32 / 60.0 / 2.5)) {
for msg in poller.iter_for_millis(17).filter(|msg| msg.is_press()) {
canvas[msg.pad()] = color;
@ -29,13 +28,17 @@ pub fn input(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller) {
}
}
pub fn draw(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller, datamatrix: &mut ZabbixLayout, cfg: &mut Config) {
pub fn draw(canvas: &mut CanvasLayout, datamatrix: &mut ZabbixLayout, cfg: &mut Config) {
println!("Refresh rate is {} seconds", cfg.refresh.unwrap());
loop {
let zabbix_data = zabbix::api::get_zabbix_problems(cfg);
let zabbix_data = get_zabbix_problems(cfg);
datamatrix.compute(&zabbix_data);
for (i, j) in datamatrix.layout.iter().enumerate() {
let p = Pad{x: i as i32, y: 1i32};
let mut y = 1i32;
for (x, j) in datamatrix.layout.iter().enumerate() {
if y % 8 == 0 {
y += 1;
};
let p = Pad { x: x as i32, y: y };
let c = match j.status {
4 => Color::RED,
3 => Color::YELLOW,
@ -49,7 +52,6 @@ pub fn draw(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller, datamatr
}
}
/*
fn _effect(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller) {
for color in (0u64..).map(|f| Color::red_green_color(f as f32 / 60.0 / 2.5)) {
for msg in poller.iter_for_millis(17).filter(|msg| msg.is_press()) {
@ -70,7 +72,7 @@ fn _effect(canvas: &mut CanvasLayout, poller: &mut CanvasLayoutPoller) {
canvas[pad] = canvas[pad].mix(surrounding_color, 0.4);
});
}
}*/
}
fn clear(canvas: &mut CanvasLayout) {
for a in 0..8 {
@ -83,4 +85,4 @@ fn clear(canvas: &mut CanvasLayout) {
}
}
let _res = canvas.flush();
}
}

39
src/zabbix/api.rs
View File

@ -2,54 +2,49 @@ use crate::config::Config;
use serde_json::{json, Value};
pub const ZABBIX_API_VERSION: &'static str = "2.0";
pub const ZABBIX_API_ID: i32 = 1;
pub fn get_zabbix_authtoken(cfg: &mut Config) {
let body = query_auth_token(&cfg.username, &cfg.password);
let response = reqwest::blocking::Client::new()
let resp = reqwest::blocking::Client::new()
.post(&cfg.server)
.json(&body)
.send()
.unwrap();
let values: Value = response.json()
.unwrap();
cfg.authtoken = Some(values["result"].as_str()
.unwrap()
.to_string());
let values: Value = resp.json().unwrap();
cfg.authtoken = Some(values["result"].as_str().unwrap().to_string());
}
pub fn get_zabbix_problems(cfg: &mut Config) -> Value {
let body = query_problems(&cfg.authtoken.as_ref()
.unwrap_or(&String::from("")),
cfg.limit.unwrap());
let response = reqwest::blocking::Client::new()
let body = query_problems(
&cfg.authtoken.as_ref().unwrap_or(&String::from("")),
cfg.limit.unwrap(),
);
let resp = reqwest::blocking::Client::new()
.post(&cfg.server)
.json(&body)
.send()
.unwrap();
response.json().unwrap()
resp.json().unwrap()
}
pub fn query_auth_token(zabbix_username: &String,
zabbix_password: &String) -> Value {
let api_function = "user.login";
let api_id = 1;
pub fn query_auth_token(zabbix_username: &String, zabbix_password: &String) -> Value {
let zabbix_api_function = "user.login";
json!({
"jsonrpc": ZABBIX_API_VERSION,
"method": api_function,
"method": zabbix_api_function,
"params": {
"user": zabbix_username,
"password": zabbix_password
},
"id": api_id
"id": ZABBIX_API_ID
})
}
pub fn query_problems(zabbix_token: &String,
zabbix_limit: i64) -> Value {
pub fn query_problems(zabbix_token: &String, zabbix_limit: i64) -> Value {
let zabbix_api_function = "problem.get";
let zabbix_api_id = 1;
json!({
"jsonrpc": ZABBIX_API_VERSION,
"method": zabbix_api_function,
@ -64,6 +59,6 @@ pub fn query_problems(zabbix_token: &String,
"limit": zabbix_limit
},
"auth": zabbix_token,
"id": zabbix_api_id
"id": ZABBIX_API_ID
})
}
}

41
src/zabbix/problems.rs
View File

@ -1,5 +1,5 @@
use serde_json::Value;
use std::fmt::{Display,Formatter,Result};
use std::fmt::{Display, Formatter, Result};
#[derive(Debug)]
pub enum ZabbixStatus {
@ -14,42 +14,29 @@ pub struct ZabbixLayout {
}
pub struct ZabbixProblems {
hostname: String,
pub status: i64,
}
impl Display for ZabbixProblems {
fn fmt(&self, f: &mut Formatter) -> Result {
write!(f,
"hostname: {}, status: {:?}",
self.hostname,
self.status)
write!(f, "status: {:?}", self.status)
}
}
impl ZabbixLayout {
pub fn compute(&mut self, input: &Value) {
self.layout = vec!();
for j in input.as_object().unwrap()["result"].as_array().unwrap()[0..8].to_vec() {
let severity = j.as_object()
.unwrap()["severity"]
.as_str()
.unwrap()
.parse::<i64>()
.unwrap();
let acknowledged = j.as_object()
.unwrap()["acknowledged"]
.as_str()
.unwrap()
.parse::<i64>()
.unwrap();
if acknowledged == 0 {
self.layout.push(
ZabbixProblems{
hostname: "test".to_string(),
status: severity
});
self.layout = Vec::new();
for j in input.as_object().unwrap()["result"].as_array().unwrap() {
let res = j.as_object().unwrap();
let severity = res["severity"].as_str().unwrap().parse::<i64>().unwrap();
let ack = res["acknowledged"]
.as_str()
.unwrap()
.parse::<i64>()
.unwrap();
if ack == 0 {
self.layout.push(ZabbixProblems { status: severity });
};
}
}
}
}