feat: auto install missing tools

Author: LinuxSuRen

cli/shell_runner.go

@@ -71,18 +71,34 @@ func (s *ShellScript) runCmdLine(cmdLine, contextDir string, keepScripts bool) (
 			s.ShellType = "bash"
 		}
 
+		deps := map[string]string{
+			s.ShellType: s.ShellType,
+		}
+		for _, cmd := range findPotentialCommands(cmdLine) {
+			deps[cmd] = cmd
+		}
+
 		is := installer.Installer{
 			Provider: "github",
 		}
-		if err = is.CheckDepAndInstall(map[string]string{
-			s.ShellType: s.ShellType,
-		}); err == nil {
+		if err = is.CheckDepAndInstall(deps); err == nil {
 			err = s.Execer.RunCommandInDir(s.ShellType, contextDir, path.Base(shellFile))
 		}
 	}
 	return
 }
 
+func findPotentialCommands(cmdLine string) (cmds []string) {
+	lines := strings.Split(cmdLine, "\n")
+	for _, line := range lines {
+		line = strings.TrimSpace(line)
+		if items := strings.Split(line, " "); len(items) > 0 && items[0] != "" {
+			cmds = append(cmds, items[0])
+		}
+	}
+	return
+}
+
 // GetTitle returns the title of this script
 func (s *ShellScript) GetTitle() string {
 	return s.Title

cli/shell_runner_fuzz_test.go

@@ -0,0 +1,22 @@
+package cli
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func FuzzInputRequest(f *testing.F) {
+	f.Add("a", "b")
+	f.Fuzz(func(t *testing.T, key, value string) {
+		ok, pair, err := isInputRequest(fmt.Sprintf("%s=%s", key, value))
+		assert.True(t, ok)
+		assert.Equal(t, key, pair[0])
+		assert.Equal(t, value, pair[1])
+		assert.Nil(t, err)
+		if err != nil {
+			t.Fail()
+		}
+	})
+}

cli/shell_runner_test.go

@@ -1,10 +1,10 @@
 package cli
 
 import (
-	"fmt"
-	"github.com/linuxsuren/http-downloader/pkg/exec"
 	"testing"
 
+	"github.com/linuxsuren/http-downloader/pkg/exec"
+
 	"github.com/stretchr/testify/assert"
 )
 
@@ -83,16 +83,33 @@ func TestIsInputRequest(t *testing.T) {
 	}
 }
 
-func FuzzInputRequest(f *testing.F) {
-	f.Add("a", "b")
-	f.Fuzz(func(t *testing.T, key, value string) {
-		ok, pair, err := isInputRequest(fmt.Sprintf("%s=%s", key, value))
-		assert.True(t, ok)
-		assert.Equal(t, key, pair[0])
-		assert.Equal(t, value, pair[1])
-		assert.Nil(t, err)
-		if err != nil {
-			t.Fail()
-		}
-	})
+func TestFindPotentialCommands(t *testing.T) {
+	tests := []struct {
+		name   string
+		cmd    string
+		expect []string
+	}{{
+		name:   "oneline cmd",
+		cmd:    "k3d create cluster",
+		expect: []string{"k3d"},
+	}, {
+		name:   "empty",
+		cmd:    "",
+		expect: nil,
+	}, {
+		name: "multiple lines",
+		cmd: `k3d create cluster
+docker ps`,
+		expect: []string{"k3d", "docker"},
+	}, {
+		name:   "with extra whitespace",
+		cmd:    " k3d   create    cluster",
+		expect: []string{"k3d"},
+	}}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			result := findPotentialCommands(tt.cmd)
+			assert.Equal(t, tt.expect, result)
+		})
+	}
 }