Signed-off-by: Lukas Wagner <l.wag...@proxmox.com> --- Notes: notifications.rs should be moved to common/src at some point. Currently it resides in pve-rs/src, since rust-analyzer breaks for me for all code in common/src.
Cargo.toml | 2 + common/pkg/Makefile | 1 + pve-rs/Cargo.toml | 1 + pve-rs/src/lib.rs | 1 + pve-rs/src/notification.rs | 128 +++++++++++++++++++++++++++++++++++++ 5 files changed, 133 insertions(+) create mode 100644 pve-rs/src/notification.rs diff --git a/Cargo.toml b/Cargo.toml index 94e29d5..fc0f7c1 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -25,6 +25,7 @@ perlmod = { version = "0.13", features = [ "exporter" ] } proxmox-acme-rs = { version = "0.4", features = ["client"] } proxmox-apt = "0.9" proxmox-http = { version = "0.8", features = ["client-sync", "client-trait"] } +proxmox-notification = "0.1" proxmox-openid = "0.9.8" proxmox-resource-scheduling = "0.2.1" proxmox-subscription = "0.3" @@ -37,3 +38,4 @@ proxmox-time = "1.1.3" #proxmox-tfa = { path = "../proxmox/proxmox-tfa" } #proxmox-time = { path = "../proxmox/proxmox-time" } #proxmox-uuid = { path = "../proxmox/proxmox-uuid" } +#proxmox-notification = { path = "../proxmox/proxmox-notification" } diff --git a/common/pkg/Makefile b/common/pkg/Makefile index bf22a7b..658d336 100644 --- a/common/pkg/Makefile +++ b/common/pkg/Makefile @@ -17,6 +17,7 @@ Proxmox/RS/CalendarEvent.pm: ../scripts/genpackage.pl perl ../scripts/genpackage.pl Common \ Proxmox::RS::APT::Repositories \ Proxmox::RS::CalendarEvent \ + Proxmox::RS::Notification \ Proxmox::RS::Subscription all: Proxmox/RS/CalendarEvent.pm diff --git a/pve-rs/Cargo.toml b/pve-rs/Cargo.toml index 0108eb2..9eafc25 100644 --- a/pve-rs/Cargo.toml +++ b/pve-rs/Cargo.toml @@ -33,6 +33,7 @@ perlmod.workspace = true proxmox-apt.workspace = true proxmox-http.workspace = true +proxmox-notification.workspace = true proxmox-openid.workspace = true proxmox-resource-scheduling.workspace = true proxmox-subscription.workspace = true diff --git a/pve-rs/src/lib.rs b/pve-rs/src/lib.rs index af3076e..7126e25 100644 --- a/pve-rs/src/lib.rs +++ b/pve-rs/src/lib.rs @@ -4,6 +4,7 @@ pub mod common; pub mod apt; +pub mod notification; pub mod openid; pub mod resource_scheduling; pub mod tfa; diff --git a/pve-rs/src/notification.rs b/pve-rs/src/notification.rs new file mode 100644 index 0000000..ab325a8 --- /dev/null +++ b/pve-rs/src/notification.rs @@ -0,0 +1,128 @@ +#[perlmod::package(name = "Proxmox::RS::Notification")] +mod export { + use anyhow::{bail, Error}; + use perlmod::Value; + use std::sync::Mutex; + + use std::collections::HashMap; + + use proxmox_notification::{ + filter::FilterModeOperator, methods, Config, Notification, Severity, + }; + + pub struct NotificationConfig { + config: Mutex<Config>, + } + + perlmod::declare_magic!(Box<NotificationConfig> : &NotificationConfig as "Proxmox::RS::Notification"); + + /// Support `dclone` so this can be put into the `ccache` of `PVE::Cluster`. + #[export(name = "STORABLE_freeze", raw_return)] + fn storable_freeze( + #[try_from_ref] this: &NotificationConfig, + cloning: bool, + ) -> Result<Value, Error> { + if !cloning { + bail!("freezing Notification config not supported!"); + } + + let mut cloned = Box::new(NotificationConfig { + config: Mutex::new(this.config.lock().unwrap().clone()), + }); + let value = Value::new_pointer::<NotificationConfig>(&mut *cloned); + let _perl = Box::leak(cloned); + Ok(value) + } + + /// Instead of `thaw` we implement `attach` for `dclone`. + #[export(name = "STORABLE_attach", raw_return)] + fn storable_attach( + #[raw] class: Value, + cloning: bool, + #[raw] serialized: Value, + ) -> Result<Value, Error> { + if !cloning { + bail!("STORABLE_attach called with cloning=false"); + } + let data = unsafe { Box::from_raw(serialized.pv_raw::<NotificationConfig>()?) }; + Ok(perlmod::instantiate_magic!(&class, MAGIC => data)) + } + + #[export(raw_return)] + fn new(#[raw] class: Value, raw_config: &str) -> Result<Value, Error> { + Ok(perlmod::instantiate_magic!(&class, MAGIC => Box::new( + NotificationConfig { + config: Mutex::new(Config::new(raw_config)?) + } + ))) + } + + #[export] + fn write(#[try_from_ref] this: &NotificationConfig) -> Result<String, Error> { + this.config.lock().unwrap().write() + } + + #[export] + fn send( + #[try_from_ref] this: &NotificationConfig, + severity: Severity, + title: &str, + message: &str, + properties: Option<HashMap<String, String>>, + ) -> Result<(), Error> { + let config = this.config.lock().unwrap(); + + let notification_bus = config.instantiate()?; + + let notification = Notification { + title: title.into(), + body: message.into(), + severity, + properties: properties.unwrap_or_default(), + }; + + notification_bus.send(¬ification)?; + + Ok(()) + } + + #[export] + fn add_sendmail_endpoint( + #[try_from_ref] this: &NotificationConfig, + name: &str, + recipient: Vec<String>, + from_address: Option<String>, + author: Option<String>, + filter: Option<String>, + ) -> Result<(), Error> { + let mut config = this.config.lock().unwrap(); + methods::add_sendmail_endpoint(&mut config, name, recipient, from_address, author, filter)?; + + Ok(()) + } + + #[export] + fn add_filter( + #[try_from_ref] this: &NotificationConfig, + name: &str, + min_severity: Option<Severity>, + sub_filter: Option<Vec<String>>, + mode: Option<FilterModeOperator>, + match_properties: Option<Vec<String>>, + invert_match: Option<bool>, + ) -> Result<(), Error> { + let mut config = this.config.lock().unwrap(); + + methods::add_filter( + &mut config, + name, + min_severity, + sub_filter, + mode, + match_properties, + invert_match, + )?; + + Ok(()) + } +} -- 2.30.2 _______________________________________________ pve-devel mailing list pve-devel@lists.proxmox.com https://lists.proxmox.com/cgi-bin/mailman/listinfo/pve-devel