upstream-gitea/modules/git/repo_ref_nogogit.go

// Copyright 2020 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

//go:build !gogit

package git

import (
	"bufio"
	"context"
	"io"
	"strings"

	"gitea.dev/modules/git/gitcmd"
)

// GetRefsFiltered returns all references of the repository that matches patterm exactly or starting with.
func (repo *Repository) GetRefsFiltered(ctx context.Context, pattern string) ([]*Reference, error) {
	refs := make([]*Reference, 0)
	cmd := gitcmd.NewCommand("for-each-ref")
	stdoutReader, stdoutReaderClose := cmd.MakeStdoutPipe()
	defer stdoutReaderClose()
	err := cmd.WithRepo(repo).
		WithPipelineFunc(func(context gitcmd.Context) error {
			bufReader := bufio.NewReader(stdoutReader)
			for {
				// The output of for-each-ref is simply a list:
				// <sha> SP <type> TAB <ref> LF
				sha, err := bufReader.ReadString(' ')
				if err == io.EOF {
					break
				}
				if err != nil {
					return err
				}
				sha = sha[:len(sha)-1]

				typ, err := bufReader.ReadString('\t')
				if err == io.EOF {
					// This should not happen, but we'll tolerate it
					break
				}
				if err != nil {
					return err
				}
				typ = typ[:len(typ)-1]

				refName, err := bufReader.ReadString('\n')
				if err == io.EOF {
					// This should not happen, but we'll tolerate it
					break
				}
				if err != nil {
					return err
				}
				refName = refName[:len(refName)-1]

				// refName cannot be HEAD but can be remotes or stash
				if strings.HasPrefix(refName, RemotePrefix) || refName == "/refs/stash" {
					continue
				}

				if pattern == "" || strings.HasPrefix(refName, pattern) {
					r := &Reference{
						Name:   refName,
						Object: MustIDFromString(sha),
						Type:   typ,
						repo:   repo,
					}
					refs = append(refs, r)
				}
			}
			return nil
		}).RunWithStderr(ctx)
	return refs, err
}