Initial wrapper

Author: sbusso

README.md

@@ -1 +1 @@
-# docker-compose-wrapper
+# Docker-Compose wrapper

docker-compose-test.yml

@@ -0,0 +1,4 @@
+version: "3.9"
+services:
+  redis:
+    image: "redis:alpine"

go.mod

@@ -0,0 +1,3 @@
+module github.com/portainer/docker-compose-wrapper
+
+go 1.15

utils.go

@@ -0,0 +1,24 @@
+package wrapper
+
+import (
+	"os/exec"
+	"path"
+	"runtime"
+)
+
+func osProgram(program string) string {
+	if runtime.GOOS == "windows" {
+		program += ".exe"
+	}
+	return program
+}
+
+func programPath(rootPath, program string) string {
+	return path.Join(rootPath, osProgram(program))
+}
+
+// IsBinaryPresent check if docker compose binary is present
+func IsBinaryPresent(program string) bool {
+	_, err := exec.LookPath(osProgram(program))
+	return err == nil
+}

utils_test.go

@@ -0,0 +1,33 @@
+package wrapper
+
+import (
+	"testing"
+)
+
+func TestIsBinaryPresent(t *testing.T) {
+	type testCase struct {
+		Name     string
+		Binary   string
+		Expected bool
+	}
+	testCases := []testCase{
+		testCase{
+			Name:     "not existing",
+			Binary:   "qwgq-er-gerw",
+			Expected: false,
+		},
+		testCase{
+			Name:     "docker-compose exists",
+			Binary:   "docker-compose",
+			Expected: true,
+		},
+	}
+
+	for _, tc := range testCases {
+		got := IsBinaryPresent(tc.Binary)
+		if got != tc.Expected {
+			t.Errorf("Error in test %s got = %v, and Expected = %v.", tc.Name, got, tc.Expected)
+		}
+	}
+
+}

wrapper.go

@@ -0,0 +1,46 @@
+package wrapper
+
+import (
+	"bytes"
+	"errors"
+	"os/exec"
+)
+
+// ComposeWrapper provide a type for managing docker compose commands
+type ComposeWrapper struct {
+	binaryPath string
+}
+
+// NewComposeWrapper initializes a new ComposeWrapper service with local docker-compose binary.
+func NewComposeWrapper() *ComposeWrapper {
+
+	return &ComposeWrapper{binaryPath: ""}
+}
+
+// Up create and start containers
+func (wrapper *ComposeWrapper) Up(filePath string) ([]byte, error) {
+	return wrapper.Command([]string{"up", "-d"}, filePath)
+}
+
+// Down stop and remove containers
+func (wrapper *ComposeWrapper) Down(filePath string) ([]byte, error) {
+	return wrapper.Command([]string{"down", "--remove-orphans"}, filePath)
+}
+
+// Command exectue a docker-compose commanåd
+func (wrapper *ComposeWrapper) Command(args []string, filePath string) ([]byte, error) {
+	program := programPath(wrapper.binaryPath, "docker-compose")
+
+	args = append([]string{"-f", filePath}, args...)
+
+	var stderr bytes.Buffer
+	cmd := exec.Command(program, args...)
+	cmd.Stderr = &stderr
+
+	output, err := cmd.Output()
+	if err != nil {
+		return nil, errors.New(stderr.String())
+	}
+
+	return output, nil
+}

wrapper_test.go

@@ -0,0 +1,22 @@
+package wrapper
+
+import (
+	"log"
+	"testing"
+)
+
+func TestCommand(t *testing.T) {
+	w := NewComposeWrapper()
+
+	file := "docker-compose-test.yml"
+	_, err := w.Up(file)
+	if err != nil {
+		log.Fatalln(err)
+	}
+
+	_, err = w.Down(file)
+	if err != nil {
+		log.Fatalln(err)
+	}
+
+}