69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
// Copyright 2015 The Gogs Authors. All rights reserved.
|
|
// Copyright 2019 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package git
|
|
|
|
import (
|
|
"path"
|
|
"strings"
|
|
|
|
"gopkg.in/src-d/go-git.v4/plumbing"
|
|
"gopkg.in/src-d/go-git.v4/plumbing/filemode"
|
|
"gopkg.in/src-d/go-git.v4/plumbing/object"
|
|
)
|
|
|
|
// GetTreeEntryByPath get the tree entries according the sub dir
|
|
func (t *Tree) GetTreeEntryByPath(relpath string) (*TreeEntry, error) {
|
|
if len(relpath) == 0 {
|
|
return &TreeEntry{
|
|
ID: t.ID,
|
|
//Type: ObjectTree,
|
|
gogitTreeEntry: &object.TreeEntry{
|
|
Name: "",
|
|
Mode: filemode.Dir,
|
|
Hash: plumbing.Hash(t.ID),
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
relpath = path.Clean(relpath)
|
|
parts := strings.Split(relpath, "/")
|
|
var err error
|
|
tree := t
|
|
for i, name := range parts {
|
|
if i == len(parts)-1 {
|
|
entries, err := tree.ListEntries()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
for _, v := range entries {
|
|
if v.Name() == name {
|
|
return v, nil
|
|
}
|
|
}
|
|
} else {
|
|
tree, err = tree.SubTree(name)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
}
|
|
return nil, ErrNotExist{"", relpath}
|
|
}
|
|
|
|
// GetBlobByPath get the blob object according the path
|
|
func (t *Tree) GetBlobByPath(relpath string) (*Blob, error) {
|
|
entry, err := t.GetTreeEntryByPath(relpath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if !entry.IsDir() {
|
|
return entry.Blob(), nil
|
|
}
|
|
|
|
return nil, ErrNotExist{"", relpath}
|
|
}
|