diff options
Diffstat (limited to 'modules')
| -rw-r--r-- | modules/git/blob.go | 23 | ||||
| -rw-r--r-- | modules/repofiles/diff_test.go | 9 |
2 files changed, 32 insertions, 0 deletions
diff --git a/modules/git/blob.go b/modules/git/blob.go index 68147673a3..df88ac2ede 100644 --- a/modules/git/blob.go +++ b/modules/git/blob.go @@ -6,6 +6,7 @@ package git import ( + "bytes" "encoding/base64" "io" "io/ioutil" @@ -50,6 +51,28 @@ func (b *Blob) GetBlobContent() (string, error) { return string(buf), nil } +// GetBlobLineCount gets line count of lob as raw text +func (b *Blob) GetBlobLineCount() (int, error) { + reader, err := b.DataAsync() + if err != nil { + return 0, err + } + defer reader.Close() + buf := make([]byte, 32*1024) + count := 0 + lineSep := []byte{'\n'} + for { + c, err := reader.Read(buf) + count += bytes.Count(buf[:c], lineSep) + switch { + case err == io.EOF: + return count, nil + case err != nil: + return count, err + } + } +} + // GetBlobContentBase64 Reads the content of the blob with a base64 encode and returns the encoded string func (b *Blob) GetBlobContentBase64() (string, error) { dataRc, err := b.DataAsync() diff --git a/modules/repofiles/diff_test.go b/modules/repofiles/diff_test.go index db2c7552c4..4e1d5b13eb 100644 --- a/modules/repofiles/diff_test.go +++ b/modules/repofiles/diff_test.go @@ -55,6 +55,15 @@ func TestGetDiffPreview(t *testing.T) { Type: 4, Content: "@@ -1,3 +1,4 @@", Comments: nil, + SectionInfo: &gitdiff.DiffLineSectionInfo{ + Path: "README.md", + LastLeftIdx: 0, + LastRightIdx: 0, + LeftIdx: 1, + RightIdx: 1, + LeftHunkSize: 3, + RightHunkSize: 4, + }, }, { LeftIdx: 1, |