From 899843978ebd7161a9659a24eff1d694f3a8e74e Mon Sep 17 00:00:00 2001
From: Erik Reider <ereider20010103@gmail.com>
Date: Tue, 11 Jan 2022 15:21:59 +0100
Subject: [PATCH] Added dry-print-both with waybar friendly output

---
 include/data.hpp |  2 ++
 src/data.cpp     | 17 +++++++++++++++++
 src/main.cpp     | 45 +++++++++++++++++++++++----------------------
 src/pulse.cpp    |  5 ++++-
 4 files changed, 46 insertions(+), 23 deletions(-)

diff --git a/include/data.hpp b/include/data.hpp
index 6a2bc51..30534ea 100644
--- a/include/data.hpp
+++ b/include/data.hpp
@@ -12,6 +12,7 @@ using namespace std;
 enum SubscriptionType {
   SUBSCRIPTION_TYPE_IDLE,
   SUBSCRIPTION_TYPE_DRY_BOTH,
+  SUBSCRIPTION_TYPE_DRY_BOTH_WAYBAR,
   SUBSCRIPTION_TYPE_DRY_SINK,
   SUBSCRIPTION_TYPE_DRY_SOURCE,
 };
@@ -49,4 +50,5 @@ struct Data {
 
  private:
   void print(bool isRunning);
+  void printWayBar(bool activeSink, bool activeSource);
 };
diff --git a/src/data.cpp b/src/data.cpp
index fb67004..8dbeda2 100644
--- a/src/data.cpp
+++ b/src/data.cpp
@@ -26,6 +26,9 @@ void Data::handleAction() {
     case SUBSCRIPTION_TYPE_DRY_BOTH:
       this->print(activeSink || activeSource);
       break;
+    case SUBSCRIPTION_TYPE_DRY_BOTH_WAYBAR:
+      this->printWayBar(activeSink, activeSource);
+      break;
     case SUBSCRIPTION_TYPE_DRY_SINK:
       this->print(activeSink);
       break;
@@ -38,3 +41,17 @@ void Data::handleAction() {
 void Data::print(bool isRunning) {
   cout << (isRunning ? "RUNNING" : "NOT RUNNING") << endl;
 }
+
+void Data::printWayBar(bool activeSink, bool activeSource) {
+  string result[2] = {activeSink ? "output" : "", activeSource ? "input" : ""};
+  string text = "";
+  for (const auto &str : result) {
+    if (!text.empty() && !str.empty()) text += "-";
+    text += str;
+  }
+  if (text.empty()) text = "none";
+  printf(
+      "{\"text\": \"\", \"alt\": \"%s\", \"tooltip\": \"\", \"class\": "
+      "\"%s\"}\n",
+      text.c_str(), text.c_str());
+}
diff --git a/src/main.cpp b/src/main.cpp
index ea351ea..391865e 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -1,33 +1,29 @@
 #include <cstring>
 
+#include "data.hpp"
 #include "pulse.hpp"
 
 void showHelp(char **argv) {
-  cout << "Usage:" << endl;
-  cout << "\t" << argv[0] << " <OPTION>" << endl;
-  cout << "Options:" << endl;
-  cout << "\t" << argv[0]
-       << "\t\t\t\t Inhibits idle if either any sink or any source is running"
-       << endl;
-  cout << "\t" << argv[0] << " -h, --help \t\t Show help options" << endl;
-  cout << "\t" << argv[0]
-       << " --dry-print-both \t Don't inhibit idle and print if either any "
-          "sink or any source is running"
-       << endl;
-  cout << "\t" << argv[0]
-       << " --dry-print-sink \t Don't inhibit idle and print if any sink is "
-          "running"
-       << endl;
-  cout << "\t" << argv[0]
-       << " --dry-print-source \t Don't inhibit idle and print if any source "
-          "is running"
-       << endl;
+  string name = basename(argv[0]);
+  cout << "Usage:\n";
+  cout << "\t" << name << " <OPTION>\n";
+  cout << "Options:\n";
+  cout << "\t " << name
+       << "\t Inhibits idle if either any sink or any source is running\n";
+  cout << "\t -h, --help \t\t\t Show help options\n";
+  cout << "\t --dry-print-both \t\t Don't inhibit idle and print if either any "
+          "sink or any source is running\n";
+  cout << "\t --dry-print-both-waybar \t Same as --dry-print-both but outputs "
+          "in a waybar friendly manner\n";
+  cout << "\t --dry-print-sink \t\t Don't inhibit idle and print if any sink is "
+          "running\n";
+  cout << "\t --dry-print-source \t\t Don't inhibit idle and print if any source "
+          "is running\n";
 }
 
 int main(int argc, char *argv[]) {
-  bool inhibitIdle = true;
-
   bool printBoth = false;
+  bool printBothWayBar = false;
   bool printSource = false;
   bool printSink = false;
 
@@ -39,6 +35,8 @@ int main(int argc, char *argv[]) {
         printSink = true;
       } else if (strcmp(argv[i], "--dry-print-both") == 0) {
         printBoth = true;
+      } else if (strcmp(argv[i], "--dry-print-both-waybar") == 0) {
+        printBothWayBar = true;
       } else {
         showHelp(argv);
         return 0;
@@ -46,12 +44,15 @@ int main(int argc, char *argv[]) {
     }
   }
 
-  if (inhibitIdle && (!printSink && !printSource && !printBoth)) {
+  if (!printSink && !printSource && !printBoth && !printBothWayBar) {
     return Pulse().init(SUBSCRIPTION_TYPE_IDLE, PA_SUBSCRIPTION_MASK_ALL,
                         EVENT_TYPE_IDLE);
   } else if (printBoth) {
     return Pulse().init(SUBSCRIPTION_TYPE_DRY_BOTH, PA_SUBSCRIPTION_MASK_ALL,
                         EVENT_TYPE_DRY_BOTH);
+  } else if (printBothWayBar) {
+    return Pulse().init(SUBSCRIPTION_TYPE_DRY_BOTH_WAYBAR,
+                        PA_SUBSCRIPTION_MASK_ALL, EVENT_TYPE_DRY_BOTH);
   } else if (printSink) {
     return Pulse().init(SUBSCRIPTION_TYPE_DRY_SINK, PA_SUBSCRIPTION_MASK_SINK,
                         EVENT_TYPE_DRY_SINK);
diff --git a/src/pulse.cpp b/src/pulse.cpp
index a9dc09b..abe58ec 100644
--- a/src/pulse.cpp
+++ b/src/pulse.cpp
@@ -10,6 +10,8 @@
 #include <cstdlib>
 #include <iostream>
 
+#include "data.hpp"
+
 int Pulse::init(SubscriptionType subscriptionType,
                 pa_subscription_mask_t pa_subscriptionType,
                 EventType eventType) {
@@ -86,7 +88,8 @@ void Pulse::subscribe_callback(pa_context *, pa_subscription_event_type_t type,
   }
   if (data->subscriptionType == SUBSCRIPTION_TYPE_IDLE) {
     eventType = EVENT_TYPE_IDLE;
-  } else if (data->subscriptionType == SUBSCRIPTION_TYPE_DRY_BOTH) {
+  } else if (data->subscriptionType == SUBSCRIPTION_TYPE_DRY_BOTH ||
+             data->subscriptionType == SUBSCRIPTION_TYPE_DRY_BOTH_WAYBAR) {
     eventType = EVENT_TYPE_DRY_BOTH;
   }
   data->eventCalled = eventType;