refactor: merge Setup into NewExecutor

executor_setup.go

@@ -0,0 +1,202 @@
+package task
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"slices"
+	"sync"
+
+	"github.com/Masterminds/semver/v3"
+	"github.com/sajari/fuzzy"
+
+	"github.com/go-task/task/v3/errors"
+	"github.com/go-task/task/v3/internal/logger"
+	"github.com/go-task/task/v3/internal/output"
+	"github.com/go-task/task/v3/internal/version"
+	"github.com/go-task/task/v3/taskfile"
+	"github.com/go-task/task/v3/taskfile/ast"
+)
+
+func (e *Executor) setup() error {
+	e.setupLogger()
+	e.setupFuzzyModel()
+	e.setupStdFiles()
+	if err := e.setupOutput(); err != nil {
+		return err
+	}
+	if err := e.setupCompiler(); err != nil {
+		return err
+	}
+	if err := e.readDotEnvFiles(); err != nil {
+		return err
+	}
+	if err := e.doVersionChecks(); err != nil {
+		return err
+	}
+	e.setupDefaults()
+	e.setupConcurrencyState()
+	return nil
+}
+
+func (e *Executor) setupFuzzyModel() {
+	if e.Taskfile == nil {
+		return
+	}
+
+	model := fuzzy.NewModel()
+	model.SetThreshold(1) // because we want to build grammar based on every task name
+
+	var words []string
+	for name, task := range e.Taskfile.Tasks.All(nil) {
+		if task.Internal {
+			continue
+		}
+		words = append(words, name)
+		words = slices.Concat(words, task.Aliases)
+	}
+
+	model.Train(words)
+	e.fuzzyModel = model
+}
+
+func (e *Executor) setupStdFiles() {
+	if e.Stdin == nil {
+		e.Stdin = os.Stdin
+	}
+	if e.Stdout == nil {
+		e.Stdout = os.Stdout
+	}
+	if e.Stderr == nil {
+		e.Stderr = os.Stderr
+	}
+}
+
+func (e *Executor) setupLogger() {
+	e.Logger = &logger.Logger{
+		Stdin:      e.Stdin,
+		Stdout:     e.Stdout,
+		Stderr:     e.Stderr,
+		Verbose:    e.Verbose,
+		Color:      e.Color,
+		AssumeYes:  e.AssumeYes,
+		AssumeTerm: e.AssumeTerm,
+	}
+}
+
+func (e *Executor) setupOutput() error {
+	if !e.OutputStyle.IsSet() {
+		e.OutputStyle = e.Taskfile.Output
+	}
+
+	var err error
+	e.Output, err = output.BuildFor(&e.OutputStyle, e.Logger)
+	return err
+}
+
+func (e *Executor) setupCompiler() error {
+	if e.UserWorkingDir == "" {
+		var err error
+		e.UserWorkingDir, err = os.Getwd()
+		if err != nil {
+			return err
+		}
+	}
+
+	e.Compiler = &Compiler{
+		Dir:            e.Dir,
+		Entrypoint:     e.Entrypoint,
+		UserWorkingDir: e.UserWorkingDir,
+		TaskfileEnv:    e.Taskfile.Env,
+		TaskfileVars:   e.Taskfile.Vars,
+		Logger:         e.Logger,
+	}
+	return nil
+}
+
+func (e *Executor) readDotEnvFiles() error {
+	if e.Taskfile == nil || len(e.Taskfile.Dotenv) == 0 {
+		return nil
+	}
+
+	if e.Taskfile.Version.LessThan(ast.V3) {
+		return nil
+	}
+
+	vars, err := e.Compiler.GetTaskfileVariables()
+	if err != nil {
+		return err
+	}
+
+	env, err := taskfile.Dotenv(vars, e.Taskfile, e.Dir)
+	if err != nil {
+		return err
+	}
+
+	for k, v := range env.All() {
+		if _, ok := e.Taskfile.Env.Get(k); !ok {
+			e.Taskfile.Env.Set(k, v)
+		}
+	}
+	return err
+}
+
+func (e *Executor) setupDefaults() {
+	if e.Taskfile.Method == "" {
+		e.Taskfile.Method = "checksum"
+	}
+	if e.Taskfile.Run == "" {
+		e.Taskfile.Run = "always"
+	}
+}
+
+func (e *Executor) setupConcurrencyState() {
+	e.executionHashes = make(map[string]context.Context)
+
+	e.taskCallCount = make(map[string]*int32, e.Taskfile.Tasks.Len())
+	e.mkdirMutexMap = make(map[string]*sync.Mutex, e.Taskfile.Tasks.Len())
+	for k := range e.Taskfile.Tasks.Keys(nil) {
+		e.taskCallCount[k] = new(int32)
+		e.mkdirMutexMap[k] = &sync.Mutex{}
+	}
+
+	if e.Concurrency > 0 {
+		e.concurrencySemaphore = make(chan struct{}, e.Concurrency)
+	}
+}
+
+func (e *Executor) doVersionChecks() error {
+	if !e.EnableVersionCheck {
+		return nil
+	}
+	// Copy the version to avoid modifying the original
+	schemaVersion := &semver.Version{}
+	*schemaVersion = *e.Taskfile.Version
+
+	// Error if the Taskfile uses a schema version below v3
+	if schemaVersion.LessThan(ast.V3) {
+		return &errors.TaskfileVersionCheckError{
+			URI:           e.Taskfile.Location,
+			SchemaVersion: schemaVersion,
+			Message:       `no longer supported. Please use v3 or above`,
+		}
+	}
+
+	// Get the current version of Task
+	// If we can't parse the version (e.g. when its "devel"), then ignore the current version checks
+	currentVersion, err := semver.NewVersion(version.GetVersion())
+	if err != nil {
+		return nil
+	}
+
+	// Error if the Taskfile uses a schema version above the current version of Task
+	if schemaVersion.GreaterThan(currentVersion) {
+		return &errors.TaskfileVersionCheckError{
+			URI:           e.Taskfile.Location,
+			SchemaVersion: schemaVersion,
+			Message:       fmt.Sprintf(`is greater than the current version of Task (%s)`, currentVersion.String()),
+		}
+	}
+
+	return nil
+}