337 lines
14 KiB
Bash
Executable File
337 lines
14 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# =============================================================================
|
|
# prefetch-images.sh — tarik SEMUA image Docker lebih dulu (agar install offline-ish)
|
|
# -----------------------------------------------------------------------------
|
|
# Daftar image TIDAK di-hardcode: diambil dari `docker compose config --images`,
|
|
# sehingga selalu sinkron dengan docker-compose.yml (termasuk image yang berada di
|
|
# balik profil, mis. dremio).
|
|
#
|
|
# CATATAN PENTING — satu image dibangun LOKAL:
|
|
# sicabe/airflow:3.3.2-om203-ol dibangun dari docker/airflow/Dockerfile.332
|
|
# (base apache/airflow:slim-3.3.2-python3.11). Ia TIDAK ada di registry publik,
|
|
# sehingga `pull` dari Docker Hub tidak mungkin.
|
|
# ✅ SEJAK 2026-10-02 image ini TERSEDIA di REGISTRY INTERNAL:
|
|
# registry.sicabe.com/airflow:3.3.2-om203-ol
|
|
# Script ini OTOMATIS ikut menariknya bila `AIRFLOW_IMAGE` menunjuk REGISTRY
|
|
# (host memuat titik, mis. `registry.sicabe.com/airflow:…`) — cukup set
|
|
# AIRFLOW_IMAGE=registry.sicabe.com/airflow:3.3.2-om203-ol
|
|
# di .env. Bila `AIRFLOW_IMAGE` dibiarkan nama LOKAL polos (`sicabe/airflow:…`),
|
|
# image DIKECUALIKAN dari pull dan wajib dibangun/di-retag manual (alur di bawah).
|
|
# Untuk host TANPA akses registry, tetap ada alur build-sekali-salin:
|
|
# 1) di SATU host: ./scripts/prefetch-images.sh
|
|
# docker build -t sicabe/airflow:3.3.2-om203-ol \
|
|
# -f docker/airflow/Dockerfile.332 docker/airflow
|
|
# ./scripts/prefetch-images.sh --save-airflow /tmp/airflow.tar
|
|
# 2) salin: scp /tmp/airflow.tar host2:/tmp/ ; scp … host3:/tmp/
|
|
# 3) di host lain: ./scripts/prefetch-images.sh --load-airflow /tmp/airflow.tar
|
|
#
|
|
# Pemakaian:
|
|
# ./scripts/prefetch-images.sh [--all | --host <nama>] [--list] [--no-base]
|
|
# ./scripts/prefetch-images.sh --save-airflow /tmp/airflow.tar
|
|
# ./scripts/prefetch-images.sh --load-airflow /tmp/airflow.tar
|
|
# =============================================================================
|
|
set -uo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
DEPLOY_DIR="$(cd "${SCRIPT_DIR}/.." && pwd)"
|
|
|
|
RED=$'\e[31m'; GRN=$'\e[32m'; YLW=$'\e[33m'; BLD=$'\e[1m'; NC=$'\e[0m'
|
|
# PENTING: semua diagnostik ke STDERR. stdout hanya untuk DATA (daftar image),
|
|
# karena collect_images() ditangkap dengan $( ) — pesan info yang ikut tertangkap
|
|
# akan dianggap nama image (bug yang ditemukan saat menguji --list).
|
|
ok() { echo -e " ${GRN}[OK]${NC} $*" >&2; }
|
|
inf() { echo -e " ${YLW}[..]${NC} $*" >&2; }
|
|
warn() { echo -e " ${YLW}[WARN]${NC} $*" >&2; }
|
|
err() { echo -e " ${RED}[!!]${NC} $*" >&2; }
|
|
|
|
MODE="all" # all | host
|
|
HOST_NAME=""
|
|
LIST_ONLY=0
|
|
# Dremio (opt-in di compose), image build lokal, dan base build Airflow DEFAULT
|
|
# DIKELUARKAN dari daftar prefetch. Ketiganya diaktifkan lewat flag bila memang
|
|
# diperlukan (mis. deployment yang menjalankan dremio / akan build Airflow).
|
|
WITH_BASE=0
|
|
WITH_DREMIO=0
|
|
SAVE_AIRFLOW=""
|
|
LOAD_AIRFLOW=""
|
|
|
|
usage() { sed -n '2,28p' "${BASH_SOURCE[0]}" | sed 's/^# \{0,1\}//'; exit 0; }
|
|
|
|
parse_args() {
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--all) MODE="all" ;;
|
|
--host) MODE="host"; HOST_NAME="${2:-}"; shift ;;
|
|
--list) LIST_ONLY=1 ;;
|
|
--no-base) WITH_BASE=0 ;; # dipertahankan: kini defaultnya sudah tanpa base
|
|
--with-base) WITH_BASE=1 ;;
|
|
--with-dremio) WITH_DREMIO=1 ;;
|
|
--save-airflow) SAVE_AIRFLOW="${2:-}"; shift ;;
|
|
--load-airflow) LOAD_AIRFLOW="${2:-}"; shift ;;
|
|
-h|--help) usage ;;
|
|
*) err "Argumen tidak dikenal: $1"; exit 2 ;;
|
|
esac
|
|
shift
|
|
done
|
|
}
|
|
|
|
require_docker() {
|
|
if ! command -v docker &>/dev/null; then
|
|
err "docker tidak ditemukan — jalankan ./scripts/prepare-host.sh lebih dulu."
|
|
exit 1
|
|
fi
|
|
}
|
|
|
|
# ─── Daftar image (tidak di-hardcode) ───────────────────────────────────────
|
|
compose_images() { # <file> [--env-file <f>] [--profile …]…
|
|
local file="$1"; shift
|
|
local out err_file
|
|
# stderr JANGAN ditelan: kalau kosong, tampilkan penyebabnya (pelajaran dari gate
|
|
# Fase 5 yang sempat menyembunyikan error compose sehingga gagal sulit dilacak).
|
|
#
|
|
# Catatan bug: versi pertama memakai `local err` + `trap … RETURN`. Trap RETURN
|
|
# berjalan SETELAH variabel local dibersihkan sehingga ${err} menjadi unbound, dan
|
|
# nama `err` juga menutupi FUNGSI err(). Kini tanpa trap + bernama err_file.
|
|
err_file="$(mktemp)"
|
|
out="$(docker compose -f "${file}" "$@" config --images 2>"${err_file}" | sort -u)"
|
|
if [[ -z "${out}" && -s "${err_file}" ]]; then
|
|
warn "docker compose gagal mendaftar image:"
|
|
sed 's/^/ /' "${err_file}" >&2
|
|
fi
|
|
rm -f "${err_file}"
|
|
printf '%s\n' "${out}"
|
|
}
|
|
|
|
# Isi ENV_ARGS (variabel global) dengan argumen env untuk compose.
|
|
# Bila .env BELUM ADA (server masih bersih — .env dibuat install.sh saat instalasi),
|
|
# pakai default dari config/environment.defaults.sh: ke-16 variabel yang diwajibkan
|
|
# compose (:?err) semuanya ada di sana, dan nilainya TIDAK memengaruhi daftar image
|
|
# (image di-pin di docker-compose.yml).
|
|
#
|
|
# PENTING: fungsi ini HARUS dipanggil di shell saat ini — bukan di dalam $( ) — supaya
|
|
# `set -a; source defaults` benar-benar berlaku untuk proses compose kita. Versi pertama
|
|
# saya memanggilnya dalam command substitution (subshell), jadi sourcing-nya sia-sia.
|
|
ENV_ARGS=()
|
|
resolve_env() { # <envfile>
|
|
local envfile="$1"
|
|
if [[ -f "${envfile}" ]]; then
|
|
ENV_ARGS=(--env-file "${envfile}")
|
|
return 0
|
|
fi
|
|
warn "berkas ${envfile} belum ada — memakai default config/environment.defaults.sh"
|
|
echo " (.env dibuat oleh install.sh saat instalasi; tidak memengaruhi daftar image)" >&2
|
|
set -a
|
|
# shellcheck disable=SC1091
|
|
. "${DEPLOY_DIR}/config/environment.defaults.sh"
|
|
set +a
|
|
ENV_ARGS=()
|
|
}
|
|
|
|
collect_images() {
|
|
local file envfile
|
|
if [[ "${MODE}" == "host" ]]; then
|
|
[[ -z "${HOST_NAME}" ]] && { err "--host memerlukan nama host."; exit 2; }
|
|
file="${DEPLOY_DIR}/generated/host-${HOST_NAME}/docker-compose.yml"
|
|
if [[ ! -f "${file}" ]]; then
|
|
err "Bundle host belum ada: ${file}"
|
|
echo " Jalankan dulu: ./install.sh --generate (atau --generate --inventory cluster.yml)"
|
|
exit 1
|
|
fi
|
|
envfile="${DEPLOY_DIR}/generated/host-${HOST_NAME}/.env"
|
|
[[ -f "${envfile}" ]] || envfile="${DEPLOY_DIR}/.env"
|
|
inf "sumber daftar image: bundle host '${HOST_NAME}'"
|
|
resolve_env "${envfile}"
|
|
compose_images "${file}" ${ENV_ARGS[@]+"${ENV_ARGS[@]}"} --profile all
|
|
else
|
|
local extra=()
|
|
if [[ ${WITH_DREMIO} -eq 1 ]]; then
|
|
extra=(--profile dremio)
|
|
inf "sumber daftar image: docker-compose.yml (profil all + dremio)"
|
|
else
|
|
inf "sumber daftar image: docker-compose.yml (profil all; dremio dikeluarkan)"
|
|
fi
|
|
resolve_env "${DEPLOY_DIR}/.env"
|
|
compose_images "${DEPLOY_DIR}/docker-compose.yml" ${ENV_ARGS[@]+"${ENV_ARGS[@]}"} \
|
|
--profile all ${extra[@]+"${extra[@]}"}
|
|
fi
|
|
}
|
|
|
|
# Base image Airflow diambil dari Dockerfile agar tidak ter-hardcode di sini.
|
|
airflow_base_image() {
|
|
local df="${DEPLOY_DIR}/docker/airflow/Dockerfile.slim"
|
|
[[ -f "${df}" ]] || return 1
|
|
grep -m1 '^FROM ' "${df}" | awk '{print $2}'
|
|
}
|
|
|
|
airflow_built_image() {
|
|
# shellcheck disable=SC1091
|
|
( set -a; [[ -f "${DEPLOY_DIR}/config/environment.defaults.sh" ]] && . "${DEPLOY_DIR}/config/environment.defaults.sh"
|
|
[[ -f "${DEPLOY_DIR}/.env" ]] && . "${DEPLOY_DIR}/.env"
|
|
echo "${AIRFLOW_IMAGE:-sicabe/airflow:3.3.2-om203-ol}" ) 2>/dev/null | tail -1
|
|
}
|
|
|
|
# Host registry dari nama image — untuk petunjuk diagnosa saat pull gagal.
|
|
# Image tanpa '/' atau dengan komponen pertama tanpa titik berasal dari Docker Hub
|
|
# (endpoint API-nya registry-1.docker.io).
|
|
registry_host() {
|
|
# Dipecah DUA baris: `local x="$1" y="${x%%/*}"` mengevaluasi RHS `y` SEBELUM `x`
|
|
# di-assign → "unbound variable" di bawah `set -u` (bug laten yang baru muncul saat
|
|
# fungsi ini dipanggil di jalur normal).
|
|
local img="$1"
|
|
local first="${img%%/*}"
|
|
# WAJIB ada '/' dulu: tanpa itu, "postgres:15" membuat ${img%%/*} = seluruh string
|
|
# dan tanda ':' adalah TAG, bukan port registry (bug yang ditemukan saat menguji).
|
|
if [[ "${img}" == */* && ( "${first}" == *.* || "${first}" == *:* || "${first}" == "localhost" ) ]]; then
|
|
echo "${first}"
|
|
else
|
|
echo "registry-1.docker.io"
|
|
fi
|
|
}
|
|
|
|
image_size_mb() {
|
|
docker image inspect -f '{{.Size}}' "$1" 2>/dev/null | awk '{printf "%d", $1/1024/1024}'
|
|
}
|
|
|
|
# ─── Aksi: simpan / muat image Airflow yang dibangun lokal ──────────────────
|
|
do_save_airflow() {
|
|
local built; built="$(airflow_built_image)"
|
|
if ! docker image inspect "${built}" &>/dev/null; then
|
|
err "Image '${built}' belum ada — bangun dulu:"
|
|
echo " docker build -t ${built} -f docker/airflow/Dockerfile.332 docker/airflow"
|
|
exit 1
|
|
fi
|
|
inf "menyimpan '${built}' ke ${SAVE_AIRFLOW} (berisi seluruh layer, termasuk base)"
|
|
docker save -o "${SAVE_AIRFLOW}" "${built}" && \
|
|
ok "tersimpan ($(du -h "${SAVE_AIRFLOW}" | cut -f1)) — salin ke host lain lalu pakai --load-airflow"
|
|
}
|
|
|
|
do_load_airflow() {
|
|
[[ -f "${LOAD_AIRFLOW}" ]] || { err "berkas tidak ditemukan: ${LOAD_AIRFLOW}"; exit 1; }
|
|
inf "memuat image dari ${LOAD_AIRFLOW}"
|
|
docker load -i "${LOAD_AIRFLOW}" && ok "image dimuat"
|
|
local built; built="$(airflow_built_image)"
|
|
docker image inspect "${built}" &>/dev/null && ok "'${built}' tersedia" || warn "image '${built}' belum terlihat"
|
|
}
|
|
|
|
# ─── Aksi: tarik image ──────────────────────────────────────────────────────
|
|
do_pull() {
|
|
local images base built built_local
|
|
images="$(collect_images)"
|
|
if [[ -z "${images}" ]]; then
|
|
err "daftar image kosong — periksa docker-compose.yml / bundle host."
|
|
exit 1
|
|
fi
|
|
|
|
base="$(airflow_base_image || true)"
|
|
built="$(airflow_built_image)"
|
|
|
|
# Image Airflow: dikecualikan HANYA bila namanya menunjuk LOKAL polos
|
|
# (`sicabe/airflow:…`, tidak ada host registry). Bila `AIRFLOW_IMAGE` menunjuk
|
|
# REGISTRY (mis. `registry.sicabe.com/airflow:…`, host memuat titik), image
|
|
# IKUT di-pull seperti image lain — sejak image dipublikasikan ke registry internal.
|
|
# Deteksi memakai registry_host(): nama Docker Hub polos -> "registry-1.docker.io".
|
|
built_local=0
|
|
if [[ -n "${built}" && "$(registry_host "${built}")" == "registry-1.docker.io" ]]; then
|
|
built_local=1
|
|
# Keluarkan dari daftar (bukan sekadar dilewati) supaya daftar = yang benar-benar diunduh.
|
|
images="$(printf '%s\n' "${images}" | grep -vxF "${built}" || true)"
|
|
fi
|
|
|
|
echo
|
|
echo -e "${BLD}Daftar image (${MODE})${NC}"
|
|
printf '%s\n' "${images}" | sed 's/^/ - /'
|
|
[[ -n "${base}" && ${WITH_BASE} -eq 1 ]] && echo " - ${base} (base build Airflow)"
|
|
if [[ -n "${built}" && ${built_local} -eq 1 ]]; then
|
|
echo " (dikecualikan: ${built} — dibangun lokal, tidak ada di Docker Hub; tersedia di registry internal)"
|
|
fi
|
|
[[ ${WITH_DREMIO} -eq 0 ]] && echo " (dikecualikan: dremio/dremio-oss — opt-in, aktifkan dengan --with-dremio)"
|
|
|
|
if [[ ${LIST_ONLY} -eq 1 ]]; then
|
|
echo
|
|
inf "--list: tidak ada yang ditarik."
|
|
return 0
|
|
fi
|
|
|
|
echo
|
|
echo -e "${BLD}Menarik image${NC}"
|
|
local failed=() okc=0
|
|
while IFS= read -r img; do
|
|
[[ -z "${img}" ]] && continue
|
|
# Image yang dibangun lokal tidak ada di registry — jangan dicoba di-pull.
|
|
if [[ ${built_local} -eq 1 && "${img}" == "${built}" ]]; then
|
|
inf "${img} — dibangun lokal, dilewati (lihat alur build-sekali-salin di bawah)"
|
|
continue
|
|
fi
|
|
if docker image inspect "${img}" &>/dev/null && [[ "${img}" != *":latest" ]]; then
|
|
ok "${img} (sudah ada)"
|
|
okc=$((okc + 1))
|
|
continue
|
|
fi
|
|
inf "pull ${img} …"
|
|
# Output pull ditangkap ke berkas, BUKAN dibuang: tanpa ini kegagalan hanya
|
|
# berbunyi "GAGAL" tanpa sebab (dialami saat minio/quay.io gagal di server).
|
|
local perr; perr="$(mktemp)"
|
|
if docker pull "${img}" >"${perr}" 2>&1; then
|
|
ok "${img}"
|
|
okc=$((okc + 1))
|
|
else
|
|
warn "GAGAL: ${img}"
|
|
tail -3 "${perr}" | sed 's/^/ /' >&2
|
|
local reg; reg="$(registry_host "${img}")"
|
|
echo " hint: periksa akses ke registry '${reg}' → curl -sI https://${reg}/v2/" >&2
|
|
failed+=("${img}")
|
|
fi
|
|
rm -f "${perr}"
|
|
done <<< "${images}"
|
|
|
|
if [[ -n "${base}" && ${WITH_BASE} -eq 1 ]]; then
|
|
if docker image inspect "${base}" &>/dev/null; then
|
|
ok "${base} (sudah ada)"
|
|
elif docker pull -q "${base}" >/dev/null 2>&1; then
|
|
ok "${base} (base build Airflow)"
|
|
else
|
|
warn "GAGAL: ${base} — build Airflow lokal akan mengunduh ulang"
|
|
failed+=("${base}")
|
|
fi
|
|
fi
|
|
|
|
echo
|
|
echo -e "${BLD}Ringkasan${NC}"
|
|
echo " berhasil/sudah ada : ${okc}"
|
|
if [[ ${#failed[@]} -gt 0 ]]; then
|
|
echo " gagal : ${#failed[@]}"
|
|
printf ' - %s\n' "${failed[@]}"
|
|
fi
|
|
|
|
# Ukuran total image yang terdaftar (perkiraan kebutuhan disk)
|
|
local total=0 size
|
|
while IFS= read -r img; do
|
|
size="$(image_size_mb "${img}")"
|
|
[[ -n "${size}" ]] && total=$((total + size))
|
|
done <<< "${images}"
|
|
echo " ukuran total : ~${total} MB (image yang terdaftar)"
|
|
|
|
if [[ ${built_local} -eq 1 ]] && ! docker image inspect "${built}" &>/dev/null; then
|
|
echo
|
|
warn "'${built}' BELUM ada dan ia TIDAK dapat di-pull (dibangun lokal)."
|
|
echo " Alur build-sekali-salin (opsi b):"
|
|
echo " di SATU host : docker build -t ${built} -f docker/airflow/Dockerfile.332 docker/airflow"
|
|
echo " ./scripts/prefetch-images.sh --save-airflow /tmp/airflow.tar"
|
|
echo " host lain : scp /tmp/airflow.tar <host>:/tmp/"
|
|
echo " ./scripts/prefetch-images.sh --load-airflow /tmp/airflow.tar"
|
|
fi
|
|
|
|
[[ ${#failed[@]} -eq 0 ]]
|
|
}
|
|
|
|
main() {
|
|
parse_args "$@"
|
|
require_docker
|
|
if [[ -n "${SAVE_AIRFLOW}" ]]; then do_save_airflow; exit $?; fi
|
|
if [[ -n "${LOAD_AIRFLOW}" ]]; then do_load_airflow; exit $?; fi
|
|
echo -e "${BLD}Prefetch image SiCABE${NC}"
|
|
do_pull
|
|
}
|
|
|
|
main "$@"
|