polybar/src/drawtypes/iconset.cpp

52 lines
1.2 KiB
C++

#include "drawtypes/iconset.hpp"
#include <algorithm>
POLYBAR_NS
namespace drawtypes {
void iconset::add(string id, label_t&& icon) {
m_icons.emplace(id, forward<decltype(icon)>(icon));
}
bool iconset::has(const string& id) {
return m_icons.find(id) != m_icons.end();
}
label_t iconset::get(const string& id, const string& fallback_id, bool fuzzy_match) {
// Try to match exactly first
auto icon = m_icons.find(id);
if (icon != m_icons.end()) {
return icon->second;
}
// If fuzzy matching is turned on, try that first before returning the fallback.
if (fuzzy_match) {
// works by finding the *longest* matching icon to the given workspace id
size_t max_size = -1;
label_t max_label;
for (auto const& icon : m_icons) {
if (id.find(icon.first) != std::string::npos) {
if (icon.first.length() > max_size || !max_label) {
max_size = icon.first.length();
max_label = icon.second;
}
}
}
if (max_label) {
return max_label;
}
}
return m_icons.find(fallback_id)->second;
}
iconset::operator bool() {
return !m_icons.empty();
}
} // namespace drawtypes
POLYBAR_NS_END