utils.go

package main

import (
	"bytes"
	"html/template"
	"log"
	"os"
	"os/exec"
	"path/filepath"
	"strings"

	"github.com/fatih/color"
)

func findRunfiles() (runfiles []Runfile) {
	tests := []string{}
	dir, _ := filepath.Split(filename)
	parts := strings.Split(dir, string(filepath.Separator))
	for i := len(parts) - 1; i > 0; i-- {
		for _, extension := range []string{"yml", "yaml"} {
			p := append(parts[:i], "Runfile."+extension)
			tests = append(tests, string(filepath.Separator)+filepath.Join(p...))
		}
	}

	for _, test := range tests {
		if _, err := os.Stat(test); err == nil {
			runfiles = append(runfiles, RunfileConstructor(test))
		}
	}

	return
}

func execute(command string) error {
	tmpl, err := template.New("test").Parse(command)
	if err != nil {
		log.Fatal(err)
	}

	var result bytes.Buffer
	err = tmpl.Execute(&result, context)
	if err != nil {
		log.Fatal(err)
	}

	color.Green("#! %s\n", result.String())

	parts := cmdRe.FindAllString(result.String(), -1)
	cmd := exec.Command(parts[0], parts[1:]...)

	cmd.Stdout = os.Stdout
	cmd.Stderr = os.Stderr

	return cmd.Run()
}

func debugPrintln(a ...interface{}) {
	c := color.New(color.Faint)
	if debug {
		c.Println(a...)
	}
}

func debugPrint(a ...interface{}) {
	c := color.New(color.Faint)
	if debug {
		c.Print(a...)
	}
}

func debugPrintf(format string, a ...interface{}) {
	c := color.New(color.Faint)
	if debug {
		c.Printf(format, a...)
	}
}