61 lines
3.2 KiB
Bash
Executable File
61 lines
3.2 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# Harness VM (10.20.0.11). Idempotent: first boot (startup script) and `make vm-sync`.
|
|
# Secrets come from Secret Manager through the VM service account (no key files, Terraform design 6.11):
|
|
# selfheal-k8s-harness-token, selfheal-k8s-ca-cert, selfheal-vm-ssh-private-key, selfheal-db-password, selfheal-harness-token
|
|
# Runner inputs (ALB DNS, AWS role ARN) come from instance metadata set by Terraform (20-compute).
|
|
set -euo pipefail
|
|
cd "$(dirname "$0")"
|
|
MD=http://metadata.google.internal/computeMetadata/v1
|
|
md() { curl -sf -H 'Metadata-Flavor: Google' "$MD/$1"; }
|
|
PROJECT="$(md project/project-id)"
|
|
ALB_DNS="$(md instance/attributes/alb-dns)"
|
|
AWS_ROLE_ARN="$(md instance/attributes/aws-role-arn)"
|
|
command -v jq >/dev/null || { apt-get update -qq && apt-get install -y -qq jq >/dev/null; }
|
|
TOKEN="$(md instance/service-accounts/default/token | jq -r .access_token)"
|
|
secret() { curl -sf -H "Authorization: Bearer ${TOKEN}" "https://secretmanager.googleapis.com/v1/projects/${PROJECT}/secrets/$1/versions/latest:access" | jq -r .payload.data | base64 -d; }
|
|
|
|
SEC=/opt/selfheal/harness/secrets
|
|
mkdir -p "$SEC"; chmod 700 /opt/selfheal/harness "$SEC"
|
|
umask 077
|
|
secret selfheal-k8s-harness-token > "$SEC/k8s-token"
|
|
secret selfheal-k8s-ca-cert > "$SEC/k8s-ca.crt"
|
|
secret selfheal-vm-ssh-private-key > "$SEC/vm-ssh-key"
|
|
secret selfheal-db-password > "$SEC/db-password"
|
|
secret selfheal-harness-token > "$SEC/harness-token"
|
|
unset TOKEN
|
|
for f in k8s-token k8s-ca.crt vm-ssh-key db-password harness-token; do [ -s "$SEC/$f" ] || { echo "empty secret file $f" >&2; exit 1; }; done
|
|
chown -R 10001:10001 "$SEC"; chmod 600 "$SEC"/* # runner runs as uid 10001
|
|
|
|
ENV_FILE=/opt/selfheal/harness.env
|
|
if [ ! -f "$ENV_FILE" ] || ! grep -q '^MEASURE_DB_PASSWORD=' "$ENV_FILE"; then
|
|
MDB_PW="$(head -c 24 /dev/urandom | base64 | tr -d '/+=')"
|
|
else
|
|
MDB_PW="$(sed -n 's/^MEASURE_DB_PASSWORD=//p' "$ENV_FILE")"
|
|
fi
|
|
HARNESS_TAG="$(cat image.tag)"
|
|
{
|
|
printf 'ALB_DNS=%s\n' "$ALB_DNS"
|
|
printf 'AWS_ROLE_ARN=%s\n' "$AWS_ROLE_ARN"
|
|
printf 'AWS_REGION=ap-northeast-1\n'
|
|
printf 'GCP_PROJECT=%s\n' "$PROJECT"
|
|
printf 'GCP_REGION=asia-northeast1\n'
|
|
printf 'LEGACY_MIG=selfheal-legacy-mig\n'
|
|
printf 'K8S_API=https://10.10.0.10:6443\n'
|
|
printf 'MIMIR_URL=http://10.20.0.10:9009/prometheus\n'
|
|
printf 'AI_DIAGNOSE_URL=http://10.20.0.12:8080/diagnose\n'
|
|
printf 'HARNESS_TAG=%s\n' "$HARNESS_TAG"
|
|
printf 'MEASURE_DB_PASSWORD=%s\n' "$MDB_PW"
|
|
} > "$ENV_FILE"
|
|
unset MDB_PW
|
|
|
|
docker compose --env-file "$ENV_FILE" -f compose.yaml up -d --pull always --remove-orphans
|
|
# runner re-reads secret files per call; recreate it anyway so a new image tag / env takes effect
|
|
docker compose --env-file "$ENV_FILE" -f compose.yaml up -d --force-recreate --no-deps runner
|
|
for i in $(seq 1 30); do
|
|
h=$(curl -s -o /dev/null -w '%{http_code}' http://127.0.0.1:8080/healthz); b=$(curl -s -o /dev/null -w '%{http_code}' http://127.0.0.1:9115/-/healthy)
|
|
a=$(curl -s -o /dev/null -w '%{http_code}' http://127.0.0.1:12345/-/ready)
|
|
if [ "$h$b$a" = "200200200" ]; then break; fi; sleep 5
|
|
done
|
|
echo "ready: runner=$h blackbox=$b alloy=$a"
|
|
docker compose --env-file "$ENV_FILE" -f compose.yaml ps --format 'table {{.Name}}\t{{.Status}}'
|