upstream-gitea/modules/git/catfile_batch_command.go

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

package git

import (
	"context"
	"strings"

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

// catFileBatchCommand implements the CatFileBatch interface using the "cat-file --batch-command" command
// for git version >= 2.36
// ref: https://git-scm.com/docs/git-cat-file#Documentation/git-cat-file.txt---batch-command
type catFileBatchCommand struct {
	ctx   context.Context
	repo  RepositoryFacade
	batch *catFileBatchCommunicator
}

var _ CatFileBatch = (*catFileBatchCommand)(nil)

func newCatFileBatchCommand(ctx context.Context, repo RepositoryFacade) *catFileBatchCommand {
	return &catFileBatchCommand{ctx: ctx, repo: repo}
}

func (b *catFileBatchCommand) getBatch(callerInfo string) *catFileBatchCommunicator {
	if b.batch != nil {
		return b.batch
	}
	b.batch = newCatFileBatch(b.ctx, b.repo, gitcmd.NewCommand("cat-file", "--batch-command").WithParentCallerInfo(callerInfo))
	return b.batch
}

func (b *catFileBatchCommand) Context() context.Context {
	return b.ctx
}

func (b *catFileBatchCommand) QueryContent(obj string) (*CatFileObject, BufferedReader, error) {
	if strings.Contains(obj, "\n") {
		setting.PanicInDevOrTesting("invalid object name with newline: %q", obj)
	}
	batch := b.getBatch(util.CallerFuncName(1))
	_, err := batch.reqWriter.Write([]byte("contents " + obj + "\n"))
	if err != nil {
		return nil, nil, err
	}
	info, err := catFileBatchParseInfoLine(batch.respReader)
	if err != nil {
		return nil, nil, err
	}
	return info, batch.respReader, nil
}

func (b *catFileBatchCommand) QueryInfo(obj string) (*CatFileObject, error) {
	if strings.Contains(obj, "\n") {
		setting.PanicInDevOrTesting("invalid object name with newline: %q", obj)
	}
	batch := b.getBatch(util.CallerFuncName(1))
	_, err := batch.reqWriter.Write([]byte("info " + obj + "\n"))
	if err != nil {
		return nil, err
	}
	return catFileBatchParseInfoLine(batch.respReader)
}

func (b *catFileBatchCommand) Close() {
	if b.batch != nil {
		b.batch.Close()
		b.batch = nil
	}
}