feat: change api to feet portainer

Author: chiptus

wrapper.go

@@ -12,29 +12,42 @@ type ComposeWrapper struct {
 }
 
 // NewComposeWrapper initializes a new ComposeWrapper service with local docker-compose binary.
-func NewComposeWrapper() *ComposeWrapper {
+func NewComposeWrapper(binaryPath string) (*ComposeWrapper, error) {
+	if !IsBinaryPresent(programPath(binaryPath, "docker-compose")) {
+		return nil, errors.New("docker-compose binary not found")
+	}
 
-	return &ComposeWrapper{binaryPath: ""}
+	return &ComposeWrapper{binaryPath: binaryPath}, nil
 }
 
 // Up create and start containers
-func (wrapper *ComposeWrapper) Up(filePath string) ([]byte, error) {
-	return wrapper.Command([]string{"up", "-d"}, filePath)
+func (wrapper *ComposeWrapper) Up(filePath, url, projectName, envFilePath string) ([]byte, error) {
+	return wrapper.Command(newUpCommand(filePath), url, projectName, envFilePath)
 }
 
 // Down stop and remove containers
-func (wrapper *ComposeWrapper) Down(filePath string) ([]byte, error) {
-	return wrapper.Command([]string{"down", "--remove-orphans"}, filePath)
+func (wrapper *ComposeWrapper) Down(filePath, url, projectName string) ([]byte, error) {
+	return wrapper.Command(newDownCommand(filePath), url, projectName, "")
 }
 
 // Command exectue a docker-compose commanåd
-func (wrapper *ComposeWrapper) Command(args []string, filePath string) ([]byte, error) {
+func (wrapper *ComposeWrapper) Command(command composeCommand, url, projectName, envFilePath string) ([]byte, error) {
 	program := programPath(wrapper.binaryPath, "docker-compose")
 
-	args = append([]string{"-f", filePath}, args...)
+	if projectName != "" {
+		command.WithProjectName(projectName)
+	}
+
+	if envFilePath != "" {
+		command.WithEnvFilePath(envFilePath)
+	}
+
+	if url != "" {
+		command.WithURL(url)
+	}
 
 	var stderr bytes.Buffer
-	cmd := exec.Command(program, args...)
+	cmd := exec.Command(program, command.ToArgs()...)
 	cmd.Stderr = &stderr
 
 	output, err := cmd.Output()
@@ -44,3 +57,39 @@ func (wrapper *ComposeWrapper) Command(args []string, filePath string) ([]byte,
 
 	return output, nil
 }
+
+type composeCommand struct {
+	command []string
+	args    []string
+}
+
+func newCommand(command []string, filePath string) composeCommand {
+	return composeCommand{
+		args:    []string{"-f", filePath},
+		command: command,
+	}
+}
+
+func newUpCommand(filePath string) composeCommand {
+	return newCommand([]string{"up", "-d"}, filePath)
+}
+
+func newDownCommand(filePath string) composeCommand {
+	return newCommand([]string{"down", "--remove-orphans"}, filePath)
+}
+
+func (command *composeCommand) WithProjectName(projectName string) {
+	command.args = append(command.args, "-p", projectName)
+}
+
+func (command *composeCommand) WithEnvFilePath(envFilePath string) {
+	command.args = append(command.args, "--env-file", envFilePath)
+}
+
+func (command *composeCommand) WithURL(url string) {
+	command.args = append(command.args, "-H", url)
+}
+
+func (command *composeCommand) ToArgs() []string {
+	return append(command.args, command.command...)
+}

wrapper_test.go

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