diff --git a/src/api/system/power.rs b/src/api/system/power.rs new file mode 100644 index 0000000..ad14ab7 --- /dev/null +++ b/src/api/system/power.rs @@ -0,0 +1,27 @@ +use crate::api::ApiError; +use crate::state::RpcState; +use jsonrpsee::types::Params; +use jsonrpsee::RpcModule; +use std::sync::Arc; + +pub fn register_methods(module: &mut RpcModule) { + module + .register_async_method("system.power.shutdown", halt_system) + .unwrap(); + + module + .register_async_method("system.power.restart", reboot_system) + .unwrap(); +} + +pub async fn halt_system<'a>(_: Params<'a>, _state: Arc) -> Result<(), ApiError> { + let systemd_manager = zbus_systemd::systemd1::ManagerProxy::new(&_state.dbus_conn).await?; + systemd_manager.halt().await?; + Ok(()) +} + +pub async fn reboot_system<'a>(_: Params<'a>, _state: Arc) -> Result<(), ApiError> { + let systemd_manager = zbus_systemd::systemd1::ManagerProxy::new(&_state.dbus_conn).await?; + systemd_manager.reboot().await?; + Ok(()) +}