forked from Shiloh/githaven
95 lines
2.8 KiB
Go
95 lines
2.8 KiB
Go
// Copyright 2015 The Gogs 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 (
|
|
"bytes"
|
|
"container/list"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// PullRequestInfo represents needed information for a pull request.
|
|
type PullRequestInfo struct {
|
|
MergeBase string
|
|
Commits *list.List
|
|
NumFiles int
|
|
}
|
|
|
|
// GetMergeBase checks and returns merge base of two branches.
|
|
func (repo *Repository) GetMergeBase(base, head string) (string, error) {
|
|
stdout, err := NewCommand("merge-base", base, head).RunInDir(repo.Path)
|
|
return strings.TrimSpace(stdout), err
|
|
}
|
|
|
|
// GetPullRequestInfo generates and returns pull request information
|
|
// between base and head branches of repositories.
|
|
func (repo *Repository) GetPullRequestInfo(basePath, baseBranch, headBranch string) (_ *PullRequestInfo, err error) {
|
|
var remoteBranch string
|
|
|
|
// We don't need a temporary remote for same repository.
|
|
if repo.Path != basePath {
|
|
// Add a temporary remote
|
|
tmpRemote := strconv.FormatInt(time.Now().UnixNano(), 10)
|
|
if err = repo.AddRemote(tmpRemote, basePath, true); err != nil {
|
|
return nil, fmt.Errorf("AddRemote: %v", err)
|
|
}
|
|
defer repo.RemoveRemote(tmpRemote)
|
|
|
|
remoteBranch = "remotes/" + tmpRemote + "/" + baseBranch
|
|
} else {
|
|
remoteBranch = baseBranch
|
|
}
|
|
|
|
prInfo := new(PullRequestInfo)
|
|
prInfo.MergeBase, err = repo.GetMergeBase(remoteBranch, headBranch)
|
|
if err == nil {
|
|
// We have a common base
|
|
logs, err := NewCommand("log", prInfo.MergeBase+"..."+headBranch, prettyLogFormat).RunInDirBytes(repo.Path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
prInfo.Commits, err = repo.parsePrettyFormatLogToList(logs)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("parsePrettyFormatLogToList: %v", err)
|
|
}
|
|
} else {
|
|
prInfo.Commits = list.New()
|
|
prInfo.MergeBase, err = GetFullCommitID(repo.Path, remoteBranch)
|
|
if err != nil {
|
|
prInfo.MergeBase = remoteBranch
|
|
}
|
|
}
|
|
|
|
// Count number of changed files.
|
|
stdout, err := NewCommand("diff", "--name-only", remoteBranch+"..."+headBranch).RunInDir(repo.Path)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
prInfo.NumFiles = len(strings.Split(stdout, "\n")) - 1
|
|
|
|
return prInfo, nil
|
|
}
|
|
|
|
// GetPatch generates and returns patch data between given revisions.
|
|
func (repo *Repository) GetPatch(base, head string) ([]byte, error) {
|
|
return NewCommand("diff", "-p", "--binary", base, head).RunInDirBytes(repo.Path)
|
|
}
|
|
|
|
// GetFormatPatch generates and returns format-patch data between given revisions.
|
|
func (repo *Repository) GetFormatPatch(base, head string) (io.Reader, error) {
|
|
stdout := new(bytes.Buffer)
|
|
stderr := new(bytes.Buffer)
|
|
|
|
if err := NewCommand("format-patch", "--binary", "--stdout", base+"..."+head).
|
|
RunInDirPipeline(repo.Path, stdout, stderr); err != nil {
|
|
return nil, concatenateError(err, stderr.String())
|
|
}
|
|
return stdout, nil
|
|
}
|