Implemented code coverage support.

Author: sflanker

go.mod

@@ -14,6 +14,7 @@ require (
 )
 
 require (
+	github.com/easy-up/go-coverage v0.0.0-20241018034313-3de592d59a78 // indirect
 	github.com/fsnotify/fsnotify v1.7.0 // indirect
 	github.com/hashicorp/hcl v1.0.0 // indirect
 	github.com/inconshreveable/mousetrap v1.1.0 // indirect

go.sum

@@ -5,6 +5,8 @@ github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc h1:U9qPSI2PIWSS1
 github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
 github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
+github.com/easy-up/go-coverage v0.0.0-20241018034313-3de592d59a78 h1:e2x+TfIgebN3zfr8wGqAYI9lK4ql7Rut6OTEhBmJr5k=
+github.com/easy-up/go-coverage v0.0.0-20241018034313-3de592d59a78/go.mod h1:fsSINOc273zPnsBaKNjNffZXZpicAArpv/cTiFYgPys=
 github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8=
 github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0=
 github.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA=

pkg/archive/bundle.go

@@ -10,6 +10,7 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"github.com/olekukonko/tablewriter"
 	"io"
 	"log/slog"
 	"os"
@@ -159,7 +160,10 @@ func (b *Bundle) Content() string {
 	sort.Sort(matrix)
 	buf := new(bytes.Buffer)
 	header := []string{"Label", "Digest", "Tags", "Size"}
-	matrix.Table(buf, header).Render()
+	table := tablewriter.NewWriter(buf)
+	table.SetHeader(header)
+	matrix.Table(table)
+	table.Render()
 	return buf.String()
 }
 

pkg/artifacts/lcov.go

@@ -0,0 +1,42 @@
+package artifacts
+
+import (
+	"errors"
+	"github.com/easy-up/go-coverage"
+	"log/slog"
+	"strings"
+)
+
+func example() (coverage.Report, error) {
+	lcovParser := coverage.New(coverage.LCOV)
+	report, err := lcovParser.Parse("./path/to/lcov.info")
+	if err != nil {
+		// Handle error
+		return coverage.Report{}, err
+	}
+	// Use the parsed report
+	return report, nil
+}
+
+func IsCoverageReport(inputFilename string) bool {
+	return strings.Contains(inputFilename, "lcov") ||
+		strings.HasSuffix(inputFilename, ".info") ||
+		strings.Contains(inputFilename, "clover") ||
+		strings.Contains(inputFilename, "cobertura") ||
+		strings.Contains(inputFilename, "coverage")
+}
+
+func GetCoverageMode(inputFilename string) (coverage.CoverageMode, error) {
+	var coverageFormat coverage.CoverageMode
+	if strings.Contains(inputFilename, "lcov") || strings.HasSuffix(inputFilename, ".info") {
+		coverageFormat = coverage.LCOV
+	} else if strings.Contains(inputFilename, "clover") {
+		coverageFormat = coverage.CLOVER
+	} else if strings.HasSuffix(inputFilename, ".xml") {
+		coverageFormat = coverage.COBERTURA
+	} else {
+		slog.Error("unsupported coverage file type, cannot be determined from filename", "filename", inputFilename)
+		return "", errors.New("failed to list coverage content")
+	}
+	return coverageFormat, nil
+}

pkg/format/matrix.go

@@ -1,8 +1,6 @@
 package format
 
 import (
-	"io"
-
 	"github.com/olekukonko/tablewriter"
 )
 
@@ -28,11 +26,8 @@ func (m *SortableMatrix) Matrix() [][]string {
 	return m.data
 }
 
-func (m *SortableMatrix) Table(w io.Writer, header []string) *tablewriter.Table {
-	table := tablewriter.NewWriter(w)
-	table.SetHeader(header)
+func (m *SortableMatrix) Table(table *tablewriter.Table) {
 	table.AppendBulk(m.data)
-	return table
 }
 
 func (m *SortableMatrix) Len() int {

pkg/gatecheck/config.go

@@ -28,6 +28,7 @@ type Config struct {
 	Cyclonedx reportWithCVEs       `json:"cyclonedx" toml:"cyclonedx" yaml:"cyclonedx"`
 	Semgrep   configSemgrepReport  `json:"semgrep"   toml:"semgrep"   yaml:"semgrep"`
 	Gitleaks  configGitleaksReport `json:"gitleaks"  toml:"gitleaks"  yaml:"gitleaks"`
+	Coverage  configCoverageReport `json:"coverage"  toml:"coverage"  yaml:"coverage"`
 }
 
 func (c *Config) String() string {
@@ -48,6 +49,12 @@ func (c *Config) String() string {
 	return contentBuf.String()
 }
 
+type configCoverageReport struct {
+	LineThreshold     float32 `json:"lineThreshold"   toml:"lineThreshold"   yaml:"lineThreshold"`
+	FunctionThreshold float32 `json:"functionThreshold" toml:"functionThreshold" yaml:"functionThreshold"`
+	BranchThreshold   float32 `json:"branchThreshold"  toml:"branchThreshold"  yaml:"branchThreshold"`
+}
+
 type configGitleaksReport struct {
 	LimitEnabled bool `json:"limitEnabled" toml:"limitEnabled" yaml:"limitEnabled"`
 }
@@ -225,6 +232,11 @@ func NewDefaultConfig() *Config {
 		Gitleaks: configGitleaksReport{
 			LimitEnabled: false,
 		},
+		Coverage: configCoverageReport{
+			LineThreshold:     0,
+			FunctionThreshold: 0,
+			BranchThreshold:   0,
+		},
 	}
 }