%post --nochroot --erroronfail --interpreter=/usr/bin/bash --log=/mnt/sysroot/var/log/anaconda/dasos.log
set -e

log() {
    echo "[EXTRA] $(date '+%Y-%m-%d %H:%M:%S') $*"
}

# Extract inst.stage2 and kernel type from boot command line
STAGE2=$(cat /proc/cmdline | tr ' ' '\n' | grep '^inst.stage2=' | cut -d= -f2- || true)
REPO_BASE="/run/install/repo"
KERNEL_TYPE=$(cat /proc/cmdline | tr ' ' '\n' | grep -i '^kernel=' | cut -d= -f2 | tr '[:upper:]' '[:lower:]')
[[ -z "$KERNEL_TYPE" ]] && KERNEL_TYPE="openeuler"

# Prepare temporary yum repo
RPMS_LIST_TODO="$(mktemp /tmp/dasos-pkg-todo.XXXXXX)"
RPM_INFO="$(mktemp /tmp/dasos-installed.XXXXXX)"
EXTRA_REPO_NAME="TmpLive_$(head /dev/urandom | tr -dc A-Za-z0-9 | head -c3)"
TMP_REPO_FILE="/tmp/${EXTRA_REPO_NAME}.repo"
mkdir -pv /tmp
rm -vf "$TMP_REPO_FILE"

if [[ "$STAGE2" =~ ^https?:// ]]; then
    log "Detected remote installation source: $STAGE2"
    cat > "$TMP_REPO_FILE" <<EOF
[${EXTRA_REPO_NAME}]
name=${EXTRA_REPO_NAME}
baseurl=${STAGE2}
enabled=1
gpgcheck=0
EOF
else
    log "Detected local installation source: $REPO_BASE"
    if [[ -d "${REPO_BASE}/Packages" && -d "${REPO_BASE}/repodata" ]]; then
        cat > "$TMP_REPO_FILE" <<EOF
[${EXTRA_REPO_NAME}]
name=${EXTRA_REPO_NAME}
baseurl=file://${REPO_BASE}
enabled=1
gpgcheck=0
EOF
    else
        log "ERROR: Missing Packages or repodata in ${REPO_BASE}"
        exit 1
    fi
fi
log "Temporary YUM repo configured: $TMP_REPO_FILE"

# List available repos
log "Available YUM repositories:"
echo "-------+ TMP REPO +-------" >> $RPM_INFO
dnf -c "$TMP_REPO_FILE" --disablerepo="*" --enablerepo="${EXTRA_REPO_NAME}" repolist >> $RPM_INFO || { log "ERROR: Failed to list YUM repos"; exit 1; }

# Generate install list
log "Generating install list to $RPMS_LIST_TODO"
echo "kernel-6.6.0-*.${KERNEL_TYPE}" > "${RPMS_LIST_TODO}"
echo "das-os-m-repos-${KERNEL_TYPE}" >> "${RPMS_LIST_TODO}"

# Install essential kernel packages
log "Installing essential kernel packages into disk"
while IFS= read -r pkg; do
    [[ -z "$pkg" ]] && continue
    log "Installing package: $pkg"
    dnf -c "$TMP_REPO_FILE" --disablerepo="*" --enablerepo="${EXTRA_REPO_NAME}" \
        --installroot=/mnt/sysimage/ -y install "$pkg" || { log "ERROR: Failed to install package $pkg"; exit 1; }
done < "$RPMS_LIST_TODO"

# Collect driver packages
echo "-------+ ${KERNEL_TYPE} driver rpm list +-------" >> "$RPM_INFO"
dnf -c "$TMP_REPO_FILE" --disablerepo="*" --enablerepo="${EXTRA_REPO_NAME}" list --available | \
awk -v sfx="$KERNEL_TYPE" '$2 ~ ("\\." sfx "$") { split($1,a,"."); print a[1] }' >> "${RPMS_LIST_TODO}.raw"

# Exclude kernel, debug, perf packages
awk '
$0 ~ /^[[:space:]]*$/ { next }
$0 ~ /^kernel(-|$)/ { next }
$0 ~ /^(bpftool|perf|python[23]-perf)$/ { next }
$0 ~ /-(debuginfo|debugsource|devel|doc|test)$/ { next }
{ print }
' "${RPMS_LIST_TODO}.raw" | sort -u > "${RPMS_LIST_TODO}.drivers"

if [[ -s "${RPMS_LIST_TODO}.drivers" ]]; then
    cat "${RPMS_LIST_TODO}.drivers" >> "$RPM_INFO"
else
    echo "[INFO] No valid driver packages found for kernel type ${KERNEL_TYPE}" >> "$RPM_INFO"
fi

# Install driver packages
INSTALL_DRIVERS=""
while IFS= read -r pkg; do
    [[ -z "$pkg" ]] && continue
    INSTALL_DRIVERS+="${pkg}-*.${KERNEL_TYPE} "
done < "${RPMS_LIST_TODO}.drivers"

if [[ -n "$INSTALL_DRIVERS" ]]; then
    log "Installing driver packages into target /mnt/sysimage..."
    dnf -c "$TMP_REPO_FILE" --disablerepo="*" --enablerepo="${EXTRA_REPO_NAME}" \
        --installroot=/mnt/sysimage/ -y install $INSTALL_DRIVERS || { log "ERROR: Failed to install driver packages"; exit 1; }
else
    log "No driver packages found to install."
fi

# Remove old kernels not matching suffix
log "Removing old kernels not matching suffix: $KERNEL_TYPE"
mapfile -t REMOVE_KERNEL_PKGS < <(chroot /mnt/sysimage rpm -qa | grep '^kernel' | grep -v "\.${KERNEL_TYPE}" || true)
for pkg in "${REMOVE_KERNEL_PKGS[@]}"; do
    log "Removing package: $pkg"
    dnf -c "$TMP_REPO_FILE" --disablerepo="*" --enablerepo="${EXTRA_REPO_NAME}" \
        --installroot=/mnt/sysimage/ -y remove "$pkg" || { log "ERROR: Failed to remove package $pkg"; exit 1; }
done

# Installed RPMs list
echo "-------+ installed rpms in Target OS +-------" >> $RPM_INFO
dnf -c "$TMP_REPO_FILE" --disablerepo="*" --enablerepo="${EXTRA_REPO_NAME}" --installroot=/mnt/sysimage/ list --installed >> $RPM_INFO || { log "ERROR: Failed to list installed packages"; exit 1; }

# Clean DNF cache and logs
log "Cleaning dnf cache and logs in /mnt/sysimage"
rm -rf /mnt/sysimage/var/cache/dnf/* /mnt/sysimage/var/log/dnf* /mnt/sysimage/var/log/yum.log || true

log "Post --nochroot dasos script completed successfully."
%end

%post --erroronfail --interpreter=/usr/bin/bash --log=/var/log/anaconda/dasos.log
set -e

log() {
    echo "[EXTRA] $(date '+%Y-%m-%d %H:%M:%S') $*"
}

# Sync console parameters from boot command line to /etc/default/grub
log "Processing console parameters for /etc/default/grub"
GRUB_DEFAULT_FILE="/etc/default/grub"
CMD_CONSOLE=$(grep -o 'console=[^ ]*' /proc/cmdline | tr '\n' ' ' | sed 's/ $//')
log "current console: ${CMD_CONSOLE}"

if [ -f "$GRUB_DEFAULT_FILE" ]; then
    if [ -n "$CMD_CONSOLE" ]; then
        # 检查 GRUB_CMDLINE_LINUX 是否存在
        if grep -q '^GRUB_CMDLINE_LINUX=' "$GRUB_DEFAULT_FILE"; then
            if grep -q 'console=' "$GRUB_DEFAULT_FILE"; then
                sed -i -r "s/(GRUB_CMDLINE_LINUX=\"[^\"]*)console=[^\"]*(\"[^\"]*)/\1${CMD_CONSOLE}\2/g" "$GRUB_DEFAULT_FILE"
            else
                sed -i -r "s/(GRUB_CMDLINE_LINUX=\"[^\"]*)\"/\1 ${CMD_CONSOLE}\"" "$GRUB_DEFAULT_FILE"
            fi
        else
            echo "GRUB_CMDLINE_LINUX=\"${CMD_CONSOLE}\"" >> "$GRUB_DEFAULT_FILE"
        fi
        log "Updated GRUB_CMDLINE_LINUX:"
        grep '^GRUB_CMDLINE_LINUX=' "$GRUB_DEFAULT_FILE"
    else
        log "No console parameter found in kernel command line, skipping update."
    fi
else
    log "ERROR: $GRUB_DEFAULT_FILE not found, cannot sync console parameters"
    exit 1
fi

# Add additional GRUB parameters
log "Adding additional GRUB parameters"
# Define additional GRUB parameters to be added
ADDITIONAL_PARAMS="crashkernel=auto rd.shell=0"

# Check and add each parameter to GRUB configuration
if [ -f "$GRUB_DEFAULT_FILE" ]; then
    for param in $ADDITIONAL_PARAMS; do
        # Check if GRUB_CMDLINE_LINUX exists
        if grep -q '^GRUB_CMDLINE_LINUX=' "$GRUB_DEFAULT_FILE"; then
            # Check if parameter already exists
            if ! grep -q "$param" "$GRUB_DEFAULT_FILE"; then
                # Add new parameter to existing configuration
                sed -i -r "s/(GRUB_CMDLINE_LINUX=\"[^\"]*)\"/\1 $param\"/g" "$GRUB_DEFAULT_FILE"
                log "Added $param to GRUB_CMDLINE_LINUX"
            else
                log "Parameter $param already exists in GRUB_CMDLINE_LINUX"
            fi
        else
            log "No console parameter found in kernel command line, skipping update."
        fi
    done
else
    log "ERROR: $GRUB_DEFAULT_FILE not found, cannot add additional GRUB parameters"
    exit 1
fi

# Update OS_NAME and OS_VERS in /etc/grub.d/10_linux
sed -i -e "s/^[[:space:]]*OS_NAME=.*$/  OS_NAME=\"\$(. \/etc\/DASOS-release; echo \${NAME})\"/g" /etc/grub.d/10_linux
sed -i -e "s/^[[:space:]]*OS_VERS=.*$/  OS_VERS=\"\$(. \/etc\/DASOS-release; echo \${VERSION}\${VERSION_ID})\"/g" /etc/grub.d/10_linux
# Regenerate grub.cfg
[ -d /boot/efi/EFI/openEuler ] && grub2-mkconfig -o /boot/efi/EFI/openEuler/grub.cfg || true
[ -d /boot/grub2 ] && grub2-mkconfig -o /boot/grub2/grub.cfg || true
# Rebuild initramfs for installed kernels
for f in /boot/initramfs-*.img; do
    fname=$(basename "$f")
    case "$fname" in
        *".openeuler."*.img|*".kylin."*.img|*".uos."*.img)
            KER_VERREL=${fname#initramfs-}
            KER_VERREL=${KER_VERREL%.img}
            log "Rebuilding initramfs for $KER_VERREL"
            dracut -f "$f" "$KER_VERREL" || { log "ERROR: dracut failed for $f"; exit 1; }
            ;;
    esac
done

log "Chroot post-install script completed successfully."
%end