coverage.go

package formatters

import (
	"bytes"
	"encoding/json"
	"strings"

	"github.com/pkg/errors"
)

type Coverage []NullInt

// MarshalJSON marshals the coverage into JSON. Since the Code Climate
// API requires this as a string "[1,2,null]" and not just a straight
// JSON array we have to do a bunch of work to coerce into that format
func (c Coverage) MarshalJSON() ([]byte, error) {
	cc := make([]interface{}, 0, len(c))
	for _, x := range c {
		cc = append(cc, x)
	}
	bb := &bytes.Buffer{}
	err := json.NewEncoder(bb).Encode(cc)
	if err != nil {
		return bb.Bytes(), err
	}
	b, err := json.Marshal(strings.TrimSpace(bb.String()))

	if err != nil {
		return b, errors.WithStack(err)
	}

	return b, nil
}

var cwp = []byte("\"")
var cws = []byte("\"")

func (c *Coverage) UnmarshalJSON(text []byte) error {
	text = bytes.TrimPrefix(text, cwp)
	text = bytes.TrimSuffix(text, cws)
	cc := make([]NullInt, 0, 1024)

	err := json.Unmarshal(text, &cc)
	if err != nil {
		return err
	}
	*c = cc
	return nil
}

func (c *Coverage) AppendNulls(quantity int) Coverage {
	newCoverage := *c
	for i := 0; i < quantity; i++ {
		newCoverage = append(newCoverage, NullInt{})
	}
	return newCoverage
}