#!/bin/bash
# Builds a combined linux-ps5 .deb from staged artifacts in /out/staging.
# Runs inside the kernel-builder container; output goes to /out.
set -e

STAGING="/out/staging"
KVER=$(ls -1 "$STAGING/lib/modules" | head -1)
VER="${KVER%%-*}"
ARCH=amd64

PKG=$(mktemp -d)
mkdir -p "$PKG/DEBIAN"
mkdir -p "$PKG/boot"
mkdir -p "$PKG/lib/modules"

cp "$STAGING/boot/bzImage"  "$PKG/boot/vmlinuz-$KVER"
cp "$STAGING/System.map"    "$PKG/boot/System.map-$KVER"
cp "$STAGING/.config"       "$PKG/boot/config-$KVER"
cp -a "$STAGING/lib/modules/$KVER" "$PKG/lib/modules/"

# Kernel headers (for out-of-tree module builds)
if [ -d "$STAGING/headers" ]; then
    cp -a "$STAGING/headers/usr" "$PKG/usr"
    # Point /lib/modules/$KVER/build at the installed headers
    HDR_DEST="/usr/lib/modules/$KVER/build"
    mkdir -p "$PKG/usr/lib/modules/$KVER"
    cp -a "$STAGING/headers/lib/modules/$KVER/build" "$PKG/usr/lib/modules/$KVER/build"
    ln -sf "$HDR_DEST" "$PKG/lib/modules/$KVER/build"
fi

cat > "$PKG/DEBIAN/control" << CTRL
Package: linux-ps5
Version: $VER
Architecture: $ARCH
Maintainer: PS5 Linux
Provides: linux-image-$KVER, linux-headers-$KVER, linux-libc-dev
Conflicts: linux-image-$KVER, linux-headers-$KVER, linux-libc-dev
Replaces: linux-image-$KVER, linux-headers-$KVER, linux-libc-dev
Description: PS5 Linux kernel $KVER (image + modules + headers + libc-dev)
CTRL

cat > "$PKG/DEBIAN/postinst" << 'POSTINST'
#!/bin/bash
set -e
KVER="$(ls -1t /lib/modules | head -1)"
echo ">> linux-ps5 postinst: kernel $KVER"

# Rebuild initramfs
if command -v update-initramfs >/dev/null 2>&1; then
    echo ">> Rebuilding initramfs with update-initramfs for $KVER"
    update-initramfs -c -k "$KVER"
elif command -v mkinitcpio >/dev/null 2>&1; then
    echo ">> Rebuilding initramfs with mkinitcpio for $KVER"
    mkinitcpio -k "$KVER" -g "/boot/initrd.img-$KVER"
elif command -v mkinitfs >/dev/null 2>&1; then
    echo ">> Rebuilding initramfs with mkinitfs for $KVER"
    mkinitfs -o "/boot/initrd.img-$KVER" "$KVER"
fi

# Copy kernel + initrd to EFI partition
if [ -d /boot/efi ]; then
    OLD_BZ=$(ls -l /boot/efi/bzImage 2>/dev/null | awk '{print $5}') || true
    OLD_INITRD=$(ls -l /boot/efi/initrd.img 2>/dev/null | awk '{print $5}') || true
    echo ">> Copying /boot/vmlinuz-$KVER -> /boot/efi/bzImage"
    cp "/boot/vmlinuz-$KVER" /boot/efi/bzImage
    NEW_BZ=$(ls -l /boot/efi/bzImage | awk '{print $5}')
    echo ">>   bzImage: ${OLD_BZ:-<new>} -> $NEW_BZ bytes"
    echo ">> Copying /boot/initrd.img-$KVER -> /boot/efi/initrd.img"
    cp "/boot/initrd.img-$KVER" /boot/efi/initrd.img
    NEW_INITRD=$(ls -l /boot/efi/initrd.img | awk '{print $5}')
    echo ">>   initrd.img: ${OLD_INITRD:-<new>} -> $NEW_INITRD bytes"
    echo ">> Kernel $KVER deployed to /boot/efi"
else
    echo ">> /boot/efi not found, skipping EFI deploy"
fi
POSTINST
chmod 755 "$PKG/DEBIAN/postinst"

dpkg-deb --build --root-owner-group "$PKG" "/out/linux-ps5_${VER}_${ARCH}.deb"
rm -rf "$PKG"