From b16c8972c70d536df0e579beee8c9054f0990d6b Mon Sep 17 00:00:00 2001
From: Matthias Richter <vrld@vrld.org>
Date: Sun, 21 Mar 2021 11:42:25 +0100
Subject: [PATCH] Add option to rewrite sway/window title
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Rewrites window title according to config option "rewrite".
"rewrite" is an object where keys are regular expressions and values are
rewrite rules if the expression matches. Rules may contain references to
captures of the expression. Regex and replacement follow ECMA-script
rules. If no regex matches, the title is left unchanged.

example:
"sway/window": {
  "rewrite": {
    "(.*) - Mozilla Firefox": " $1",
    "(.*) - zsh": " $1",
  }
}
---
 include/modules/sway/window.hpp |  1 +
 src/modules/sway/window.cpp     | 22 +++++++++++++++++++++-
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/include/modules/sway/window.hpp b/include/modules/sway/window.hpp
index 40aaa1a0..0f7ae317 100644
--- a/include/modules/sway/window.hpp
+++ b/include/modules/sway/window.hpp
@@ -22,6 +22,7 @@ class Window : public ALabel, public sigc::trackable {
   std::tuple<std::size_t, int, std::string, std::string> getFocusedNode(const Json::Value& nodes,
                                                                         std::string&       output);
   void                                                   getTree();
+  std::string                                            rewriteTitle(const std::string& title);
 
   const Bar&       bar_;
   std::string      window_;
diff --git a/src/modules/sway/window.cpp b/src/modules/sway/window.cpp
index d8f113f7..3b424c2f 100644
--- a/src/modules/sway/window.cpp
+++ b/src/modules/sway/window.cpp
@@ -1,5 +1,6 @@
 #include "modules/sway/window.hpp"
 #include <spdlog/spdlog.h>
+#include <regex>
 
 namespace waybar::modules::sway {
 
@@ -56,7 +57,7 @@ auto Window::update() -> void {
     bar_.window.get_style_context()->remove_class("solo");
     bar_.window.get_style_context()->remove_class("empty");
   }
-  label_.set_markup(fmt::format(format_, fmt::arg("title", window_),
+  label_.set_markup(fmt::format(format_, fmt::arg("title", rewriteTitle(window_)),
                                 fmt::arg("app_id", app_id_)));
   if (tooltipEnabled()) {
     label_.set_tooltip_text(window_);
@@ -131,4 +132,23 @@ void Window::getTree() {
   }
 }
 
+std::string Window::rewriteTitle(const std::string& title)
+{
+  const auto& rules = config_["rewrite"];
+  if (!rules.isObject()) {
+    return title;
+  }
+
+  for (auto it = rules.begin(); it != rules.end(); ++it) {
+    if (it.key().isString() && it->isString()) {
+      const std::regex rule{it.key().asString()};
+      if (std::regex_match(title, rule)) {
+        return std::regex_replace(title, rule, it->asString());
+      }
+    }
+  }
+
+  return title;
+}
+
 }  // namespace waybar::modules::sway