git_repo.go

package releasedir

import (
	gopath "path"
	"strings"

	bosherr "github.com/cloudfoundry/bosh-utils/errors"
	boshsys "github.com/cloudfoundry/bosh-utils/system"
)

type FSGitRepo struct {
	dirPath    string
	gitDirPath string // .git dir
	runner     boshsys.CmdRunner
	fs         boshsys.FileSystem
}

func NewFSGitRepo(dirPath string, runner boshsys.CmdRunner, fs boshsys.FileSystem) FSGitRepo {
	return FSGitRepo{
		dirPath:    dirPath,
		gitDirPath: gopath.Join(dirPath, ".git"),
		runner:     runner,
		fs:         fs,
	}
}

func (r FSGitRepo) Init() error {
	_, _, _, err := r.runner.RunCommand("git", "init", r.dirPath)
	if err != nil {
		return bosherr.WrapErrorf(err, "Initing git repo")
	}

	ignoreTpl := `config/private.yml
blobs
dev_releases
releases/*.tgz
releases/**/*.tgz
.dev_builds
.final_builds/jobs/**/*.tgz
.final_builds/packages/**/*.tgz
.DS_Store
.idea
*.swp
*~
*#
#*
`

	err = r.fs.WriteFileString(gopath.Join(r.dirPath, ".gitignore"), ignoreTpl)
	if err != nil {
		return bosherr.WrapError(err, "Creating .gitignore file")
	}

	return nil
}

func (r FSGitRepo) LastCommitSHA() (string, error) {
	stdout, stderr, _, err := r.runner.RunCommand(
		"git", "--git-dir", r.gitDirPath, "rev-parse", "--short", "HEAD")
	if err != nil {
		if r.isNotGitRepo(stderr) {
			return "non-git", nil
		}

		if strings.Contains(stderr, "Needed a single revision") {
			return "empty", nil
		}

		return "", bosherr.WrapErrorf(err, "Checking last commit SHA")
	}

	return strings.TrimSpace(stdout), nil
}

func (r FSGitRepo) MustNotBeDirty(force bool) (bool, error) {
	stdout, stderr, _, err := r.runner.RunCommand(
		"git", "--git-dir", r.gitDirPath, "status", "--short")
	if err != nil {
		if r.isNotGitRepo(stderr) {
			return false, nil
		}

		return false, bosherr.WrapErrorf(err, "Checking status")
	}

	// Strip out newline which is added if there are any changes
	stdout = strings.TrimSpace(stdout)

	if len(stdout) > 0 {
		if force {
			return true, nil
		}
		return true, bosherr.Errorf("Git repository has local modifications:\n\n%s", stdout)
	}

	return false, nil
}

func (r FSGitRepo) isNotGitRepo(stderr string) bool {
	return strings.Contains(stderr, "Not a git repository")
}