might as well do this

This commit is contained in:
Mars 2025-04-30 03:00:48 -04:00
parent f6a7bee047
commit 3ad961a571
Signed by: pupbrained
GPG key ID: 0FF5B8826803F895
6 changed files with 424 additions and 227 deletions

View file

@ -1,29 +1,28 @@
#ifdef __linux__ #ifdef __linux__
// clang-format off // clang-format off
#include <cstring> // std::strlen #include <cstring> // std::strlen
#include <dbus/dbus.h> #include <dbus/dbus.h> // DBus::{DBusConnection, DBusMessage, DBusMessageIter, etc.}
#include <expected> // std::{unexpected, expected} #include <expected> // std::{unexpected, expected}
#include <format> // std::{format, format_to_n} #include <format> // std::{format, format_to_n}
#include <fstream> // std::ifstream #include <fstream> // std::ifstream
#include <climits> // PATH_MAX #include <limits> // std::numeric_limits
#include <limits> // std::numeric_limits #include <string> // std::{getline, string (String)}
#include <string> // std::{getline, string (String)} #include <string_view> // std::string_view (StringView)
#include <string_view> // std::string_view (StringView) #include <sys/socket.h> // ucred, getsockopt, SOL_SOCKET, SO_PEERCRED
#include <sys/socket.h> // ucred, getsockopt, SOL_SOCKET, SO_PEERCRED #include <sys/statvfs.h> // statvfs
#include <sys/statvfs.h> // statvfs #include <sys/sysinfo.h> // sysinfo
#include <sys/sysinfo.h> // sysinfo #include <sys/utsname.h> // utsname, uname
#include <sys/utsname.h> // utsname, uname #include <unistd.h> // readlink
#include <unistd.h> // readlink
#include "src/core/util/helpers.hpp" #include "src/core/util/helpers.hpp"
#include "src/core/util/logging.hpp" #include "src/core/util/logging.hpp"
#include "src/wrappers/dbus.hpp"
#include "src/wrappers/wayland.hpp" #include "src/wrappers/wayland.hpp"
#include "src/wrappers/xcb.hpp" #include "src/wrappers/xcb.hpp"
#include "os.hpp"
#include "linux/pkg_count.hpp" #include "linux/pkg_count.hpp"
#include "os.hpp"
// clang-format on // clang-format on
using namespace util::types; using namespace util::types;
@ -179,81 +178,6 @@ namespace {
return String(compositorNameView); return String(compositorNameView);
} }
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wold-style-cast"
fn GetMprisPlayers(DBusConnection* connection) -> Vec<String> {
Vec<String> mprisPlayers;
DBusError err;
dbus_error_init(&err);
// Create a method call to org.freedesktop.DBus.ListNames
DBusMessage* msg = dbus_message_new_method_call(
"org.freedesktop.DBus", // target service
"/org/freedesktop/DBus", // object path
"org.freedesktop.DBus", // interface name
"ListNames" // method name
);
if (!msg) {
debug_log("Failed to create message for ListNames.");
return mprisPlayers;
}
// Send the message and block until we get a reply.
DBusMessage* reply = dbus_connection_send_with_reply_and_block(connection, msg, -1, &err);
dbus_message_unref(msg);
if (dbus_error_is_set(&err)) {
debug_log("DBus error in ListNames: {}", err.message);
dbus_error_free(&err);
return mprisPlayers;
}
if (!reply) {
debug_log("No reply received for ListNames.");
return mprisPlayers;
}
// The expected reply signature is "as" (an array of strings)
DBusMessageIter iter;
if (!dbus_message_iter_init(reply, &iter)) {
debug_log("Reply has no arguments.");
dbus_message_unref(reply);
return mprisPlayers;
}
if (dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_ARRAY) {
debug_log("Reply argument is not an array.");
dbus_message_unref(reply);
return mprisPlayers;
}
// Iterate over the array of strings
DBusMessageIter subIter;
dbus_message_iter_recurse(&iter, &subIter);
while (dbus_message_iter_get_arg_type(&subIter) != DBUS_TYPE_INVALID) {
if (dbus_message_iter_get_arg_type(&subIter) == DBUS_TYPE_STRING) {
const char* name = nullptr;
dbus_message_iter_get_basic(&subIter, static_cast<void*>(&name));
if (name && std::string_view(name).contains("org.mpris.MediaPlayer2"))
mprisPlayers.emplace_back(name);
}
dbus_message_iter_next(&subIter);
}
dbus_message_unref(reply);
return mprisPlayers;
}
#pragma clang diagnostic pop
fn GetActivePlayer(const Vec<String>& mprisPlayers) -> Option<String> {
if (!mprisPlayers.empty())
return mprisPlayers.front();
return None;
}
} // namespace } // namespace
namespace os { namespace os {
@ -306,166 +230,134 @@ namespace os {
return info.totalram * info.mem_unit; return info.totalram * info.mem_unit;
} }
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wold-style-cast"
fn GetNowPlaying() -> Result<MediaInfo, DraconisError> { fn GetNowPlaying() -> Result<MediaInfo, DraconisError> {
DBusError err; Result<dbus::ConnectionGuard, DraconisError> connectionResult = dbus::BusGet(DBUS_BUS_SESSION);
dbus_error_init(&err);
// Connect to the session bus if (!connectionResult)
DBusConnection* connection = dbus_bus_get(DBUS_BUS_SESSION, &err); return Err(connectionResult.error());
if (!connection) dbus::ConnectionGuard& connection = *connectionResult;
if (dbus_error_is_set(&err)) {
error_log("DBus connection error: {}", err.message);
DraconisError error = DraconisError(DraconisErrorCode::ApiUnavailable, err.message); Option<String> activePlayer = None;
dbus_error_free(&err);
return Err(error); {
Result<dbus::MessageGuard, DraconisError> listNamesResult = dbus::MessageNewMethodCall(
"org.freedesktop.DBus", "/org/freedesktop/DBus", "org.freedesktop.DBus", "ListNames"
);
if (!listNamesResult)
return Err(listNamesResult.error());
dbus::MessageGuard& listNames = *listNamesResult;
Result<dbus::MessageGuard, DraconisError> listNamesReplyResult =
dbus::ConnectionSendWithReplyAndBlock(connection, listNames, 100);
if (!listNamesReplyResult)
return Err(listNamesReplyResult.error());
dbus::MessageGuard& listNamesReply = *listNamesReplyResult;
dbus::MessageIter iter;
if (dbus::MessageIterInit(listNamesReply, &iter) && dbus::MessageIterGetArgType(&iter) == DBUS_TYPE_ARRAY) {
dbus::MessageIter subIter;
dbus::MessageIterRecurse(&iter, &subIter);
while (dbus::MessageIterGetArgType(&subIter) != DBUS_TYPE_INVALID) {
if (Option<String> name = dbus::MessageIterGetString(&subIter))
if (name->find("org.mpris.MediaPlayer2") != String::npos) {
activePlayer = std::move(*name);
break;
}
dbus::MessageIterNext(&subIter);
}
} else {
return Err(DraconisError(DraconisErrorCode::ParseError, "Invalid DBus ListNames reply format"));
} }
Vec<String> mprisPlayers = GetMprisPlayers(connection);
if (mprisPlayers.empty()) {
dbus_connection_unref(connection);
return Err(DraconisError(DraconisErrorCode::NotFound, "No MPRIS players found"));
} }
Option<String> activePlayer = GetActivePlayer(mprisPlayers); if (!activePlayer)
return Err(DraconisError(DraconisErrorCode::NotFound, "No active MPRIS players found"));
if (!activePlayer.has_value()) { Result<dbus::MessageGuard, DraconisError> msgResult = dbus::MessageNewMethodCall(
dbus_connection_unref(connection); activePlayer->c_str(), "/org/mpris/MediaPlayer2", "org.freedesktop.DBus.Properties", "Get"
return Err(DraconisError(DraconisErrorCode::NotFound, "No active MPRIS player found"));
}
// Prepare a call to the Properties.Get method to fetch "Metadata"
DBusMessage* msg = dbus_message_new_method_call(
activePlayer->c_str(), // target service (active player)
"/org/mpris/MediaPlayer2", // object path
"org.freedesktop.DBus.Properties", // interface
"Get" // method name
); );
if (!msg) { if (!msgResult)
dbus_connection_unref(connection); return Err(msgResult.error());
return Err(DraconisError(DraconisErrorCode::InternalError, "Failed to create DBus message"));
}
const char* interfaceName = "org.mpris.MediaPlayer2.Player"; dbus::MessageGuard& msg = *msgResult;
const char* propertyName = "Metadata";
if (!dbus_message_append_args( if (!dbus::MessageAppendArgs(
msg, DBUS_TYPE_STRING, &interfaceName, DBUS_TYPE_STRING, &propertyName, DBUS_TYPE_INVALID msg, DBUS_TYPE_STRING, "org.mpris.MediaPlayer2.Player", DBUS_TYPE_STRING, "Metadata", DBUS_TYPE_INVALID
)) { ))
dbus_message_unref(msg);
dbus_connection_unref(connection);
return Err(DraconisError(DraconisErrorCode::InternalError, "Failed to append arguments to DBus message")); return Err(DraconisError(DraconisErrorCode::InternalError, "Failed to append arguments to DBus message"));
}
// Call the method and block until reply is received. Result<dbus::MessageGuard, DraconisError> replyResult =
DBusMessage* reply = dbus_connection_send_with_reply_and_block(connection, msg, -1, &err); dbus::ConnectionSendWithReplyAndBlock(connection, msg, 100);
dbus_message_unref(msg);
if (dbus_error_is_set(&err)) { if (!replyResult)
error_log("DBus error in Properties.Get: {}", err.message); return Err(replyResult.error());
DraconisError error = DraconisError(DraconisErrorCode::ApiUnavailable, err.message); dbus::MessageGuard& reply = *replyResult;
dbus_error_free(&err);
dbus_connection_unref(connection);
return Err(error); Option<String> title = None;
} Option<String> artist = None;
if (!reply) { dbus::MessageIter propIter;
dbus_connection_unref(connection); if (!dbus::MessageIterInit(reply, &propIter))
return Err(DraconisError(DraconisErrorCode::ApiUnavailable, "No reply received for Properties.Get")); return Err(DraconisError(DraconisErrorCode::ParseError, "Properties.Get reply has no arguments"));
}
// The reply should contain a variant holding a dictionary ("a{sv}") if (dbus::MessageIterGetArgType(&propIter) != DBUS_TYPE_VARIANT)
DBusMessageIter iter; return Err(DraconisError(DraconisErrorCode::ParseError, "Properties.Get reply argument is not a variant"));
if (!dbus_message_iter_init(reply, &iter)) { dbus::MessageIter variantIter;
dbus_message_unref(reply); dbus::MessageIterRecurse(&propIter, &variantIter);
dbus_connection_unref(connection);
return Err(DraconisError(DraconisErrorCode::InternalError, "Reply has no arguments"));
}
if (dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_VARIANT) { if (dbus::MessageIterGetArgType(&variantIter) != DBUS_TYPE_ARRAY ||
dbus_message_unref(reply); dbus_message_iter_get_element_type(&variantIter) != DBUS_TYPE_DICT_ENTRY)
dbus_connection_unref(connection); return Err(
return Err(DraconisError(DraconisErrorCode::InternalError, "Reply argument is not a variant")); DraconisError(DraconisErrorCode::ParseError, "Metadata variant content is not a dictionary array (a{sv})")
} );
// Recurse into the variant to get the dictionary dbus::MessageIter dictIter;
DBusMessageIter variantIter; dbus::MessageIterRecurse(&variantIter, &dictIter);
dbus_message_iter_recurse(&iter, &variantIter);
if (dbus_message_iter_get_arg_type(&variantIter) != DBUS_TYPE_ARRAY) { while (dbus::MessageIterGetArgType(&dictIter) == DBUS_TYPE_DICT_ENTRY) {
dbus_message_unref(reply); dbus::MessageIter entryIter;
dbus_connection_unref(connection); dbus::MessageIterRecurse(&dictIter, &entryIter);
return Err(DraconisError(DraconisErrorCode::InternalError, "Variant argument is not an array"));
}
String title; Option<String> key = dbus::MessageIterGetString(&entryIter);
String artist;
DBusMessageIter arrayIter;
dbus_message_iter_recurse(&variantIter, &arrayIter);
// Iterate over each dictionary entry (each entry is of type dict entry) if (!key) {
while (dbus_message_iter_get_arg_type(&arrayIter) != DBUS_TYPE_INVALID) { dbus::MessageIterNext(&dictIter);
if (dbus_message_iter_get_arg_type(&arrayIter) == DBUS_TYPE_DICT_ENTRY) { continue;
DBusMessageIter dictEntry;
dbus_message_iter_recurse(&arrayIter, &dictEntry);
// Get the key (a string)
const char* key = nullptr;
if (dbus_message_iter_get_arg_type(&dictEntry) == DBUS_TYPE_STRING)
dbus_message_iter_get_basic(&dictEntry, static_cast<void*>(&key));
// Move to the value (a variant)
dbus_message_iter_next(&dictEntry);
if (dbus_message_iter_get_arg_type(&dictEntry) == DBUS_TYPE_VARIANT) {
DBusMessageIter valueIter;
dbus_message_iter_recurse(&dictEntry, &valueIter);
if (key && std::string_view(key) == "xesam:title") {
if (dbus_message_iter_get_arg_type(&valueIter) == DBUS_TYPE_STRING) {
const char* val = nullptr;
dbus_message_iter_get_basic(&valueIter, static_cast<void*>(&val));
if (val)
title = val;
}
} else if (key && std::string_view(key) == "xesam:artist") {
// Expect an array of strings
if (dbus_message_iter_get_arg_type(&valueIter) == DBUS_TYPE_ARRAY) {
DBusMessageIter subIter;
dbus_message_iter_recurse(&valueIter, &subIter);
if (dbus_message_iter_get_arg_type(&subIter) == DBUS_TYPE_STRING) {
const char* val = nullptr;
dbus_message_iter_get_basic(&subIter, static_cast<void*>(&val));
if (val)
artist = val;
}
}
}
}
} }
dbus_message_iter_next(&arrayIter); if (!dbus::MessageIterNext(&entryIter) || dbus::MessageIterGetArgType(&entryIter) != DBUS_TYPE_VARIANT) {
dbus::MessageIterNext(&dictIter);
continue;
}
dbus::MessageIter valueVariantIter;
dbus::MessageIterRecurse(&entryIter, &valueVariantIter);
if (*key == "xesam:title")
title = dbus::MessageIterGetString(&valueVariantIter);
else if (*key == "xesam:artist")
if (dbus::MessageIterGetArgType(&valueVariantIter) == DBUS_TYPE_ARRAY &&
dbus_message_iter_get_element_type(&valueVariantIter) == DBUS_TYPE_STRING) {
dbus::MessageIter artistArrayIter;
dbus::MessageIterRecurse(&valueVariantIter, &artistArrayIter);
artist = dbus::MessageIterGetString(&artistArrayIter);
}
dbus::MessageIterNext(&dictIter);
} }
dbus_message_unref(reply); return MediaInfo(std::move(title), std::move(artist));
dbus_connection_unref(connection);
return MediaInfo(artist, title);
} }
#pragma clang diagnostic pop
fn GetWindowManager() -> Option<String> { fn GetWindowManager() -> Option<String> {
if (Result<String, DraconisError> waylandResult = GetWaylandCompositor()) if (Result<String, DraconisError> waylandResult = GetWaylandCompositor())

View file

@ -1,3 +1,6 @@
#ifdef __linux__
// clang-format off
#include "src/os/linux/pkg_count.hpp" #include "src/os/linux/pkg_count.hpp"
#include <SQLiteCpp/SQLiteCpp.h> #include <SQLiteCpp/SQLiteCpp.h>
@ -9,6 +12,7 @@
#include "src/core/util/logging.hpp" #include "src/core/util/logging.hpp"
#include "src/core/util/types.hpp" #include "src/core/util/types.hpp"
// clang-format on
using util::error::DraconisError, util::error::DraconisErrorCode; using util::error::DraconisError, util::error::DraconisErrorCode;
using util::types::u64, util::types::i64, util::types::Result, util::types::Err, util::types::String, using util::types::u64, util::types::i64, util::types::Result, util::types::Err, util::types::String,
@ -199,10 +203,9 @@ namespace {
cacheTimePoint.time_since_epoch() >= dbModTime.time_since_epoch()) { cacheTimePoint.time_since_epoch() >= dbModTime.time_since_epoch()) {
// Use cacheTimePoint for logging as well // Use cacheTimePoint for logging as well
debug_log( debug_log(
"Using valid {} package count cache (DB file unchanged since {}). Count: {}", "Using valid {} package count cache (DB file unchanged since {}).",
pmId, pmId,
std::format("{:%F %T %Z}", floor<seconds>(cacheTimePoint)), // Format the time_point std::format("{:%F %T %Z}", floor<seconds>(cacheTimePoint))
count
); );
return count; return count;
} }
@ -337,4 +340,6 @@ namespace os::linux {
return totalCount; return totalCount;
} }
} // namespace os::linux } // namespace os::linux
#endif // __linux__

View file

@ -1,10 +1,14 @@
#pragma once #pragma once
#ifdef __linux__
// clang-format off
#include <filesystem> #include <filesystem>
#include "src/core/util/defs.hpp" #include "src/core/util/defs.hpp"
#include "src/core/util/error.hpp" #include "src/core/util/error.hpp"
#include "src/core/util/types.hpp" #include "src/core/util/types.hpp"
// clang-format on
namespace os::linux { namespace os::linux {
using util::error::DraconisError; using util::error::DraconisError;
@ -51,4 +55,6 @@ namespace os::linux {
// Get total package count from all available package managers // Get total package count from all available package managers
fn GetTotalPackageCount() -> Result<u64, DraconisError>; fn GetTotalPackageCount() -> Result<u64, DraconisError>;
} // namespace os::linux } // namespace os::linux
#endif // __linux__

282
src/wrappers/dbus.hpp Normal file
View file

@ -0,0 +1,282 @@
#pragma once
#ifdef __linux__
// clang-format off
#include <dbus/dbus.h> // DBus Library
#include <utility> // std::exchange
#include <format> // std::format
#include <cstdarg> // va_list, va_start, va_end
#include "src/core/util/defs.hpp"
#include "src/core/util/error.hpp"
#include "src/core/util/types.hpp"
// clang-format on
namespace dbus {
using util::error::DraconisError, util::error::DraconisErrorCode;
using util::types::Option, util::types::Result, util::types::Err, util::types::String, util::types::i32,
util::types::None;
/**
* @brief RAII wrapper for DBusError. Automatically initializes and frees.
*/
class ErrorGuard {
DBusError m_Err {};
bool m_IsInitialized = false;
public:
ErrorGuard() : m_IsInitialized(true) { dbus_error_init(&m_Err); }
~ErrorGuard() {
if (m_IsInitialized)
dbus_error_free(&m_Err);
}
ErrorGuard(const ErrorGuard&) = delete;
fn operator=(const ErrorGuard&)->ErrorGuard& = delete;
ErrorGuard(ErrorGuard&& other) noexcept : m_Err(other.m_Err), m_IsInitialized(other.m_IsInitialized) {
other.m_IsInitialized = false;
dbus_error_init(&other.m_Err);
}
fn operator=(ErrorGuard&& other) noexcept -> ErrorGuard& {
if (this != &other) {
if (m_IsInitialized) {
dbus_error_free(&m_Err);
}
m_Err = other.m_Err;
m_IsInitialized = other.m_IsInitialized;
other.m_IsInitialized = false;
dbus_error_init(&other.m_Err);
}
return *this;
}
[[nodiscard]] fn isSet() const -> bool { return m_IsInitialized && dbus_error_is_set(&m_Err); }
[[nodiscard]] fn message() const -> const char* { return isSet() ? m_Err.message : ""; }
[[nodiscard]] fn name() const -> const char* { return isSet() ? m_Err.name : ""; }
[[nodiscard]] fn get() -> DBusError* { return &m_Err; }
[[nodiscard]] fn get() const -> const DBusError* { return &m_Err; }
[[nodiscard]] fn toDraconisError(const DraconisErrorCode code = DraconisErrorCode::PlatformSpecific) const
-> DraconisError {
if (isSet())
return { code, std::format("D-Bus Error: {} ({})", message(), name()) };
return { DraconisErrorCode::InternalError, "Attempted to convert non-set DBusErrorGuard" };
}
};
/**
* @brief RAII wrapper for DBusConnection. Automatically unrefs.
*/
class ConnectionGuard {
DBusConnection* m_Conn = nullptr;
public:
explicit ConnectionGuard(DBusConnection* conn = nullptr) : m_Conn(conn) {}
~ConnectionGuard() {
if (m_Conn)
dbus_connection_unref(m_Conn);
}
ConnectionGuard(const ConnectionGuard&) = delete;
fn operator=(const ConnectionGuard&)->ConnectionGuard& = delete;
ConnectionGuard(ConnectionGuard&& other) noexcept : m_Conn(std::exchange(other.m_Conn, nullptr)) {}
fn operator=(ConnectionGuard&& other) noexcept -> ConnectionGuard& {
if (this != &other) {
if (m_Conn)
dbus_connection_unref(m_Conn);
m_Conn = std::exchange(other.m_Conn, nullptr);
}
return *this;
}
[[nodiscard]] fn get() const -> DBusConnection* { return m_Conn; }
explicit operator bool() const { return m_Conn != nullptr; }
[[nodiscard]] fn release() -> DBusConnection* { return std::exchange(m_Conn, nullptr); }
};
/**
* @brief RAII wrapper for DBusMessage. Automatically unrefs.
*/
class MessageGuard {
DBusMessage* m_Msg = nullptr;
public:
explicit MessageGuard(DBusMessage* msg = nullptr) : m_Msg(msg) {}
~MessageGuard() {
if (m_Msg)
dbus_message_unref(m_Msg);
}
MessageGuard(const MessageGuard&) = delete;
fn operator=(const MessageGuard&)->MessageGuard& = delete;
MessageGuard(MessageGuard&& other) noexcept : m_Msg(std::exchange(other.m_Msg, nullptr)) {}
fn operator=(MessageGuard&& other) noexcept -> MessageGuard& {
if (this != &other) {
if (m_Msg)
dbus_message_unref(m_Msg);
m_Msg = std::exchange(other.m_Msg, nullptr);
}
return *this;
}
[[nodiscard]] fn get() const -> DBusMessage* { return m_Msg; }
explicit operator bool() const { return m_Msg != nullptr; }
[[nodiscard]] fn release() -> DBusMessage* { return std::exchange(m_Msg, nullptr); }
};
/**
* @brief Connects to a D-Bus bus type.
* @param bus_type The type of bus (e.g., DBUS_BUS_SESSION, DBUS_BUS_SYSTEM).
* @return Result containing a DBusConnectionGuard on success, or DraconisError on failure.
*/
inline fn BusGet(const DBusBusType bus_type) -> Result<ConnectionGuard, DraconisError> {
ErrorGuard err;
DBusConnection* rawConn = dbus_bus_get(bus_type, err.get());
if (err.isSet())
return Err(err.toDraconisError(DraconisErrorCode::ApiUnavailable));
if (!rawConn)
return Err(DraconisError(DraconisErrorCode::ApiUnavailable, "dbus_bus_get returned null without setting error"));
return ConnectionGuard(rawConn);
}
/**
* @brief Creates a new D-Bus method call message.
* @param destination Service name (e.g., "org.freedesktop.Notifications").
* @param path Object path (e.g., "/org/freedesktop/Notifications").
* @param interface Interface name (e.g., "org.freedesktop.Notifications").
* @param method Method name (e.g., "Notify").
* @return Result containing a DBusMessageGuard on success, or DraconisError on failure.
*/
inline fn MessageNewMethodCall(const char* destination, const char* path, const char* interface, const char* method)
-> Result<MessageGuard, DraconisError> {
DBusMessage* rawMsg = dbus_message_new_method_call(destination, path, interface, method);
if (!rawMsg)
return Err(DraconisError(DraconisErrorCode::InternalError, "dbus_message_new_method_call failed (allocation?)"));
return MessageGuard(rawMsg);
}
/**
* @brief Sends a message and waits for a reply.
* @param connection The D-Bus connection guard.
* @param message The D-Bus message guard to send.
* @param timeout_milliseconds Timeout duration.
* @return Result containing the reply DBusMessageGuard on success, or DraconisError on failure.
*/
inline fn ConnectionSendWithReplyAndBlock(
const ConnectionGuard& connection,
const MessageGuard& message,
const i32 timeout_milliseconds = 100
) -> Result<MessageGuard, DraconisError> {
ErrorGuard err;
DBusMessage* rawReply =
dbus_connection_send_with_reply_and_block(connection.get(), message.get(), timeout_milliseconds, err.get());
if (err.isSet())
return Err(err.toDraconisError(DraconisErrorCode::ApiUnavailable));
if (!rawReply)
return Err(DraconisError(
DraconisErrorCode::ApiUnavailable,
"dbus_connection_send_with_reply_and_block returned null without setting error"
));
return MessageGuard(rawReply);
}
/**
* @brief Appends arguments to a D-Bus message using varargs.
* @param message The message guard.
* @param first_arg_type The D-Bus type code of the first argument.
* @param ... Subsequent arguments (type code, value pointer, type code, value pointer...).
* End with DBUS_TYPE_INVALID.
* @return True on success, false on failure (e.g., allocation error).
*/
inline fn MessageAppendArgs(const MessageGuard& message, const int first_arg_type, ...) -> bool {
va_list args;
va_start(args, first_arg_type);
const bool result = dbus_message_append_args_valist(message.get(), first_arg_type, args);
va_end(args);
return result;
}
using MessageIter = DBusMessageIter;
/**
* @brief Initializes a message iterator.
* @param message The message guard.
* @param iter Pointer to the iterator to initialize.
* @return True if iterator is valid, false otherwise.
*/
inline fn MessageIterInit(const MessageGuard& message, MessageIter* iter) -> bool {
return dbus_message_iter_init(message.get(), iter);
}
/**
* @brief Recurses into a container-type argument.
* @param iter The current iterator.
* @param sub_iter Pointer to the sub-iterator to initialize.
*/
inline fn MessageIterRecurse(MessageIter* iter, MessageIter* sub_iter) -> void {
dbus_message_iter_recurse(iter, sub_iter);
}
/**
* @brief Gets the D-Bus type code of the current argument.
* @param iter The iterator.
* @return The type code (e.g., DBUS_TYPE_STRING, DBUS_TYPE_INVALID).
*/
inline fn MessageIterGetArgType(MessageIter* iter) -> int { return dbus_message_iter_get_arg_type(iter); }
/**
* @brief Gets the value of a basic-typed argument.
* @param iter The iterator.
* @param value Pointer to store the retrieved value. Must match the argument type.
*/
inline fn MessageIterGetBasic(MessageIter* iter, void* value) -> void { dbus_message_iter_get_basic(iter, value); }
/**
* @brief Advances the iterator to the next argument.
* @param iter The iterator.
* @return True if successful, false if at the end.
*/
inline fn MessageIterNext(MessageIter* iter) -> bool { return dbus_message_iter_next(iter); }
/**
* @brief Helper to safely get a string argument from an iterator.
* @param iter The iterator positioned at a DBUS_TYPE_STRING argument.
* @return An Option containing the string value, or None if the type is wrong or value is null.
*/
inline fn MessageIterGetString(MessageIter* iter) -> Option<String> {
if (MessageIterGetArgType(iter) == DBUS_TYPE_STRING) {
const char* strPtr = nullptr;
MessageIterGetBasic(iter, static_cast<void*>(&strPtr));
if (strPtr)
return String(strPtr);
}
return None;
}
} // namespace dbus
#endif // __linux__

View file

@ -1,9 +1,13 @@
#pragma once #pragma once
#include <wayland-client.h> #ifdef __linux__
// clang-format off
#include <wayland-client.h> // Wayland client library
#include "src/core/util/defs.hpp" #include "src/core/util/defs.hpp"
#include "src/core/util/types.hpp" #include "src/core/util/types.hpp"
// clang-format on
struct wl_display; struct wl_display;
@ -56,3 +60,5 @@ namespace wl {
[[nodiscard]] fn fd() const -> util::types::i32 { return get_fd(m_Display); } [[nodiscard]] fn fd() const -> util::types::i32 { return get_fd(m_Display); }
}; };
} // namespace wl } // namespace wl
#endif // __linux__

View file

@ -1,9 +1,13 @@
#pragma once #pragma once
#include <xcb/xcb.h> #ifdef __linux__
// clang-format off
#include <xcb/xcb.h> // XCB library
#include "src/core/util/defs.hpp" #include "src/core/util/defs.hpp"
#include "src/core/util/types.hpp" #include "src/core/util/types.hpp"
// clang-format on
namespace xcb { namespace xcb {
using util::types::u8, util::types::i32, util::types::CStr, util::types::None; using util::types::u8, util::types::i32, util::types::CStr, util::types::None;
@ -166,3 +170,5 @@ namespace xcb {
[[nodiscard]] fn operator*() const->T& { return *m_Reply; } [[nodiscard]] fn operator*() const->T& { return *m_Reply; }
}; };
} // namespace xcb } // namespace xcb
#endif // __linux__