forgejo/modules/git/blame.go

139 lines
3 KiB
Go
Raw Normal View History

2019-04-20 04:47:00 +02:00
// Copyright 2019 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT
2019-04-20 04:47:00 +02:00
package git
2019-04-20 04:47:00 +02:00
import (
"bufio"
"context"
2019-04-20 04:47:00 +02:00
"fmt"
"io"
"os"
"regexp"
)
// 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 *Command
output io.WriteCloser
reader io.ReadCloser
bufferedReader *bufio.Reader
done chan error
lastSha *string
2019-04-20 04:47:00 +02:00
}
var shaLineRegex = regexp.MustCompile("^([a-z0-9]{40})")
// NextPart returns next part of blame (sequential code lines with the same commit)
2019-04-20 04:47:00 +02:00
func (r *BlameReader) NextPart() (*BlamePart, error) {
var blamePart *BlamePart
if r.lastSha != nil {
2019-06-12 21:41:28 +02:00
blamePart = &BlamePart{*r.lastSha, make([]string, 0)}
2019-04-20 04:47:00 +02:00
}
var line []byte
var isPrefix bool
var err error
for err != io.EOF {
line, isPrefix, err = r.bufferedReader.ReadLine()
if err != nil && err != io.EOF {
return blamePart, err
}
2019-04-20 04:47:00 +02:00
if len(line) == 0 {
// isPrefix will be false
2019-04-20 04:47:00 +02:00
continue
}
lines := shaLineRegex.FindSubmatch(line)
2019-04-20 04:47:00 +02:00
if lines != nil {
sha1 := string(lines[1])
2019-04-20 04:47:00 +02:00
if blamePart == nil {
2019-06-12 21:41:28 +02:00
blamePart = &BlamePart{sha1, make([]string, 0)}
2019-04-20 04:47:00 +02:00
}
if blamePart.Sha != sha1 {
r.lastSha = &sha1
// need to munch to end of line...
for isPrefix {
_, isPrefix, err = r.bufferedReader.ReadLine()
if err != nil && err != io.EOF {
return blamePart, err
}
}
2019-04-20 04:47:00 +02:00
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 = r.bufferedReader.ReadLine()
if err != nil && err != io.EOF {
return blamePart, err
}
2019-04-20 04:47:00 +02:00
}
}
r.lastSha = nil
return blamePart, nil
}
// Close BlameReader - don't run NextPart after invoking that
func (r *BlameReader) Close() error {
err := <-r.done
r.bufferedReader = nil
_ = r.reader.Close()
_ = r.output.Close()
return err
2019-04-20 04:47:00 +02:00
}
// CreateBlameReader creates reader for given repository, commit and file
func CreateBlameReader(ctx context.Context, repoPath, commitID, file string) (*BlameReader, error) {
cmd := NewCommandContextNoGlobals(ctx, "blame", "--porcelain").
AddDynamicArguments(commitID).
AddDashesAndList(file).
SetDescription(fmt.Sprintf("GetBlame [repo_path: %s]", repoPath))
reader, stdout, err := os.Pipe()
2019-04-20 04:47:00 +02:00
if err != nil {
return nil, err
2019-04-20 04:47:00 +02:00
}
done := make(chan error, 1)
2019-04-20 04:47:00 +02:00
go func(cmd *Command, dir string, stdout io.WriteCloser, done chan error) {
if err := cmd.Run(&RunOpts{
UseContextTimeout: true,
Dir: dir,
Stdout: stdout,
Stderr: os.Stderr,
}); err == nil {
stdout.Close()
}
done <- err
}(cmd, repoPath, stdout, done)
2019-04-20 04:47:00 +02:00
bufferedReader := bufio.NewReader(reader)
2019-04-20 04:47:00 +02:00
return &BlameReader{
cmd: cmd,
output: stdout,
reader: reader,
bufferedReader: bufferedReader,
done: done,
2019-04-20 04:47:00 +02:00
}, nil
}