go/repo_root/repo_root.go

package repo_root

import (
	"fmt"
	"os"
	"strings"
)

// Return the path to the repo root. Note that this will return an error if
// the CWD is not inside a checkout, so this cannot run on production servers.
func Get() (string, error) {
	dir, err := os.Getwd()
	if err != nil {
		return "", err
	}
	prefixes := []string{"go.skia.org/infra", "buildbot"}
	for _, prefix := range prefixes {
		if strings.Contains(dir, prefix) {
			return strings.Split(dir, prefix)[0] + prefix, nil
		}
	}
	// If this function is used outside of tests, please remove the following.
	if d := os.Getenv("WORKSPACE_DIR"); d != "" {
		return d, nil
	}
	return "", fmt.Errorf("No repo root found; are we running inside a checkout?")
}