codeci.go

package main 

import (
	"fmt"
	"io/ioutil"
	"gopkg.in/yaml.v2"
	"strings"
	"os/exec"
    "os"
)

const version = "0.2.1"

func check(e error){
	if e != nil {
		panic(e)
	}
}

type CodeCi struct {
	Os string `yaml:"os"`
	Language string `yaml:"language"`
    Image string `yaml:"image"`
	Script []string `yaml:"script"`
}

func dockerfileName() string {
    return "Dockerfile.codeci"
}

func dockercomposeName() string {
    return "docker-compose.codeci.yml"
}

func onlytestName() string {
    return "onlytest.codeci.sh"
}

func testName() string {
    return "test.codeci.sh"
}

func createTestScript(codeci CodeCi) string {
    jobInfo := []string{"echo 'Job Node Info: '", "echo \n", "echo 'uname -a'", "uname -a", "echo \n", "echo 'df -h'", "df -h", "echo \n", "echo 'free -m'", "free -m", "echo \n", "echo 'bash --version'", "bash --version", "echo \n", "echo 'lscpu'", "lscpu", "echo \n", "echo 'lsb_release -a'", "lsb_release -a", "echo \n", "echo 'service --status-all'", "service --status-all", "echo \n", "echo 'dpkg -l'", "dpkg -l", "echo \n", "echo \n"}
    s := []string{"#!/bin/bash", "\n", "\n", strings.Join(jobInfo, "\n") , "\n", "echo 'running your commands: '", "\n", "sleep 1", "\n","set -ex", "\n", strings.Join(codeci.Script, "\n"), "\n"}
    return strings.Join(s, "")
}

func createDockerFile(codeci CodeCi) string{
    if codeci.Image != "" {
        s := []string{"FROM ", codeci.Image, "\n", "ADD . /app\nWORKDIR /app\nCMD [\"bash\", \"", testName(),"\"]", "\n"}
        return strings.Join(s, "")
    }
    if codeci.Language == "none" {
        s := []string{"FROM therickys93/", codeci.Os, "\n", "ADD . /app\nWORKDIR /app\nCMD [\"bash\", \"", testName(),"\"]", "\n"}
        return strings.Join(s, "")
    } else {
        s := []string{"FROM therickys93/", codeci.Os, codeci.Language, "\n", "ADD . /app\nWORKDIR /app\nCMD [\"bash\", \"", testName(),"\"]", "\n"}
        return strings.Join(s, "")
    }
} 

func codeCIWhalesay() string {
    return "image: docker/whalesay\nscript:\n   - cowsay Hello CodeCI!"
}

func main() {
    filename := "codeci.yml"
    data := []byte{}
    if len(os.Args) > 1 {
        if os.Args[1] == "--version" {
            fmt.Printf("%s version: %s\n", os.Args[0], version)
            os.Exit(0)
        } else if os.Args[1] == "--help" {
            fmt.Printf("usage: %s --> runs the build and search for the codeci.yml\n", os.Args[0])
            fmt.Printf("usage: %s --version --> show the current version\n", os.Args[0])
            fmt.Printf("usage: %s -f codeci.whateveryouwant.yml --> specify the name of your codeci file\n", os.Args[0])
            fmt.Printf("usage: %s test --> test the installation\n", os.Args[0])
            os.Exit(0)
        } else if os.Args[1] == "-f" {
            if strings.HasPrefix(os.Args[2], "codeci") && strings.HasSuffix(os.Args[2], ".yml") {
                filename = os.Args[2]
            } else {
                os.Exit(1)
            }
        } else if os.Args[1] == "test" {
            data = []byte(codeCIWhalesay())
        } else {
            os.Exit(1)
        }
    }
    filenames := []string{"./", filename}
    var err error
    if strings.EqualFold(string(data), "") {
        data, err = ioutil.ReadFile(strings.Join(filenames, ""))
        check(err)
    }
    fmt.Printf("reading the provided codeci.yml file...\n\n")
	fmt.Print(string(data))
    fmt.Printf("\n\n")
	var codeci CodeCi

	err = yaml.Unmarshal([]byte(string(data)), &codeci)
	check(err)

    fmt.Printf("Creating temp files...\n")
	// create the test.sh file
	d1 := []byte(createTestScript(codeci))
    err = ioutil.WriteFile(testName(), d1, 0644)
    check(err)

    // create the Dockerfile
    d1 = []byte(createDockerFile(codeci))
    err = ioutil.WriteFile(dockerfileName(), d1, 0644) 
    check(err)

    // create the docker-compose.yml file
    s := []string{"sut:\n", "  build: .\n", "  dockerfile: ", dockerfileName(), "\n"}
    d1 = []byte(strings.Join(s, ""))
    err = ioutil.WriteFile(dockercomposeName(), d1, 0644)
    check(err)

    // create the onlytest.sh file
    s = []string{"#!/bin/bash", "\n", "\n", 
        "docker-compose -f ", dockercomposeName() ," -p ci build", "\n", 
        "echo running the script...", "\n", 
        "echo -e '\n'", "\n", 
        "docker-compose -f ", dockercomposeName(), " -p ci up -d", "\n", 
        "docker logs -f ci_sut_1", "\n", 
        "echo -e '\n'", "\n",
        "echo 'BUILD EXIT CODE:'",  "\n", 
        "docker wait ci_sut_1", "\n", 
        "EXIT_STATUS=$(docker wait ci_sut_1)", "\n",
        "if [ $EXIT_STATUS == 0 ]; then echo -e '\nBUILD SUCCESS\n'; else echo -e '\nBUILD FAILED\n'; fi", "\n",
        "docker-compose -f ", dockercomposeName(), " -p ci kill", "\n", 
        "docker rm ci_sut_1", "\n", 
        "docker rmi ci_sut", "\n",
        "exit $EXIT_STATUS"}
    d1 = []byte(strings.Join(s, ""))
    err = ioutil.WriteFile(onlytestName(), d1, 0644)
    check(err)

    // run the script onlytest.sh
    fmt.Print("run the build...\n")
    cmd := exec.Command("/bin/bash", onlytestName())
    cmd.Stdout = os.Stdout
    cmd.Stderr = os.Stderr
    err = cmd.Start()
    exitstatus := 0
    if err != nil {
        exitstatus = 1
    }
    err = cmd.Wait()
    if err != nil {
        exitstatus = 2
    }

    // remove all the files
    fmt.Print("removing the temp files...\n")
    // remove test file
    os.Remove(testName())
    // remove Dockerfile
    os.Remove(dockerfileName())
    // remove only test file
    os.Remove(onlytestName())
    // remove docker compose file
    os.Remove(dockercomposeName())

    fmt.Print("done!\n")

    os.Exit(exitstatus)
}