internal/git/gitcmd/decoder.go

package gitcmd

import (
	"bufio"
	"fmt"
	"io"
	"strings"

	"gitlab.com/gitlab-org/gitaly/v16/internal/git"
)

// ShowRefDecoder parses the output format of git-show-ref
type ShowRefDecoder struct {
	r   *bufio.Reader
	err error
}

// NewShowRefDecoder returns a new ShowRefDecoder
func NewShowRefDecoder(r io.Reader) *ShowRefDecoder {
	return &ShowRefDecoder{
		r: bufio.NewReader(r),
	}
}

// Decode reads and parses the next reference. Returns io.EOF when the end of
// the reader has been reached.
func (d *ShowRefDecoder) Decode(ref *git.Reference) error {
	if d.err != nil {
		return d.err
	}

	line, err := d.r.ReadString('\n')
	d.err = err

	if len(line) > 0 && line[len(line)-1] == '\n' {
		line = line[:len(line)-1]
	}

	splits := strings.SplitN(line, " ", 2)
	if len(splits) != 2 {
		if d.err != nil {
			return d.err
		}
		return fmt.Errorf("show-ref decoder: invalid line: %q", line)
	}

	*ref = git.NewReference(git.ReferenceName(splits[1]), git.ObjectID(splits[0]))

	return nil
}