mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
99 lines
3.1 KiB
Go
99 lines
3.1 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package git
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/hanzoai/git/modules/git/gitcmd"
|
|
"github.com/hanzoai/git/modules/setting"
|
|
"github.com/hanzoai/git/modules/util"
|
|
)
|
|
|
|
// GetRefs returns all references of the repository.
|
|
func (repo *Repository) GetRefs() ([]*Reference, error) {
|
|
return repo.GetRefsFiltered("")
|
|
}
|
|
|
|
// ListOccurrences lists all refs of the given refType the given commit appears in sorted by creation date DESC
|
|
// refType should only be a literal "branch" or "tag" and nothing else
|
|
func (repo *Repository) ListOccurrences(ctx context.Context, refType, commitSHA string) ([]string, error) {
|
|
cmd := gitcmd.NewCommand()
|
|
switch refType {
|
|
case "branch":
|
|
cmd.AddArguments("branch")
|
|
case "tag":
|
|
cmd.AddArguments("tag")
|
|
default:
|
|
return nil, util.NewInvalidArgumentErrorf(`can only use "branch" or "tag" for refType, but got %q`, refType)
|
|
}
|
|
stdout, _, err := cmd.AddArguments("--no-color", "--sort=-creatordate", "--contains").
|
|
AddDynamicArguments(commitSHA).WithDir(repo.Path).RunStdString(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
refs := strings.Split(strings.TrimSpace(stdout), "\n")
|
|
if refType == "branch" {
|
|
return parseBranches(refs), nil
|
|
}
|
|
return parseTags(refs), nil
|
|
}
|
|
|
|
func parseBranches(refs []string) []string {
|
|
results := make([]string, 0, len(refs))
|
|
for _, ref := range refs {
|
|
if strings.HasPrefix(ref, "* ") { // current branch (main branch)
|
|
results = append(results, ref[len("* "):])
|
|
} else if strings.HasPrefix(ref, " ") { // all other branches
|
|
results = append(results, ref[len(" "):])
|
|
} else if ref != "" {
|
|
results = append(results, ref)
|
|
}
|
|
}
|
|
return results
|
|
}
|
|
|
|
func parseTags(refs []string) []string {
|
|
results := make([]string, 0, len(refs))
|
|
for _, ref := range refs {
|
|
if ref != "" {
|
|
results = append(results, ref)
|
|
}
|
|
}
|
|
return results
|
|
}
|
|
|
|
// UnstableGuessRefByShortName does the best guess to see whether a "short name" provided by user is a branch, tag or commit.
|
|
// It could guess wrongly if the input is already ambiguous. For example:
|
|
// * "refs/heads/the-name" vs "refs/heads/refs/heads/the-name"
|
|
// * "refs/tags/1234567890" vs commit "1234567890"
|
|
// In most cases, it SHOULD AVOID using this function, unless there is an irresistible reason (eg: make API friendly to end users)
|
|
// If the function is used, the caller SHOULD CHECK the ref type carefully.
|
|
func (repo *Repository) UnstableGuessRefByShortName(shortName string) RefName {
|
|
if repo.IsBranchExist(shortName) {
|
|
return RefNameFromBranch(shortName)
|
|
}
|
|
if repo.IsTagExist(shortName) {
|
|
return RefNameFromTag(shortName)
|
|
}
|
|
if strings.HasPrefix(shortName, "refs/") {
|
|
if repo.IsReferenceExist(shortName) {
|
|
return RefName(shortName)
|
|
}
|
|
}
|
|
commit, err := repo.GetCommit(shortName)
|
|
if err == nil {
|
|
commitIDString := commit.ID.String()
|
|
// make sure the "shortName" is either partial commit ID, or it is HEAD
|
|
if strings.HasPrefix(commitIDString, shortName) || shortName == RefNameHead {
|
|
return RefName(commitIDString)
|
|
} else {
|
|
setting.PanicInDevOrTesting("abuse of UnstableGuessRefByShortName, queried %s, got %s", shortName, commitIDString)
|
|
}
|
|
}
|
|
return ""
|
|
}
|