optimize functions.sh

Backstage.sh

@@ -1,26 +1,116 @@
 #!/bin/bash
 export PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"
 
-RED="\033[31m"
-GREEN="\033[32m"
-YELLOW="\033[33m"
-BLUE="\033[36m"
-BLANK="\033[0m"
+color_print() {
+    awk '
+    function red_exit(output) {
+        output = escape_string(output)
+        printf("%s%s%s", colors["red"], output, colors["no_color"])
+        exit 1
+    }
 
-colorEcho() {
-    COLOR=$1
-    shift # delete first parameter
-    echo -e "${COLOR}${@}${BLANK}"
-    echo
+    # parameters are local variables
+    function escape_string(input, i, c, output) {
+        output = ""
+
+        for(;;) {
+            i = match(input, /\\/)
+            if(i == 0) {
+                break
+            }
+
+            output = output substr(input, 0, i - 1)
+
+            c = substr(input, i + 1, 1)
+            switch(c) {
+                case "a":
+                    output = output "\a"
+                    break
+                case "b":
+                    output = output "\b"
+                    break
+                case "f":
+                    output = output "\f"
+                    break
+                case "n":
+                    output = output "\n"
+                    break
+                case "r":
+                    output = output "\r"
+                    break
+                case "t":
+                    output = output "\t"
+                    break
+                case "v":
+                    output = output "\v"
+                    break
+                default:
+                    output = output c
+            }
+
+            input = substr(input, i + 2)
+        }
+        output = output input
+
+        return output
+    }
+    
+    BEGIN {
+        colors["no_color"] = "\033[0m"
+        colors["black"] = "\033[0;30m"
+        colors["red"] = "\033[0;31m"
+        colors["green"] = "\033[0;32m"
+        colors["orange"] = "\033[0;33m"
+        colors["blue"] = "\033[0;34m"
+        colors["purple"] = "\033[0;35m"
+        colors["cyan"] = "\033[0;36m"
+        colors["light_gray"] = "\033[0;37m"
+        colors["dark_gray"] = "\033[1;30m"
+        colors["light_red"] = "\033[1;31m"
+        colors["light_green"] = "\033[1;32m"
+        colors["yellow"] = "\033[1;33m"
+        colors["light_blue"] = "\033[1;34m"
+        colors["light_purple"] = "\033[1;35m"
+        colors["light_cyan"] = "\033[1;36m"
+        colors["white"] = "\033[1;37m"
+
+        if(ARGC < 3 || ARGC % 2 == 0) {
+            red_exit("The number of parameters must be no less than 3 and an odd number.\n")
+        }
+        
+        for(i = 1; i < ARGC; i += 2) {
+            color = ARGV[i]
+            output = ARGV[i + 1]
+
+            # check color exists
+            if(!(color in colors)) {
+                red_exit("invalid color: " color "\n")
+            }
+
+            # color print
+            output = escape_string(output)
+            printf("%s%s%s", colors[color], output, colors["no_color"])
+        }
+
+        exit 0
+    }' "$@"
 }
 
-colorRead() {
-    COLOR=$1
-    OUTPUT=$2
-    VARIABLE=$3
-    echo -e -n "$COLOR$OUTPUT${BLANK}: "
-    read $VARIABLE </dev/tty
-    echo
+color_println() {
+    color_print "$@\n"
+}
+
+color_read() {
+    color=$1
+    output="$2"
+    varaible="$3"
+
+    color_print "$color" "$output: "
+    if echo "$SHELL" | grep -q "bash"; then
+        read -e $varaible </dev/tty
+    else
+        read $varaible </dev/tty
+    fi
 }
 
 cmd_need() {
@@ -30,7 +120,7 @@ cmd_need() {
     for cmd in $1; do
         if ! command -v $cmd >/dev/null 2>&1; then
             # check if auto install
-            if command apt >/dev/null 2>&1; then
+            if command -v apt >/dev/null 2>&1; then
                 # apt install package need update first
                 if [ "update_flag" = "0" ]; then
                     apt update >/dev/null 2>&1
@@ -39,20 +129,20 @@ cmd_need() {
 
                 package=$(dpkg -S bin/$cmd 2>&1 | grep "bin/$cmd$" | awk -F':' '{print $1}')
                 if [ ! -z "$package" ]; then
-                    colorEcho $BLUE "正在安装 $cmd ..."
+                    color_println "cyan" "正在安装 $cmd ..."
                     apt install $package -y >/dev/null 2>&1
                     continue
                 fi
-            elif command yum >/dev/null 2>&1; then
+            elif command -v yum >/dev/null 2>&1; then
                 package=$(yum whatprovides *bin/$cmd 2>&1 | grep " : " | awk -F' : ' '{print $1}' | sed -n '1p')
                 if [ ! -z "$package" ]; then
-                    colorEcho $BLUE "正在安装 $cmd ..."
+                    color_println "cyan" "正在安装 $cmd ..."
                     yum install $package -y >/dev/null 2>&1
                     continue
                 fi
             fi
 
-            colorEcho $RED "找不到 $cmd 命令"
+            color_println "red" "找不到 $cmd 命令"
             exit_flag=1
         fi
     done
@@ -67,32 +157,32 @@ install_tool() {
     if [ -d "/usr" -a ! -z "$(command -v apt-get yum)" ]; then
         wp="/usr/local/$key"
     else
-        colorRead $YELLOW "请输入安装目录，例如：/usr/local，不能是/bin " wp
+        color_println "yellow" "请输入安装目录，例如：/usr/local，不能是/bin " wp
         [ -z "$wp" ] && exit 1
         wp=$(echo "$wp/$key" | sed 's|///*|/|g') # 连续的 / 变为一个
     fi
     file="$key.tar"
     if [ -d "$wp" ]; then
-        colorEcho $YELLOW "正在卸载 $key ..."
+        color_println "cyan" "正在卸载 $key ..."
         bash $wp/uninstall.sh >/dev/null 2>&1
     fi
-    colorEcho $YELLOW "正在安装 $key 到 $wp ..."
+    color_println "cyan" "正在安装 $key 到 $wp ..."
     rm -rf $wp
     mkdir -p $wp
     curl -L "https://raw.githubusercontent.com/FH0/nubia/master/linux-tools/$file" |
         tar -C $wp -xf-
-    sed -i "s|wp=.*|wp=\"$wp\"|g" $wp/*.sh # 修改路径
+    sed -i "s|^wp=.*|wp=\"$wp\"|g" $wp/*.sh # 修改路径
     bash $wp/install.sh
 }
 
 check_environment() {
     if [ "${EUID:-$(id -u)}" != "0" ]; then
-        colorEcho $RED "请切换到 root 用户后再执行此脚本！"
+        color_println "red" "请切换到 root 用户后再执行此脚本！"
         exit 1
     fi
 
     if [ "$(uname -r | awk -F '.' '{print $1}')" -lt "3" ]; then
-        colorEcho $RED "内核不支持，请升级内核或更新系统！"
+        color_println "red" "内核不支持，请升级内核或更新系统！"
         exit 1
     fi
 }
@@ -111,18 +201,18 @@ panel() {
         tools_add "$tool" "$tool"
     done
 
-    colorEcho $BLUE "欢迎使用 JZDH 集合脚本"
+    color_println "cyan" "欢迎使用 JZDH 集合脚本"
     var=1
     echo -e "$tools" | grep -Ev '^$' | while read tool; do
         tool_path="$(echo "$tool" | awk '{print $NF}')"
         tool_name="$(echo "$tool" | awk '{$NF=""; print $0}')"
         if [ -d "/usr/local/$tool_path" ]; then
-            printf "%3s. 安装 ${GREEN}$tool_name${BLANK}\n" "$((var++))"
+            color_println "no_color" " $((var++)). 安装 " "green" "$tool_name"
         else
-            printf "%3s. 安装 $tool_name\n" "$((var++))"
+            color_println "no_color" " $((var++)). 安装 " "no_color" "$tool_name"
         fi
     done
-    echo && colorRead ${YELLOW} '请选择' panel_choice
+    echo && color_read "yellow" '请选择' panel_choice
     [ -z "$panel_choice" ] && clear && exit 0
     for J in $panel_choice; do
         install_tool $(echo -e "$tools" | sed -n "${J}p" | awk '{print $NF}')