From 9270168f17b6ca1f6716117ea99c65531cab4fd1 Mon Sep 17 00:00:00 2001
From: Tom Chauveau <tom.chauveau@epitech.eu>
Date: Sat, 5 Jun 2021 23:09:53 +0200
Subject: [PATCH] Add docker-compose package in stdlib

Signed-off-by: Tom Chauveau <tom.chauveau@epitech.eu>
---
 stdlib/docker/compose/client.cue  |  16 ++++
 stdlib/docker/compose/compose.cue | 152 ++++++++++++++++++++++++++++++
 2 files changed, 168 insertions(+)
 create mode 100644 stdlib/docker/compose/client.cue
 create mode 100644 stdlib/docker/compose/compose.cue

diff --git a/stdlib/docker/compose/client.cue b/stdlib/docker/compose/client.cue
new file mode 100644
index 00000000..99eec7ba
--- /dev/null
+++ b/stdlib/docker/compose/client.cue
@@ -0,0 +1,16 @@
+package compose
+
+import (
+	"dagger.io/alpine"
+)
+
+// A container image to run the docker-compose client
+#Client: alpine.#Image & {
+	package: {
+		bash:             true
+		jq:               true
+		curl:             true
+		"openssh-client": true
+		"docker-compose": true
+	}
+}
diff --git a/stdlib/docker/compose/compose.cue b/stdlib/docker/compose/compose.cue
new file mode 100644
index 00000000..d67eb7e8
--- /dev/null
+++ b/stdlib/docker/compose/compose.cue
@@ -0,0 +1,152 @@
+package compose
+
+import (
+	"strconv"
+	"dagger.io/dagger"
+	"dagger.io/dagger/op"
+)
+
+#Up: {
+	ssh?: {
+		// ssh host
+		host: string @dagger(input)
+
+		// ssh user
+		user: string @dagger(input)
+
+		// ssh port
+		port: *22 | int @dagger(input)
+
+		// private key
+		key: dagger.#Secret @dagger(input)
+
+		// fingerprint
+		fingerprint?: string @dagger(input)
+
+		// ssh key passphrase
+		keyPassphrase?: dagger.#Secret @dagger(input)
+	}
+
+	// Accept either a contaxt, a docker-compose or both together
+	context?:     dagger.#Artifact @dagger(input)
+	composeFile?: string           @dagger(input)
+
+	// Image registries
+	registries: [...{
+		target?:  string
+		username: string
+		secret:   dagger.#Secret
+	}] @dagger(input)
+
+	#code: #"""
+			if [ -n "$DOCKER_HOSTNAME" ]; then
+				export DOCKER_HOST="ssh://$DOCKER_USERNAME@$DOCKER_HOSTNAME:$DOCKER_PORT"
+				# Start ssh-agent
+				eval $(ssh-agent) > /dev/null
+				# Add key
+				if [ -f "/key" ]; then
+					message="$(ssh-keygen -y -f /key < /dev/null 2>&1)" || {
+						>&2 echo "$message"
+						exit 1
+					}
+					ssh-add /key > /dev/null
+					if [ "$?" != 0 ]; then
+						exit 1
+					fi
+				fi
+				if [[ ! -z $FINGERPRINT ]]; then
+					mkdir -p "$HOME"/.ssh
+					# Add user's fingerprint to known hosts
+					echo "$FINGERPRINT" >> "$HOME"/.ssh/known_hosts
+				else
+					# Add host to known hosts
+					ssh -i /key -o "UserKnownHostsFile "$HOME"/.ssh/known_hosts" -o "StrictHostKeyChecking accept-new" -p "$DOCKER_PORT" "$DOCKER_USERNAME"@"$DOCKER_HOSTNAME" /bin/true > /dev/null 2>&1
+				fi
+			fi
+
+			cd /context
+			docker-compose build
+			docker-compose up -d
+		"""#
+
+	#up: [
+		op.#Load & {from: #Client},
+
+		for registry in registries {
+			op.#DockerLogin & {registry}
+		},
+
+		if context != _|_ {
+			op.#Copy & {
+				from: context
+				dest: "/context/"
+			}
+		},
+
+		if context == _|_ {
+			op.#Mkdir & {
+				path: "/context/"
+			}
+		},
+
+		if composeFile != _|_ {
+			op.#WriteFile & {
+				content: composeFile
+				dest:    "/context/docker-compose.yaml"
+			}
+		},
+
+		if ssh.keyPassphrase != _|_ {
+			op.#WriteFile & {
+				content: #"""
+					#!/bin/bash
+					cat /passphrase
+					"""#
+				dest: "/get_passphrase"
+				mode: 0o500
+			}
+		},
+
+		op.#WriteFile & {
+			content: #code
+			dest:    "/entrypoint.sh"
+		},
+
+		op.#Exec & {
+			always: true
+			args: [
+				"/bin/sh",
+				"--noprofile",
+				"--norc",
+				"-eo",
+				"pipefail",
+				"/entrypoint.sh",
+			]
+			env: {
+				if ssh != _|_ {
+					DOCKER_HOSTNAME: ssh.host
+					DOCKER_USERNAME: ssh.user
+					DOCKER_PORT:     strconv.FormatInt(ssh.port, 10)
+					if ssh.keyPassphrase != _|_ {
+						SSH_ASKPASS: "/get_passphrase"
+						DISPLAY:     "1"
+					}
+					if ssh.fingerprint != _|_ {
+						FINGERPRINT: ssh.fingerprint
+					}
+				}
+			}
+			mount: {
+				if ssh == _|_ {
+					"/var/run/docker.sock": "docker.sock"
+				}
+				if ssh.key != _|_ {
+					"/key": secret: ssh.key
+				}
+				if ssh.keyPassphrase != _|_ {
+					"/passphrase": secret: ssh.keyPassphrase
+				}
+			}
+		},
+	]
+}