123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
use deno_core::serde::Deserialize;use deno_core::serde_json;use notify_rust::{self, error::Error as NotifyRustError, Notification};
#[derive(Debug, Deserialize)]#[serde(tag = "type")]enum NotificationTimeout {    /// Do not expire, user will have to close this manually.    Never,    /// Expire after n milliseconds.    Milliseconds { value: u32 },}
impl Into<notify_rust::Timeout> for NotificationTimeout {    fn into(self) -> notify_rust::Timeout {        match self {            NotificationTimeout::Never => notify_rust::Timeout::Never,            NotificationTimeout::Milliseconds { value } => {                notify_rust::Timeout::Milliseconds(value)            }        }    }}
#[derive(Debug, Deserialize)]struct NotificationOptions {    #[serde(rename = "_title")]    title: String,
    #[serde(rename = "_subtitle")]    subtitle: Option<String>,
    #[serde(rename = "_body")]    body: Option<String>,
    #[serde(rename = "_icon")]    icon: Option<String>,
    #[serde(rename = "_soundName")]    sound_name: Option<String>,
    #[serde(rename = "_timeout")]    timeout: Option<NotificationTimeout>,}
/// Read the given NotificationOptions and send it as a notification.fn send_notification(options: NotificationOptions) -> Result<(), NotifyRustError> {    let mut notification = Notification::new();
    // Set Notification options    notification.summary(&options.title);
    if let Some(subtitle) = &options.subtitle {        notification.subtitle(subtitle);    }
    if let Some(body) = &options.body {        notification.body(body);    }
    if let Some(icon) = &options.icon {        notification.icon(icon);    }
    if let Some(sound_name) = &options.sound_name {        notification.sound_name(sound_name);    }
    if let Some(timeout) = options.timeout {        notification.timeout(timeout);    }
    // Display it    match notification.show() {        Ok(_) => Ok(()),        Err(error) => Err(error),    }}
// Set the app bundle on macos#[cfg(target_os = "macos")]fn set_bundle() {    use notify_rust::error::{ApplicationError, MacOsError};
    match notify_rust::set_application("com.apple.Terminal") {        // Success        Ok(_) => {}        // If already set, ignore        Err(MacOsError::Application(ApplicationError::AlreadySet(_))) => {}        // Another error, report it (but don't fail)        Err(err) => eprintln!("Error setting application bundle: {}", err),    }}
/// Convert the given str to a buffer pointer to be returned by ffi./// Inspired by https://github.com/denoland/deno_bindgenfn to_result_str(str: String) -> *const u8 {    let length = (str.len() as u32).to_be_bytes();    let mut v = length.to_vec();
    let bytes = str.as_bytes();    v.extend_from_slice(bytes);
    std::mem::forget(str);    let result = v.as_ptr();    // Leak the result to JS land.    std::mem::forget(v);    result}
#[no_mangle]pub extern "C" fn notify_send(ptr: *const u8, len: usize) -> *const u8 {    let buf = unsafe { std::slice::from_raw_parts(ptr, len) };
    // Try parsing JSON into NotificationOptions object    let options: NotificationOptions = match serde_json::from_slice(buf) {        Ok(opt) => opt,        Err(error) => {            return to_result_str(format!(                r#"{{"type":"error","when":"parsing_input","message":"{}"}}"#,                error.to_string().replace("\"", "\\\"")            ))        }    };
    // Set the app bundle (if necessary)    #[cfg(target_os = "macos")]    set_bundle();
    // Try sending the notification    match send_notification(options) {        Ok(_) => to_result_str(r#"{"type":"success"}"#.to_string()),        Err(error) => to_result_str(format!(            r#"{{"type":"error","when":"sending_notification","message":"{}"}}"#,            error.to_string().replace("\"", "\\\"")        )),    }}