local.go

package main

import (
	"context"
	"fmt"
	"os/exec"
	"runtime"
	"time"
)

// LocalExecutor runs commands on the local machine.
type LocalExecutor struct {
	shell string
}

func NewLocalExecutor(shell string) *LocalExecutor {
	if shell == "" {
		if runtime.GOOS == "windows" {
			shell = "cmd"
		} else {
			shell = "bash"
		}
	}
	return &LocalExecutor{shell: shell}
}

func (e *LocalExecutor) Exec(command string, timeout int) (string, error) {
	var ctx context.Context
	var cancel context.CancelFunc

	if timeout > 0 {
		ctx, cancel = context.WithTimeout(context.Background(), time.Duration(timeout)*time.Second)
		defer cancel()
	} else {
		ctx = context.Background()
	}

	var cmd *exec.Cmd
	if e.shell == "cmd" {
		cmd = exec.CommandContext(ctx, "cmd", "/C", command)
	} else {
		cmd = exec.CommandContext(ctx, e.shell, "-c", command)
	}

	output, err := cmd.CombinedOutput()
	if ctx.Err() == context.DeadlineExceeded {
		return string(output), fmt.Errorf("command timed out after %d seconds", timeout)
	}
	if err != nil {
		return string(output) + "\nError: " + err.Error(), nil
	}

	return string(output), nil
}

func (e *LocalExecutor) Close() error {
	return nil
}