1
0
mirror of https://github.com/go-task/task.git synced 2025-01-26 05:27:15 +02:00
task/internal/experiments/experiments.go
2023-12-20 19:55:25 -06:00

59 lines
1.2 KiB
Go

package experiments
import (
"fmt"
"io"
"os"
"strings"
"text/tabwriter"
"github.com/joho/godotenv"
"github.com/go-task/task/v3/internal/logger"
)
const envPrefix = "TASK_X_"
// A list of experiments.
var (
GentleForce bool
RemoteTaskfiles bool
AnyVariables bool
)
func init() {
readDotEnv()
GentleForce = parseEnv("GENTLE_FORCE")
RemoteTaskfiles = parseEnv("REMOTE_TASKFILES")
AnyVariables = parseEnv("ANY_VARIABLES")
}
func parseEnv(xName string) bool {
envName := fmt.Sprintf("%s%s", envPrefix, xName)
return os.Getenv(envName) == "1"
}
func readDotEnv() {
env, _ := godotenv.Read()
// If the env var is an experiment, set it.
for key, value := range env {
if strings.HasPrefix(key, envPrefix) {
os.Setenv(key, value)
}
}
}
func printExperiment(w io.Writer, l *logger.Logger, name string, value bool) {
l.FOutf(w, logger.Yellow, "* ")
l.FOutf(w, logger.Green, name)
l.FOutf(w, logger.Default, ": \t%t\n", value)
}
func List(l *logger.Logger) error {
w := tabwriter.NewWriter(os.Stdout, 0, 8, 0, ' ', 0)
printExperiment(w, l, "GENTLE_FORCE", GentleForce)
printExperiment(w, l, "REMOTE_TASKFILES", RemoteTaskfiles)
printExperiment(w, l, "ANY_VARIABLES", AnyVariables)
return w.Flush()
}