polybar/src/drawtypes/label.cpp

244 lines
7.1 KiB
C++

#include <utility>
#include "drawtypes/label.hpp"
POLYBAR_NS
namespace drawtypes {
string label::get() const {
return m_tokenized;
}
label::operator bool() {
return !m_text.empty();
}
label_t label::clone() {
vector<token> tokens;
if (!m_tokens.empty()) {
std::back_insert_iterator<decltype(tokens)> back_it(tokens);
std::copy(m_tokens.begin(), m_tokens.end(), back_it);
}
return factory_util::shared<label>(m_text, m_foreground, m_background, m_underline, m_overline, m_font, m_padding, m_margin,
m_maxlen, m_ellipsis, move(tokens));
}
void label::reset_tokens() {
m_tokenized = m_text;
}
bool label::has_token(const string& token) {
return m_text.find(token) != string::npos;
}
void label::replace_token(const string& token, string replacement) {
if (!has_token(token)) {
return;
}
for (auto&& tok : m_tokens) {
if (token == tok.token) {
if (tok.max != 0 && replacement.length() > tok.max) {
replacement = replacement.erase(tok.max) + tok.suffix;
} else if (tok.min != 0 && replacement.length() < tok.min) {
replacement.insert(0, tok.min - replacement.length(), ' ');
}
m_tokenized = string_util::replace_all(m_tokenized, token, move(replacement));
}
}
}
void label::replace_defined_values(const label_t& label) {
if (!label->m_foreground.empty()) {
m_foreground = label->m_foreground;
}
if (!label->m_background.empty()) {
m_background = label->m_background;
}
if (!label->m_underline.empty()) {
m_underline = label->m_underline;
}
if (!label->m_overline.empty()) {
m_overline = label->m_overline;
}
if (label->m_font != 0) {
m_font = label->m_font;
}
if (label->m_padding.left != 0) {
m_padding.left = label->m_padding.left;
}
if (label->m_padding.right != 0) {
m_padding.right = label->m_padding.right;
}
if (label->m_margin.left != 0) {
m_margin.left = label->m_margin.left;
}
if (label->m_margin.right != 0) {
m_margin.right = label->m_margin.right;
}
if (label->m_maxlen != 0) {
m_maxlen = label->m_maxlen;
m_ellipsis = label->m_ellipsis;
}
}
void label::copy_undefined(const label_t& label) {
if (m_foreground.empty() && !label->m_foreground.empty()) {
m_foreground = label->m_foreground;
}
if (m_background.empty() && !label->m_background.empty()) {
m_background = label->m_background;
}
if (m_underline.empty() && !label->m_underline.empty()) {
m_underline = label->m_underline;
}
if (m_overline.empty() && !label->m_overline.empty()) {
m_overline = label->m_overline;
}
if (m_font == 0 && label->m_font != 0) {
m_font = label->m_font;
}
if (m_padding.left == 0 && label->m_padding.left != 0) {
m_padding.left = label->m_padding.left;
}
if (m_padding.right == 0 && label->m_padding.right != 0) {
m_padding.right = label->m_padding.right;
}
if (m_margin.left == 0 && label->m_margin.left != 0) {
m_margin.left = label->m_margin.left;
}
if (m_margin.right == 0 && label->m_margin.right != 0) {
m_margin.right = label->m_margin.right;
}
if (m_maxlen == 0 && label->m_maxlen != 0) {
m_maxlen = label->m_maxlen;
m_ellipsis = label->m_ellipsis;
}
}
/**
* Create a label by loading values from the configuration
*/
label_t load_label(const config& conf, const string& section, string name, bool required, string def) {
vector<token> tokens;
size_t start, end, pos;
name = string_util::ltrim(string_util::rtrim(name, '>'), '<');
string text;
struct side_values padding, margin;
if (required) {
text = conf.get<string>(section, name);
} else {
text = conf.get<string>(section, name, move(def));
}
size_t len{text.size()};
if (len > 2 && text[0] == '"' && text[len - 1] == '"') {
text = text.substr(1, len - 2);
}
const auto get_left_right = [&](string key) {
auto value = conf.get<int>(section, key, 0);
auto left = conf.get<int>(section, key + "-left", value);
auto right = conf.get<int>(section, key + "-right", value);
return side_values{static_cast<uint16_t>(left), static_cast<uint16_t>(right)};
};
padding = get_left_right(name + "-padding");
margin = get_left_right(name + "-margin");
string line{text};
while ((start = line.find('%')) != string::npos && (end = line.find('%', start + 1)) != string::npos) {
auto token_str = line.substr(start, end - start + 1);
// ignore false positives (lemonbar-style declarations)
if (token_str[1] == '{') {
line.erase(0, start + 1);
continue;
}
line.erase(start, end - start + 1);
tokens.emplace_back(token{token_str, 0, 0});
auto& token = tokens.back();
// find min delimiter
if ((pos = token_str.find(':')) == string::npos) {
continue;
}
// strip min/max specifiers from the label string token
token.token = token_str.substr(0, pos) + '%';
text = string_util::replace(text, token_str, token.token);
try {
token.min = std::stoul(&token_str[pos + 1], nullptr, 10);
} catch (const std::invalid_argument& err) {
continue;
}
// find max delimiter
if ((pos = token_str.find(':', pos + 1)) == string::npos) {
continue;
}
try {
token.max = std::stoul(&token_str[pos + 1], nullptr, 10);
} catch (const std::invalid_argument& err) {
continue;
}
// ignore max lengths less than min
if (token.max < token.min) {
token.max = 0;
}
// find suffix delimiter
if ((pos = token_str.find(':', pos + 1)) != string::npos) {
token.suffix = token_str.substr(pos + 1, end - pos - 1);
}
}
// clang-format off
return make_shared<label>(text,
conf.get<string>(section, name + "-foreground", ""),
conf.get<string>(section, name + "-background", ""),
conf.get<string>(section, name + "-underline", ""),
conf.get<string>(section, name + "-overline", ""),
conf.get<int>(section, name + "-font", 0),
padding,
margin,
conf.get<size_t>(section, name + "-maxlen", 0),
conf.get<bool>(section, name + "-ellipsis", true),
move(tokens));
// clang-format on
}
/**
* Create a label by loading optional values from the configuration
*/
label_t load_optional_label(const config& conf, string section, string name, string def) {
return load_label(conf, move(section), move(name), false, move(def));
}
/**
* Create an icon by loading values from the configuration
*/
icon_t load_icon(const config& conf, string section, string name, bool required, string def) {
return load_label(conf, move(section), move(name), required, move(def));
}
/**
* Create an icon by loading optional values from the configuration
*/
icon_t load_optional_icon(const config& conf, string section, string name, string def) {
return load_icon(conf, move(section), move(name), false, move(def));
}
}
POLYBAR_NS_END