grim/convey
Clone
Summary
Browse
Changes
Graph
Move the config loaders to their own packages outside of the config directory
2017-07-25, Gary Kramlich
f2fe901ca161
Move the config loaders to their own packages outside of the config directory
/*
* Convey
* Copyright 2016-2017 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
docker
import
(
"fmt"
"io/ioutil"
"os"
"os/user"
"path/filepath"
"strings"
"github.com/aphistic/gomol"
"bitbucket.org/rw_grim/convey/environment"
"bitbucket.org/rw_grim/convey/state"
"bitbucket.org/rw_grim/convey/tasks"
)
type
Run
struct
{
Command
string
`yaml:"command"`
EntryPoint
string
`yaml:"entrypoint"`
Environment
[]
string
`yaml:"environment"`
Image
string
`yaml:"image"`
WorkDir
string
`yaml:"workdir"`
WorkSpace
string
`yaml:"workspace"`
Script
[]
string
`yaml:"script"`
Shell
string
`yaml:"shell"`
Labels
[]
string
`yaml:"labels"`
}
func
(
r
*
Run
)
UnmarshalYAML
(
unmarshal
func
(
interface
{})
error
)
error
{
type
rawRun
Run
raw
:=
rawRun
{
Shell
:
"/bin/sh"
}
if
err
:=
unmarshal
(
&
raw
);
err
!=
nil
{
return
err
}
*
r
=
Run
(
raw
)
return
nil
}
const
runTemplate
=
`run --rm
{{if .UID}} -e UID={{.UID}}{{end}}
{{if .GID}} -e GID={{.GID}}{{end}}
-v {{.WorkspacePath}}:{{.WorkspaceMount}}
-e CONVEY_WORKSPACE={{.WorkspaceMount}}
{{if .CPUShares }} --cpu-shares {{.CPUShares}}{{end}}
{{if .Memory }} --memory {{.Memory}}{{end}}
{{if .ScriptFile }} -v {{.ScriptFile}}:{{.ScriptFile}}{{end}}
{{if .SSHAgent }} -e SSH_AUTH_SOCK -v {{.SSHAuthSock}}:{{.SSHAuthSock}}{{end}}
{{if .EntryPoint}} --entrypoint {{.EntryPoint}}{{end}}
{{if .WorkDir}} -w {{.WorkDir}}{{end}}
{{if .Network}} --network {{.Network}}{{end}}
{{range .Labels}} -l '{{.}}'{{end}}
{{range .Environment}} -e {{.}}{{end}}
{{.Image}}{{if .Command}} {{.Command}}{{end}}`
func
(
r
*
Run
)
Execute
(
name
string
,
logger
*
gomol
.
LogAdapter
,
env
[]
string
,
st
*
state
.
State
)
error
{
fullEnv
:=
environment
.
Merge
(
env
,
r
.
Environment
)
fullEnv
=
environment
.
Merge
(
fullEnv
,
st
.
Environment
)
user
,
err
:=
user
.
Current
()
if
err
!=
nil
{
return
err
}
// now expand the environment
for
idx
,
v
:=
range
fullEnv
{
fullEnv
[
idx
]
=
environment
.
Mapper
(
v
,
fullEnv
)
}
// assign a default workspace location
workSpace
:=
r
.
WorkSpace
if
workSpace
==
""
{
workSpace
=
"/workspace"
}
// initialize some variables
entryPoint
:=
r
.
EntryPoint
commandArg
:=
environment
.
Mapper
(
r
.
Command
,
fullEnv
)
scriptFile
:=
""
// if we're using a script defined in the yaml, create it and override
// some variables
if
len
(
r
.
Script
)
>
0
{
entryPoint
=
r
.
Shell
if
entryPoint
==
""
{
entryPoint
=
"/bin/sh"
}
// figure out the current working directory
pwd
,
err
:=
os
.
Getwd
()
if
err
!=
nil
{
return
err
}
// now get the absolute path
absPwd
,
err
:=
filepath
.
Abs
(
pwd
)
if
err
!=
nil
{
return
err
}
// create the temp file to write the script to
script
,
err
:=
ioutil
.
TempFile
(
absPwd
,
"convey-script-"
)
if
err
!=
nil
{
return
err
}
// set scriptFile to the name of the temp file
scriptFile
=
script
.
Name
()
// set the run command argument to the script file
commandArg
=
scriptFile
// remove the file when the function exits
defer
os
.
Remove
(
scriptFile
)
// iterate the script and run the environment variable exapansion on each line
for
idx
,
item
:=
range
r
.
Script
{
r
.
Script
[
idx
]
=
environment
.
Mapper
(
item
,
fullEnv
)
}
// write the script to the file
script
.
WriteString
(
strings
.
Join
(
r
.
Script
,
"\n"
))
script
.
Close
()
// make the script executable to the user
os
.
Chmod
(
scriptFile
,
0700
)
}
// build the dict for the template
params
:=
map
[
string
]
interface
{}{
"Command"
:
commandArg
,
"CPUShares"
:
st
.
CPUShares
,
"Environment"
:
fullEnv
,
"EntryPoint"
:
entryPoint
,
"GID"
:
user
.
Gid
,
"Image"
:
environment
.
Mapper
(
r
.
Image
,
fullEnv
),
"Labels"
:
environment
.
SliceMapper
(
r
.
Labels
,
fullEnv
),
"Memory"
:
st
.
Memory
,
"Network"
:
st
.
Network
.
Name
,
"ScriptFile"
:
scriptFile
,
"SSHAgent"
:
st
.
EnableSSHAgent
,
"SSHAuthSock"
:
os
.
Getenv
(
"SSH_AUTH_SOCK"
),
"UID"
:
user
.
Uid
,
"WorkDir"
:
environment
.
Mapper
(
r
.
WorkDir
,
fullEnv
),
"WorkspacePath"
:
environment
.
Mapper
(
st
.
Workspace
.
MountPoint
,
fullEnv
),
"WorkspaceMount"
:
environment
.
Mapper
(
workSpace
,
fullEnv
),
}
// run the command
return
Docker
(
name
,
runTemplate
,
params
,
st
)
}
func
(
r
*
Run
)
New
()
tasks
.
Task
{
return
&
Run
{}
}
func
(
r
*
Run
)
Valid
()
error
{
if
r
.
Image
==
""
{
return
fmt
.
Errorf
(
"no image specified"
)
}
return
nil
}