context.go

package git

import (
	"path/filepath"

	"github.com/opentable/sous/lib"
	"github.com/opentable/sous/util/firsterr"
)

// SourceContext gathers together a number of bits of information about the
// repository such as its current branch, revision, nearest tag, nearest semver
// tag, etc.
func (r *Repo) SourceContext() (*sous.SourceContext, error) {
	var (
		revision, branch,
		nearestTagName, nearestTagRevision,
		repoRelativeDir string
		files, modifiedFiles, newFiles []string
		allTags                        []sous.Tag
		remotes                        Remotes
		//unpushedCommits                []string
	)
	c := r.Client
	if err := firsterr.Parallel().Set(
		func(err *error) { branch, *err = c.CurrentBranch() },
		func(err *error) { revision, *err = c.Revision() },
		func(err *error) {
			repoRelativeDir, *err = filepath.Rel(r.Root, c.Sh.Dir())
			if repoRelativeDir == "." {
				repoRelativeDir = ""
			}
		},
		func(err *error) {
			allTags, *err = r.Client.ListTags()
			if *err != nil || len(allTags) == 0 {
				return
			}
			nearestTagName, *err = c.NearestTag()
			if *err != nil {
				return
			}

			ntr, terr := c.RevisionAt(nearestTagName)

			if terr == nil {
				nearestTagRevision = ntr
			}
		},
		func(err *error) { files, *err = c.ListFiles() },
		func(err *error) { modifiedFiles, *err = c.ModifiedFiles() },
		func(err *error) { newFiles, *err = c.NewFiles() },
		func(err *error) { remotes, *err = c.ListRemotes() },
		//func(err *error) { unpushedCommits, *err = c.ListUnpushedCommits() },
	); err != nil {
		return nil, err
	}

	primaryRemoteURL := guessPrimaryRemote(remotes)
	return &sous.SourceContext{
		RootDir:            r.Root,
		OffsetDir:          repoRelativeDir,
		Branch:             branch,
		Revision:           revision,
		Files:              files,
		ModifiedFiles:      modifiedFiles,
		NewFiles:           newFiles,
		Tags:               allTags,
		NearestTagName:     nearestTagName,
		NearestTagRevision: nearestTagRevision,
		PrimaryRemoteURL:   primaryRemoteURL,
		RemoteURLs:         allFetchURLs(remotes),
		DirtyWorkingTree:   len(modifiedFiles)+len(newFiles) != 0,
	}, nil
}

func allFetchURLs(remotes Remotes) []string {
	var remURLs []string
	for _, r := range remotes {
		u, err := CanonicalRepoURL(r.FetchURL)
		if err != nil {
			continue
		}
		remURLs = append(remURLs, u)
	}
	return remURLs
}

func guessPrimaryRemote(remotes map[string]Remote) string {
	primaryRemote, ok := remotes["upstream"]
	if !ok {
		primaryRemote, ok = remotes["origin"]
	}
	if !ok {
		return ""
	}
	if primaryRemote.FetchURL == "" {
		return ""
	}
	// We don't care about this error, empty string is
	// an acceptable return value.
	u, _ := CanonicalRepoURL(primaryRemote.FetchURL)
	return u
}