#!/bin/sh
# instalador-initramfs.sh — empaqueta el instalador ENTERO en un initramfs que corre en RAM.
#
# ── PARA QUÉ ───────────────────────────────────────────────────────────────────────────────────
# Es lo que hace posible reemplazar el Linux que está corriendo **sin un USB**: se salta a este
# initramfs por `kexec` (`takana kernel kexec --jump`, ADR 0017), el sistema viejo muere en el salto
# y su disco queda libre ⇒ recién entonces su raíz se puede formatear. El mismo initramfs sirve
# arrancado por QEMU o por un medio cualquiera; el salto es sólo una forma de llegar.
#
# ── POR QUÉ EL PAYLOAD VIAJA ADENTRO Y NO EN EL DISCO ──────────────────────────────────────────
# Porque el disco es justamente lo que se va a formatear. Dejar el rootfs en la partición destino y
# formatearla después es morderse la cola; dejarlo en otra partición ata la instalación a que exista
# esa otra partición. En RAM no depende de nada: el precio es que el initramfs pesa lo que pese el
# rootfs, y por eso el script COMPRUEBA que entre en la memoria de la máquina antes de armarlo.
#
# Uso:
#   ROOTFS=<product-rootfs> scripts/instalador-initramfs.sh --root-part /dev/nvme0n1p2 --esp /dev/nvme0n1p1
#   # o sin destino fijo: lo decide el instalador al arrancar (--use-free-space)
#
# Salida: work/takana-instalador.cpio.gz  (+ el `takana kernel kexec` que lo salta, impreso al final)
set -eu
ROOT="$(cd "$(dirname "$0")/.." && pwd)"; cd "$ROOT"

ARGS_INSTALADOR=""
while [ $# -gt 0 ]; do ARGS_INSTALADOR="$ARGS_INSTALADOR $1"; shift; done
[ -n "$ARGS_INSTALADOR" ] || { echo "uso: $0 <argumentos que se le pasarán a takana-install-partition.sh>" >&2; exit 2; }

OUT="${OUT:-work/takana-instalador.cpio.gz}"
STAGE="${STAGE:-work/.instalador-initramfs-stage}"
IRD_TREE="${IRD_TREE:-work/.instalador-initramfs-tree}"
KERNEL="${KERNEL:-$(ls -dt store/*-linux-metal/boot/bzImage /store/*-linux-metal/boot/bzImage 2>/dev/null | head -1)}"
[ -n "${KERNEL:-}" ] && [ -r "$KERNEL" ] || { echo "no encuentro un bzImage de linux-metal (set KERNEL=)" >&2; exit 1; }

. "$(dirname "$0")/lib/herramientas-desde-store.sh"
MKE2FS=$(herramienta e2fsprogs sbin/mke2fs mke2fs)  || exit 1
SFDISK=$(herramienta util-linux sbin/sfdisk sfdisk) || exit 1
MMD=$(herramienta    mtools usr/bin/mmd   mmd)      || exit 1
MCOPY=$(herramienta  mtools usr/bin/mcopy mcopy)    || exit 1
MDIR=$(herramienta   mtools usr/bin/mdir  mdir)     || exit 1
CPIO=$(herramienta   libarchive usr/bin/bsdcpio cpio) || exit 1
GZIP=$(herramienta   gzip bin/gzip gzip)            || exit 1

# ── 1) el stage (rootfs a instalar + initramfs de pivote) lo arma el script validado ───────────
echo "==> stage del sistema a instalar"
STAGE_ONLY=1 KEEP_STORE=1 STAGE="$STAGE" ./scripts/install-image-efi.sh
[ -d "$STAGE/root" ] && [ -r "$STAGE/initramfs.cpio.gz" ] || { echo "el stage no quedó completo" >&2; exit 1; }

# ── 2) ¿entra en la RAM de esta máquina? ───────────────────────────────────────────────────────
# Un initramfs vive ENTERO en memoria, y el kernel no lo avisa con elegancia: se queda sin RAM al
# desempaquetar y el arranque muere sin un mensaje que diga por qué. Medirlo antes es la diferencia
# entre «no entra, usá el USB» y un cuelgue mudo en la máquina del usuario.
_mib_payload=$(du -sm "$STAGE/root" | cut -f1)
_mib_ram=$(( $(awk '/^MemTotal:/ {print $2}' /proc/meminfo) / 1024 ))
echo "    payload ${_mib_payload} MiB · RAM de esta máquina ${_mib_ram} MiB"
if [ "$(( _mib_payload * 100 / _mib_ram ))" -gt 40 ]; then
    echo "instalador-initramfs: el payload es más del 40 % de la RAM ($_mib_payload de $_mib_ram MiB)." >&2
    echo "                      En RAM van el initramfs comprimido Y su contenido desempaquetado." >&2
    echo "                      Este camino no sirve para este rootfs: instalá desde un medio." >&2
    exit 1
fi

# ── 3) el árbol del initramfs ──────────────────────────────────────────────────────────────────
echo "==> árbol del initramfs-instalador"
rm -rf "$IRD_TREE"; mkdir -p "$IRD_TREE/bin" "$IRD_TREE/dev" "$IRD_TREE/proc" "$IRD_TREE/sys" \
    "$IRD_TREE/instalador/lib" "$IRD_TREE/payload"
BUSYBOX="${BUSYBOX:-$STAGE/root/bin/busybox}"
[ -r "$BUSYBOX" ] || { echo "no encuentro busybox estático en $BUSYBOX" >&2; exit 1; }
cp "$BUSYBOX" "$IRD_TREE/bin/busybox"; chmod 0755 "$IRD_TREE/bin/busybox"
for a in sh mount umount mkdir echo sleep ls cat sed awk grep cut tr basename dirname readlink \
         find sync reboot dmesg blkid setsid tee du wc head tail sort printf test sha256sum; do
    ln -sf busybox "$IRD_TREE/bin/$a"
done
# Las herramientas de disco NO salen de busybox: son las SELLADAS del corpus, estáticas musl. Dentro
# del initramfs no hay store del que resolverlas por hash, así que viajan copiadas y el instalador
# las recibe por `TOOL_<binario>` — el override que existe exactamente para esto.
for t in "$MKE2FS" "$SFDISK" "$MMD" "$MCOPY" "$MDIR"; do
    cp "$t" "$IRD_TREE/bin/$(basename "$t")"; chmod 0755 "$IRD_TREE/bin/$(basename "$t")"
done
TAKANA_MUSL="${TAKANA_MUSL:-$ROOT/target/x86_64-unknown-linux-musl/release/takana}"
[ -x "$TAKANA_MUSL" ] || TAKANA_MUSL="$ROOT/target/release/takana"
cp "$TAKANA_MUSL" "$IRD_TREE/bin/takana"; chmod 0755 "$IRD_TREE/bin/takana"
cp scripts/takana-install-partition.sh "$IRD_TREE/instalador/"
cp scripts/lib/herramientas-desde-store.sh "$IRD_TREE/instalador/lib/"
cp "$KERNEL" "$IRD_TREE/payload/bzImage"
mkdir -p "$IRD_TREE/payload/stage"
cp -a "$STAGE/root" "$IRD_TREE/payload/stage/root"
cp "$STAGE/initramfs.cpio.gz" "$IRD_TREE/payload/stage/initramfs.cpio.gz"
printf '%s\n' "$ARGS_INSTALADOR" > "$IRD_TREE/instalador/args"

# ── 4) /init — lo que corre cuando el kernel nuevo toma el control ─────────────────────────────
cat > "$IRD_TREE/init" <<'INIT'
#!/bin/busybox sh
# PID 1 del instalador. No pivota a ningún disco: instala y reinicia.
export PATH=/bin
mount -t proc     proc /proc 2>/dev/null || true
mount -t sysfs    sys  /sys  2>/dev/null || true
mount -t devtmpfs dev  /dev  2>/dev/null || true
# ⚠ El log va a tty0 **y** al serial, por lo mismo que el initramfs de pivote: el cmdline horneado
# deja /dev/console en ttyS0 y en un metal sin puerto serie todo esto sería invisible.
log() { echo "$*"; for d in /dev/tty0 /dev/ttyS0; do echo "$*" > "$d" 2>/dev/null || true; done; }
rescate() {
    log "TAKANA-INSTALADOR-FALLO: $*"
    log "  bloques visibles: $(ls /dev/sd* /dev/nvme* /dev/vd* /dev/mmcblk* 2>/dev/null | tr '\n' ' ')"
    log "  shell de rescate en tty1 — probá: blkid ; dmesg | tail -30"
    setsid -c sh -c 'exec /bin/sh </dev/tty1 >/dev/tty1 2>&1' >/dev/null 2>&1 &
    exec sh
}
log "TAKANA-INSTALADOR: arrancado en RAM. El sistema viejo YA NO CORRE: su disco está libre."

# Los discos tardan en enumerar (USB y NVMe sobre todo). Misma espera larga que el pivote, y por el
# mismo motivo: en QEMU aparecen al instante y la carrera sólo se ve en metal.
i=0
while [ "$i" -lt 60 ]; do
    ls /dev/nvme* /dev/sd* /dev/vd* /dev/mmcblk* >/dev/null 2>&1 && break
    i=$(( i + 1 )); [ $(( i % 5 )) -eq 0 ] && log "TAKANA-INSTALADOR: esperando discos… ${i}s"
    sleep 1
done

cd /instalador
ARGS=$(cat /instalador/args)
log "TAKANA-INSTALADOR: takana-install-partition.sh $ARGS"
# Las herramientas viajan en /bin: el instalador las toma por TOOL_<binario> en vez de buscarlas en
# un store que acá no existe. STAGE_LISTO dice que el árbol ya vino armado.
export TOOL_mke2fs=/bin/mke2fs TOOL_sfdisk=/bin/sfdisk TOOL_mmd=/bin/mmd \
       TOOL_mcopy=/bin/mcopy TOOL_mdir=/bin/mdir
export STAGE=/payload/stage STAGE_LISTO=1 KERNEL=/payload/bzImage HAMMER=/bin/takana
# shellcheck disable=SC2086
if sh /instalador/takana-install-partition.sh $ARGS --yes 2>&1 | tee /dev/tty0 > /dev/ttyS0; then
    log "TAKANA-INSTALADOR-OK: instalado. Reiniciando en 10 s (sacá el medio si lo hubiera)."
    sync; sleep 10; reboot -f
else
    rescate "el instalador salió con error — NADA se reinicia solo, mirá el log de arriba"
fi
INIT
chmod +x "$IRD_TREE/init"

# ── 5) empaquetar ──────────────────────────────────────────────────────────────────────────────
mkdir -p "$(dirname "$OUT")"
( cd "$IRD_TREE" && find . -print0 | "$CPIO" --null -o -H newc --owner=root:root 2>/dev/null | "$GZIP" -1 ) > "$OUT"
echo "==> $OUT: $(du -h "$OUT" | cut -f1)"
echo
echo "Para SALTAR a él sin pasar por el firmware (el userspace de esta máquina muere ACÁ):"
echo "  sudo takana kernel kexec --probe          # ¿se puede? exit 0 = sí"
echo "  sudo takana kernel kexec <linux-metal> --initrd $OUT \\"
echo "       --cmdline 'console=tty0 console=ttyS0,115200' --jump"
echo
echo "Para probarlo SIN saltar (y sin riesgo), en QEMU:"
echo "  qemu-system-x86_64 -m 4096 -kernel $KERNEL -initrd $OUT \\"
echo "       -append 'console=ttyS0' -drive file=<disco.img>,format=raw -nographic"