Got rson working, added format and verbose options
This commit is contained in:
parent
3b92de541a
commit
6fe5d22745
@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "snot"
|
name = "snot"
|
||||||
version = "0.1.0"
|
version = "0.1.1"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
authors = ["candifloss <candifloss.cc>"]
|
authors = ["candifloss <candifloss.cc>"]
|
||||||
|
|
||||||
@ -9,5 +9,6 @@ zbus = "4.4.0"
|
|||||||
zvariant = "4.2.0"
|
zvariant = "4.2.0"
|
||||||
tokio = { version = "1.40.0", features = ["full"] }
|
tokio = { version = "1.40.0", features = ["full"] }
|
||||||
futures-util = "0.3.30"
|
futures-util = "0.3.30"
|
||||||
|
serde = { version = "1.0", features = ["derive"] }
|
||||||
serde_json = "1.0.128"
|
serde_json = "1.0.128"
|
||||||
rson_rs = "0.2.1"
|
rson_rs = "0.2.1"
|
@ -1,21 +1,19 @@
|
|||||||
// This module deals with converting the notification object into rson format, which can be used instead of json if preferred
|
// 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 crate::notification::Notification;
|
||||||
use rson_rs::ser::to_string;
|
use rson_rs::ser::to_string as rson_string;
|
||||||
use rson_rs::value::Value;
|
|
||||||
use rson_rs::value::Value::Map;
|
|
||||||
|
|
||||||
impl Notification {
|
impl Notification {
|
||||||
pub fn actions_rson(&self) -> Value {
|
pub fn actions_rson(&self) -> String {
|
||||||
if self.actions().is_empty() {
|
if self.actions().is_empty() {
|
||||||
Value::from_str("{actions: Null}")
|
String::new()
|
||||||
} else {
|
} else {
|
||||||
// Wrap the map into a rson_rs::value::Value
|
rson_string(&self.actions()).unwrap()
|
||||||
self.actions()
|
|
||||||
.iter()
|
|
||||||
.map(|(id, label)| {
|
|
||||||
(id.clone(), Value::from_str(label)) // Create key-value pairs: id -> label
|
|
||||||
})
|
|
||||||
.collect::<Map<_, _>>() // Collect into a rson_rs::value::Value::Map
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
pub fn hints_rson(&self) -> String {
|
||||||
|
rson_string(&self.hints()).unwrap()
|
||||||
|
}
|
||||||
|
pub fn rson(&self) -> String {
|
||||||
|
rson_string(&self).unwrap()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
43
src/main.rs
43
src/main.rs
@ -1,7 +1,7 @@
|
|||||||
pub mod formats {
|
pub mod formats {
|
||||||
pub mod json;
|
pub mod json;
|
||||||
pub mod plain;
|
pub mod plain;
|
||||||
//pub mod rson;
|
pub mod rson;
|
||||||
}
|
}
|
||||||
mod notification;
|
mod notification;
|
||||||
use notification::{to_notif, Notification};
|
use notification::{to_notif, Notification};
|
||||||
@ -32,12 +32,18 @@ fn server_properties() -> HashMap<String, String> {
|
|||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() -> Result<()> {
|
async fn main() -> Result<()> {
|
||||||
let args: Vec<String> = env::args().collect();
|
let args: Vec<String> = env::args().collect();
|
||||||
let op_format: String = if args.is_empty() || args[1] == "j" {
|
let op_format: &str = if args.is_empty() || args[1] == "j" {
|
||||||
"j".to_string() // Default value, json format
|
"j" // Default value, json format
|
||||||
|
} else if args[1] == "p" {
|
||||||
|
"p" // Plain format
|
||||||
|
} else if args[1] == "r" {
|
||||||
|
"r" // rson format
|
||||||
} else {
|
} else {
|
||||||
"p".to_string() // Plain format
|
"j"
|
||||||
};
|
};
|
||||||
|
|
||||||
|
let verbose: bool = (args.len() > 2) && (args[2] == "v");
|
||||||
|
|
||||||
let connection = Connection::session().await?;
|
let connection = Connection::session().await?;
|
||||||
connection
|
connection
|
||||||
.request_name(NOTIF_INTERFACE) // Requesting dbus for this service name. Any other services/procs using this name should be stopped/disabled before this
|
.request_name(NOTIF_INTERFACE) // Requesting dbus for this service name. Any other services/procs using this name should be stopped/disabled before this
|
||||||
@ -64,9 +70,13 @@ async fn main() -> Result<()> {
|
|||||||
let properties = server_properties();
|
let properties = server_properties();
|
||||||
// Reply with the properties
|
// Reply with the properties
|
||||||
connection.reply(&msg, &properties).await?;
|
connection.reply(&msg, &properties).await?;
|
||||||
|
if verbose {
|
||||||
println!("GetAll request received for interface: {interface_name}");
|
println!("GetAll request received for interface: {interface_name}");
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
|
if verbose {
|
||||||
println!("Unknown interface requested: {interface_name}");
|
println!("Unknown interface requested: {interface_name}");
|
||||||
|
}
|
||||||
// Reply with an error
|
// Reply with an error
|
||||||
connection
|
connection
|
||||||
.reply_error(
|
.reply_error(
|
||||||
@ -81,16 +91,20 @@ async fn main() -> Result<()> {
|
|||||||
// Client requested server information. Respond with: (Server_name, author, software_version, dbus_spec_version)
|
// Client requested server information. Respond with: (Server_name, author, software_version, dbus_spec_version)
|
||||||
let response = (SERVER_NAME, VENDOR, VERSION, SPEC_VERSION);
|
let response = (SERVER_NAME, VENDOR, VERSION, SPEC_VERSION);
|
||||||
connection.reply(&msg, &response).await?;
|
connection.reply(&msg, &response).await?;
|
||||||
|
if verbose {
|
||||||
println!("Request received: {member}\n\tName: {SERVER_NAME}, Vendor: {VENDOR}, Version: {VERSION}, DBus spec version: {SPEC_VERSION}");
|
println!("Request received: {member}\n\tName: {SERVER_NAME}, Vendor: {VENDOR}, Version: {VERSION}, DBus spec version: {SPEC_VERSION}");
|
||||||
// Remove this LATER
|
// Remove this LATER
|
||||||
}
|
}
|
||||||
|
}
|
||||||
"GetCapabilities" => {
|
"GetCapabilities" => {
|
||||||
// Client requested server capabilities. Respond with the supported capabilities
|
// Client requested server capabilities. Respond with the supported capabilities
|
||||||
let capabilities = vec!["actions", "body", "body-hyperlinks"]; // Add more LATER
|
let capabilities = vec!["actions", "body", "body-hyperlinks"]; // Add more LATER
|
||||||
connection.reply(&msg, &capabilities).await?;
|
connection.reply(&msg, &capabilities).await?;
|
||||||
|
if verbose {
|
||||||
println!("Request received: {member}\n\tCapabilities: {capabilities:?}");
|
println!("Request received: {member}\n\tCapabilities: {capabilities:?}");
|
||||||
// Remove this LATER
|
// Remove this LATER
|
||||||
}
|
}
|
||||||
|
}
|
||||||
"Notify" => {
|
"Notify" => {
|
||||||
// New notification received. Now, respond to the client with a notification ID
|
// New notification received. Now, respond to the client with a notification ID
|
||||||
notification_id += 1; // This could be incremented or generated.
|
notification_id += 1; // This could be incremented or generated.
|
||||||
@ -102,10 +116,19 @@ async fn main() -> Result<()> {
|
|||||||
// Convert the msg body to a Notification object
|
// Convert the msg body to a Notification object
|
||||||
let notif: Notification = to_notif(&msg_body)?;
|
let notif: Notification = to_notif(&msg_body)?;
|
||||||
// Handle the notif
|
// Handle the notif
|
||||||
if op_format == "j" {
|
match op_format {
|
||||||
println!("JSON!\n{}\n", ¬if.json()); // Print the json version
|
"j" => {
|
||||||
} else {
|
println!("{}", ¬if.json()); // Print the json version
|
||||||
println!("New notification!\n{}\n", ¬if.plain()); // Print the plain version
|
}
|
||||||
|
"r" => {
|
||||||
|
println!("{}", ¬if.rson()); // Print the plain version
|
||||||
|
}
|
||||||
|
"p" => {
|
||||||
|
println!("{}\n", ¬if.plain()); // Print the plain version
|
||||||
|
}
|
||||||
|
_ => {
|
||||||
|
println!("Onkown output format.");
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
"CloseNotification" => {
|
"CloseNotification" => {
|
||||||
@ -114,17 +137,21 @@ async fn main() -> Result<()> {
|
|||||||
|
|
||||||
// Tracking notifications by their IDs, closing them, and other features may be implemented later
|
// Tracking notifications by their IDs, closing them, and other features may be implemented later
|
||||||
// close_notification(notification_id);
|
// close_notification(notification_id);
|
||||||
|
if verbose {
|
||||||
println!("Closing notification with ID: {notification_id}");
|
println!("Closing notification with ID: {notification_id}");
|
||||||
|
}
|
||||||
|
|
||||||
// Respond to the client, acknowledging the closure
|
// Respond to the client, acknowledging the closure
|
||||||
connection.reply(&msg, &()).await?;
|
connection.reply(&msg, &()).await?;
|
||||||
}
|
}
|
||||||
_ => {
|
_ => {
|
||||||
|
if verbose {
|
||||||
println!("Unhandled method: {member}"); // Other methods are either irrelevant or unhandled at this stage of development
|
println!("Unhandled method: {member}"); // Other methods are either irrelevant or unhandled at this stage of development
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
@ -1,10 +1,10 @@
|
|||||||
// use serde::Serialize;
|
use serde::Serialize;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use zbus::{message::Body, Result};
|
use zbus::{message::Body, Result};
|
||||||
use zvariant::OwnedValue;
|
use zvariant::OwnedValue;
|
||||||
|
|
||||||
// A notificaion object
|
// A notificaion object
|
||||||
// #[derive(Serialize)] // To help with json
|
#[derive(Serialize)] // To help with json, rson
|
||||||
pub struct Notification {
|
pub struct Notification {
|
||||||
// The application that sent the notification
|
// The application that sent the notification
|
||||||
app_name: String,
|
app_name: String,
|
||||||
|
Loading…
Reference in New Issue
Block a user