Implement http.#Wait to wait for a URL to be ready. It creates a dependency in the Cue tree

Signed-off-by: guillaume <guillaume.derouville@gmail.com>

stdlib/http/http.cue

@@ -6,6 +6,7 @@ import (
 
 	"alpha.dagger.io/alpine"
 	"alpha.dagger.io/dagger"
+	"alpha.dagger.io/dagger/op"
 	"alpha.dagger.io/os"
 )
 
@@ -84,3 +85,70 @@ import (
 		"statusCode": strconv.Atoi(statusCode)
 	}
 }
+
+// URL listener
+// Creates a dependency on URL
+#Wait: {
+	// URL to listen
+	url: string
+
+	// Waiting time between checks (sec.)
+	interval: int | *30
+	
+	// Max amount of retries
+	retries: int | *3
+
+	// Max initialization time (sec.)
+	startPeriod: int | *0
+
+	// Time until timeout (sec.)
+	timeout: int | *30
+
+	#up: [
+		op.#Load & {
+			from: alpine.#Image & {
+				package: bash: "=~5.1"
+				package: curl: true
+			}
+		},
+		op.#Exec & {
+			args: ["/bin/bash", "-c",
+				#"""
+					# (f: str -> int)
+					nb_retries=$(($NB_RETRIES+0))
+					starting_period=$(($START_PERIOD+0))
+
+					status="0"
+					SECONDS=0
+					# START_PERIOD implementation
+					while [ $SECONDS -lt $starting_period ]; do
+						status="$(curl --connect-timeout 1 -s -o /dev/null -w ''%{http_code}'' $HEALTH_URL)"
+						if [ "$status" == "200" ]; then
+							exit 0;
+						fi
+						sleep 1;
+					done
+
+					# TIMEOUT, INTERVAL, INTERVAL implementation
+					for ((i=0;i<NB_RETRIES;i++)); do
+						status="$(curl --connect-timeout $TIMEOUT -s -o /dev/null -w ''%{http_code}'' $HEALTH_URL)"
+						if [ "$status" == "200" ]; then
+							exit 0;
+						fi
+						sleep "$INTERVAL";
+					done
+
+					exit 1;
+					"""#,
+			]
+			always: true
+			env: {
+				HEALTH_URL: url
+				INTERVAL: "\(interval)"
+				NB_RETRIES: "\(retries)"
+				START_PERIOD: "\(startPeriod)"
+				TIMEOUT: "\(timeout)"
+			}
+		},
+	]
+}

stdlib/http/tests/http.cue

@@ -2,13 +2,47 @@ package http
 
 import (
 	"alpha.dagger.io/alpine"
+	"alpha.dagger.io/dagger"
+	"alpha.dagger.io/docker"
 	"alpha.dagger.io/os"
+	"alpha.dagger.io/random"
 )
 
+
+TestDockersocket: dagger.#Stream & dagger.#Input
+
+TestSuffix: random.#String & {
+	seed: ""
+}
+
+TestRun: docker.#Run & {
+	name:   "daggerci-test-wait-\(TestSuffix.out)"
+	ref:    "nginx"
+	socket: TestDockersocket
+	ports: ["8080:80"]
+}
+
+// Waits for TestRun to finish initializing
+Testhealth: #Wait & {
+	url: "http://localhost:8080/"
+}
+
+TestWait: {
+	query: os.#Container & {
+		image: alpine.#Image & {
+			package: bash: "=~5.1"
+			package: curl: true
+		}
+		command: #"""
+			test "$(curl -L --fail --silent --show-error --write-out "%{http_code}" "$URL" -o /dev/null)" = "200"
+			"""#
+		env: URL: Testhealth.url
+	}
+}
+
 TestRequest: {
 	req: #Get & {
-		url: "https://api.github.com/"
-		request: header: Accept: "application/json"
+		url: Testhealth.url
 	}
 
 	testRaw: os.#Container & {
@@ -24,9 +58,8 @@ TestRequest: {
 		shell: args: ["--noprofile", "--norc", "-eo", "pipefail", "-c"]
 		command: #Command
 	}
-}
-
-#Command: #"""
-	test "$(cat /content.json | jq -r .current_user_url)" = 'https://api.github.com/user'
+	#Command: #"""
+	cat /content.json | grep -q nginx >/dev/null
 	test "$STATUS" = "200"
 	"""#
+}