summaryrefslogtreecommitdiffstats
path: root/modules/git/blame.go
diff options
context:
space:
mode:
authorLunny Xiao <xiaolunwen@gmail.com>2019-06-27 02:15:26 +0800
committertechknowlogick <techknowlogick@gitea.io>2019-06-26 14:15:26 -0400
commitedc94c70413048107ea728ff330f32ca3de6df88 (patch)
treecf5bcc5ba8ee4b2e585dc14a6cf7440ce88bc073 /modules/git/blame.go
parent161e12e157a48f9bcd2de50c187c77444aa2a9a8 (diff)
downloadgitea-edc94c70413048107ea728ff330f32ca3de6df88.tar.gz
gitea-edc94c70413048107ea728ff330f32ca3de6df88.zip
Monitor all git commands; move blame to git package and replace git as a variable (#6864)
* monitor all git commands; move blame to git package and replace git as a variable * use git command but not other commands * fix build * move exec.Command to git.NewCommand * fix fmt * remove unrelated changes * remove unrelated changes * refactor IsEmpty and add tests * fix tests * fix tests * fix tests * fix tests * remove gitLogger * fix fmt * fix isEmpty * fix lint * fix tests
Diffstat (limited to 'modules/git/blame.go')
-rw-r--r--modules/git/blame.go123
1 files changed, 123 insertions, 0 deletions
diff --git a/modules/git/blame.go b/modules/git/blame.go
new file mode 100644
index 0000000000..548236b657
--- /dev/null
+++ b/modules/git/blame.go
@@ -0,0 +1,123 @@
+// 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 (
+ "bufio"
+ "fmt"
+ "io"
+ "os"
+ "os/exec"
+ "regexp"
+
+ "code.gitea.io/gitea/modules/process"
+)
+
+// BlamePart represents block of blame - continuous lines with one sha
+type BlamePart struct {
+ Sha string
+ Lines []string
+}
+
+// BlameReader returns part of file blame one by one
+type BlameReader struct {
+ cmd *exec.Cmd
+ pid int64
+ output io.ReadCloser
+ scanner *bufio.Scanner
+ lastSha *string
+}
+
+var shaLineRegex = regexp.MustCompile("^([a-z0-9]{40})")
+
+// NextPart returns next part of blame (sequencial code lines with the same commit)
+func (r *BlameReader) NextPart() (*BlamePart, error) {
+ var blamePart *BlamePart
+
+ scanner := r.scanner
+
+ if r.lastSha != nil {
+ blamePart = &BlamePart{*r.lastSha, make([]string, 0)}
+ }
+
+ for scanner.Scan() {
+ line := scanner.Text()
+
+ // Skip empty lines
+ if len(line) == 0 {
+ continue
+ }
+
+ lines := shaLineRegex.FindStringSubmatch(line)
+ if lines != nil {
+ sha1 := lines[1]
+
+ if blamePart == nil {
+ blamePart = &BlamePart{sha1, make([]string, 0)}
+ }
+
+ if blamePart.Sha != sha1 {
+ r.lastSha = &sha1
+ return blamePart, nil
+ }
+ } else if line[0] == '\t' {
+ code := line[1:]
+
+ blamePart.Lines = append(blamePart.Lines, code)
+ }
+ }
+
+ r.lastSha = nil
+
+ return blamePart, nil
+}
+
+// Close BlameReader - don't run NextPart after invoking that
+func (r *BlameReader) Close() error {
+ process.GetManager().Remove(r.pid)
+
+ if err := r.cmd.Wait(); err != nil {
+ return fmt.Errorf("Wait: %v", err)
+ }
+
+ return nil
+}
+
+// CreateBlameReader creates reader for given repository, commit and file
+func CreateBlameReader(repoPath, commitID, file string) (*BlameReader, error) {
+ _, err := OpenRepository(repoPath)
+ if err != nil {
+ return nil, err
+ }
+
+ return createBlameReader(repoPath, GitExecutable, "blame", commitID, "--porcelain", "--", file)
+}
+
+func createBlameReader(dir string, command ...string) (*BlameReader, error) {
+ cmd := exec.Command(command[0], command[1:]...)
+ cmd.Dir = dir
+ cmd.Stderr = os.Stderr
+
+ stdout, err := cmd.StdoutPipe()
+ if err != nil {
+ return nil, fmt.Errorf("StdoutPipe: %v", err)
+ }
+
+ if err = cmd.Start(); err != nil {
+ return nil, fmt.Errorf("Start: %v", err)
+ }
+
+ pid := process.GetManager().Add(fmt.Sprintf("GetBlame [repo_path: %s]", dir), cmd)
+
+ scanner := bufio.NewScanner(stdout)
+
+ return &BlameReader{
+ cmd,
+ pid,
+ stdout,
+ scanner,
+ nil,
+ }, nil
+}