grim/convey
Clone
Summary
Browse
Changes
Graph
Move the map of runtime.Task's to the Runtime instance
2021-12-23, Gary Kramlich
bd445f33e472
Move the map of runtime.Task's to the Runtime instance
// 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"
"hg.sr.ht/~grim/govcs"
"hg.sr.ht/~grim/govcs/hg"
"keep.imfreedom.org/grim/convey/consts"
"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
())
vars
:=
map
[
string
]
string
{
"BRANCH"
:
repo
.
Branch
(),
"BRANCH_NORMALIZED"
:
normalize
.
Normalize
(
repo
.
Branch
()),
"COMMIT"
:
repo
.
Commit
(),
"COMMIT_SHORT"
:
repo
.
ShortCommit
(),
"REMOTE"
:
repo
.
Remote
(
""
),
"VCS"
:
name
,
name
+
"_BRANCH"
:
repo
.
Branch
(),
name
+
"_BRANCH_NORMALIZED"
:
normalize
.
Normalize
(
repo
.
Branch
()),
name
+
"_COMMIT"
:
repo
.
Commit
(),
name
+
"_COMMIT_SHORT"
:
repo
.
ShortCommit
(),
name
+
"_REMOTE"
:
repo
.
Remote
(
""
),
}
// check for vcs specific values. If this grows past 1, make it a switch
// type.
if
hg
,
ok
:=
repo
.(
*
hg
.
Mercurial
);
ok
{
vars
[
"BOOKMARK"
]
=
hg
.
Bookmark
()
vars
[
"HG_BOOKMARK"
]
=
hg
.
Bookmark
()
}
for
k
,
v
:=
range
vars
{
if
err
:=
e
.
setEnvOS
(
k
,
v
);
err
!=
nil
{
return
err
}
}
return
nil
}
// LoadDefaults will load the default environment variables for a convey run.
func
(
e
Environment
)
LoadDefaults
(
wd
string
)
error
{
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
}