feat(custom): escape option

Alexays · Mar 1, 2019 · 79b0367 · 79b0367
1 parent 2c411b9
commit 79b0367
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/src/modules/custom.cpp b/src/modules/custom.cpp
@@ -124,7 +124,11 @@ void waybar::modules::Custom::parseOutputRaw()
   int i = 0;
   while (getline(output, line)) {
     if (i == 0) {
-      text_ = line;
+      if (config_["escape"].isBool() && config_["escape"].asBool()) {
+        text_ = Glib::Markup::escape_text(line);
+      } else {
+        text_ = line;
+      }
       tooltip_ = line;
       class_ = "";
     } else if (i == 1) {
@@ -144,8 +148,16 @@ void waybar::modules::Custom::parseOutputJson()
   std::string line;
   while (getline(output, line)) {
     auto parsed = parser_.parse(line);
-    text_ = parsed["text"].asString();
-    alt_ = parsed["alt"].asString();
+    if (config_["escape"].isBool() && config_["escape"].asBool()) {
+      text_ = Glib::Markup::escape_text(parsed["text"].asString());
+    } else {
+      text_ = parsed["text"].asString();
+    }
+    if (config_["escape"].isBool() && config_["escape"].asBool()) {
+      alt_ = Glib::Markup::escape_text(parsed["alt"].asString());
+    } else {
+      alt_ = parsed["alt"].asString();
+    }
     tooltip_ = parsed["tooltip"].asString();
     class_ = parsed["class"].asString();
     if (!parsed["percentage"].asString().empty() && parsed["percentage"].isUInt()) {