aboutsummaryrefslogtreecommitdiffstats
path: root/models/renderhelper/commit_checker.go
blob: 4815643e67348a2aa68d0a2b3a5c2c18f7b07768 (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
// Copyright 2024 The Gitea Authors. All rights reserved.
// SPDX-License-Identifier: MIT

package renderhelper

import (
	"context"
	"io"

	"code.gitea.io/gitea/modules/git"
	"code.gitea.io/gitea/modules/gitrepo"
	"code.gitea.io/gitea/modules/log"
)

type commitChecker struct {
	ctx           context.Context
	commitCache   map[string]bool
	gitRepoFacade gitrepo.Repository

	gitRepo       *git.Repository
	gitRepoCloser io.Closer
}

func newCommitChecker(ctx context.Context, gitRepo gitrepo.Repository) *commitChecker {
	return &commitChecker{ctx: ctx, commitCache: make(map[string]bool), gitRepoFacade: gitRepo}
}

func (c *commitChecker) Close() error {
	if c != nil && c.gitRepoCloser != nil {
		return c.gitRepoCloser.Close()
	}
	return nil
}

func (c *commitChecker) IsCommitIDExisting(commitID string) bool {
	exist, inCache := c.commitCache[commitID]
	if inCache {
		return exist
	}

	if c.gitRepo == nil {
		r, closer, err := gitrepo.RepositoryFromContextOrOpen(c.ctx, c.gitRepoFacade)
		if err != nil {
			log.Error("unable to open repository: %s Error: %v", gitrepo.RepoGitURL(c.gitRepoFacade), err)
			return false
		}
		c.gitRepo, c.gitRepoCloser = r, closer
	}

	exist = c.gitRepo.IsReferenceExist(commitID) // Don't use IsObjectExist since it doesn't support short hashs with gogit edition.
	c.commitCache[commitID] = exist
	return exist
}