123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133 |
- // Copyright 2020 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- //go:build !gogit
-
- package git
-
- import (
- "io"
- "math"
- "strings"
- )
-
- // Tree represents a flat directory listing.
- type Tree struct {
- ID ObjectID
- ResolvedID ObjectID
- repo *Repository
-
- // parent tree
- ptree *Tree
-
- entries Entries
- entriesParsed bool
-
- entriesRecursive Entries
- entriesRecursiveParsed bool
- }
-
- // ListEntries returns all entries of current tree.
- func (t *Tree) ListEntries() (Entries, error) {
- if t.entriesParsed {
- return t.entries, nil
- }
-
- if t.repo != nil {
- wr, rd, cancel := t.repo.CatFileBatch(t.repo.Ctx)
- defer cancel()
-
- _, _ = wr.Write([]byte(t.ID.String() + "\n"))
- _, typ, sz, err := ReadBatchLine(rd)
- if err != nil {
- return nil, err
- }
- if typ == "commit" {
- treeID, err := ReadTreeID(rd, sz)
- if err != nil && err != io.EOF {
- return nil, err
- }
- _, _ = wr.Write([]byte(treeID + "\n"))
- _, typ, sz, err = ReadBatchLine(rd)
- if err != nil {
- return nil, err
- }
- }
- if typ == "tree" {
- t.entries, err = catBatchParseTreeEntries(t.ID.Type(), t, rd, sz)
- if err != nil {
- return nil, err
- }
- t.entriesParsed = true
- return t.entries, nil
- }
-
- // Not a tree just use ls-tree instead
- for sz > math.MaxInt32 {
- discarded, err := rd.Discard(math.MaxInt32)
- sz -= int64(discarded)
- if err != nil {
- return nil, err
- }
- }
- for sz > 0 {
- discarded, err := rd.Discard(int(sz))
- sz -= int64(discarded)
- if err != nil {
- return nil, err
- }
- }
- }
-
- stdout, _, runErr := NewCommand(t.repo.Ctx, "ls-tree", "-l").AddDynamicArguments(t.ID.String()).RunStdBytes(&RunOpts{Dir: t.repo.Path})
- if runErr != nil {
- if strings.Contains(runErr.Error(), "fatal: Not a valid object name") || strings.Contains(runErr.Error(), "fatal: not a tree object") {
- return nil, ErrNotExist{
- ID: t.ID.String(),
- }
- }
- return nil, runErr
- }
-
- var err error
- t.entries, err = parseTreeEntries(t.repo.objectFormat, stdout, t)
- if err == nil {
- t.entriesParsed = true
- }
-
- return t.entries, err
- }
-
- // listEntriesRecursive returns all entries of current tree recursively including all subtrees
- // extraArgs could be "-l" to get the size, which is slower
- func (t *Tree) listEntriesRecursive(extraArgs TrustedCmdArgs) (Entries, error) {
- if t.entriesRecursiveParsed {
- return t.entriesRecursive, nil
- }
-
- stdout, _, runErr := NewCommand(t.repo.Ctx, "ls-tree", "-t", "-r").
- AddArguments(extraArgs...).
- AddDynamicArguments(t.ID.String()).
- RunStdBytes(&RunOpts{Dir: t.repo.Path})
- if runErr != nil {
- return nil, runErr
- }
-
- var err error
- t.entriesRecursive, err = parseTreeEntries(t.repo.objectFormat, stdout, t)
- if err == nil {
- t.entriesRecursiveParsed = true
- }
-
- return t.entriesRecursive, err
- }
-
- // ListEntriesRecursiveFast returns all entries of current tree recursively including all subtrees, no size
- func (t *Tree) ListEntriesRecursiveFast() (Entries, error) {
- return t.listEntriesRecursive(nil)
- }
-
- // ListEntriesRecursiveWithSize returns all entries of current tree recursively including all subtrees, with size
- func (t *Tree) ListEntriesRecursiveWithSize() (Entries, error) {
- return t.listEntriesRecursive(TrustedCmdArgs{"--long"})
- }
|