123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159 |
- // Copyright 2015 The Gogs Authors. All rights reserved.
- // Copyright 2018 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package git
-
- import (
- "context"
- "errors"
- "fmt"
- "strings"
- )
-
- // BranchPrefix base dir of the branch information file store on git
- const BranchPrefix = "refs/heads/"
-
- // IsReferenceExist returns true if given reference exists in the repository.
- func IsReferenceExist(ctx context.Context, repoPath, name string) bool {
- _, _, err := NewCommand(ctx, "show-ref", "--verify").AddDashesAndList(name).RunStdString(&RunOpts{Dir: repoPath})
- return err == nil
- }
-
- // IsBranchExist returns true if given branch exists in the repository.
- func IsBranchExist(ctx context.Context, repoPath, name string) bool {
- return IsReferenceExist(ctx, repoPath, BranchPrefix+name)
- }
-
- // Branch represents a Git branch.
- type Branch struct {
- Name string
- Path string
-
- gitRepo *Repository
- }
-
- // GetHEADBranch returns corresponding branch of HEAD.
- func (repo *Repository) GetHEADBranch() (*Branch, error) {
- if repo == nil {
- return nil, fmt.Errorf("nil repo")
- }
- stdout, _, err := NewCommand(repo.Ctx, "symbolic-ref", "HEAD").RunStdString(&RunOpts{Dir: repo.Path})
- if err != nil {
- return nil, err
- }
- stdout = strings.TrimSpace(stdout)
-
- if !strings.HasPrefix(stdout, BranchPrefix) {
- return nil, fmt.Errorf("invalid HEAD branch: %v", stdout)
- }
-
- return &Branch{
- Name: stdout[len(BranchPrefix):],
- Path: stdout,
- gitRepo: repo,
- }, nil
- }
-
- func GetDefaultBranch(ctx context.Context, repoPath string) (string, error) {
- stdout, _, err := NewCommand(ctx, "symbolic-ref", "HEAD").RunStdString(&RunOpts{Dir: repoPath})
- if err != nil {
- return "", err
- }
- stdout = strings.TrimSpace(stdout)
- if !strings.HasPrefix(stdout, BranchPrefix) {
- return "", errors.New("the HEAD is not a branch: " + stdout)
- }
- return strings.TrimPrefix(stdout, BranchPrefix), nil
- }
-
- // GetBranch returns a branch by it's name
- func (repo *Repository) GetBranch(branch string) (*Branch, error) {
- if !repo.IsBranchExist(branch) {
- return nil, ErrBranchNotExist{branch}
- }
- return &Branch{
- Path: repo.Path,
- Name: branch,
- gitRepo: repo,
- }, nil
- }
-
- // GetBranches returns a slice of *git.Branch
- func (repo *Repository) GetBranches(skip, limit int) ([]*Branch, int, error) {
- brs, countAll, err := repo.GetBranchNames(skip, limit)
- if err != nil {
- return nil, 0, err
- }
-
- branches := make([]*Branch, len(brs))
- for i := range brs {
- branches[i] = &Branch{
- Path: repo.Path,
- Name: brs[i],
- gitRepo: repo,
- }
- }
-
- return branches, countAll, nil
- }
-
- // DeleteBranchOptions Option(s) for delete branch
- type DeleteBranchOptions struct {
- Force bool
- }
-
- // DeleteBranch delete a branch by name on repository.
- func (repo *Repository) DeleteBranch(name string, opts DeleteBranchOptions) error {
- cmd := NewCommand(repo.Ctx, "branch")
-
- if opts.Force {
- cmd.AddArguments("-D")
- } else {
- cmd.AddArguments("-d")
- }
-
- cmd.AddDashesAndList(name)
- _, _, err := cmd.RunStdString(&RunOpts{Dir: repo.Path})
-
- return err
- }
-
- // CreateBranch create a new branch
- func (repo *Repository) CreateBranch(branch, oldbranchOrCommit string) error {
- cmd := NewCommand(repo.Ctx, "branch")
- cmd.AddDashesAndList(branch, oldbranchOrCommit)
-
- _, _, err := cmd.RunStdString(&RunOpts{Dir: repo.Path})
-
- return err
- }
-
- // AddRemote adds a new remote to repository.
- func (repo *Repository) AddRemote(name, url string, fetch bool) error {
- cmd := NewCommand(repo.Ctx, "remote", "add")
- if fetch {
- cmd.AddArguments("-f")
- }
- cmd.AddDynamicArguments(name, url)
-
- _, _, err := cmd.RunStdString(&RunOpts{Dir: repo.Path})
- return err
- }
-
- // RemoveRemote removes a remote from repository.
- func (repo *Repository) RemoveRemote(name string) error {
- _, _, err := NewCommand(repo.Ctx, "remote", "rm").AddDynamicArguments(name).RunStdString(&RunOpts{Dir: repo.Path})
- return err
- }
-
- // GetCommit returns the head commit of a branch
- func (branch *Branch) GetCommit() (*Commit, error) {
- return branch.gitRepo.GetBranchCommit(branch.Name)
- }
-
- // RenameBranch rename a branch
- func (repo *Repository) RenameBranch(from, to string) error {
- _, _, err := NewCommand(repo.Ctx, "branch", "-m").AddDynamicArguments(from, to).RunStdString(&RunOpts{Dir: repo.Path})
- return err
- }
|