137 lines
3.4 KiB
Rust
137 lines
3.4 KiB
Rust
use std::collections::HashMap;
|
|
|
|
use tracing::{warn, error};
|
|
use reqwest::StatusCode;
|
|
use serde::Serialize;
|
|
use serde_repr::*;
|
|
use pollster::FutureExt as _;
|
|
|
|
use crate::{presence::OnPresence, config::NtfyConfig};
|
|
|
|
pub struct Ntfy {
|
|
base_url: String,
|
|
topic: String
|
|
}
|
|
|
|
#[derive(Serialize_repr)]
|
|
#[repr(u8)]
|
|
enum Priority {
|
|
Min = 1,
|
|
Low,
|
|
Default,
|
|
High,
|
|
Max,
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
#[serde(rename_all = "snake_case", tag = "action")]
|
|
enum ActionType {
|
|
Broadcast {
|
|
#[serde(skip_serializing_if = "HashMap::is_empty")]
|
|
extras: HashMap<String, String>
|
|
},
|
|
// View,
|
|
// Http
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
struct Action {
|
|
#[serde(flatten)]
|
|
action: ActionType,
|
|
label: String,
|
|
clear: Option<bool>,
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
struct Notification {
|
|
topic: String,
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
title: Option<String>,
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
message: Option<String>,
|
|
#[serde(skip_serializing_if = "Vec::is_empty")]
|
|
tags: Vec<String>,
|
|
#[serde(skip_serializing_if = "Option::is_none")]
|
|
priority: Option<Priority>,
|
|
#[serde(skip_serializing_if = "Vec::is_empty")]
|
|
actions: Vec<Action>,
|
|
}
|
|
|
|
impl Notification {
|
|
fn new(topic: &str) -> Self {
|
|
Self { topic: topic.to_owned(), title: None, message: None, tags: Vec::new(), priority: None, actions: Vec::new() }
|
|
}
|
|
|
|
fn set_title(mut self, title: &str) -> Self {
|
|
self.title = Some(title.to_owned());
|
|
self
|
|
}
|
|
|
|
fn set_message(mut self, message: &str) -> Self {
|
|
self.message = Some(message.to_owned());
|
|
self
|
|
}
|
|
|
|
fn add_tag(mut self, tag: &str) -> Self {
|
|
self.tags.push(tag.to_owned());
|
|
self
|
|
}
|
|
|
|
fn set_priority(mut self, priority: Priority) -> Self {
|
|
self.priority = Some(priority);
|
|
self
|
|
}
|
|
|
|
fn add_action(mut self, action: Action) -> Self {
|
|
self.actions.push(action);
|
|
self
|
|
}
|
|
}
|
|
|
|
impl Ntfy {
|
|
pub fn new(config: NtfyConfig) -> Self {
|
|
Self { base_url: config.url, topic: config.topic.unwrap() }
|
|
}
|
|
}
|
|
|
|
impl OnPresence for Ntfy {
|
|
fn on_presence(&mut self, presence: bool) {
|
|
// Setup extras for the broadcast
|
|
let extras = HashMap::from([
|
|
("cmd".into(), "presence".into()),
|
|
("state".into(), if presence { "0" } else { "1" }.into()),
|
|
]);
|
|
|
|
// Create broadcast action
|
|
let action = Action {
|
|
action: ActionType::Broadcast { extras },
|
|
label: if presence { "Set away" } else { "Set home" }.to_owned(),
|
|
clear: Some(true)
|
|
};
|
|
|
|
// Create the notification
|
|
let notification = Notification::new(&self.topic)
|
|
.set_title("Presence")
|
|
.set_message(if presence { "Home" } else { "Away" })
|
|
.add_tag("house")
|
|
.add_action(action)
|
|
.set_priority(Priority::Low);
|
|
|
|
// Create the request
|
|
let req = reqwest::Client::new()
|
|
.post(self.base_url.clone())
|
|
.body(serde_json::to_string(¬ification).unwrap());
|
|
|
|
// Send the notification
|
|
let res = req.send().block_on();
|
|
if let Err(err) = res {
|
|
error!("Something went wrong while sending the notifcation: {err}");
|
|
} else if let Ok(res) = res {
|
|
let status = res.status();
|
|
if !status.is_success() {
|
|
warn!("Received status {status} when sending notification");
|
|
}
|
|
}
|
|
}
|
|
}
|