mirror of
https://github.com/go-gitea/gitea.git
synced 2024-12-04 14:46:57 -05:00
229 lines
5.2 KiB
Go
229 lines
5.2 KiB
Go
|
package git
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"path"
|
||
|
"sort"
|
||
|
"strings"
|
||
|
|
||
|
"golang.org/x/crypto/openpgp"
|
||
|
"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/format/index"
|
||
|
"gopkg.in/src-d/go-git.v4/plumbing/object"
|
||
|
"gopkg.in/src-d/go-git.v4/storage"
|
||
|
|
||
|
"gopkg.in/src-d/go-billy.v4"
|
||
|
)
|
||
|
|
||
|
// Commit stores the current contents of the index in a new commit along with
|
||
|
// a log message from the user describing the changes.
|
||
|
func (w *Worktree) Commit(msg string, opts *CommitOptions) (plumbing.Hash, error) {
|
||
|
if err := opts.Validate(w.r); err != nil {
|
||
|
return plumbing.ZeroHash, err
|
||
|
}
|
||
|
|
||
|
if opts.All {
|
||
|
if err := w.autoAddModifiedAndDeleted(); err != nil {
|
||
|
return plumbing.ZeroHash, err
|
||
|
}
|
||
|
}
|
||
|
|
||
|
idx, err := w.r.Storer.Index()
|
||
|
if err != nil {
|
||
|
return plumbing.ZeroHash, err
|
||
|
}
|
||
|
|
||
|
h := &buildTreeHelper{
|
||
|
fs: w.Filesystem,
|
||
|
s: w.r.Storer,
|
||
|
}
|
||
|
|
||
|
tree, err := h.BuildTree(idx)
|
||
|
if err != nil {
|
||
|
return plumbing.ZeroHash, err
|
||
|
}
|
||
|
|
||
|
commit, err := w.buildCommitObject(msg, opts, tree)
|
||
|
if err != nil {
|
||
|
return plumbing.ZeroHash, err
|
||
|
}
|
||
|
|
||
|
return commit, w.updateHEAD(commit)
|
||
|
}
|
||
|
|
||
|
func (w *Worktree) autoAddModifiedAndDeleted() error {
|
||
|
s, err := w.Status()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
for path, fs := range s {
|
||
|
if fs.Worktree != Modified && fs.Worktree != Deleted {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
if _, err := w.Add(path); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (w *Worktree) updateHEAD(commit plumbing.Hash) error {
|
||
|
head, err := w.r.Storer.Reference(plumbing.HEAD)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
name := plumbing.HEAD
|
||
|
if head.Type() != plumbing.HashReference {
|
||
|
name = head.Target()
|
||
|
}
|
||
|
|
||
|
ref := plumbing.NewHashReference(name, commit)
|
||
|
return w.r.Storer.SetReference(ref)
|
||
|
}
|
||
|
|
||
|
func (w *Worktree) buildCommitObject(msg string, opts *CommitOptions, tree plumbing.Hash) (plumbing.Hash, error) {
|
||
|
commit := &object.Commit{
|
||
|
Author: *opts.Author,
|
||
|
Committer: *opts.Committer,
|
||
|
Message: msg,
|
||
|
TreeHash: tree,
|
||
|
ParentHashes: opts.Parents,
|
||
|
}
|
||
|
|
||
|
if opts.SignKey != nil {
|
||
|
sig, err := w.buildCommitSignature(commit, opts.SignKey)
|
||
|
if err != nil {
|
||
|
return plumbing.ZeroHash, err
|
||
|
}
|
||
|
commit.PGPSignature = sig
|
||
|
}
|
||
|
|
||
|
obj := w.r.Storer.NewEncodedObject()
|
||
|
if err := commit.Encode(obj); err != nil {
|
||
|
return plumbing.ZeroHash, err
|
||
|
}
|
||
|
return w.r.Storer.SetEncodedObject(obj)
|
||
|
}
|
||
|
|
||
|
func (w *Worktree) buildCommitSignature(commit *object.Commit, signKey *openpgp.Entity) (string, error) {
|
||
|
encoded := &plumbing.MemoryObject{}
|
||
|
if err := commit.Encode(encoded); err != nil {
|
||
|
return "", err
|
||
|
}
|
||
|
r, err := encoded.Reader()
|
||
|
if err != nil {
|
||
|
return "", err
|
||
|
}
|
||
|
var b bytes.Buffer
|
||
|
if err := openpgp.ArmoredDetachSign(&b, signKey, r, nil); err != nil {
|
||
|
return "", err
|
||
|
}
|
||
|
return b.String(), nil
|
||
|
}
|
||
|
|
||
|
// buildTreeHelper converts a given index.Index file into multiple git objects
|
||
|
// reading the blobs from the given filesystem and creating the trees from the
|
||
|
// index structure. The created objects are pushed to a given Storer.
|
||
|
type buildTreeHelper struct {
|
||
|
fs billy.Filesystem
|
||
|
s storage.Storer
|
||
|
|
||
|
trees map[string]*object.Tree
|
||
|
entries map[string]*object.TreeEntry
|
||
|
}
|
||
|
|
||
|
// BuildTree builds the tree objects and push its to the storer, the hash
|
||
|
// of the root tree is returned.
|
||
|
func (h *buildTreeHelper) BuildTree(idx *index.Index) (plumbing.Hash, error) {
|
||
|
const rootNode = ""
|
||
|
h.trees = map[string]*object.Tree{rootNode: {}}
|
||
|
h.entries = map[string]*object.TreeEntry{}
|
||
|
|
||
|
for _, e := range idx.Entries {
|
||
|
if err := h.commitIndexEntry(e); err != nil {
|
||
|
return plumbing.ZeroHash, err
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return h.copyTreeToStorageRecursive(rootNode, h.trees[rootNode])
|
||
|
}
|
||
|
|
||
|
func (h *buildTreeHelper) commitIndexEntry(e *index.Entry) error {
|
||
|
parts := strings.Split(e.Name, "/")
|
||
|
|
||
|
var fullpath string
|
||
|
for _, part := range parts {
|
||
|
parent := fullpath
|
||
|
fullpath = path.Join(fullpath, part)
|
||
|
|
||
|
h.doBuildTree(e, parent, fullpath)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (h *buildTreeHelper) doBuildTree(e *index.Entry, parent, fullpath string) {
|
||
|
if _, ok := h.trees[fullpath]; ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
if _, ok := h.entries[fullpath]; ok {
|
||
|
return
|
||
|
}
|
||
|
|
||
|
te := object.TreeEntry{Name: path.Base(fullpath)}
|
||
|
|
||
|
if fullpath == e.Name {
|
||
|
te.Mode = e.Mode
|
||
|
te.Hash = e.Hash
|
||
|
} else {
|
||
|
te.Mode = filemode.Dir
|
||
|
h.trees[fullpath] = &object.Tree{}
|
||
|
}
|
||
|
|
||
|
h.trees[parent].Entries = append(h.trees[parent].Entries, te)
|
||
|
}
|
||
|
|
||
|
type sortableEntries []object.TreeEntry
|
||
|
|
||
|
func (sortableEntries) sortName(te object.TreeEntry) string {
|
||
|
if te.Mode == filemode.Dir {
|
||
|
return te.Name + "/"
|
||
|
}
|
||
|
return te.Name
|
||
|
}
|
||
|
func (se sortableEntries) Len() int { return len(se) }
|
||
|
func (se sortableEntries) Less(i int, j int) bool { return se.sortName(se[i]) < se.sortName(se[j]) }
|
||
|
func (se sortableEntries) Swap(i int, j int) { se[i], se[j] = se[j], se[i] }
|
||
|
|
||
|
func (h *buildTreeHelper) copyTreeToStorageRecursive(parent string, t *object.Tree) (plumbing.Hash, error) {
|
||
|
sort.Sort(sortableEntries(t.Entries))
|
||
|
for i, e := range t.Entries {
|
||
|
if e.Mode != filemode.Dir && !e.Hash.IsZero() {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
path := path.Join(parent, e.Name)
|
||
|
|
||
|
var err error
|
||
|
e.Hash, err = h.copyTreeToStorageRecursive(path, h.trees[path])
|
||
|
if err != nil {
|
||
|
return plumbing.ZeroHash, err
|
||
|
}
|
||
|
|
||
|
t.Entries[i] = e
|
||
|
}
|
||
|
|
||
|
o := h.s.NewEncodedObject()
|
||
|
if err := t.Encode(o); err != nil {
|
||
|
return plumbing.ZeroHash, err
|
||
|
}
|
||
|
|
||
|
return h.s.SetEncodedObject(o)
|
||
|
}
|