From ba90e973603a81d5e82e584e10025c07cbe8dc47 Mon Sep 17 00:00:00 2001
From: wy-luke <wykj.luke@gmail.com>
Date: Mon, 4 Sep 2023 03:00:03 +0800
Subject: [PATCH] fix: command line arguments parse

---
 README_CN.md    |   2 +-
 sd-installer.sh | 100 ++++++++++++++++++++++++------------------------
 2 files changed, 51 insertions(+), 51 deletions(-)

diff --git a/README_CN.md b/README_CN.md
index 5136cf3..a0c3578 100644
--- a/README_CN.md
+++ b/README_CN.md
@@ -22,7 +22,7 @@
    /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/wy-luke/StableDiffusion-Installer-For-Mac/main/sd-installer.sh)"
    ```
 
-   国内用户存在网络问题时，可以尝试使用以下命令。同时，在提示 `是否存在网络连通问题？ [y/n]` 时，输入 `y`，然后按**回车**。
+   国内用户存在网络问题时，可以尝试使用以下命令。同时，在提示 `网络是否顺畅? 默认y [y/n]` 时，输入 `n`，然后按**回车**。
 
    ```bash
    /bin/bash -c "$(curl -fsSL https://raw.fastgit.org/wy-luke/StableDiffusion-Installer-For-Mac/main/sd-installer.sh)"
diff --git a/sd-installer.sh b/sd-installer.sh
index caed55f..1ef33a5 100644
--- a/sd-installer.sh
+++ b/sd-installer.sh
@@ -12,55 +12,6 @@ net_connected=true
 brew_installer_path="$tmp_path/brew_installer.sh"
 test_mode=0 # Only for test. 0 for no test, 1 for yes-test, 2 for no-test
 
-# Parse command line arguments
-while [[ $# -gt 0 ]]; do
-    case "$1" in
-    -0 | -1 | -2)
-        test_mode=${1:0}
-        shift
-        ;;
-    # -h | --help)
-    #     show_help
-    #     exit 0
-    #     ;;
-    *)
-        echo_red "未知选项: $1"
-        # show_help
-        exit 1
-        ;;
-    esac
-done
-
-if [$test_mode != 0]; then
-    if [$test_mode == 1]; then
-        net_connected=true
-        echo_green "Yes-test"
-    else
-        net_connected=false
-        echo_green "No-test"
-    fi
-else
-    echo_green "For non-Chinese users, you could just ignore this and press the Enter key"
-    read -rp "是否存在网络连通问题? 默认n [y/n] " net_choice
-    net_choice=${net_choice:-n}
-    if [[ $net_choice == [yY] ]]; then
-        net_connected=false
-        echo_green "将会设置国内镜像源"
-    else
-        echo_green "网络通畅, 正常安装"
-    fi
-fi
-
-brew_installer_url="https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh"
-sd_installer_url="https://raw.githubusercontent.com/wy-luke/StableDiffusion-Installer-For-Mac/main/sd-installer.sh"
-sd_webui_url="https://github.com/AUTOMATIC1111/stable-diffusion-webui.git"
-
-if ! $net_connected; then
-    brew_installer_url="https://raw.fastgit.org/Homebrew/install/HEAD/install.sh"
-    sd_installer_url="https://raw.fastgit.org/wy-luke/StableDiffusion-Installer-For-Mac/main/sd-installer.sh"
-    sd_webui_url="https://hub.fastgit.xyz/AUTOMATIC1111/stable-diffusion-webui.git"
-fi
-
 function clean_up {
     echo "############ Clean ###################################"
     rm -rf $tmp_path
@@ -68,7 +19,7 @@ function clean_up {
 
 # Define a function to handle errors
 function handle_error {
-    if [$test_mode != 0]; then
+    if ["$test_mode" != 0]; then
         echo_red "测试失败, 不重试"
         exit 1
     fi
@@ -106,6 +57,55 @@ function echo_red {
     echo -e "\033[31m$1\033[0m"
 }
 
+# Parse command line arguments
+while [[ $# -gt 0 ]]; do
+    case "$1" in
+    -1 | -2)
+        test_mode=${1:1}
+        shift
+        ;;
+    # -h | --help)
+    #     show_help
+    #     exit 0
+    #     ;;
+    *)
+        echo_red "未知选项: $1"
+        # show_help
+        exit 1
+        ;;
+    esac
+done
+
+if ["$test_mode" != 0]; then
+    if ["$test_mode" == 1]; then
+        net_connected=true
+        echo_green "Yes-test"
+    else
+        net_connected=false
+        echo_green "No-test"
+    fi
+else
+    echo_green "For non-Chinese users, you could just ignore this and press the Enter key"
+    read -rp "网络是否顺畅? 默认y [y/n] " net_choice
+    net_choice=${net_choice:-y}
+    if [[ $net_choice == [nN] ]]; then
+        net_connected=false
+        echo_green "将会设置国内镜像源"
+    else
+        echo_green "网络通畅, 正常安装"
+    fi
+fi
+
+brew_installer_url="https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh"
+sd_installer_url="https://raw.githubusercontent.com/wy-luke/StableDiffusion-Installer-For-Mac/main/sd-installer.sh"
+sd_webui_url="https://github.com/AUTOMATIC1111/stable-diffusion-webui.git"
+
+if ! $net_connected; then
+    brew_installer_url="https://raw.fastgit.org/Homebrew/install/HEAD/install.sh"
+    sd_installer_url="https://raw.fastgit.org/wy-luke/StableDiffusion-Installer-For-Mac/main/sd-installer.sh"
+    sd_webui_url="https://hub.fastgit.xyz/AUTOMATIC1111/stable-diffusion-webui.git"
+fi
+
 echo "############ 开始安装 Stable Diffusion web UI #########" && echo
 
 if [ ! -d $tmp_path ]; then