authelia/suites/docker.go

62 lines
1.9 KiB
Go

package suites
import (
"fmt"
"os"
"os/exec"
"strings"
"github.com/clems4ever/authelia/utils"
log "github.com/sirupsen/logrus"
)
// DockerEnvironment represent a docker environment
type DockerEnvironment struct {
dockerComposeFiles []string
}
// NewDockerEnvironment create a new docker environment
func NewDockerEnvironment(files []string) *DockerEnvironment {
return &DockerEnvironment{dockerComposeFiles: files}
}
func (de *DockerEnvironment) createCommandWithStdout(cmd string) *exec.Cmd {
dockerCmdLine := "docker-compose -f " + strings.Join(de.dockerComposeFiles, " -f ") + " " + cmd
log.Trace(dockerCmdLine)
return utils.CommandWithStdout("bash", "-c", dockerCmdLine)
}
func (de *DockerEnvironment) createCommand(cmd string) *exec.Cmd {
dockerCmdLine := "docker-compose -f " + strings.Join(de.dockerComposeFiles, " -f ") + " " + cmd
log.Trace(dockerCmdLine)
return exec.Command("bash", "-c", dockerCmdLine)
}
// Up spawn a docker environment
func (de *DockerEnvironment) Up(suitePath string) error {
cmd := de.createCommandWithStdout("up -d")
cmd.Env = append(os.Environ(), "SUITE_PATH="+suitePath)
return cmd.Run()
}
// Restart restarts a service
func (de *DockerEnvironment) Restart(suitePath, service string) error {
cmd := de.createCommandWithStdout(fmt.Sprintf("restart %s", service))
cmd.Env = append(os.Environ(), "SUITE_PATH="+suitePath)
return cmd.Run()
}
// Down spawn a docker environment
func (de *DockerEnvironment) Down(suitePath string) error {
cmd := de.createCommandWithStdout("down -v")
cmd.Env = append(os.Environ(), "SUITE_PATH="+suitePath)
return cmd.Run()
}
// Logs get logs of a given service of the environment
func (de *DockerEnvironment) Logs(service string, flags []string) (string, error) {
cmd := de.createCommand("logs " + strings.Join(flags, " ") + " " + service)
content, err := cmd.Output()
return string(content), err
}