Add shell completion to .zshrc (#1265)

Author: deliahu

get-cli.sh

@@ -31,7 +31,7 @@ function main() {
   rm -rf $cortex_sh_tmp_dir && mkdir -p $cortex_sh_tmp_dir
 
   if command -v curl >/dev/null; then
-    curl -s -o $cortex_sh_tmp_dir/cortex https://s3-us-west-2.amazonaws.com/get-cortex/$CORTEX_VERSION_BRANCH_STABLE/cli/$parsed_os/cortex
+    curl -s -w "" -o $cortex_sh_tmp_dir/cortex https://s3-us-west-2.amazonaws.com/get-cortex/$CORTEX_VERSION_BRANCH_STABLE/cli/$parsed_os/cortex
   elif command -v wget >/dev/null; then
     wget -q -O $cortex_sh_tmp_dir/cortex https://s3-us-west-2.amazonaws.com/get-cortex/$CORTEX_VERSION_BRANCH_STABLE/cli/$parsed_os/cortex
   else
@@ -82,6 +82,15 @@ function get_bash_profile_path() {
   echo ""
 }
 
+function get_zsh_profile_path() {
+  if [ -f $HOME/.zshrc ]; then
+    echo $HOME/.zshrc
+    return
+  fi
+
+  echo ""
+}
+
 function guess_if_bash_completion_installed() {
   if [ -f $HOME/.bash_profile ]; then
     if grep -q -e "^[^#]*bash-completion" -e "^[^#]*bash_completion" "$HOME/.bash_profile"; then
@@ -102,12 +111,15 @@ function guess_if_bash_completion_installed() {
 
 function update_bash_profile() {
   bash_profile_path=$(get_bash_profile_path)
+  zsh_profile_path=$(get_zsh_profile_path)
   maybe_bash_completion_installed=$(guess_if_bash_completion_installed)
+  did_locate_shell_profile="false"
 
   if [ "$bash_profile_path" != "" ]; then
+    did_locate_shell_profile="true"
     if ! grep -Fxq "source <(cortex completion bash)" "$bash_profile_path"; then
       echo
-      read -p "Would you like to modify your bash profile ($bash_profile_path) to enable cortex command completion and the cx alias? [y/n] " -r
+      read -p "Would you like to modify your bash profile ($bash_profile_path) to enable cortex command completion and the cx alias in bash? [y/n] " -r
       echo
       if [ "$REPLY" = "y" ] || [ "$REPLY" = "Y" ] || [ "$REPLY" = "yes" ] || [ "$REPLY" = "Yes" ] || [ "$REPLY" = "YES" ]; then
         echo -e "\nsource <(cortex completion bash)" >> $bash_profile_path
@@ -118,20 +130,30 @@ function update_bash_profile() {
           echo -e "Note: \`bash-completion\` must be installed on your system for cortex command completion to function properly"
         fi
       else
-        echo -e "Your bash profile has not been modified. If you would like to modify it manually, add this line to your bash profile:"
-        echo "  source <(cortex completion bash)"
-        if [ ! "$maybe_bash_completion_installed" = "true" ]; then
-          echo "Note: \`bash-completion\` must be installed on your system for cortex command completion to function properly"
-        fi
+        echo -e "Your bash profile has not been modified (run \`cortex completion --help\` to show how to enable bash completion manually)"
       fi
     fi
-  else
-    echo -e "\nIf your would like to enable cortex command completion and the cx alias, add this line to your bash profile:"
-    echo "  source <(cortex completion bash)"
-    if [ ! "$maybe_bash_completion_installed" = "true" ]; then
-      echo "Note: \`bash-completion\` must be installed on your system for cortex command completion to function properly"
+  fi
+
+  if [ "$zsh_profile_path" != "" ]; then
+    did_locate_shell_profile="true"
+    if ! grep -Fxq "source <(cortex completion zsh)" "$zsh_profile_path"; then
+      echo
+      read -p "Would you like to modify your zsh profile ($zsh_profile_path) to enable cortex command completion and the cx alias in zsh? [y/n] " -r
+      echo
+      if [ "$REPLY" = "y" ] || [ "$REPLY" = "Y" ] || [ "$REPLY" = "yes" ] || [ "$REPLY" = "Yes" ] || [ "$REPLY" = "YES" ]; then
+        echo -e "\nsource <(cortex completion zsh)" >> $zsh_profile_path
+        echo -e "✓ Your zsh profile has been updated"
+        echo -e "\nStart a new zsh shell for completion to take effect. If completion still doesn't work, you can try adding this line to the top of $zsh_profile_path: autoload -Uz compinit && compinit"
+      else
+        echo -e "Your zsh profile has not been modified (run \`cortex completion --help\` to show how to enable zsh completion manually)"
+      fi
     fi
   fi
+
+  if [ "$did_locate_shell_profile" = "false" ]; then
+    echo -e "\nIf you would like to enable cortex bash completion and the cx alias, run \`cortex completion --help\` for instructions"
+  fi
 }
 
 main