aboutsummaryrefslogtreecommitdiffstats
path: root/modules/lfs/pointer_scanner_gogit.go
blob: 7e8b812f466dee98e91b798172178a54041be335 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
// Copyright 2021 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.

//go:build gogit
// +build gogit

package lfs

import (
	"context"
	"fmt"

	"code.gitea.io/gitea/modules/git"

	"github.com/go-git/go-git/v5/plumbing/object"
)

// SearchPointerBlobs scans the whole repository for LFS pointer files
func SearchPointerBlobs(ctx context.Context, repo *git.Repository, pointerChan chan<- PointerBlob, errChan chan<- error) {
	gitRepo := repo.GoGitRepo()

	err := func() error {
		blobs, err := gitRepo.BlobObjects()
		if err != nil {
			return fmt.Errorf("lfs.SearchPointerBlobs BlobObjects: %w", err)
		}

		return blobs.ForEach(func(blob *object.Blob) error {
			select {
			case <-ctx.Done():
				return ctx.Err()
			default:
			}

			if blob.Size > blobSizeCutoff {
				return nil
			}

			reader, err := blob.Reader()
			if err != nil {
				return fmt.Errorf("lfs.SearchPointerBlobs blob.Reader: %w", err)
			}
			defer reader.Close()

			pointer, _ := ReadPointer(reader)
			if pointer.IsValid() {
				pointerChan <- PointerBlob{Hash: blob.Hash.String(), Pointer: pointer}
			}

			return nil
		})
	}()

	if err != nil {
		select {
		case <-ctx.Done():
		default:
			errChan <- err
		}
	}

	close(pointerChan)
	close(errChan)
}