travis-exec.go

package main

import (
	"os"
	"io/ioutil"
	"github.com/smallfish/simpleyaml"
	"fmt"
	"strings"
	"os/exec"
	"bufio"
)

func setEnv(envvar string) {
	fmt.Println("\n[+] Adding to environment variables: " + envvar)
	vars := strings.Split(envvar, " ")

	for i := range vars {
		currentVar := strings.Split(vars[i], "=")

		// If PATH => prefix
		if currentVar[0] == "PATH" {
			os.Setenv(currentVar[0], strings.Replace(currentVar[1], "\"", "", -1) + ":" + os.Getenv("PATH"))
		} else {
			os.Setenv(currentVar[0], currentVar[1])
		}
		//fmt.Println(os.Getenv(currentVar[0]))
	}
}

func runCmd(command string) {
	fmt.Println("\n[+] Running command: " + command)

	// Prepare cmd to run with args
	cmdArgs := strings.Fields(command)
	cmd := exec.Command(cmdArgs[0], cmdArgs[1:len(cmdArgs)]...)

	// Stream lines output to stdout
	stdout, _ := cmd.StdoutPipe()
	cmd.Start()

	scanner := bufio.NewScanner(stdout)
	scanner.Split(bufio.ScanLines)
	for scanner.Scan() {
		m := scanner.Text()
		fmt.Println(m)
	}
	cmd.Wait()
}

func runSection(yaml simpleyaml.Yaml, key string) {
	list, _ := yaml.Get(key).Array()
	for item := range list {
		current, _ := yaml.Get(key).GetIndex(item).String()
		runCmd(current)
	}
}

func checkTravisLifecycleExists(yaml *simpleyaml.Yaml, travisLifecycle map[string]bool, lifecycle string)  {
	if _, ok := travisLifecycle[lifecycle]; ok {
		runSection(*yaml, lifecycle)
	}
}

func main() {
	var travisField string
	var travisKeys []string
	travisLifecycle := map[string]bool {
		"before_install": true,
		"install": true,
		"before_script": true,
		"script": true,
		"after_success": true,
	}

	// Check args
	if len(os.Args) < 2 {
		fmt.Println("Usage: travis-exec <travis_file> <optional travis_field>")
		fmt.Println("Example: go run tests/k8s-euft/travis-exec.go .travis.yml install")
		os.Exit(1)
	}

	filename := os.Args[1]
	if len(os.Args) == 3 {
		travisField = os.Args[2]
	}

	source, err := ioutil.ReadFile(filename)
	if err != nil {
		panic(err)
	}

	// Loading Yaml config
	yaml, err := simpleyaml.NewYaml(source)
	if err != nil {
		panic(err)
	}

	// Run all tests (install/script...) for all envs or the specified one
	allEnv, _ := yaml.Get("env").Array()
	if len(travisField) == 0 {
		travisKeys, _ = yaml.GetMapKeys()
	} else {
		travisKeys = append(travisKeys, travisField)
	}

	// for each env, do:
	for currentIndex := range allEnv {
		envName, _ := yaml.Get("env").GetIndex(currentIndex).String()
		setEnv(envName)
		// run lifecycles in travis order
		checkTravisLifecycleExists(yaml, travisLifecycle, "before_install")
		checkTravisLifecycleExists(yaml, travisLifecycle, "install")
		checkTravisLifecycleExists(yaml, travisLifecycle, "before_script")
		checkTravisLifecycleExists(yaml, travisLifecycle, "script")
		checkTravisLifecycleExists(yaml, travisLifecycle, "after_success")
	}
}