From c9bc79c956467155f8446e158588c75d202f350d Mon Sep 17 00:00:00 2001
From: scorpion-26 <dev.scorpion26@gmail.com>
Date: Tue, 26 Mar 2024 22:09:38 +0100
Subject: [PATCH] Toggle audio sink/source on icon click

A click on the mic/speaker icon now mutes/unmutes the sink/source.

Implements https://github.com/scorpion-26/gBar/issues/81
---
 src/Bar.cpp      | 22 ++++++++++++++++++----
 src/PulseAudio.h | 28 ++++++++++++++++++++++++++++
 src/System.cpp   |  9 +++++++++
 src/System.h     |  2 ++
 4 files changed, 57 insertions(+), 4 deletions(-)

diff --git a/src/Bar.cpp b/src/Bar.cpp
index 3c06d9d..3283d6c 100644
--- a/src/Bar.cpp
+++ b/src/Bar.cpp
@@ -286,8 +286,8 @@ namespace Bar
 
         Widget* audioSlider;
         Widget* micSlider;
-        Text* audioIcon;
-        Text* micIcon;
+        Button* audioIcon;
+        Button* micIcon;
         void OnChangeVolumeSink(Slider&, double value)
         {
             System::SetVolumeSink(value);
@@ -315,6 +315,18 @@ namespace Bar
             System::SetVolumeSource(micVolume);
         }
 
+        void OnToggleSink(Button& button)
+        {
+            System::AudioInfo info = System::GetAudioInfo();
+            System::SetMutedSink(!info.sinkMuted);
+        }
+
+        void OnToggleSource(Button& button)
+        {
+            System::AudioInfo info = System::GetAudioInfo();
+            System::SetMutedSource(!info.sourceMuted);
+        }
+
         TimerResult UpdateAudio(Widget&)
         {
             System::AudioInfo info = System::GetAudioInfo();
@@ -344,7 +356,7 @@ namespace Bar
                 }
                 else
                 {
-                    ((Slider*)micSlider)->SetValue(info.sinkVolume);
+                    ((Slider*)micSlider)->SetValue(info.sourceVolume);
                 }
                 if (info.sourceMuted)
                 {
@@ -605,18 +617,20 @@ namespace Bar
             Utils::SetTransform(*box, {-1, false, SideToAlignment(side)});
             box->SetOrientation(Utils::GetOrientation());
             {
-                auto icon = Widget::Create<Text>();
+                auto icon = Widget::Create<Button>();
                 icon->SetAngle(Utils::GetAngle());
                 switch (type)
                 {
                 case AudioType::Input:
                     icon->SetClass("mic-icon");
                     icon->SetText(Config::Get().speakerHighIcon);
+                    icon->OnClick(DynCtx::OnToggleSource);
                     DynCtx::micIcon = icon.get();
                     break;
                 case AudioType::Output:
                     icon->SetClass("audio-icon");
                     icon->SetText(Config::Get().micHighIcon);
+                    icon->OnClick(DynCtx::OnToggleSink);
                     if (!RotatedIcons())
                         Utils::SetTransform(*icon, {-1, true, Alignment::Fill, 0, 6});
                     DynCtx::audioIcon = icon.get();
diff --git a/src/PulseAudio.h b/src/PulseAudio.h
index a90c8ed..8bb0d3d 100644
--- a/src/PulseAudio.h
+++ b/src/PulseAudio.h
@@ -234,6 +234,34 @@ namespace PulseAudio
         system(cmd.c_str());
     }
 
+    inline void SetMutedSink(bool muted)
+    {
+        if (muted)
+        {
+            LOG("Audio: Mute sink");
+            system("pamixer --mute");
+        }
+        else
+        {
+            LOG("Audio: Unmute sink");
+            system("pamixer --unmute");
+        }
+    }
+
+    inline void SetMutedSource(bool muted)
+    {
+        if (muted)
+        {
+            LOG("Audio: Mute source");
+            system("pamixer --default-source --mute");
+        }
+        else
+        {
+            LOG("Audio: Unmute source");
+            system("pamixer --default-source --unmute");
+        }
+    }
+
     inline void Shutdown()
     {
         pa_mainloop_free(mainLoop);
diff --git a/src/System.cpp b/src/System.cpp
index 09f0e94..3150d34 100644
--- a/src/System.cpp
+++ b/src/System.cpp
@@ -484,6 +484,15 @@ namespace System
         PulseAudio::SetVolumeSource(volume);
     }
 
+    void SetMutedSink(bool muted)
+    {
+        PulseAudio::SetMutedSink(muted);
+    }
+    void SetMutedSource(bool muted)
+    {
+        PulseAudio::SetMutedSource(muted);
+    }
+
 #ifdef WITH_WORKSPACES
     void PollWorkspaces(const std::string& monitor, uint32_t numWorkspaces)
     {
diff --git a/src/System.h b/src/System.h
index 668aa10..50d7efa 100644
--- a/src/System.h
+++ b/src/System.h
@@ -85,6 +85,8 @@ namespace System
     AudioInfo GetAudioInfo();
     void SetVolumeSink(double volume);
     void SetVolumeSource(double volume);
+    void SetMutedSink(bool muted);
+    void SetMutedSource(bool muted);
 
 #ifdef WITH_WORKSPACES
     enum class WorkspaceStatus