8 "github.com/edwarnicke/exechelper"
16 type Container struct {
21 volumes map[string]Volume
22 envVars map[string]string
25 func NewContainer(yamlInput ContainerConfig) (*Container, error) {
26 containerName := yamlInput["name"].(string)
27 if len(containerName) == 0 {
28 err := fmt.Errorf("container name must not be blank")
32 var container = new(Container)
33 container.volumes = make(map[string]Volume)
34 container.envVars = make(map[string]string)
35 container.name = containerName
37 if image, ok := yamlInput["image"]; ok {
38 container.image = image.(string)
40 container.image = "hs-test/vpp"
43 if isOptional, ok := yamlInput["is-optional"]; ok {
44 container.isOptional = isOptional.(bool)
46 container.isOptional = false
49 if _, ok := yamlInput["volumes"]; ok {
50 r:= strings.NewReplacer("$HST_DIR", workDir)
51 for _, volu := range yamlInput["volumes"].([]interface{}) {
52 volumeMap := volu.(ContainerConfig)
53 hostDir := r.Replace(volumeMap["host-dir"].(string))
54 containerDir := volumeMap["container-dir"].(string)
55 container.addVolume(hostDir, containerDir)
57 if isDefaultWorkDir, ok := volumeMap["is-default-work-dir"]; ok &&
58 isDefaultWorkDir.(bool) &&
59 len(container.workDir) == 0 {
60 container.workDir = containerDir
66 if _, ok := yamlInput["vars"]; ok {
67 for _, envVar := range yamlInput["vars"].([]interface{}) {
68 container.addEnvVar(envVar)
74 func (c *Container) getRunCommand() string {
75 syncPath := fmt.Sprintf(" -v %s:/tmp/sync", c.getSyncPath())
76 cmd := "docker run --cap-add=all -d --privileged --network host --rm"
78 cmd += c.getVolumesAsCliOption()
79 cmd += c.getEnvVarsAsCliOption()
80 cmd += " --name " + c.name + " " + c.image
84 func (c *Container) run() error {
86 return fmt.Errorf("run container failed: name is blank")
89 exechelper.Run(fmt.Sprintf("mkdir -p /tmp/%s/sync", c.name))
90 cmd := c.getRunCommand()
91 err := exechelper.Run(cmd)
93 return fmt.Errorf("container run failed: %s", err)
99 func (c *Container) addVolume(hostDir string, containerDir string) {
101 volume.hostDir = hostDir
102 volume.containerDir = containerDir
103 c.volumes[hostDir] = volume
106 func (c *Container) getVolumeByHostDir(hostDir string) Volume {
107 return c.volumes[hostDir]
110 func (c *Container) getVolumesAsCliOption() string {
113 if len(c.volumes) > 0 {
114 for _, volume := range c.volumes {
115 cliOption += fmt.Sprintf(" -v %s:%s", volume.hostDir, volume.containerDir)
122 func (c *Container) getWorkDirAsCliOption() string {
123 if len(c.workDir) == 0 {
126 return fmt.Sprintf(" --workdir=\"%s\"", c.workDir)
129 func (c *Container) addEnvVar(envVar interface{}) {
130 envVarMap := envVar.(ContainerConfig)
131 name := envVarMap["name"].(string)
132 value := envVarMap["value"].(string)
133 c.envVars[name] = value
136 func (c *Container) getEnvVarsAsCliOption() string {
138 if len(c.envVars) == 0 {
142 for name, value := range c.envVars {
143 cliOption += fmt.Sprintf(" -e %s=%s", name, value)
148 func (c *Container) getSyncPath() string {
149 return fmt.Sprintf("/tmp/%s/sync", c.name)
152 func (c *Container) exec(command string) (string, error) {
153 cliCommand := "docker exec -d " + c.name + " " + command
154 byteOutput, err := exechelper.CombinedOutput(cliCommand)
155 return string(byteOutput), err
158 func (c *Container) execAction(args string) (string, error) {
159 syncFile := c.getSyncPath() + "/rc"
162 workDir := c.getWorkDirAsCliOption()
163 cmd := fmt.Sprintf("docker exec -d %s %s hs-test %s",
167 err := exechelper.Run(cmd)
171 res, err := waitForSyncFile(syncFile)
173 return "", fmt.Errorf("failed to read sync file while executing 'hs-test %s': %v", args, err)
175 o := res.StdOutput + res.ErrOutput
177 return o, fmt.Errorf("cmd resulted in non-zero value %d: %s", res.Code, res.Desc)
182 func (c *Container) stop() error {
183 return exechelper.Run("docker stop " + c.name)