From 833d946d719926445ac2ce7b2584e8ca1b405cc8 Mon Sep 17 00:00:00 2001
From: Ole Claussen <claussen.ole@gmail.com>
Date: Thu, 27 Aug 2020 16:48:31 +0000
Subject: [PATCH] provision docker environment automatically

---
 pkg/configuration/plugin.go | 145 ++++++++++++++++++++++++++++++++++++
 plugin/plugin.go            |   5 +-
 2 files changed, 148 insertions(+), 2 deletions(-)
 create mode 100644 pkg/configuration/plugin.go

diff --git a/pkg/configuration/plugin.go b/pkg/configuration/plugin.go
new file mode 100644
index 0000000..7c59952
--- /dev/null
+++ b/pkg/configuration/plugin.go
@@ -0,0 +1,145 @@
+package configuration
+
+import (
+	"archive/tar"
+	"io"
+	"io/ioutil"
+	"os"
+	"path/filepath"
+
+	dockerapi "github.com/docker/docker/api/types"
+	docker "github.com/docker/docker/client"
+	"github.com/dodo-cli/dodo-core/pkg/plugin"
+	"github.com/dodo-cli/dodo-core/pkg/plugin/configuration"
+	"github.com/dodo-cli/dodo-core/pkg/types"
+	"github.com/dodo-cli/dodo-docker/pkg/client"
+	log "github.com/hashicorp/go-hclog"
+	"golang.org/x/net/context"
+)
+
+// TODO: a lot of this should be consolidated with runtime and client
+
+const (
+	envHost       = "DOCKER_HOST"
+	envApiVersion = "DOCKER_API_VERSION"
+	envCertPath   = "DOCKER_CERT_PATH"
+	envVerify     = "DOCKER_TLS_VERIFY"
+)
+
+type Configuration struct {
+	client     *docker.Client
+	host       string
+	apiVersion string
+	certPath   string
+}
+
+func (p *Configuration) Type() plugin.Type {
+	return configuration.Type
+}
+
+func (p *Configuration) Init() error {
+	dockerClient, err := client.GetDockerClient()
+	if err != nil {
+		return err
+	}
+
+	p.client = dockerClient
+	p.host = os.Getenv(envHost)
+	p.apiVersion = os.Getenv(envApiVersion)
+	p.certPath = os.Getenv(envCertPath)
+
+	return nil
+}
+
+func (p *Configuration) UpdateConfiguration(backdrop *types.Backdrop) (*types.Backdrop, error) {
+	env := []*types.Environment{}
+
+	if len(p.apiVersion) > 0 {
+		env = append(env, &types.Environment{Key: envApiVersion, Value: p.apiVersion})
+	}
+
+	if len(p.host) > 0 {
+		env = append(env, &types.Environment{Key: envHost, Value: p.host})
+	}
+
+	if len(p.certPath) > 0 {
+		env = append(env,
+			&types.Environment{Key: envCertPath, Value: p.certPath},
+			&types.Environment{Key: envVerify, Value: "1"},
+		)
+	}
+
+	return &types.Backdrop{Environment: env}, nil
+}
+
+func (p *Configuration) Provision(containerID string) error {
+	if len(p.certPath) == 0 {
+		return nil
+	}
+
+	caPath := filepath.Join(p.certPath, "ca.pem")
+	ca, err := ioutil.ReadFile(caPath)
+	if err != nil {
+		return err
+	}
+
+	if err := p.uploadFile(containerID, caPath, ca); err != nil {
+		return err
+	}
+
+	certPath := filepath.Join(p.certPath, "cert.pem")
+	cert, err := ioutil.ReadFile(certPath)
+	if err != nil {
+		return err
+	}
+
+	if err := p.uploadFile(containerID, certPath, cert); err != nil {
+		return err
+	}
+
+	keyPath := filepath.Join(p.certPath, "key.pem")
+	key, err := ioutil.ReadFile(keyPath)
+	if err != nil {
+		return err
+	}
+
+	if err := p.uploadFile(containerID, keyPath, key); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func (p *Configuration) uploadFile(containerID string, path string, contents []byte) error {
+	reader, writer := io.Pipe()
+	defer reader.Close()
+	defer writer.Close()
+
+	go func() {
+		if err := p.client.CopyToContainer(
+			context.Background(),
+			containerID,
+			"/",
+			reader,
+			dockerapi.CopyToContainerOptions{},
+		); err != nil {
+			log.L().Error("could not upload file to container", "error", err)
+		}
+	}()
+
+	tarWriter := tar.NewWriter(writer)
+	defer tarWriter.Close()
+
+	err := tarWriter.WriteHeader(&tar.Header{
+		Name: path,
+		Mode: 0644,
+		Size: int64(len(contents)),
+	})
+	if err != nil {
+		return err
+	}
+
+	_, err = tarWriter.Write(contents)
+
+	return err
+}
diff --git a/plugin/plugin.go b/plugin/plugin.go
index e9df6cc..d977b22 100644
--- a/plugin/plugin.go
+++ b/plugin/plugin.go
@@ -2,14 +2,15 @@ package plugin
 
 import (
 	"github.com/dodo-cli/dodo-core/pkg/plugin"
+	"github.com/dodo-cli/dodo-docker/pkg/configuration"
 	"github.com/dodo-cli/dodo-docker/pkg/runtime"
 )
 
 func RunMe() int {
-	plugin.ServePlugins(&runtime.ContainerRuntime{})
+	plugin.ServePlugins(&runtime.ContainerRuntime{}, &configuration.Configuration{})
 	return 0
 }
 
 func IncludeMe() {
-	plugin.IncludePlugins(&runtime.ContainerRuntime{})
+	plugin.IncludePlugins(&runtime.ContainerRuntime{}, &configuration.Configuration{})
 }