forgejo-federation/modules/git/blame.go

150 lines
3.6 KiB
Go
Raw Normal View History

2019-04-20 08:17:00 +05:30
// 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
2019-04-20 08:17:00 +05:30
import (
"bufio"
"context"
2019-04-20 08:17:00 +05:30
"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
output io.ReadCloser
reader *bufio.Reader
lastSha *string
cancel context.CancelFunc // Cancels the context that this reader runs in
finished process.FinishedFunc // Tells the process manager we're finished and it can remove the associated process from the process table
2019-04-20 08:17:00 +05:30
}
var shaLineRegex = regexp.MustCompile("^([a-z0-9]{40})")
// NextPart returns next part of blame (sequential code lines with the same commit)
2019-04-20 08:17:00 +05:30
func (r *BlameReader) NextPart() (*BlamePart, error) {
var blamePart *BlamePart
reader := r.reader
2019-04-20 08:17:00 +05:30
if r.lastSha != nil {
2019-06-13 01:11:28 +05:30
blamePart = &BlamePart{*r.lastSha, make([]string, 0)}
2019-04-20 08:17:00 +05:30
}
var line []byte
var isPrefix bool
var err error
for err != io.EOF {
line, isPrefix, err = reader.ReadLine()
if err != nil && err != io.EOF {
return blamePart, err
}
2019-04-20 08:17:00 +05:30
if len(line) == 0 {
// isPrefix will be false
2019-04-20 08:17:00 +05:30
continue
}
lines := shaLineRegex.FindSubmatch(line)
2019-04-20 08:17:00 +05:30
if lines != nil {
sha1 := string(lines[1])
2019-04-20 08:17:00 +05:30
if blamePart == nil {
2019-06-13 01:11:28 +05:30
blamePart = &BlamePart{sha1, make([]string, 0)}
2019-04-20 08:17:00 +05:30
}
if blamePart.Sha != sha1 {
r.lastSha = &sha1
// need to munch to end of line...
for isPrefix {
_, isPrefix, err = reader.ReadLine()
if err != nil && err != io.EOF {
return blamePart, err
}
}
2019-04-20 08:17:00 +05:30
return blamePart, nil
}
} else if line[0] == '\t' {
code := line[1:]
blamePart.Lines = append(blamePart.Lines, string(code))
}
// need to munch to end of line...
for isPrefix {
_, isPrefix, err = reader.ReadLine()
if err != nil && err != io.EOF {
return blamePart, err
}
2019-04-20 08:17:00 +05:30
}
}
r.lastSha = nil
return blamePart, nil
}
// Close BlameReader - don't run NextPart after invoking that
func (r *BlameReader) Close() error {
defer r.finished() // Only remove the process from the process table when the underlying command is closed
r.cancel() // However, first cancel our own context early
_ = r.output.Close()
2019-04-20 08:17:00 +05:30
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(ctx context.Context, repoPath, commitID, file string) (*BlameReader, error) {
return createBlameReader(ctx, repoPath, GitExecutable, "blame", commitID, "--porcelain", "--", file)
2019-04-20 08:17:00 +05:30
}
func createBlameReader(ctx context.Context, dir string, command ...string) (*BlameReader, error) {
// Here we use the provided context - this should be tied to the request performing the blame so that it does not hang around.
ctx, cancel, finished := process.GetManager().AddContext(ctx, fmt.Sprintf("GetBlame [repo_path: %s]", dir))
cmd := exec.CommandContext(ctx, command[0], command[1:]...)
2019-04-20 08:17:00 +05:30
cmd.Dir = dir
cmd.Stderr = os.Stderr
stdout, err := cmd.StdoutPipe()
if err != nil {
defer finished()
2019-04-20 08:17:00 +05:30
return nil, fmt.Errorf("StdoutPipe: %v", err)
}
if err = cmd.Start(); err != nil {
defer finished()
_ = stdout.Close()
2019-04-20 08:17:00 +05:30
return nil, fmt.Errorf("Start: %v", err)
}
reader := bufio.NewReader(stdout)
2019-04-20 08:17:00 +05:30
return &BlameReader{
cmd: cmd,
output: stdout,
reader: reader,
cancel: cancel,
finished: finished,
2019-04-20 08:17:00 +05:30
}, nil
}