1
0
mirror of https://github.com/go-task/task.git synced 2025-04-23 12:18:57 +02:00

refactor: onyl read Taskvars file once

This commit is contained in:
Andrey Nering 2017-07-03 21:16:10 -03:00
parent 196d3cb13d
commit b8fe8d465e
3 changed files with 30 additions and 34 deletions

View File

@ -26,7 +26,6 @@ func (e *Executor) ReadTaskfile() error {
if err != nil { if err != nil {
switch err.(type) { switch err.(type) {
case taskFileNotFound: case taskFileNotFound:
return nil
default: default:
return err return err
} }
@ -34,6 +33,9 @@ func (e *Executor) ReadTaskfile() error {
if err := mergo.MapWithOverwrite(&e.Tasks, osTasks); err != nil { if err := mergo.MapWithOverwrite(&e.Tasks, osTasks); err != nil {
return err return err
} }
if err := e.readTaskvarsFile(); err != nil {
return err
}
return nil return nil
} }
@ -49,3 +51,27 @@ func (e *Executor) readTaskfileData(path string) (tasks map[string]*Task, err er
} }
return nil, taskFileNotFound{path} return nil, taskFileNotFound{path}
} }
func (e *Executor) readTaskvarsFile() error {
file := filepath.Join(e.Dir, TaskvarsFilePath)
if b, err := ioutil.ReadFile(file + ".yml"); err == nil {
if err := yaml.Unmarshal(b, &e.taskvars); err != nil {
return err
}
return nil
}
if b, err := ioutil.ReadFile(file + ".json"); err == nil {
if err := json.Unmarshal(b, &e.taskvars); err != nil {
return err
}
return nil
}
if b, err := ioutil.ReadFile(file + ".toml"); err == nil {
if err := toml.Unmarshal(b, &e.taskvars); err != nil {
return err
}
return nil
}
return nil
}

View File

@ -30,6 +30,7 @@ type Executor struct {
Stdout io.Writer Stdout io.Writer
Stderr io.Writer Stderr io.Writer
taskvars Params
watchingFiles map[string]struct{} watchingFiles map[string]struct{}
} }

View File

@ -2,9 +2,7 @@ package task
import ( import (
"bytes" "bytes"
"encoding/json"
"errors" "errors"
"io/ioutil"
"os" "os"
"path/filepath" "path/filepath"
"runtime" "runtime"
@ -13,9 +11,7 @@ import (
"github.com/go-task/task/execext" "github.com/go-task/task/execext"
"github.com/BurntSushi/toml"
"github.com/Masterminds/sprig" "github.com/Masterminds/sprig"
"gopkg.in/yaml.v2"
) )
var ( var (
@ -63,16 +59,14 @@ func (e *Executor) getVariables(task string, params Params) (map[string]string,
} }
localVariables[key] = val localVariables[key] = val
} }
if fileVariables, err := e.readTaskvarsFile(); err == nil { if e.taskvars != nil {
for key, value := range fileVariables { for key, value := range e.taskvars {
val, err := e.handleDynamicVariableContent(value) val, err := e.handleDynamicVariableContent(value)
if err != nil { if err != nil {
return nil, err return nil, err
} }
localVariables[key] = val localVariables[key] = val
} }
} else {
return nil, err
} }
for key, value := range getEnvironmentVariables() { for key, value := range getEnvironmentVariables() {
localVariables[key] = value localVariables[key] = value
@ -163,28 +157,3 @@ func getEnvironmentVariables() map[string]string {
} }
return m return m
} }
func (e *Executor) readTaskvarsFile() (map[string]string, error) {
file := filepath.Join(e.Dir, TaskvarsFilePath)
var variables map[string]string
if b, err := ioutil.ReadFile(file + ".yml"); err == nil {
if err := yaml.Unmarshal(b, &variables); err != nil {
return nil, err
}
return variables, nil
}
if b, err := ioutil.ReadFile(file + ".json"); err == nil {
if err := json.Unmarshal(b, &variables); err != nil {
return nil, err
}
return variables, nil
}
if b, err := ioutil.ReadFile(file + ".toml"); err == nil {
if err := toml.Unmarshal(b, &variables); err != nil {
return nil, err
}
return variables, nil
}
return variables, nil
}