Method NewReferenceIter

git/ref_iter.go:20–78

NewReferenceIter returns an iterator that iterates over all of the references in `repo`.

(ctx context.Context)

// NewReferenceIter returns an iterator that iterates over all of the
// references in `repo`.
func (repo *Repository) NewReferenceIter(ctx context.Context) (*ReferenceIter, error) {
	iter := ReferenceIter{
		refCh: make(chan Reference),
		errCh: make(chan error),
	}

	p := pipe.New()
	p.Add(
		// Output all references and their values:
		pipe.CommandStage(
			"git-for-each-ref",
			repo.GitCommand(
				"for-each-ref",
				"--format=%(objectname) %(objecttype) %(objectsize) %(refname)",
			),
		),

		// Read the references and send them to `iter.refCh`, then close
		// the channel.
		pipe.Function(
			"parse-refs",
			func(ctx context.Context, env pipe.Env, stdin io.Reader, stdout io.Writer) error {
				defer close(iter.refCh)

				in := bufio.NewReader(stdin)
				for {
					line, err := in.ReadBytes('\n')
					if err != nil {
						if err == io.EOF {
							return nil
						}
						return fmt.Errorf("reading 'git for-each-ref' output: %w", err)
					}

					ref, err := ParseReference(string(line[:len(line)-1]))
					if err != nil {
						return fmt.Errorf("parsing 'git for-each-ref' output: %w", err)
					}
					select {
					case iter.refCh <- ref:
					case <-ctx.Done():
						return ctx.Err()
					}
				}
			},
		),
	)

	err := p.Start(ctx)
	if err != nil {
		return nil, err
	}

	go func() {
		iter.errCh <- p.Wait()
	}()

	return &iter, nil