graphql-engine/cli/util/git.go
Aravind Shankar bb63d7e60e
cli: allow managing actions (#3859)
Co-authored-by: Rishichandra Wawhal <rishichandra.wawhal@gmail.com>
Co-authored-by: Rikin Kachhia <54616969+rikinsk@users.noreply.github.com>
Co-authored-by: Aravind <aravindkp@outlook.in>
Co-authored-by: Anon Ray <ecthiender@users.noreply.github.com>
Co-authored-by: Shahidh K Muhammed <muhammedshahid.k@gmail.com>
2020-02-24 21:44:46 +05:30

103 lines
2.2 KiB
Go

package util
import (
"os"
"path/filepath"
"gopkg.in/src-d/go-git.v4"
"gopkg.in/src-d/go-git.v4/plumbing"
)
// Default Codegen Assets constants
const (
ActionsCodegenOrg string = "hasura/codegen-assets"
ActionsCodegenRepoURI = "https://github.com/hasura/codegen-assets.git"
ActionsCodegenDirName = "actions-codegen-assets"
)
// Default init-templates repo constants
const (
InitTemplatesRepoURI string = "https://github.com/hasura/graphql-engine-install-manifests.git"
InitTemplatesDirName = "init-templates"
)
type GitUtil struct {
URI string
Path string
// Optional
ReferenceName plumbing.ReferenceName
}
func NewGitUtil(uri string, path string, refName string) *GitUtil {
cfg := &GitUtil{
URI: uri,
Path: path,
ReferenceName: plumbing.HEAD,
}
if refName != "" {
cfg.ReferenceName = plumbing.NewBranchReferenceName(refName)
}
return cfg
}
func (g *GitUtil) EnsureCloned() error {
if ok, err := g.IsGitCloned(); err != nil {
return err
} else if !ok {
_, err := git.PlainClone(g.Path, false, &git.CloneOptions{
URL: g.URI,
ReferenceName: g.ReferenceName,
})
if err != nil && err != git.ErrRepositoryAlreadyExists {
return err
}
}
return nil
}
func (g *GitUtil) IsGitCloned() (bool, error) {
f, err := os.Stat(filepath.Join(g.Path, ".git"))
if os.IsNotExist(err) {
return false, nil
}
return err == nil && f.IsDir(), err
}
// EnsureUpdated will ensure the destination path exists and is up to date.
func (g *GitUtil) EnsureUpdated() error {
if err := g.EnsureCloned(); err != nil {
return err
}
return g.updateAndCleanUntracked()
}
func (g *GitUtil) updateAndCleanUntracked() error {
repo, err := git.PlainOpen(g.Path)
if err != nil {
return err
}
err = repo.Fetch(&git.FetchOptions{})
if err != nil && err != git.NoErrAlreadyUpToDate {
return err
}
wt, err := repo.Worktree()
if err != nil {
return err
}
err = wt.Pull(&git.PullOptions{})
if err != nil && err != git.NoErrAlreadyUpToDate {
return err
}
err = wt.Reset(&git.ResetOptions{
Commit: plumbing.ZeroHash,
Mode: git.HardReset,
})
if err != nil {
return err
}
return wt.Clean(&git.CleanOptions{
Dir: true,
})
}