Compare commits

..

No commits in common. "78a84f4b4815223bef6a065aa877b3703becb289" and "ca9499db5fdcbfb6b96d86765178fc17ddf3da36" have entirely different histories.

8 changed files with 124 additions and 116 deletions

View File

@ -9,7 +9,6 @@ zbus = "4.4.0"
zvariant = "4.2.0"
tokio = { version = "1.40.0", features = ["full"] }
futures-util = "0.3.30"
serde = { version = "1.0.210", features = ["derive"] }
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0.128"
rson_rs = "0.2.1"
argh = "0.1.12"
rson_rs = "0.2.1"

View File

@ -8,20 +8,6 @@ Inspired by [`tiramisu`](https://github.com/Sweets/tiramisu)
- Do one thing and do it well([DOTADIW](https://en.wikipedia.org/w/index.php?title=Unix_philosophy&useskin=vector#Do_One_Thing_and_Do_It_Well)) & [KISS](https://en.wikipedia.org/wiki/KISS_Principle) principle: no extra complicated features
- (Not really a feature) Written in [`rust`](https://www.rust-lang.org/) using the [`zbus`](https://docs.rs/zbus/latest/zbus/) crate
## Usage
```bash
snot --help
Usage: snot [-f <format>] [-v]
Print desktop notifications
Options:
-f, --format select output format: r(rson), j(json, default), p(plain)
-v, --verbose verbose mode
--help display usage information
```
## Supported formats
- Plain text - Print the output text. (✓ Just print it)
@ -33,6 +19,12 @@ Options:
- Better handling of `json` and `rson` data
- Better ways to work with other programs
## Usage
```bash
snot [r|j|p] [v]
```
## Why this project?
- Something simple to work with [`EWW`](https://github.com/elkowar/eww) widgets

View File

@ -3,7 +3,68 @@ use crate::notification::Notification;
use serde_json::{json, Value}; // json!() macro and Value type
impl Notification {
// Actions, as json
pub fn actions_json(&self) -> Value {
if self.actions().is_empty() {
json!({}) // Empty JSON object if no actions
} else {
serde_json::Value::Object(
// Wrap the map into a serde_json::Value
self.actions()
.iter()
.map(|(id, label)| {
(id.clone(), json!(label)) // Create key-value pairs: id -> label
})
.collect::<serde_json::Map<_, _>>(), // Collect into a serde_json::Map
)
}
}
// Default action, as json string
pub fn default_action_json(&self) -> Value {
match self.default_action() {
None => serde_json::Value::from("None"),
Some(actn) => serde_json::Value::String(actn.0),
}
}
// Hints, as json
pub fn hints_json(&self) -> Value {
if self.hints().is_empty() {
json!({}) // Empty JSON object if no hints
} else {
serde_json::Value::Object(
self.hints()
.iter()
.map(|(key, value)| {
(key.clone(), json!(value.to_string())) // Convert hint value to string
})
.collect::<serde_json::Map<_, _>>(), // Collect into a serde_json::Map
)
}
}
// The notification as a json object
pub fn json(&self) -> Value {
json!(self)
// Initialize
let mut notif_json: Value = json!({
"app_name": &self.app_name(),
"replace_id": &self.replace_id(),
"icon": &self.icon(),
"summary": &self.summary(),
"body": &self.body(),
"hints": &self.hints_json(),
"expiration_timeout": self.expir_timeout(),
"urgency": self.urgency(),
});
// Conditionally add the Actions fields
if let Value::Object(ref mut map) = notif_json {
if !self.actions().is_empty() {
map.insert("actions".to_string(), self.actions_json());
map.insert("default_action".to_string(), self.default_action_json());
}
}
notif_json
}
}

View File

@ -1,10 +1,19 @@
// This module deals with converting the notification object into rson format, which can be used instead of json if preferred
use crate::notification::Notification;
use rson_rs::ser::to_string as rson_string;
use rson_rs::de::Error as RsonError;
impl Notification {
pub fn rson(&self) -> Result<String, RsonError> {
rson_string(self).map_err(|e| RsonError::Message(format!("RSON serialization error: {}", e)))
pub fn actions_rson(&self) -> String {
if self.actions().is_empty() {
String::new()
} else {
rson_string(&self.actions()).unwrap()
}
}
pub fn hints_rson(&self) -> String {
rson_string(&self.hints()).unwrap()
}
pub fn rson(&self) -> String {
rson_string(&self).unwrap()
}
}

View File

@ -1,41 +0,0 @@
use serde::{Serialize, Serializer};
/*
use std::collections::HashMap;
use std::hash::BuildHasher;
use zvariant::OwnedValue; */
pub fn serialize_actions<S>(actions: &[String], serializer: S) -> Result<S::Ok, S::Error>
where
S: Serializer,
{
let mut map = serde_json::Map::new();
// Assuming actions are in pairs: [id, label, id, label, ...]
for chunk in actions.chunks(2) {
if let [id, label] = chunk {
map.insert(id.clone(), serde_json::Value::String(label.clone()));
}
}
map.serialize(serializer)
}
/*
pub fn serialize_hints<S, H>(
hints: &HashMap<String, OwnedValue, H>,
serializer: S,
) -> Result<S::Ok, S::Error>
where
S: Serializer,
H: BuildHasher,
{
let mut map = serde_json::Map::new();
for (key, value) in hints {
// Customize OwnedValue serialization as needed
map.insert(key.clone(), serde_json::Value::String(value.to_string()));
}
map.serialize(serializer)
}
*/

View File

@ -2,19 +2,18 @@ pub mod formats {
pub mod json;
pub mod plain;
pub mod rson;
pub mod serde;
}
mod notification;
use notification::{to_notif, Notification};
use std::collections::HashMap;
pub mod optparse;
use std::env;
use futures_util::stream::TryStreamExt;
use zbus::{message::Body, Connection, Result};
const SERVER_NAME: &str = "SNot"; // Server software name
const VENDOR: &str = "candifloss.cc"; // Server software vendor
const VERSION: &str = "0.1.2"; // Server software version
const VERSION: &str = "0.1.0"; // Server software version
const SPEC_VERSION: &str = "0.42"; // DBus specification version
const NOTIF_INTERFACE: &str = "org.freedesktop.Notifications"; // DBus interface name
@ -32,28 +31,18 @@ fn server_properties() -> HashMap<String, String> {
#[tokio::main]
async fn main() -> Result<()> {
// Options
let args: optparse::Cli = argh::from_env();
// Format: r|j|p
let op_format = match args.format {
Some(value) => {
// Reject invalid format option
if ["r", "j", "p"].contains(&value.as_str()) {
value.clone()
} else {
// Exit with error code
return Err(zbus::Error::from(std::io::Error::new(
std::io::ErrorKind::InvalidInput,
"Invalid output format",
)));
}
}
None => "j".to_string(),
let args: Vec<String> = env::args().collect();
let op_format: &str = if args.is_empty() || args[1] == "j" {
"j" // Default value, json format
} else if args[1] == "p" {
"p" // Plain format
} else if args[1] == "r" {
"r" // rson format
} else {
"j"
};
// Verbose mode. Off/false by default
let verbose = args.verbose;
let verbose: bool = (args.len() > 2) && (args[2] == "v");
let connection = Connection::session().await?;
connection
@ -85,6 +74,9 @@ async fn main() -> Result<()> {
println!("GetAll request received for interface: {interface_name}");
}
} else {
if verbose {
println!("Unknown interface requested: {interface_name}");
}
// Reply with an error
connection
.reply_error(
@ -93,9 +85,6 @@ async fn main() -> Result<()> {
&"Unknown interface".to_string(),
)
.await?;
if verbose {
println!("Unknown interface requested: {interface_name}");
}
}
}
"GetServerInformation" => {
@ -127,27 +116,23 @@ async fn main() -> Result<()> {
// Convert the msg body to a Notification object
let notif: Notification = to_notif(&msg_body)?;
// Handle the notif
match op_format.as_str() {
match op_format {
"j" => {
println!("{}", &notif.json()); // Print the json version
}
"r" => {
// Print the rson version
match notif.rson() {
Ok(rson_string) => println!("{}", rson_string),
Err(e) => eprintln!("Failed to convert to RSON: {}", e),
}
println!("{}", &notif.rson()); // Print the plain version
}
"p" => {
println!("{}\n", &notif.plain()); // Print the plain version
}
_ => {
println!("Onkown output format."); // This is probably unreachable
println!("Onkown output format.");
}
}
}
"CloseNotification" => {
// Client sent a 'close' signal. Extract notification ID of the notif to be closed from the message body
// Client sent a close signal. Extract notification ID of the notif to be closed from the message body
let notification_id: u32 = msg.body().deserialize()?; // This method has only one parameter, the id
// Tracking notifications by their IDs, closing them, and other features may be implemented later

View File

@ -1,4 +1,3 @@
use crate::formats::serde::serialize_actions;
use serde::Serialize;
use std::collections::HashMap;
use zbus::{message::Body, Result};
@ -18,10 +17,8 @@ pub struct Notification {
// Notification content/body
body: String,
// Action requests that can be sent back to the client - "Reply," "Mark as Read," "Play/Pause/Next," "Snooze/Dismiss," etc.
#[serde(serialize_with = "serialize_actions")]
actions: Vec<String>,
// Useful extra data - notif type, urgency, notif sound, icon data, etc.
//#[serde(serialize_with = "serialize_hints")]
hints: HashMap<String, OwnedValue>,
// Seconds till this notif expires. Optional
expir_timeout: i32,
@ -51,11 +48,30 @@ impl Notification {
// Key:Val pairs of actions
pub fn actions(&self) -> Vec<(String, String)> {
self.actions
.chunks(2)
.map(|chunk| (chunk[0].clone(), chunk[1].clone()))
.collect()
let mut actions: Vec<(String, String)> = vec![];
let acts = &self.actions;
let act_len = acts.len();
let mut i = 0;
while i < act_len {
// Action ID, used by the sender to id the clicked action
let action_id = &acts[i];
// Localised human-readable string that describes the action
let action_label = &acts[i + 1];
// Pair of (id, label)
let action_pair = (action_id.to_owned(), action_label.to_owned());
// Add it to the Vec
actions.push(action_pair);
i += 2;
}
actions
}
/*
pub fn actions(&self) -> Vec<(String, String)> {
self.actions
.chunks(2)
.map(|chunk| (chunk[0].clone(), chunk[1].clone()))
.collect()
} */
// Hints
pub fn hints(&self) -> &HashMap<String, OwnedValue> {

View File

@ -1,13 +0,0 @@
use argh::FromArgs;
#[derive(FromArgs)]
/// Print desktop notifications
pub struct Cli {
/// select output format: r(rson), j(json, default), p(plain)
#[argh(option, short = 'f')]
pub format: Option<String>,
/// verbose mode
#[argh(switch, short = 'v')]
pub verbose: bool,
}