tnr 0.2.0

A macOS notification library and CLI tool using terminal-notifier
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142

use std::process::Command;

#[derive(Debug, Clone)]
pub enum Type {
    Error,
    Warning,
    Info,
    Success,
}

impl Type {
    pub fn from_str(s: &str) -> Option<Self> {
        match s.to_lowercase().as_str() {
            "error" => Some(Self::Error),
            "warning" => Some(Self::Warning),
            "info" => Some(Self::Info),
            "success" => Some(Self::Success),
            _ => None,
        }
    }
    
    pub fn get_emoji(&self) -> &'static str {
        match self {
            Self::Error => "❌",
            Self::Warning => "⚠️",
            Self::Info => "ℹ️",
            Self::Success => "✅",
        }
    }
    
    pub fn get_default_title(&self) -> &'static str {
        match self {
            Self::Error => "Error",
            Self::Warning => "Warning",
            Self::Info => "Information",
            Self::Success => "Success",
        }
    }
    
    pub fn get_sound(&self) -> &'static str {
        match self {
            Self::Error => "Sosumi",
            Self::Warning => "Funk",
            Self::Info => "Glass",
            Self::Success => "Hero",
        }
    }
}

#[derive(Debug, Clone)]
pub struct Notify {
    pub title: String,
    pub message: String,
    pub notification_type: Option<Type>,
    pub url: Option<String>,
}

impl Notify {
    pub fn new(message: impl Into<String>) -> Self {
        Self {
            title: "Notification".to_string(),
            message: message.into(),
            notification_type: None,
            url: None,
        }
    }
    
    pub fn with_title(mut self, title: impl Into<String>) -> Self {
        self.title = title.into();
        self
    }
    
    pub fn with_type(mut self, notification_type: Type) -> Self {
        self.notification_type = Some(notification_type);
        self
    }
    
    pub fn with_url(mut self, url: impl Into<String>) -> Self {
        self.url = Some(url.into());
        self
    }
    
    pub fn send(&self) -> Result<(), TnrError> {
        send_notification(self)
    }
}

#[derive(Debug)]
pub enum TnrError {
    ExecutionFailed(String),
    CommandNotFound(String),
}

impl std::fmt::Display for TnrError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            TnrError::ExecutionFailed(msg) => write!(f, "Failed to send notification: {}", msg),
            TnrError::CommandNotFound(msg) => write!(f, "Command not found: {}", msg),
        }
    }
}

impl std::error::Error for TnrError {}

pub fn send_notification(notification: &Notify) -> Result<(), TnrError> {
    let title = if notification.title.is_empty() {
        match &notification.notification_type {
            Some(nt) => format!("{} {}", nt.get_emoji(), nt.get_default_title()),
            None => "Notification".to_string(),
        }
    } else {
        notification.title.clone()
    };
    
    let message = match &notification.notification_type {
        Some(nt) => format!("{} {}", nt.get_emoji(), notification.message),
        None => notification.message.clone(),
    };
    
    let mut cmd = Command::new("terminal-notifier");
    cmd.arg("-title")
        .arg(&title)
        .arg("-message")
        .arg(&message);
    
    if let Some(nt) = &notification.notification_type {
        cmd.arg("-sound").arg(nt.get_sound());
    }
    
    if let Some(url) = &notification.url {
        cmd.arg("-open").arg(url);
    }
    
    let output = cmd.output()
        .map_err(|e| TnrError::CommandNotFound(format!("terminal-notifier: {}. Please install with: brew install terminal-notifier", e)))?;
    
    if output.status.success() {
        Ok(())
    } else {
        Err(TnrError::ExecutionFailed(String::from_utf8_lossy(&output.stderr).to_string()))
    }
}