fix: insert error handling in list-all & download plugin scripts

lib/commands/command-install.bash

@@ -173,24 +173,27 @@ install_tool_version() {
       )
     fi
 
-    (
-      # shellcheck disable=SC2031
-      export ASDF_INSTALL_TYPE=$install_type
-      # shellcheck disable=SC2031
-      export ASDF_INSTALL_VERSION=$version
-      # shellcheck disable=SC2031
-      export ASDF_INSTALL_PATH=$install_path
-      # shellcheck disable=SC2031
-      export ASDF_DOWNLOAD_PATH=$download_path
-      # shellcheck disable=SC2031
-      export ASDF_CONCURRENCY=$concurrency
-      mkdir "$install_path"
-      asdf_run_hook "pre_asdf_install_${plugin_name}" "$full_version"
-      bash "${plugin_path}"/bin/install
-    )
-
-    local exit_code=$?
-    if [ $exit_code -eq 0 ]; then
+    local download_exit_code=$?
+    if [ $download_exit_code -eq 0 ]; then
+      (
+        # shellcheck disable=SC2031
+        export ASDF_INSTALL_TYPE=$install_type
+        # shellcheck disable=SC2031
+        export ASDF_INSTALL_VERSION=$version
+        # shellcheck disable=SC2031
+        export ASDF_INSTALL_PATH=$install_path
+        # shellcheck disable=SC2031
+        export ASDF_DOWNLOAD_PATH=$download_path
+        # shellcheck disable=SC2031
+        export ASDF_CONCURRENCY=$concurrency
+        mkdir "$install_path"
+        asdf_run_hook "pre_asdf_install_${plugin_name}" "$full_version"
+        bash "${plugin_path}"/bin/install
+      )
+    fi
+
+    local install_exit_code=$?
+    if [ $install_exit_code -eq 0 ] && [ $download_exit_code -eq 0 ]; then
       # Remove download directory if --keep-download flag or always_keep_download config setting are not set
       always_keep_download=$(get_asdf_config_value "always_keep_download")
       if [ ! "$keep_download" = "true" ] && [ ! "$always_keep_download" = "yes" ] && [ -d "$download_path" ]; then

lib/commands/command-list-all.bash

@@ -8,7 +8,15 @@ list_all_command() {
   check_if_plugin_exists "$plugin_name"
 
   local versions
-  versions=$(bash "${plugin_path}/bin/list-all")
+  # Capture return code to allow error handling
+  return_code=0 && versions=$(bash "${plugin_path}/bin/list-all" 2>&1) || return_code=$?
+
+  if [[ $return_code -ne 0 ]]; then
+    # Printing all output to allow plugin to handle error formatting
+    printf "Plugin %s's list-all callback script failed with output:\\n" "${plugin_name}" >&2
+    printf "%s\\n" "${versions}" >&2
+    exit 1
+  fi
 
   if [[ $query ]]; then
     versions=$(tr ' ' '\n' <<<"$versions" |

test/fixtures/dummy_broken_plugin/bin/download

@@ -0,0 +1,4 @@
+#!/usr/bin/env bash
+
+echo "Download failed!"
+exit 1

test/fixtures/dummy_broken_plugin/bin/install

@@ -0,0 +1,3 @@
+#!/usr/bin/env bash
+
+echo "Unused script"

test/fixtures/dummy_broken_plugin/bin/list-all

@@ -0,0 +1,4 @@
+#!/usr/bin/env bash
+
+echo "List-all failed!"
+exit 1

test/install_command.bats

@@ -6,6 +6,7 @@ setup() {
   setup_asdf_dir
   install_dummy_legacy_plugin
   install_dummy_plugin
+  install_dummy_broken_plugin
 
   PROJECT_DIR=$HOME/project
   mkdir $PROJECT_DIR
@@ -258,3 +259,12 @@ EOM
   [ -d $ASDF_DIR/downloads/dummy/1.1.0 ]
   [ $(cat $ASDF_DIR/installs/dummy/1.1.0/version) = "1.1.0" ]
 }
+
+@test "install_command fails when download script exits with non-zero code" {
+  run asdf install dummy-broken 1.0.0
+  echo $output
+  [ "$status" -eq 1 ]
+  [ ! -d $ASDF_DIR/downloads/dummy-broken/1.1.0 ]
+  [ ! -d $ASDF_DIR/installs/dummy-broken/1.1.0 ]
+  [ "$output" == "Download failed!" ]
+}

test/list_command.bats

@@ -5,6 +5,7 @@ load test_helpers
 setup() {
   setup_asdf_dir
   install_dummy_plugin
+  install_dummy_broken_plugin
 }
 
 teardown() {
@@ -15,7 +16,8 @@ teardown() {
   run asdf install dummy 1.0.0
   run asdf install dummy 1.1.0
   run asdf list
-  [ "$(echo -e "dummy\n  1.0.0\n  1.1.0")" == "$output" ]
+  [[ "$output" == "$(echo -e "dummy\n  1.0.0\n  1.1.0")"* ]]
+  [[ "$output" == *"$(echo -e "dummy-broken\n  No versions installed")" ]]
   [ "$status" -eq 0 ]
 }
 
@@ -26,7 +28,10 @@ teardown() {
   run asdf install dummy 1.0.0
   run asdf install tummy 2.0.0
   run asdf list
-  [ "$(echo -e "dummy\n  1.0.0\nmummy\n  No versions installed\ntummy\n  2.0.0")" == "$output" ]
+  [[ "$output" == "$(echo -e "dummy\n  1.0.0")"* ]]
+  [[ "$output" == *"$(echo -e "dummy-broken\n  No versions installed")"* ]]
+  [[ "$output" == *"$(echo -e "mummy\n  No versions installed")"* ]]
+  [[ "$output" == *"$(echo -e "tummy\n  2.0.0")" ]]
   [ "$status" -eq 0 ]
 }
 
@@ -49,3 +54,11 @@ teardown() {
   [ "$(echo -e "1.0.0\n1.1.0")" == "$output" ]
   [ "$status" -eq 0 ]
 }
+
+@test "list_all_command fails when list-all script exits with non-zero code" {
+  run asdf list-all dummy-broken
+  echo $output
+  [ "$status" -eq 1 ]
+  [[ "$output" == "Plugin dummy-broken's list-all callback script failed with output:"* ]]
+  [[ "$output" == *"List-all failed!" ]]
+}

test/test_helpers.bash

@@ -33,6 +33,12 @@ install_mock_legacy_plugin() {
   cp -r "$BATS_TEST_DIRNAME/fixtures/dummy_legacy_plugin" "$location/plugins/$plugin_name"
 }
 
+install_mock_broken_plugin() {
+  local plugin_name=$1
+  local location="${2:-$ASDF_DIR}"
+  cp -r "$BATS_TEST_DIRNAME/fixtures/dummy_broken_plugin" "$location/plugins/$plugin_name"
+}
+
 install_mock_plugin_repo() {
   local plugin_name=$1
   local location="${BASE_DIR}/repo-${plugin_name}"
@@ -59,6 +65,10 @@ install_dummy_legacy_plugin() {
   install_mock_legacy_plugin "legacy-dummy"
 }
 
+install_dummy_broken_plugin() {
+  install_mock_broken_plugin "dummy-broken"
+}
+
 install_dummy_version() {
   install_mock_plugin_version "dummy" "$1"
 }