HGKeeper

* Convey

* 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

import (

"os"

"os/exec"

"strings"

"bitbucket.org/rw_grim/convey/normalize"

)

func addGitEnv(envvar string, args ...string) string {

cmd := exec.Command("git", args...)

output, _ := cmd.Output()

trimmed := strings.TrimSpace(string(output))

os.Setenv(envvar, trimmed)

return trimmed

}

func initializeGit() []string {

os.Setenv("GIT_CONFIG_NOSYSTEM", "1")

addGitEnv("GIT_COMMIT", "rev-parse", "HEAD")

addGitEnv("GIT_COMMIT_SHORT", "rev-parse", "--short", "HEAD")

// add the branch and a normalized version

branch := addGitEnv("GIT_BRANCH", "rev-parse", "--abbrev-ref", "HEAD")

os.Setenv("GIT_BRANCH_NORMALIZED", normalize.Normalize(branch))

return []string{"GIT_COMMIT", "GIT_COMMIT_SHORT", "GIT_BRANCH", "GIT_BRANCH_NORMALIZED"}

}