From e742d169c29e5c116499785b83201a137eac6bfc Mon Sep 17 00:00:00 2001
From: Shihaam Abdul Rahman <shihaam@shihaam.me>
Date: Sun, 20 Aug 2023 12:00:53 +0500
Subject: [PATCH] mOdUlAr

---
 scripts/tginput.sh | 40 ++++++++++++++++++++++++++--------------
 1 file changed, 26 insertions(+), 14 deletions(-)

diff --git a/scripts/tginput.sh b/scripts/tginput.sh
index c6f89ac..0d63168 100755
--- a/scripts/tginput.sh
+++ b/scripts/tginput.sh
@@ -6,30 +6,42 @@ if [[ $XDG_SESSION_TYPE != "x11" ]]; then
 fi
 
 get_tg_window_id(){
-tg_window_id=$(wmctrl -lx | grep telegram-desktop | awk '{print $1}')
+	tg_window_id=$(wmctrl -lx | grep telegram-desktop | awk '{print $1}')
 }
 
 get_mouse_location(){
-mouse_xy=$(xdotool getmouselocation --shell)
-mouse_x=$(printf "%s\n" "$mouse_xy" | grep X= | cut -d= -f2)
-mouse_y=$(printf "%s\n" "$mouse_xy" | grep Y= | cut -d= -f2)
+	mouse_xy=$(xdotool getmouselocation --shell)
+	mouse_x=$(printf "%s\n" "$mouse_xy" | grep X= | cut -d= -f2)
+	mouse_y=$(printf "%s\n" "$mouse_xy" | grep Y= | cut -d= -f2)
 }
 
 return_mouse(){
-xdotool mousemove $mouse_x $mouse_y
+	xdotool mousemove $mouse_x $mouse_y
+}
+
+lock_tg(){
+	xdotool windowactivate --sync $tg_window_id key ctrl+l
+}
+
+readpin(){
+	tgpin=$(cat ~/.local/.tgpin.txt)
+}
+
+unlock_tg(){
+	xdotool windowactivate --sync $tg_window_id key $tgpin Return
 }
 
 if [ "$1" == "lock" ]; then
-    get_tg_window_id
-    get_mouse_location
-    xdotool windowactivate --sync $tg_window_id key ctrl+l
-    return_mouse
+	get_tg_window_id
+	get_mouse_location
+	lock_tg
+	return_mouse
 elif [ "$1" == "unlock" ]; then
-    get_tg_window_id
-    get_mouse_location
-    tgpin=$(cat ~/.local/.tgpin.txt)
-    xdotool windowactivate --sync $tg_window_id key $tgpin Return
-    return_mouse
+	get_tg_window_id
+	get_mouse_location
+	readpin
+	unlock_tg
+	return_mouse
 else
     echo "Invalid argument. Please use 'lock' or 'unlock'."
     exit 1