diff options
author | Lunny Xiao <xiaolunwen@gmail.com> | 2019-06-27 02:15:26 +0800 |
---|---|---|
committer | techknowlogick <techknowlogick@gitea.io> | 2019-06-26 14:15:26 -0400 |
commit | edc94c70413048107ea728ff330f32ca3de6df88 (patch) | |
tree | cf5bcc5ba8ee4b2e585dc14a6cf7440ce88bc073 /modules/git/blame.go | |
parent | 161e12e157a48f9bcd2de50c187c77444aa2a9a8 (diff) | |
download | gitea-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.go | 123 |
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 +} |