From f1acea521bbacb034ae5bb7380e78c51853faa34 Mon Sep 17 00:00:00 2001
From: Lauri Niskanen <ape@ape3000.com>
Date: Sun, 7 Jul 2024 06:47:06 +0300
Subject: [PATCH] pulseaudio: Add 'sink-mapping' config option

This commit adds a new optional config to the pulseaudio modules. It
maps sink names to other sink names so that if the current sink is a
key, the sink named by the value is considered to be the current sink
instead of it.

E.g.
  "sink-mapping": {
    "easyeffects_sink": "speakers_sink"
  }
---
 include/util/audio_backend.hpp    |  4 +++-
 man/waybar-pulseaudio.5.scd       |  4 ++++
 src/modules/pulseaudio.cpp        |  1 +
 src/modules/pulseaudio_slider.cpp |  3 ++-
 src/util/audio_backend.cpp        | 17 +++++++++++++++++
 5 files changed, 27 insertions(+), 2 deletions(-)

diff --git a/include/util/audio_backend.hpp b/include/util/audio_backend.hpp
index 2f53103e5..b9a9e0fff 100644
--- a/include/util/audio_backend.hpp
+++ b/include/util/audio_backend.hpp
@@ -48,6 +48,7 @@ class AudioBackend {
   std::string default_source_name_;
 
   std::vector<std::string> ignored_sinks_;
+  std::map<std::string, std::string> sink_mapping_;
 
   std::function<void()> on_updated_cb_ = NOOP;
 
@@ -69,6 +70,7 @@ class AudioBackend {
   void changeVolume(ChangeType change_type, double step = 1, uint16_t max_volume = 100);
 
   void setIgnoredSinks(const Json::Value& config);
+  void setSinkMapping(const Json::Value& config);
 
   std::string getSinkPortName() const { return port_name_; }
   std::string getFormFactor() const { return form_factor_; }
@@ -93,4 +95,4 @@ class AudioBackend {
   bool isBluetooth();
 };
 
-}  // namespace waybar::util
\ No newline at end of file
+}  // namespace waybar::util
diff --git a/man/waybar-pulseaudio.5.scd b/man/waybar-pulseaudio.5.scd
index 232e84a00..4f585d4ec 100644
--- a/man/waybar-pulseaudio.5.scd
+++ b/man/waybar-pulseaudio.5.scd
@@ -113,6 +113,10 @@ Additionally, you can control the volume by scrolling *up* or *down* while the c
 	typeof: array ++
 	Sinks in this list will not be shown as active sink by Waybar. Entries should be the sink's description field.
 
+*sink-mapping*: ++
+	typeof: object ++
+	Sinks named by the values of this mapping will be considered to be the current sink instead of the sinks named by the respective keys.
+
 *menu*: ++
 	typeof: string ++
 	Action that popups the menu.
diff --git a/src/modules/pulseaudio.cpp b/src/modules/pulseaudio.cpp
index 255ca571f..0189daaa4 100644
--- a/src/modules/pulseaudio.cpp
+++ b/src/modules/pulseaudio.cpp
@@ -7,6 +7,7 @@ waybar::modules::Pulseaudio::Pulseaudio(const std::string &id, const Json::Value
 
   backend = util::AudioBackend::getInstance([this] { this->dp.emit(); });
   backend->setIgnoredSinks(config_["ignored-sinks"]);
+  backend->setSinkMapping(config_["sink-mapping"]);
 }
 
 bool waybar::modules::Pulseaudio::handleScroll(GdkEventScroll *e) {
diff --git a/src/modules/pulseaudio_slider.cpp b/src/modules/pulseaudio_slider.cpp
index bf85584ed..7e642d291 100644
--- a/src/modules/pulseaudio_slider.cpp
+++ b/src/modules/pulseaudio_slider.cpp
@@ -6,6 +6,7 @@ PulseaudioSlider::PulseaudioSlider(const std::string& id, const Json::Value& con
     : ASlider(config, "pulseaudio-slider", id) {
   backend = util::AudioBackend::getInstance([this] { this->dp.emit(); });
   backend->setIgnoredSinks(config_["ignored-sinks"]);
+  backend->setSinkMapping(config_["sink-mapping"]);
 
   if (config_["target"].isString()) {
     std::string target = config_["target"].asString();
@@ -79,4 +80,4 @@ void PulseaudioSlider::onValueChanged() {
   backend->changeVolume(volume, min_, max_);
 }
 
-}  // namespace waybar::modules
\ No newline at end of file
+}  // namespace waybar::modules
diff --git a/src/util/audio_backend.cpp b/src/util/audio_backend.cpp
index 3d90b6d5a..c267816f9 100644
--- a/src/util/audio_backend.cpp
+++ b/src/util/audio_backend.cpp
@@ -155,6 +155,13 @@ void AudioBackend::sinkInfoCb(pa_context * /*context*/, const pa_sink_info *i, i
     }
   }
 
+  if (const auto mapping = backend->sink_mapping_.find(backend->current_sink_name_);
+      mapping != backend->sink_mapping_.end()) {
+    if (i->name == mapping->second) {
+      backend->current_sink_name_ = i->name;
+    }
+  }
+
   if (backend->current_sink_name_ == i->name) {
     backend->current_sink_running_ = i->state == PA_SINK_RUNNING;
   }
@@ -292,4 +299,14 @@ void AudioBackend::setIgnoredSinks(const Json::Value &config) {
   }
 }
 
+void AudioBackend::setSinkMapping(const Json::Value &config) {
+  if (config.isObject()) {
+    for (auto it = config.begin(); it != config.end(); ++it) {
+      if (it.key().isString() && it->isString()) {
+        sink_mapping_.emplace(it.key().asString(), it->asString());
+      }
+    }
+  }
+}
+
 }  // namespace waybar::util