polybar/src/x11/xresources.cpp

#include <X11/Xlib.h>
#include <cstring>
#include <utility>

#include "utils/factory.hpp"
#include "x11/connection.hpp"
#include "x11/xresources.hpp"

POLYBAR_NS

/**
 * Create instance
 */
xresource_manager::make_type xresource_manager::make() {
  return *factory_util::singleton<xresource_manager>(static_cast<Display*>(connection::make()));
}

/**
 * Construct manager instance
 */
xresource_manager::xresource_manager(Display* dsp) {
  XrmInitialize();

  if ((m_manager = XResourceManagerString(dsp)) != nullptr) {
    m_db = XrmGetStringDatabase(m_manager);
  }
}

/**
 * Deconstruct instance
 */
xresource_manager::~xresource_manager() {
  if (m_manager != nullptr) {
    XrmDestroyDatabase(m_db);
  }
}

/**
 * Get string value from the X resource db
 */
string xresource_manager::get_string(string name, string fallback) const {
  auto result = load_value(move(name), "String", 64);
  if (result.empty()) {
    return fallback;
  }
  return result;
}

/**
 * Get float value from the X resource db
 */
float xresource_manager::get_float(string name, float fallback) const {
  auto result = load_value(move(name), "String", 64);
  if (result.empty()) {
    return fallback;
  }
  return strtof(result.c_str(), nullptr);
}

/**
 * Get integer value from the X resource db
 */
int xresource_manager::get_int(string name, int fallback) const {
  auto result = load_value(move(name), "String", 64);
  if (result.empty()) {
    return fallback;
  }
  return atoi(result.c_str());
}

/**
 * Query the database for given key
 */
string xresource_manager::load_value(const string& key, const string& res_type, size_t n) const {
  if (m_manager != nullptr) {
    char* type = nullptr;
    XrmValue ret{};

    if (XrmGetResource(m_db, key.c_str(), res_type.c_str(), &type, &ret)) {
      if (ret.addr != nullptr && !std::strncmp(res_type.c_str(), type, n)) {
        return {ret.addr};
      }
    }
  }
  return "";
}

POLYBAR_NS_END