pedroMMM/goss

package outputs

import (
    "encoding/json"
    "fmt"
    "io"
    "time"

    "github.com/aelsabbahy/goss/resource"
    "github.com/aelsabbahy/goss/util"
    "github.com/fatih/color"
)

type Json struct{}

func (r Json) Output(w io.Writer, results <-chan []resource.TestResult,
    startTime time.Time, outConfig util.OutputConfig) (exitCode int) {

    var pretty bool
    pretty = util.IsValueInList("pretty", outConfig.FormatOptions)

    color.NoColor = true
    testCount := 0
    failed := 0
    var resultsOut []map[string]interface{}
    for resultGroup := range results {
        for _, testResult := range resultGroup {
            if !testResult.Successful {
                failed++
            }
            m := struct2map(testResult)
            m["summary-line"] = humanizeResult(testResult)
            m["duration"] = int64(m["duration"].(float64))
            resultsOut = append(resultsOut, m)
            testCount++
        }
    }

    summary := make(map[string]interface{})
    duration := time.Since(startTime)
    summary["test-count"] = testCount
    summary["failed-count"] = failed
    summary["total-duration"] = duration
    summary["summary-line"] = fmt.Sprintf("Count: %d, Failed: %d, Duration: %.3fs", testCount, failed, duration.Seconds())

    out := make(map[string]interface{})
    out["results"] = resultsOut
    out["summary"] = summary

    var j []byte
    if pretty {
        j, _ = json.MarshalIndent(out, "", "    ")
    } else {
        j, _ = json.Marshal(out)
    }

    fmt.Fprintln(w, string(j))

    if failed > 0 {
        return 1
    }

    return 0
}

func init() {
    RegisterOutputer("json", &Json{}, []string{"pretty"})
}

func struct2map(i interface{}) map[string]interface{} {
    out := make(map[string]interface{})
    j, _ := json.Marshal(i)
    json.Unmarshal(j, &out)
    return out
}