grim/convey

Dump govcs in here for now until i can sort out this out
redux
2021-10-18, Gary Kramlich
c1a95d79fcc5
Dump govcs in here for now until i can sort out this out
// Convey
// Copyright 2016-2018 Gary Kramlich <grim@reaperworld.com>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
// Package environment provides utilities for managing environment variables.
package environment
import (
"os"
"strings"
"time"
"keep.imfreedom.org/grim/convey/consts"
"keep.imfreedom.org/grim/convey/govcs"
"keep.imfreedom.org/grim/convey/govcs/hg"
"keep.imfreedom.org/grim/convey/normalize"
)
// setEnvOS will add the given envar by name only to the environment but also
// put value into the OS's environmanet variable of the same name.
func (e Environment) setEnvOS(name, value string) error {
if err := os.Setenv(name, value); err != nil {
return err
}
e[name] = ""
return nil
}
func (e Environment) loadVCS(wd string) error {
repo, err := govcs.Detect(wd)
if err != nil {
return nil
}
name := strings.ToUpper(repo.Name())
if err := e.setEnvOS("VCS", name); err != nil {
return err
}
if err := e.setEnvOS("COMMIT", repo.Commit()); err != nil {
return err
}
if err := e.setEnvOS("COMMIT_SHORT", repo.ShortCommit()); err != nil {
return err
}
if err := e.setEnvOS("COMMIT", repo.Commit()); err != nil {
return err
}
if err := e.setEnvOS("BRANCH", repo.Branch()); err != nil {
return err
}
if err := e.setEnvOS("BRANCH_NORMALIZED", normalize.Normalize(repo.Branch())); err != nil {
return err
}
if err := e.setEnvOS("REMOTE", repo.Remote("")); err != nil {
return err
}
// check for vcs specific values. If this grows past 1, make it a switch
// type.
if hg, ok := repo.(*hg.Mercurial); ok {
if err := e.setEnvOS("BOOKMARK", hg.Bookmark()); err != nil {
return err
}
}
return nil
}
// LoadDefaults will load the default environment variables for a convey run.
func (e Environment) LoadDefaults(wd string) error {
oldHome := os.Getenv("HOME")
e["HOME"] = "/tmp"
defer func() {
e["HOME"] = oldHome
}()
e["RUN_TIME"] = time.Now().UTC().Format("2006-01-02T15:04:05-0700")
e["CONVEY_VERSION"] = consts.Version
err := e.loadVCS(wd)
if err != nil {
return err
}
return nil
}