From f39deaf1df28b1cf26d2fa955bc72f1578d0370d Mon Sep 17 00:00:00 2001
From: Ark74 <ark@switnet.org>
Date: Tue, 24 Oct 2023 20:55:02 -0600
Subject: [PATCH] Add welcome and color to script.

---
 simple-nerd-dictation-installer.sh | 74 ++++++++++++++++++++++++------
 1 file changed, 59 insertions(+), 15 deletions(-)

diff --git a/simple-nerd-dictation-installer.sh b/simple-nerd-dictation-installer.sh
index a6ec799..426ce47 100644
--- a/simple-nerd-dictation-installer.sh
+++ b/simple-nerd-dictation-installer.sh
@@ -9,15 +9,45 @@
 # - https://github.com/ideasman42/nerd-dictation
 
 set -e
+# Reset
+Color_Off='\e[0m'       # Text Reset
+# Regular Colors
+Black='\e[0;30m'        # Black
+Red='\e[0;31m'          # Red
+Green='\e[0;32m'        # Green
+Yellow='\e[0;33m'       # Yellow
+Blue='\e[0;34m'         # Blue
+Purple='\e[0;35m'       # Purple
+Cyan='\e[0;36m'         # Cyan
+
+printwc() {
+    printf "%b$2%b" "$1" "${Color_Off}"
+}
+print_title() {
+printwc "${Blue}" "\n#--------------------------------------------------"
+printwc "${Blue}" "\n# $1"
+printwc "${Blue}" "\n#--------------------------------------------------\n"
+}
+printf '
+########################################################################
+                    Welcome to Nerd-Dictation Installer
+########################################################################
+# Install nerd-dictation offline speech2text tool script using VOSK backend/model.
+#
+# Follow the installer repo at:
+# - https://forge.switnet.net/Ark74/simple-nerd-dictation-installer
+#
+# Nerd Dictation repo at:
+# - https://github.com/ideasman42/nerd-dictation\n\n'
 
 if [ "$(id -u)" = 0 ]; then
-   echo "Please don't use with root privileges, sudo will be used only when necessary!"
+   printwc "${Yellow}" "Please don't use with root privileges, sudo will be used only when necessary!"
    exit 0
 fi
 if dpkg-architecture -e amd64 ;then
-    echo "Supported architecture."
+    printwc "${Green}" "> amd64: supported architecture."
 else
-    echo "Warning: Only run this script on amd64 arch."
+    printwc "${Red}" "> Warning: Only run this script on amd64 arch."
     exit
 fi
 
@@ -31,7 +61,9 @@ CHINESE_VOSK_MODEL="https://alphacephei.com/vosk/models/vosk-model-small-cn-0.22
 NERD_DIC_GIT_REPO="https://github.com/ideasman42/nerd-dictation.git"
 OPT_VOSK="/opt/vosk-python"
 
-# Select language.
+#--------------------------------------------------
+print_title "# Select language."
+#--------------------------------------------------
 printf "\n> Vosk Model Language selection.\n"
 PS3='Select one of the available language models to use on nerd-dictation: '
 options=("English" "Spanish" "French" "German" "Chinese" "Exit")
@@ -68,15 +100,19 @@ do
             exit 0
             break
             ;;
-        *) echo "Invalid option $REPLY, choose one number of the above.";;
+        *) printwc "${Yellow}" "Invalid option $REPLY, choose one number of the above.";;
     esac
 done
 VOSK_FOLDER="$(basename $VOSK_MODEL|sed 's|.zip||')"
 
-# Install dependencies
+#--------------------------------------------------
+print_title "Install dependencies"
+#--------------------------------------------------
 sudo apt-get install -y git ssh python3-setuptools pulseaudio-utils xdotool
 
-# Install vosk python3 module.
+#--------------------------------------------------
+print_title "Install vosk python3 module."
+#--------------------------------------------------
 [ -d "$OPT_VOSK" ] && sudo rm -rf "$OPT_VOSK"
 sudo mkdir "$OPT_VOSK"
 cat << EOF  | sudo tee "$OPT_VOSK"/setup.py
@@ -94,39 +130,47 @@ sudo python3 "$OPT_VOSK"/setup.py install
 if [ $? = 0 ]; then
     sudo rm "$OPT_VOSK"/setup.py
 else
-    echo "something went wrong, please check setup.py file"
+    printwc "${Red}" "Something went wrong, please check setup.py file"
     exit
 fi
 
-# Cleaning existing local directories and binary link.
+#--------------------------------------------------
+print_title "Cleaning existing local directories and binary link."
+#--------------------------------------------------
 rm -rf "$HOME"/.local/nerd-dictation
 rm -f  "$HOME"/.local/bin/nerd-dictation
 rm -rf "$HOME"/.config/nerd-dictation
 
-# Set repo as local package.
+#--------------------------------------------------
+print_title "Set repo as local package."
+#--------------------------------------------------
 cd "$HOME"/.local
 git clone "$NERD_DIC_GIT_REPO"
 cd nerd-dictation
 
-# Setup language model
+#--------------------------------------------------
+print_title "Setup language model"
+#--------------------------------------------------
 wget "$VOSK_MODEL"
 unzip "$(basename $VOSK_MODEL)"
 if [ $? = 0 ]; then
     rm "$(basename $VOSK_MODEL)"
 else
-    echo "something went wrong, please check $(basename $VOSK_MODEL) file"
+    printwc "${Red}" "Something went wrong, please check $(basename $VOSK_MODEL) file"
     exit
 fi
 mv "$VOSK_FOLDER" model
 mkdir "$HOME"/.config/nerd-dictation
 mv model "$HOME"/.config/nerd-dictation/
 
-# Setup local bin for nerd-dictation
+#--------------------------------------------------
+print_title "Setup local bin for nerd-dictation"
+#--------------------------------------------------
 [ ! -d "$HOME"/.local/bin ] && mkdir "$HOME"/.local/bin
 ln -s "$HOME"/.local/nerd-dictation/nerd-dictation "$HOME"/.local/bin
 
 if [ $? = 0 ]; then
-    echo "Completed script!\nPlease exit and login for nerd-dictation to be used on the terminal."
+    printwc "${Green}" "Completed script!\nPlease exit and login for nerd-dictation to be used on the terminal."
 else
-    echo "Something went wrong please check."
+    printwc "${Red}" "Something went wrong please check the previous output."
 fi