repo_config.go

package gitwrapper

import (
	"errors"
	"os"
	"regexp"
	"strings"
)

var (
	ErrNoRemoteRepo    = errors.New("no remote repository")
	ErrNoDefaultBranch = errors.New("no default branch")

	defaultBranches = []string{
		"develop",
		"master",
		"main",
	}
)

type RepoConfig struct {
	RemoteHost    string
	Organization  string
	Repository    string
	Branches      []string
	CurrentBranch string
	DefaultBranch string
	CommitHash    string
}

func NewRepoConfig(dir string) (*RepoConfig, error) {
	if err := os.Chdir(dir); err != nil {
		return nil, err
	}

	var repoConfig = new(RepoConfig)
	var err error

	if repoConfig.RemoteHost, repoConfig.Organization, repoConfig.Repository, err = getRemoteInfo(); err != nil {
		return nil, err
	}

	if repoConfig.Branches, repoConfig.CurrentBranch, repoConfig.DefaultBranch, err = getBranchInfo(); err != nil {
		return nil, err
	}

	if repoConfig.CommitHash, err = runGitCmd(false, "git rev-parse --short HEAD"); err != nil {
		return nil, err
	}

	return repoConfig, nil
}

func getRemoteInfo() (string, string, string, error) {
	o, err := runGitCmd(false, "git remote -v")
	if err != nil {
		return "", "", "", err
	}
	r, _ := regexp.Compile(`(?m)^origin\s+(?:ssh://)?(?:git@)?(github\.com|gitee\.com)(?:-.*)?(?::|/)(.*)/(.*|.*\.git)\s+\(fetch\)$`)
	matches := r.FindStringSubmatch(o)
	if len(matches) == 0 {
		return "", "", "", ErrNoRemoteRepo
	}
	return matches[1], matches[2], strings.Split(matches[3], ".git")[0], nil
}

func getBranchInfo() (branches []string, curBranch string, defBranch string, err error) {
	bb, err := getAllBranches()
	if err != nil {
		return
	}

	for _, b := range bb {
		branches = append(branches, b.Name)
		if b.IsCurrent {
			curBranch = b.Name
		}

		if b.IsRemote && isDefaultBranch(b.Name) {
			defBranch = b.Name
			break
		}
	}
	if defBranch == "" {
		err = ErrNoDefaultBranch
		return
	}

	return
}

func isDefaultBranch(name string) bool {
	for _, branch := range defaultBranches {
		if name == branch {
			return true
		}
	}
	return false
}