grim/convey
Clone
Summary
Browse
Changes
Graph
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
}