Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add some quality-of-life gdbus helpers #1558

Merged
merged 6 commits into from
Nov 2, 2024
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
77 changes: 59 additions & 18 deletions examples/gio_dbus_register_object/main.rs
Original file line number Diff line number Diff line change
@@ -1,5 +1,8 @@
use gio::prelude::*;
use std::sync::mpsc::{channel, Receiver, Sender};
use gio::{prelude::*, IOErrorEnum};
use std::{
sync::mpsc::{channel, Receiver, Sender},
time::Duration,
};

const EXAMPLE_XML: &str = r#"
<node>
Expand All @@ -8,10 +11,48 @@ const EXAMPLE_XML: &str = r#"
<arg type='s' name='name' direction='in'/>
<arg type='s' name='greet' direction='out'/>
</method>
<method name='SlowHello'>
<arg type='s' name='name' direction='in'/>
<arg type='u' name='delay' direction='in'/>
<arg type='s' name='greet' direction='out'/>
</method>
</interface>
</node>
"#;

#[derive(Debug, glib::Variant)]
struct Hello {
name: String,
}

#[derive(Debug, glib::Variant)]
struct SlowHello {
name: String,
delay: u32,
}

#[derive(Debug)]
enum HelloMethod {
Hello(Hello),
SlowHello(SlowHello),
}

impl DBusMethodCall for HelloMethod {
fn parse_call(
swsnr marked this conversation as resolved.
Show resolved Hide resolved
_obj_path: &str,
swsnr marked this conversation as resolved.
Show resolved Hide resolved
_interface: &str,
method: &str,
params: glib::Variant,
) -> Result<Self, glib::Error> {
match method {
"Hello" => Ok(params.get::<Hello>().map(Self::Hello)),
"SlowHello" => Ok(params.get::<SlowHello>().map(Self::SlowHello)),
_ => Err(glib::Error::new(IOErrorEnum::Failed, "No such method")),
}
.and_then(|p| p.ok_or_else(|| glib::Error::new(IOErrorEnum::Failed, "Invalid parameters")))
}
}

fn on_startup(app: &gio::Application, tx: &Sender<gio::RegistrationId>) {
let connection = app.dbus_connection().expect("connection");

Expand All @@ -22,25 +63,25 @@ fn on_startup(app: &gio::Application, tx: &Sender<gio::RegistrationId>) {

if let Ok(id) = connection
.register_object("/com/github/gtk_rs/examples/HelloWorld", &example)
.method_call(glib::clone!(
#[strong]
app,
move |_, _, _, _, method, params, invocation| {
match method {
"Hello" => {
if let Some((name,)) = <(String,)>::from_variant(&params) {
let greet = format!("Hello {name}!");
println!("{greet}");
invocation.return_value(Some(&(greet,).to_variant()));
} else {
invocation.return_error(gio::IOErrorEnum::Failed, "Invalid parameters");
}
.typed_method_call::<HelloMethod>()
.invoke_and_return_future_local(|_, sender, call| {
println!("Method call from {sender}");
async {
match call {
HelloMethod::Hello(Hello { name }) => {
let greet = format!("Hello {name}!");
println!("{greet}");
Ok(Some(greet.to_variant()))
}
HelloMethod::SlowHello(SlowHello { name, delay }) => {
glib::timeout_future(Duration::from_secs(delay as u64)).await;
let greet = format!("Hello {name} after {delay} seconds!");
println!("{greet}");
Ok(Some(greet.to_variant()))
}
_ => unreachable!(),
}
app.quit();
}
))
})
.build()
{
println!("Registered object");
Expand Down
120 changes: 116 additions & 4 deletions gio/src/dbus_connection.rs
Original file line number Diff line number Diff line change
@@ -1,13 +1,110 @@
// Take a look at the license at the top of the repository in the LICENSE file.

use std::{boxed::Box as Box_, num::NonZeroU32};

use glib::{prelude::*, translate::*};
use std::{boxed::Box as Box_, future::Future, marker::PhantomData, num::NonZeroU32};

use crate::{
ffi, ActionGroup, DBusConnection, DBusInterfaceInfo, DBusMessage, DBusMethodInvocation,
DBusSignalFlags, MenuModel,
};
use glib::{prelude::*, translate::*};

pub trait DBusMethodCall: Sized {
fn parse_call(
obj_path: &str,
interface: &str,
method: &str,
params: glib::Variant,
) -> Result<Self, glib::Error>;
}

// rustdoc-stripper-ignore-next
/// Handle method invocations.
pub struct MethodCallBuilder<'a, T> {
registration: RegistrationBuilder<'a>,
capture_type: PhantomData<T>,
}

impl<'a, T: DBusMethodCall> MethodCallBuilder<'a, T> {
// rustdoc-stripper-ignore-next
/// Handle invocation of a parsed method call.
///
/// For each DBus method call parse the call, and then invoke the given closure
/// with
///
/// 1. the DBus connection object,
/// 2. the name of the sender of the method call,
/// 3. the parsed call, and
/// 4. the method invocation object.
///
/// The closure **must** return a value through the invocation object in all
/// code paths, using any of its `return_` functions, such as
/// [`DBusMethodInvocation::return_result`] or
/// [`DBusMethodInvocation::return_future_local`], to finish the call.
///
/// If direct access to the invocation object is not needed,
/// [`invoke_and_return`] and [`invoke_and_return_future_local`] provide a
/// safer interface where the callback returns a result directly.
pub fn invoke<F>(self, f: F) -> RegistrationBuilder<'a>
where
F: Fn(DBusConnection, &str, T, DBusMethodInvocation) + 'static,
{
self.registration.method_call(
move |connection, sender, obj_path, interface, method, params, invocation| {
match T::parse_call(obj_path, interface, method, params) {
Ok(call) => f(connection, sender, call, invocation),
Err(error) => invocation.return_gerror(error),
}
},
)
}

// rustdoc-stripper-ignore-next
/// Handle invocation of a parsed method call.
///
/// For each DBus method call parse the call, and then invoke the given closure
/// with
///
/// 1. the DBus connection object,
/// 2. the name of the sender of the method call, and
/// 3. the parsed call.
///
/// The return value of the closure is then returned on the method call.
/// If the returned variant value is not a tuple, it is automatically wrapped
/// in a single element tuple, as DBus methods must always return tuples.
/// See [`DBusMethodInvocation::return_result`] for details.
pub fn invoke_and_return<F>(self, f: F) -> RegistrationBuilder<'a>
where
F: Fn(DBusConnection, &str, T) -> Result<Option<glib::Variant>, glib::Error> + 'static,
{
self.invoke(move |connection, sender, call, invocation| {
invocation.return_result(f(connection, sender, call))
})
}

// rustdoc-stripper-ignore-next
/// Handle an async invocation of a parsed method call.
///
/// For each DBus method call parse the call, and then invoke the given closure
/// with
///
/// 1. the DBus connection object,
/// 2. the name of the sender of the method call, and
/// 3. the parsed call.
///
/// The output of the future is then returned on the method call.
/// If the returned variant value is not a tuple, it is automatically wrapped
/// in a single element tuple, as DBus methods must always return tuples.
/// See [`DBusMethodInvocation::return_future_local`] for details.
pub fn invoke_and_return_future_local<F, Fut>(self, f: F) -> RegistrationBuilder<'a>
where
F: Fn(DBusConnection, &str, T) -> Fut + 'static,
Fut: Future<Output = Result<Option<glib::Variant>, glib::Error>> + 'static,
{
self.invoke(move |connection, sender, call, invocation| {
invocation.return_future_local(f(connection, sender, call));
})
}
}

#[derive(Debug, Eq, PartialEq)]
pub struct RegistrationId(NonZeroU32);
Expand All @@ -22,6 +119,8 @@ pub struct FilterId(NonZeroU32);
#[derive(Debug, Eq, PartialEq)]
pub struct SignalSubscriptionId(NonZeroU32);

// rustdoc-stripper-ignore-next
/// Build a registered DBus object, by handling different parts of DBus.
#[must_use = "The builder must be built to be used"]
pub struct RegistrationBuilder<'a> {
connection: &'a DBusConnection,
Expand All @@ -38,7 +137,7 @@ pub struct RegistrationBuilder<'a> {
Option<Box_<dyn Fn(DBusConnection, &str, &str, &str, &str, glib::Variant) -> bool>>,
}

impl RegistrationBuilder<'_> {
impl<'a> RegistrationBuilder<'a> {
pub fn method_call<
F: Fn(DBusConnection, &str, &str, &str, &str, glib::Variant, DBusMethodInvocation) + 'static,
>(
Expand All @@ -49,6 +148,19 @@ impl RegistrationBuilder<'_> {
self
}

// rustdoc-stripper-ignore-next
/// Handle method calls on this object.
///
/// Return a builder for method calls which parses method names and
/// parameters with the given [`DBusMethodCall`] and then allows to dispatch
/// the parsed call either synchronously or asynchronously.
pub fn typed_method_call<T: DBusMethodCall>(self) -> MethodCallBuilder<'a, T> {
MethodCallBuilder {
registration: self,
capture_type: Default::default(),
}
}

#[doc(alias = "get_property")]
pub fn property<F: Fn(DBusConnection, &str, &str, &str, &str) -> glib::Variant + 'static>(
mut self,
Expand Down
42 changes: 41 additions & 1 deletion gio/src/dbus_method_invocation.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
// Take a look at the license at the top of the repository in the LICENSE file.

use glib::{prelude::*, translate::*};
use glib::{prelude::*, translate::*, VariantTy};

use crate::{ffi, DBusMethodInvocation};

Expand All @@ -26,4 +26,44 @@ impl DBusMethodInvocation {
);
}
}

// rustdoc-stripper-ignore-next
/// Return a result for this invocation.
///
/// If `Ok` return the contained value with [`return_value`]. If the return
/// value is not a tuple, automatically convert it to a one-element tuple, as
/// DBus return values must be tuples.
///
/// If `Err` return the contained error with [`return_gerror`].
pub fn return_result(self, result: Result<Option<glib::Variant>, glib::Error>) {
match result {
Ok(Some(value)) if !value.is_type(VariantTy::TUPLE) => {
let tupled = glib::Variant::tuple_from_iter(std::iter::once(value));
self.return_value(Some(&tupled));
}
Ok(value) => self.return_value(value.as_ref()),
Err(error) => self.return_gerror(error),
}
}

// rustdoc-stripper-ignore-next
/// Return an async result for this invocation.
///
/// Spawn the given future on the thread-default main context, and return the
/// the result with [`return_result`]. Specifically, if a variant is returned
/// that is not a tuple it is automatically wrapped into a tuple.
///
/// The given `Future` does not have to be `Send`.
///
/// This can be called only from the thread where the main context is running, e.g.
/// from any other `Future` that is executed on this main context, or after calling
/// `with_thread_default` or `acquire` on the main context.
pub fn return_future_local<F>(self, f: F) -> glib::JoinHandle<()>
where
F: std::future::Future<Output = Result<Option<glib::Variant>, glib::Error>> + 'static,
{
glib::spawn_future_local(async move {
self.return_result(f.await);
})
}
}
9 changes: 5 additions & 4 deletions gio/src/prelude.rs
Original file line number Diff line number Diff line change
Expand Up @@ -35,10 +35,11 @@ pub use crate::{
action_map::ActionMapExtManual, application::ApplicationExtManual, auto::traits::*,
cancellable::CancellableExtManual, converter::ConverterExtManual,
data_input_stream::DataInputStreamExtManual, datagram_based::DatagramBasedExtManual,
dbus_proxy::DBusProxyExtManual, file::FileExtManual, file_enumerator::FileEnumeratorExtManual,
inet_address::InetAddressExtManual, input_stream::InputStreamExtManual,
io_stream::IOStreamExtManual, list_model::ListModelExtManual,
output_stream::OutputStreamExtManual, pollable_input_stream::PollableInputStreamExtManual,
dbus_connection::DBusMethodCall, dbus_proxy::DBusProxyExtManual, file::FileExtManual,
file_enumerator::FileEnumeratorExtManual, inet_address::InetAddressExtManual,
input_stream::InputStreamExtManual, io_stream::IOStreamExtManual,
list_model::ListModelExtManual, output_stream::OutputStreamExtManual,
pollable_input_stream::PollableInputStreamExtManual,
pollable_output_stream::PollableOutputStreamExtManual, settings::SettingsExtManual,
simple_proxy_resolver::SimpleProxyResolverExtManual, socket::SocketExtManual,
socket_control_message::SocketControlMessageExtManual,
Expand Down
2 changes: 1 addition & 1 deletion glib/src/source.rs
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ impl FromGlib<u32> for SourceId {
}

// rustdoc-stripper-ignore-next
/// Process identificator
/// Process identifier
#[derive(Copy, Clone, Debug, Eq, PartialEq)]
#[doc(alias = "GPid")]
pub struct Pid(pub ffi::GPid);
Expand Down
Loading