mirror of
https://github.com/polybar/polybar.git
synced 2026-03-05 15:59:37 +00:00
98 lines
3.4 KiB
C++
98 lines
3.4 KiB
C++
#include "lemonbuddy.hpp"
|
|
#include "registry.hpp"
|
|
#include "modules/base.hpp"
|
|
#include "utils/config.hpp"
|
|
#include "utils/string.hpp"
|
|
|
|
std::string ModuleFormatter::Format::decorate(Builder *builder, std::string output)
|
|
{
|
|
if (this->offset != 0) builder->offset(this->offset);
|
|
|
|
if (this->margin > 0) builder->space(this->margin);
|
|
|
|
if (!this->bg.empty()) builder->background(this->bg);
|
|
if (!this->fg.empty()) builder->color(this->fg);
|
|
if (!this->ul.empty()) builder->underline(this->ul);
|
|
if (!this->ol.empty()) builder->overline(this->ol);
|
|
|
|
if (this->padding > 0) builder->space(this->padding);
|
|
|
|
builder->append(output);
|
|
|
|
if (this->padding > 0) builder->space(this->padding);
|
|
|
|
if (!this->ol.empty()) builder->overline_close();
|
|
if (!this->ul.empty()) builder->underline_close();
|
|
if (!this->fg.empty()) builder->color_close();
|
|
if (!this->bg.empty()) builder->background_close();
|
|
|
|
if (this->margin > 0) builder->space(this->margin);
|
|
|
|
return builder->flush();
|
|
}
|
|
|
|
void ModuleFormatter::add(std::string name, std::string fallback, std::vector<std::string>&& tags, std::vector<std::string>&& whitelist)
|
|
{
|
|
auto format = std::make_unique<Format>();
|
|
|
|
format->value = config::get<std::string>(this->module_name, name, fallback);
|
|
format->fg = config::get<std::string>(this->module_name, name +"-foreground", "");
|
|
format->bg = config::get<std::string>(this->module_name, name +"-background", "");
|
|
format->ul = config::get<std::string>(this->module_name, name +"-underline", "");
|
|
format->ol = config::get<std::string>(this->module_name, name +"-overline", "");
|
|
format->spacing = config::get<int>(this->module_name, name +"-spacing", DEFAULT_SPACING);
|
|
format->padding = config::get<int>(this->module_name, name +"-padding", 0);
|
|
format->margin = config::get<int>(this->module_name, name +"-margin", 0);
|
|
format->offset = config::get<int>(this->module_name, name +"-offset", 0);
|
|
format->tags.swap(tags);
|
|
|
|
for (auto &&tag : string::split(format->value, ' ')) {
|
|
if (tag[0] != '<' || tag[tag.length()-1] != '>')
|
|
continue;
|
|
if (std::find(format->tags.begin(), format->tags.end(), tag) != format->tags.end())
|
|
continue;
|
|
if (std::find(whitelist.begin(), whitelist.end(), tag) != whitelist.end())
|
|
continue;
|
|
throw UndefinedFormatTag("["+ this->module_name +"] Undefined \""+ name +"\" tag: "+ tag);
|
|
}
|
|
|
|
this->formats.insert(std::make_pair(name, std::move(format)));
|
|
}
|
|
|
|
std::unique_ptr<ModuleFormatter::Format>& ModuleFormatter::get(std::string format_name)
|
|
{
|
|
auto format = this->formats.find(format_name);
|
|
if (format == this->formats.end())
|
|
throw UndefinedFormat("Format \""+ format_name +"\" has not been added");
|
|
return format->second;
|
|
}
|
|
|
|
bool ModuleFormatter::has(std::string tag, std::string format_name)
|
|
{
|
|
auto format = this->formats.find(format_name);
|
|
if (format == this->formats.end())
|
|
throw UndefinedFormat(format_name);
|
|
return format->second->value.find(tag) != std::string::npos;
|
|
}
|
|
|
|
bool ModuleFormatter::has(std::string tag)
|
|
{
|
|
for (auto &&format : this->formats)
|
|
if (format.second->value.find(tag) != std::string::npos)
|
|
return true;
|
|
return false;
|
|
}
|
|
|
|
namespace modules
|
|
{
|
|
void broadcast_module_update(std::string module_name)
|
|
{
|
|
log_trace("Broadcasting module update for => "+ module_name);
|
|
get_registry()->notify(module_name);
|
|
}
|
|
|
|
std::string get_tag_name(std::string tag) {
|
|
return tag.length() < 2 ? "" : tag.substr(1, tag.length()-2);
|
|
}
|
|
}
|