upstream-gitea/modules/git/repo_tree_gogit.go

// Copyright 2015 The Gogs Authors. All rights reserved.
// Copyright 2019 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

//go:build gogit

package git

import (
	"context"
	"errors"

	"gitea.dev/modules/git/gitcmd"

	"github.com/go-git/go-git/v5/plumbing"
)

func (repo *Repository) getTree(_ context.Context, id ObjectID) (*Tree, error) {
	gogitTree, err := repo.gogitRepo.TreeObject(plumbing.Hash(id.RawValue()))
	if err != nil {
		if errors.Is(err, plumbing.ErrObjectNotFound) {
			return nil, ErrNotExist{
				ID: id.String(),
			}
		}
		return nil, err
	}

	tree := newTree(id)
	tree.resolvedGogitTreeObject = gogitTree
	return tree, nil
}

// GetTree find the tree object in the repository.
func (repo *Repository) GetTree(ctx context.Context, idStr string) (*Tree, error) {
	objectFormat, err := repo.GetObjectFormat(ctx)
	if err != nil {
		return nil, err
	}

	if len(idStr) != objectFormat.FullLength() {
		res, _, err := gitcmd.NewCommand("rev-parse", "--verify").
			AddDynamicArguments(idStr).
			WithRepo(repo).
			RunStdString(ctx)
		if err != nil {
			return nil, err
		}
		if len(res) > 0 {
			idStr = res[:len(res)-1]
		}
	}
	id, err := NewIDFromString(idStr)
	if err != nil {
		return nil, err
	}
	commitObject, err := repo.gogitRepo.CommitObject(plumbing.Hash(id.RawValue()))
	if err == nil {
		id = ParseGogitHash(commitObject.TreeHash)
	}
	treeObject, err := repo.getTree(ctx, id)
	if err != nil {
		return nil, err
	}
	return treeObject, nil
}