160 lines
3.1 KiB
CUE
160 lines
3.1 KiB
CUE
package http
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strconv"
|
|
|
|
"alpha.dagger.io/alpine"
|
|
"alpha.dagger.io/dagger"
|
|
"alpha.dagger.io/dagger/op"
|
|
"alpha.dagger.io/os"
|
|
)
|
|
|
|
#Get: #Do & {method: "GET"}
|
|
#Post: #Do & {method: "POST"}
|
|
#Put: #Do & {method: "PUT"}
|
|
#Delete: #Do & {method: "DELETE"}
|
|
#Do: {
|
|
url: string
|
|
method: "GET" | "POST" | "PUT" | "DELETE" | "PATH" | "HEAD"
|
|
|
|
request: {
|
|
body: string | *""
|
|
header: [string]: string | [...string]
|
|
token: dagger.#Secret | *null
|
|
}
|
|
|
|
ctr: os.#Container & {
|
|
image: alpine.#Image & {
|
|
package: curl: true
|
|
package: bash: true
|
|
package: jq: true
|
|
}
|
|
shell: path: "/bin/bash"
|
|
always: true
|
|
|
|
env: {
|
|
METHOD: method
|
|
HEADERS: json.Marshal(request.header)
|
|
BODY: request.body
|
|
URL: url
|
|
}
|
|
if request.token != null {
|
|
secret: "/token": request.token
|
|
}
|
|
command: #"""
|
|
curlArgs=(
|
|
"$URL"
|
|
-L --fail --silent --show-error
|
|
--write-out "%{http_code}"
|
|
-X "$METHOD"
|
|
-o /response
|
|
)
|
|
|
|
[ -n "$BODY" ] && curlArgs+=("-d" "'$BODY'")
|
|
|
|
headers="$(echo $HEADERS | jq -r 'to_entries | map(.key + ": " + (.value | tostring) + "\n") | add')"
|
|
while read h; do
|
|
curlArgs+=("-H" "'$h'")
|
|
done <<< "$headers"
|
|
if [ -e /token ]; then
|
|
curlArgs+=("-H" "Authorization: bearer $(cat /token)")
|
|
fi
|
|
curl "${curlArgs[@]}" > /status
|
|
"""#
|
|
}
|
|
|
|
statusCode: {
|
|
os.#File & {
|
|
from: ctr
|
|
path: "/status"
|
|
}
|
|
}.contents @dagger(output)
|
|
|
|
body: {
|
|
os.#File & {
|
|
from: ctr
|
|
path: "/response"
|
|
}
|
|
}.contents @dagger(output)
|
|
|
|
// Force os.#File exec before Atoi
|
|
response: {
|
|
"body": body
|
|
"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
|
|
|
|
// Env variables
|
|
env: [string]: string
|
|
|
|
#up: [
|
|
op.#Load & {
|
|
from: alpine.#Image & {
|
|
package: bash: true
|
|
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)"
|
|
for k, v in env {
|
|
"\(k)": v
|
|
}
|
|
}
|
|
},
|
|
]
|
|
}
|