From f6cfdec084b9fd314a5707e52ee0273afd5cb3d6 Mon Sep 17 00:00:00 2001
From: Cristian Miranda <crism60@gmail.com>
Date: Wed, 17 Jan 2024 21:02:18 -0300
Subject: [PATCH] :sparkles: do-not-disturb: add DND utility

---
 home/.config/polybar/config.ini               |  6 ++---
 .../scripts/{dunstctl => do-not-disturb}      |  2 ++
 home/bin/do-not-disturb                       | 22 +++++++++++++++++++
 home/bin/lock-screen                          | 12 +++++-----
 home/bin/toggle-app-audio                     | 14 ++++++++++++
 5 files changed, 47 insertions(+), 9 deletions(-)
 rename home/.config/polybar/scripts/{dunstctl => do-not-disturb} (87%)
 create mode 100755 home/bin/do-not-disturb
 create mode 100755 home/bin/toggle-app-audio

diff --git a/home/.config/polybar/config.ini b/home/.config/polybar/config.ini
index 51776476..2c9ee24b 100644
--- a/home/.config/polybar/config.ini
+++ b/home/.config/polybar/config.ini
@@ -444,11 +444,11 @@ click-right = alacritty --class i3FloatingWindow -e $HOME/bin/vpn --disconnect
 
 [module/dunst]
 type = custom/script
-interval = 2
+interval = 1
 format = <label>
 label = "%output%"
-exec = .config/polybar/scripts/dunstctl
-click-left = .config/polybar/scripts/dunstctl --toggle
+exec = .config/polybar/scripts/do-not-disturb
+click-left = .config/polybar/scripts/do-not-disturb --toggle
 
 [module/spotify]
 type = custom/script
diff --git a/home/.config/polybar/scripts/dunstctl b/home/.config/polybar/scripts/do-not-disturb
similarity index 87%
rename from home/.config/polybar/scripts/dunstctl
rename to home/.config/polybar/scripts/do-not-disturb
index c5e9ca58..9b5e30c9 100755
--- a/home/.config/polybar/scripts/dunstctl
+++ b/home/.config/polybar/scripts/do-not-disturb
@@ -7,8 +7,10 @@ case "$1" in
 --toggle)
     muted=$(dunstctl is-paused 2>/dev/null)
     if [[ $muted == "true" ]]; then
+        $HOME/bin/do-not-disturb off
         dunstctl set-paused false 2>/dev/null
     else
+        $HOME/bin/do-not-disturb on
         dunstctl set-paused true 2>/dev/null
     fi
     ;;
diff --git a/home/bin/do-not-disturb b/home/bin/do-not-disturb
new file mode 100755
index 00000000..597acd84
--- /dev/null
+++ b/home/bin/do-not-disturb
@@ -0,0 +1,22 @@
+#!/bin/bash
+
+MODE=$1 # on/off
+
+APPS=(
+    "discord"
+    "spotify"
+    "telegram-desktop"
+    "whatsapp-for-linux"
+    "zoom"
+)
+
+if [ "$MODE" == "on" ]; then
+    MODE="mute"
+
+else
+    MODE="unmute"
+fi
+
+for APP in "${APPS[@]}"; do
+    $HOME/bin/toggle-app-audio $APP $MODE
+done
diff --git a/home/bin/lock-screen b/home/bin/lock-screen
index 869a9905..dd596ad3 100755
--- a/home/bin/lock-screen
+++ b/home/bin/lock-screen
@@ -14,12 +14,12 @@ XIDLEHOOK_LOCK_SOCKET=/tmp/xidlehook.sock
 
 NO_FORK="--nofork"
 
-function pauseNotifications() {
-    dunstctl set-paused true
+function doNotDisturbOn() {
+    $HOME/bin/do-not-disturb on
 }
 
-function resumeNotifications() {
-    dunstctl set-paused false
+function doNotDisturbOff() {
+    $HOME/bin/do-not-disturb off
 }
 
 function turnScreenOff() {
@@ -82,7 +82,7 @@ function lockScreen() {
     if pgrep -x "i3lock" >/dev/null; then
         echo "Computer already locked. Aborting..."
     else
-        pauseNotifications
+        doNotDisturbOn
         muteVolume
         turnScreenOff
 
@@ -90,7 +90,7 @@ function lockScreen() {
         runLocker
         openrgb --mode static --color 0313fc
 
-        resumeNotifications
+        doNotDisturbOff
         unmuteVolume
     fi
 }
diff --git a/home/bin/toggle-app-audio b/home/bin/toggle-app-audio
new file mode 100755
index 00000000..471a0cb4
--- /dev/null
+++ b/home/bin/toggle-app-audio
@@ -0,0 +1,14 @@
+#!/bin/bash
+
+APP_NAME=$1
+ACTION=$2
+SINK_INPUTS=$(pactl list sink-inputs | grep -i -E -B 20 "(application.name = \"$APP_NAME\"|device.description = \"$APP_NAME\")" | grep "Sink Input #" | awk '{print $3}' | sed 's/#//g')
+
+for SINK in $SINK_INPUTS; do
+  MUTE_STATUS=0
+  if [ "$ACTION" == "mute" ]; then
+    MUTE_STATUS=1
+  fi
+
+  pactl set-sink-input-mute $SINK $MUTE_STATUS
+done