feat: add y/n dialog before modifying rcfile (#35)

install.sh

@@ -11,37 +11,37 @@ PLATFORM=$(uname -s)
 ARCH=$(uname -m)
 
 if [ $PLATFORM = "Darwin" ]; then
-  PLATFORM="apple-darwin"
+    PLATFORM="apple-darwin"
 elif [ $PLATFORM = "Linux" ]; then
-  PLATFORM="unknown-linux-musl"
+    PLATFORM="unknown-linux-musl"
 fi
 
 if [ $ARCH == "arm64" ] || [ $ARCH == "aarch64" ]; then
-  ARCH="aarch64"
+    ARCH="aarch64"
 fi
 
 BINARY="yen-rs-${ARCH}-${PLATFORM}"
 
+DOWNLOAD_URL="https://github.com/${REPO}/releases/latest/download/${BINARY}"
 printf "This script will automatically download and install yen for you.\nGetting it from this url: $DOWNLOAD_URL\nThe binary will be installed into '$INSTALL_DIR'\n"
 
 if ! hash curl 2> /dev/null; then
-  echo "error: you do not have 'curl' installed which is required for this script."
-  exit 1
+    echo "error: you do not have 'curl' installed which is required for this script."
+    exit 1
 fi
 
 TEMP_FILE=$(mktemp "${TMPDIR:-/tmp}/.yen_install.XXXXXXXX")
 
 cleanup() {
-  rm -f "$TEMP_FILE"
+    rm -f "$TEMP_FILE"
 }
 
 trap cleanup EXIT
 
-DOWNLOAD_URL="https://github.com/${REPO}/releases/latest/download/${BINARY}"
 HTTP_CODE=$(curl -SL --progress-bar "$DOWNLOAD_URL" --output "$TEMP_FILE" --write-out "%{http_code}")
 if [ ${HTTP_CODE} -lt 200 ] || [ ${HTTP_CODE} -gt 299 ]; then
-  echo "error: '${DOWNLOAD_URL}' is not available"
-  exit 1
+    echo "error: '${DOWNLOAD_URL}' is not available"
+    exit 1
 fi
 
 # Move yen to the install directory
@@ -52,17 +52,17 @@ mv "$TEMP_FILE" "$INSTALL_DIR/yen"
 USERPATH_URL="https://yen.tushar.lol/userpath.pyz"
 HTTP_CODE=$(curl -SL --progress-bar "$USERPATH_URL" --output "$TEMP_FILE" --write-out "%{http_code}")
 if [ ${HTTP_CODE} -lt 200 ] || [ ${HTTP_CODE} -gt 299 ]; then
-  echo "error: '${USERPATH_URL}' is not available"
-  exit 1
+    echo "error: '${USERPATH_URL}' is not available"
+    exit 1
 fi
 mkdir -p "$INSTALL_DIR"
 mv "$TEMP_FILE" "$INSTALL_DIR/userpath.pyz"
 
 MICROVENV_URL="https://yen.tushar.lol/microvenv.py"
 HTTP_CODE=$(curl -SL --progress-bar "$MICROVENV_URL" --output "$TEMP_FILE" --write-out "%{http_code}")
 if [ ${HTTP_CODE} -lt 200 ] || [ ${HTTP_CODE} -gt 299 ]; then
-  echo "error: '${MICROVENV_URL}' is not available"
-  exit 1
+    echo "error: '${MICROVENV_URL}' is not available"
+    exit 1
 fi
 mkdir -p "$INSTALL_DIR"
 mv "$TEMP_FILE" "$INSTALL_DIR/microvenv.py"
@@ -71,12 +71,25 @@ mv "$TEMP_FILE" "$INSTALL_DIR/microvenv.py"
 update_shell() {
     FILE=$1
     LINE=$2
+    FLAGS=$3
 
     if [ -f "$FILE" ]; then
-        if ! grep -Fxq "$LINE" "$FILE"
-        then
-            echo "Updating '${FILE}'"
-            echo "$LINE" >> "$FILE"
+        if ! grep -Fxq "$LINE" "$FILE"; then
+            if [[ $FLAGS == *"-y"* ]]; then
+                echo "$LINE" >> "$FILE"
+            else
+                while true; do
+                    printf "Install 'yen' in '${FILE}'? [y/n] "
+                    old_stty_cfg=$(stty -g)
+                    stty raw -echo ; yn=$(head -c 1) ; stty $old_stty_cfg
+                    echo
+                    case $yn in
+                        [Yy]* ) echo "$LINE" >> "$FILE"; break;;
+                        [Nn]* ) break;;
+                        * ) echo "Please answer [y]es or [n]o.";;
+                    esac
+                done
+            fi
         fi
     fi
 }
@@ -89,17 +102,17 @@ case "$(basename "$SHELL")" in
             BASH_FILE=~/.bashrc
         fi
         LINE="export PATH=\$PATH:${INSTALL_DIR}"
-        update_shell $BASH_FILE "$LINE"
+        update_shell $BASH_FILE "$LINE" "$@"
         ;;
 
     fish)
         LINE="fish_add_path ${INSTALL_DIR}"
-        update_shell ~/.config/fish/config.fish "$LINE"
+        update_shell ~/.config/fish/config.fish "$LINE" "$@"
         ;;
 
     zsh)
         LINE="export PATH=\$PATH:${INSTALL_DIR}"
-        update_shell ~/.zshrc "$LINE"
+        update_shell ~/.zshrc "$LINE" "$@"
         ;;
 
     *)
@@ -111,4 +124,4 @@ chmod +x "$INSTALL_DIR/yen"
 
 echo "Please restart or source your shell."
 
-}; __wrap__
+}; __wrap__ "$@"