Support tools like stringer (#277)

Author: anttiharju

internal/choose/choose.go

@@ -4,9 +4,10 @@ import (
 	"context"
 
 	"github.com/anttiharju/vmatch/internal/doctor"
-	"github.com/anttiharju/vmatch/internal/inject"
 	"github.com/anttiharju/vmatch/internal/language"
 	"github.com/anttiharju/vmatch/internal/linter"
+	"github.com/anttiharju/vmatch/internal/scripts"
+	"github.com/anttiharju/vmatch/internal/symlinks"
 	"github.com/anttiharju/vmatch/pkg/version"
 )
 
@@ -15,19 +16,23 @@ func firstArgIs(arg string, args []string) bool {
 }
 
 func Wrapper(ctx context.Context, args []string) int {
-	exitCode := inject.Scripts()
+	exitCode := scripts.Inject()
 	if exitCode != 0 {
 		return exitCode
 	}
 
-	if firstArgIs("go", args) {
-		wrappedLanguage := language.Wrap("go")
+	defer func() {
+		symlinks.Sync()
+	}()
+
+	if firstArgIs(string(scripts.Golang), args) {
+		wrappedLanguage := language.Wrap(scripts.Golang)
 
 		return wrappedLanguage.Run(ctx, args[1:])
 	}
 
-	if firstArgIs("golangci-lint", args) {
-		wrappedLinter := linter.Wrap("golangci-lint")
+	if firstArgIs(string(scripts.GolangCILint), args) {
+		wrappedLinter := linter.Wrap(scripts.GolangCILint)
 
 		return wrappedLinter.Run(ctx, args[1:])
 	}

internal/find/find.go

@@ -11,9 +11,9 @@ import (
 	"strings"
 
 	"github.com/anttiharju/vmatch/internal/exitcode"
-	"github.com/anttiharju/vmatch/internal/find"
-	"github.com/anttiharju/vmatch/internal/install"
-	"github.com/anttiharju/vmatch/internal/wrapper"
+	"github.com/anttiharju/vmatch/internal/scripts"
+	"github.com/anttiharju/vmatch/pkg/install"
+	"github.com/anttiharju/vmatch/pkg/wrapper"
 )
 
 type WrappedLanguage struct {
@@ -45,15 +45,10 @@ func validateVersion(version string) (string, error) {
 	return version, nil
 }
 
-func Wrap(name string) *WrappedLanguage {
-	baseWrapper := wrapper.BaseWrapper{Name: name}
+func Wrap(script scripts.Script) *WrappedLanguage {
+	baseWrapper := wrapper.BaseWrapper{Name: string(script)}
 
-	desiredVersion, err := find.Version("go.mod", languageParser, validateVersion)
-	if err != nil {
-		baseWrapper.ExitWithPrintln(exitcode.VersionReadFileIssue, err.Error())
-	}
-
-	err = baseWrapper.GenerateInstallPath(desiredVersion)
+	err := baseWrapper.GenerateInstallPath("go.mod", languageParser, validateVersion)
 	if err != nil {
 		baseWrapper.ExitWithPrintln(exitcode.InstallPathIssue, err.Error())
 	}

internal/inject/inject.go

@@ -9,8 +9,8 @@ import (
 	"strings"
 
 	"github.com/anttiharju/vmatch/internal/exitcode"
-	"github.com/anttiharju/vmatch/internal/find"
-	"github.com/anttiharju/vmatch/internal/wrapper"
+	"github.com/anttiharju/vmatch/internal/scripts"
+	"github.com/anttiharju/vmatch/pkg/wrapper"
 )
 
 type WrappedLinter struct {
@@ -33,15 +33,10 @@ func validateVersion(version string) (string, error) {
 	return version, nil
 }
 
-func Wrap(name string) *WrappedLinter {
-	baseWrapper := wrapper.BaseWrapper{Name: name}
+func Wrap(script scripts.Script) *WrappedLinter {
+	baseWrapper := wrapper.BaseWrapper{Name: string(script)}
 
-	desiredVersion, err := find.Version(".golangci-version", linterParser, validateVersion)
-	if err != nil {
-		baseWrapper.ExitWithPrintln(exitcode.VersionReadFileIssue, err.Error())
-	}
-
-	err = baseWrapper.GenerateInstallPath(desiredVersion)
+	err := baseWrapper.GenerateInstallPath(".golangci-version", linterParser, validateVersion)
 	if err != nil {
 		baseWrapper.ExitWithPrintln(exitcode.InstallPathIssue, err.Error())
 	}

internal/language/language.go

@@ -0,0 +1,92 @@
+package scripts
+
+import (
+	"embed"
+	"fmt"
+	"io/fs"
+	"os"
+	"path/filepath"
+)
+
+type Script string
+
+const (
+	Golang         Script = "go"
+	GolangCILint   Script = "golangci-lint"
+	GolangCILintV2 Script = "golangci-lint-v2"
+)
+
+func Scripts() []Script {
+	return []Script{Golang, GolangCILint, GolangCILintV2}
+}
+
+func (s Script) File() string {
+	switch s {
+	case Golang:
+		return "go.sh"
+	case GolangCILint, GolangCILintV2:
+		return "golangci-lint.sh"
+	default:
+		return ""
+	}
+}
+
+func exists(path string) bool {
+	_, err := os.Stat(path)
+
+	return err == nil
+}
+
+func Inject() int {
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error: failed to get home directory: %v\n", err)
+
+		return 1
+	}
+
+	binDir := filepath.Join(homeDir, ".vmatch", "bin")
+	if !exists(binDir) {
+		if err := os.MkdirAll(binDir, 0o755); err != nil {
+			fmt.Fprintf(os.Stderr, "Error: failed to create directory %s: %v\n", binDir, err)
+
+			return 1
+		}
+	}
+
+	scripts := Scripts()
+	for _, script := range scripts {
+		if err := createScript(binDir, script); err != nil {
+			fmt.Fprintf(os.Stderr, "Error: failed to create script %s: %v\n", script, err)
+
+			return 1
+		}
+	}
+
+	return 0
+}
+
+//go:embed go.sh golangci-lint.sh
+var scripts embed.FS
+
+func createScript(binDir string, script Script) error {
+	name := string(script)
+	sourcePath := script.File()
+	destPath := filepath.Join(binDir, string(script))
+
+	if exists(destPath) {
+		return nil
+	}
+
+	content, err := fs.ReadFile(scripts, sourcePath)
+	if err != nil {
+		return fmt.Errorf("failed to read embedded script %s: %w", name, err)
+	}
+
+	//nolint:gosec // using 0o755 instead of 0o600 is intentional here
+	if err := os.WriteFile(destPath, content, 0o755); err != nil {
+		return fmt.Errorf("failed to write script %s: %w", name, err)
+	}
+
+	return nil
+}