1
0
mirror of https://github.com/go-task/task.git synced 2025-11-25 22:32:55 +02:00

Merge branch 'stdout-redir'

This commit is contained in:
Andrey Nering
2017-07-01 15:33:44 -03:00
9 changed files with 94 additions and 54 deletions

View File

@@ -3,6 +3,7 @@ package main
import ( import (
"fmt" "fmt"
"log" "log"
"os"
"github.com/go-task/task" "github.com/go-task/task"
@@ -52,7 +53,11 @@ hello:
} }
if init { if init {
if err := task.InitTaskfile(); err != nil { wd, err := os.Getwd()
if err != nil {
log.Fatal(err)
}
if err := task.InitTaskfile(wd); err != nil {
log.Fatal(err) log.Fatal(err)
} }
return return
@@ -61,6 +66,10 @@ hello:
e := task.Executor{ e := task.Executor{
Force: force, Force: force,
Watch: watch, Watch: watch,
Stdin: os.Stdin,
Stdout: os.Stdout,
Stderr: os.Stderr,
} }
if err := e.ReadTaskfile(); err != nil { if err := e.ReadTaskfile(); err != nil {
log.Fatal(err) log.Fatal(err)

View File

@@ -2,7 +2,6 @@ package task
import ( import (
"fmt" "fmt"
"os"
"sort" "sort"
"text/tabwriter" "text/tabwriter"
) )
@@ -12,10 +11,10 @@ func (e *Executor) printExistingTasksHelp() {
if len(tasks) == 0 { if len(tasks) == 0 {
return return
} }
fmt.Println("Available tasks for this project:") e.println("Available tasks for this project:")
// Format in tab-separated columns with a tab stop of 8. // Format in tab-separated columns with a tab stop of 8.
w := tabwriter.NewWriter(os.Stdout, 0, 8, 0, '\t', 0) w := tabwriter.NewWriter(e.Stdout, 0, 8, 0, '\t', 0)
for _, task := range tasks { for _, task := range tasks {
fmt.Fprintln(w, fmt.Sprintf("- %s:\t%s", task, e.Tasks[task].Desc)) fmt.Fprintln(w, fmt.Sprintf("- %s:\t%s", task, e.Tasks[task].Desc))
} }

View File

@@ -4,6 +4,7 @@ import (
"io/ioutil" "io/ioutil"
"log" "log"
"os" "os"
"path/filepath"
) )
const defaultTaskfile = `# github.com/go-task/task const defaultTaskfile = `# github.com/go-task/task
@@ -14,14 +15,16 @@ default:
` `
// InitTaskfile Taskfile creates a new Taskfile // InitTaskfile Taskfile creates a new Taskfile
func InitTaskfile() error { func InitTaskfile(path string) error {
for _, f := range []string{"Taskfile.yml", "Taskfile.toml", "Taskfile.json"} { for _, f := range []string{"Taskfile.yml", "Taskfile.toml", "Taskfile.json"} {
f = filepath.Join(path, f)
if _, err := os.Stat(f); err == nil { if _, err := os.Stat(f); err == nil {
return ErrTaskfileAlreadyExists return ErrTaskfileAlreadyExists
} }
} }
if err := ioutil.WriteFile("Taskfile.yml", []byte(defaultTaskfile), 0666); err != nil { f := filepath.Join(path, "Taskfile.yml")
if err := ioutil.WriteFile(f, []byte(defaultTaskfile), 0666); err != nil {
return err return err
} }
log.Printf("Taskfile.yml created in the current directory") log.Printf("Taskfile.yml created in the current directory")

13
log.go Normal file
View File

@@ -0,0 +1,13 @@
package task
import (
"fmt"
)
func (e *Executor) println(args ...interface{}) {
fmt.Fprintln(e.Stdout, args...)
}
func (e *Executor) printfln(format string, args ...interface{}) {
fmt.Fprintf(e.Stdout, format+"\n", args...)
}

26
task.go
View File

@@ -4,7 +4,7 @@ import (
"bytes" "bytes"
"context" "context"
"fmt" "fmt"
"log" "io"
"os" "os"
"path/filepath" "path/filepath"
"strings" "strings"
@@ -26,6 +26,10 @@ type Executor struct {
Force bool Force bool
Watch bool Watch bool
Stdin io.Reader
Stdout io.Writer
Stderr io.Writer
watchingFiles map[string]struct{} watchingFiles map[string]struct{}
} }
@@ -52,6 +56,16 @@ func (e *Executor) Run(args ...string) error {
return ErrCyclicDependencyDetected return ErrCyclicDependencyDetected
} }
if e.Stdin == nil {
e.Stdin = os.Stdin
}
if e.Stdout == nil {
e.Stdout = os.Stdout
}
if e.Stderr == nil {
e.Stderr = os.Stderr
}
// check if given tasks exist // check if given tasks exist
for _, a := range args { for _, a := range args {
if _, ok := e.Tasks[a]; !ok { if _, ok := e.Tasks[a]; !ok {
@@ -93,7 +107,7 @@ func (e *Executor) RunTask(ctx context.Context, name string) error {
return err return err
} }
if upToDate { if upToDate {
log.Printf(`task: Task "%s" is up to date`, name) e.printfln(`task: Task "%s" is up to date`, name)
return nil return nil
} }
} }
@@ -231,13 +245,13 @@ func (e *Executor) runCommand(ctx context.Context, task string, i int) error {
Command: c, Command: c,
Dir: dir, Dir: dir,
Env: envs, Env: envs,
Stdin: os.Stdin, Stdin: e.Stdin,
Stderr: os.Stderr, Stderr: e.Stderr,
} }
if t.Set == "" { if t.Set == "" {
log.Println(c) e.println(c)
opts.Stdout = os.Stdout opts.Stdout = e.Stdout
if err = execext.RunCommand(opts); err != nil { if err = execext.RunCommand(opts); err != nil {
return err return err
} }

View File

@@ -4,7 +4,6 @@ import (
"bytes" "bytes"
"io/ioutil" "io/ioutil"
"os" "os"
"os/exec"
"path/filepath" "path/filepath"
"strings" "strings"
"testing" "testing"
@@ -37,7 +36,9 @@ func TestDeps(t *testing.T) {
} }
e := &task.Executor{ e := &task.Executor{
Dir: dir, Dir: dir,
Stdout: ioutil.Discard,
Stderr: ioutil.Discard,
} }
assert.NoError(t, e.ReadTaskfile()) assert.NoError(t, e.ReadTaskfile())
assert.NoError(t, e.Run("default")) assert.NoError(t, e.Run("default"))
@@ -69,7 +70,9 @@ func TestVars(t *testing.T) {
} }
e := &task.Executor{ e := &task.Executor{
Dir: dir, Dir: dir,
Stdout: ioutil.Discard,
Stderr: ioutil.Discard,
} }
assert.NoError(t, e.ReadTaskfile()) assert.NoError(t, e.ReadTaskfile())
assert.NoError(t, e.Run("default")) assert.NoError(t, e.Run("default"))
@@ -101,7 +104,9 @@ func TestTaskCall(t *testing.T) {
} }
e := &task.Executor{ e := &task.Executor{
Dir: dir, Dir: dir,
Stdout: ioutil.Discard,
Stderr: ioutil.Discard,
} }
assert.NoError(t, e.ReadTaskfile()) assert.NoError(t, e.ReadTaskfile())
assert.NoError(t, e.Run("default")) assert.NoError(t, e.Run("default"))
@@ -122,23 +127,23 @@ func TestStatus(t *testing.T) {
if _, err := os.Stat(file); err == nil { if _, err := os.Stat(file); err == nil {
t.Errorf("File should not exists: %v", err) t.Errorf("File should not exists: %v", err)
} }
c := exec.Command("task", "gen-foo")
c.Dir = dir e := &task.Executor{
if err := c.Run(); err != nil { Dir: dir,
t.Error(err) Stdout: ioutil.Discard,
Stderr: ioutil.Discard,
} }
assert.NoError(t, e.ReadTaskfile())
assert.NoError(t, e.Run("gen-foo"))
if _, err := os.Stat(file); err != nil { if _, err := os.Stat(file); err != nil {
t.Errorf("File should exists: %v", err) t.Errorf("File should exists: %v", err)
} }
buff := bytes.NewBuffer(nil) buff := bytes.NewBuffer(nil)
c = exec.Command("task", "gen-foo") e.Stdout, e.Stderr = buff, buff
c.Dir = dir assert.NoError(t, e.Run("gen-foo"))
c.Stderr = buff
c.Stdout = buff
if err := c.Run(); err != nil {
t.Error(err)
}
if buff.String() != `task: Task "gen-foo" is up to date`+"\n" { if buff.String() != `task: Task "gen-foo" is up to date`+"\n" {
t.Errorf("Wrong output message: %s", buff.String()) t.Errorf("Wrong output message: %s", buff.String())
} }
@@ -153,11 +158,10 @@ func TestInit(t *testing.T) {
t.Errorf("Taskfile.yml should not exists") t.Errorf("Taskfile.yml should not exists")
} }
c := exec.Command("task", "--init") if err := task.InitTaskfile(dir); err != nil {
c.Dir = dir
if err := c.Run(); err != nil {
t.Error(err) t.Error(err)
} }
if _, err := os.Stat(file); err != nil { if _, err := os.Stat(file); err != nil {
t.Errorf("Taskfile.yml should exists") t.Errorf("Taskfile.yml should exists")
} }

View File

@@ -36,7 +36,7 @@ func (e *Executor) handleDynamicVariableContent(value string) (string, error) {
Command: strings.TrimPrefix(value, "$"), Command: strings.TrimPrefix(value, "$"),
Dir: e.Dir, Dir: e.Dir,
Stdout: buff, Stdout: buff,
Stderr: os.Stderr, Stderr: e.Stderr,
} }
if err := execext.RunCommand(opts); err != nil { if err := execext.RunCommand(opts); err != nil {
return "", err return "", err

View File

@@ -58,19 +58,19 @@ func (r *Runner) binTest(op syntax.BinTestOperator, x, y string) bool {
} }
return re.MatchString(x) return re.MatchString(x)
case syntax.TsNewer: case syntax.TsNewer:
i1, i2 := stat(x), stat(y) i1, i2 := stat(r.Dir, x), stat(r.Dir, y)
if i1 == nil || i2 == nil { if i1 == nil || i2 == nil {
return false return false
} }
return i1.ModTime().After(i2.ModTime()) return i1.ModTime().After(i2.ModTime())
case syntax.TsOlder: case syntax.TsOlder:
i1, i2 := stat(x), stat(y) i1, i2 := stat(r.Dir, x), stat(r.Dir, y)
if i1 == nil || i2 == nil { if i1 == nil || i2 == nil {
return false return false
} }
return i1.ModTime().Before(i2.ModTime()) return i1.ModTime().Before(i2.ModTime())
case syntax.TsDevIno: case syntax.TsDevIno:
i1, i2 := stat(x), stat(y) i1, i2 := stat(r.Dir, x), stat(r.Dir, y)
return os.SameFile(i1, i2) return os.SameFile(i1, i2)
case syntax.TsEql: case syntax.TsEql:
return atoi(x) == atoi(y) return atoi(x) == atoi(y)
@@ -95,40 +95,40 @@ func (r *Runner) binTest(op syntax.BinTestOperator, x, y string) bool {
} }
} }
func stat(name string) os.FileInfo { func stat(dir, name string) os.FileInfo {
info, _ := os.Stat(name) info, _ := os.Stat(filepath.Join(dir, name))
return info return info
} }
func statMode(name string, mode os.FileMode) bool { func statMode(dir, name string, mode os.FileMode) bool {
info := stat(name) info := stat(dir, name)
return info != nil && info.Mode()&mode != 0 return info != nil && info.Mode()&mode != 0
} }
func (r *Runner) unTest(op syntax.UnTestOperator, x string) bool { func (r *Runner) unTest(op syntax.UnTestOperator, x string) bool {
switch op { switch op {
case syntax.TsExists: case syntax.TsExists:
return stat(x) != nil return stat(r.Dir, x) != nil
case syntax.TsRegFile: case syntax.TsRegFile:
info := stat(x) info := stat(r.Dir, x)
return info != nil && info.Mode().IsRegular() return info != nil && info.Mode().IsRegular()
case syntax.TsDirect: case syntax.TsDirect:
return statMode(x, os.ModeDir) return statMode(r.Dir, x, os.ModeDir)
//case syntax.TsCharSp: //case syntax.TsCharSp:
//case syntax.TsBlckSp: //case syntax.TsBlckSp:
case syntax.TsNmPipe: case syntax.TsNmPipe:
return statMode(x, os.ModeNamedPipe) return statMode(r.Dir, x, os.ModeNamedPipe)
case syntax.TsSocket: case syntax.TsSocket:
return statMode(x, os.ModeSocket) return statMode(r.Dir, x, os.ModeSocket)
case syntax.TsSmbLink: case syntax.TsSmbLink:
info, _ := os.Lstat(x) info, _ := os.Lstat(x)
return info != nil && info.Mode()&os.ModeSymlink != 0 return info != nil && info.Mode()&os.ModeSymlink != 0
case syntax.TsSticky: case syntax.TsSticky:
return statMode(x, os.ModeSticky) return statMode(r.Dir, x, os.ModeSticky)
case syntax.TsUIDSet: case syntax.TsUIDSet:
return statMode(x, os.ModeSetuid) return statMode(r.Dir, x, os.ModeSetuid)
case syntax.TsGIDSet: case syntax.TsGIDSet:
return statMode(x, os.ModeSetgid) return statMode(r.Dir, x, os.ModeSetgid)
//case syntax.TsGrpOwn: //case syntax.TsGrpOwn:
//case syntax.TsUsrOwn: //case syntax.TsUsrOwn:
//case syntax.TsModif: //case syntax.TsModif:
@@ -149,7 +149,7 @@ func (r *Runner) unTest(op syntax.UnTestOperator, x string) bool {
_, err := exec.LookPath(filepath.Join(r.Dir, x)) _, err := exec.LookPath(filepath.Join(r.Dir, x))
return err == nil return err == nil
case syntax.TsNoEmpty: case syntax.TsNoEmpty:
info := stat(x) info := stat(r.Dir, x)
return info != nil && info.Size() > 0 return info != nil && info.Size() > 0
//case syntax.TsFdTerm: //case syntax.TsFdTerm:
case syntax.TsEmpStr: case syntax.TsEmpStr:

View File

@@ -2,8 +2,6 @@ package task
import ( import (
"context" "context"
"fmt"
"log"
"strings" "strings"
"time" "time"
@@ -13,12 +11,12 @@ import (
// watchTasks start watching the given tasks // watchTasks start watching the given tasks
func (e *Executor) watchTasks(args ...string) error { func (e *Executor) watchTasks(args ...string) error {
log.Printf("task: Started watching for tasks: %s", strings.Join(args, ", ")) e.printfln("task: Started watching for tasks: %s", strings.Join(args, ", "))
// run tasks on init // run tasks on init
for _, a := range args { for _, a := range args {
if err := e.RunTask(context.Background(), a); err != nil { if err := e.RunTask(context.Background(), a); err != nil {
fmt.Println(err) e.println(err)
break break
} }
} }
@@ -32,7 +30,7 @@ func (e *Executor) watchTasks(args ...string) error {
go func() { go func() {
for { for {
if err := e.registerWatchedFiles(watcher, args); err != nil { if err := e.registerWatchedFiles(watcher, args); err != nil {
log.Printf("Error watching files: %v", err) e.printfln("Error watching files: %v", err)
} }
time.Sleep(time.Second * 2) time.Sleep(time.Second * 2)
} }
@@ -44,12 +42,12 @@ loop:
case <-watcher.Events: case <-watcher.Events:
for _, a := range args { for _, a := range args {
if err := e.RunTask(context.Background(), a); err != nil { if err := e.RunTask(context.Background(), a); err != nil {
fmt.Println(err) e.println(err)
continue loop continue loop
} }
} }
case err := <-watcher.Errors: case err := <-watcher.Errors:
fmt.Println(err) e.println(err)
continue loop continue loop
} }
} }