#!/bin/sh

GRUB_PATH=boot/grub/grub.cfg
ISOLINUX_PATH=isolinux/live.cfg

# Detect serial console type from existing GRUB config
if grep -q "console=ttyAMA" "${GRUB_PATH}"; then
    SERIAL_TTY="ttyAMA0"
else
    SERIAL_TTY="ttyS0"
fi

KVM_CONSOLE="console=${SERIAL_TTY},115200 console=tty0"
SERIAL_CONSOLE="console=tty0 console=${SERIAL_TTY},115200"

# Grub.cfg Update
GRUB_MENUENTRY=$(sed -e '/menuentry.*hotkey.*/,/^}/!d' -e 's/--hotkey=l//g' $GRUB_PATH)

# Update KVM menuentry name
sed -i 's/"Live system \((.*vyos)\)"/"Live system \1 - KVM console"/' $GRUB_PATH

# Insert serial menuentry
echo "$GRUB_MENUENTRY" | sed \
    -e 's/"Live system \((.*vyos)\)"/"Live system \1 - Serial console"/' \
    -e "s/$KVM_CONSOLE/$SERIAL_CONSOLE/g" >> $GRUB_PATH

# Live.cfg Update
ISOLINUX_MENUENTRY=$(sed -e '/label live-vyos$/,/^\tappend.*/!d' $ISOLINUX_PATH)

# Update KVM menuentry name
sed -i 's/Live system \((.*vyos)\)/Live system \1 - KVM console/' $ISOLINUX_PATH

# Insert serial menuentry
echo "\n$ISOLINUX_MENUENTRY" | sed \
    -e 's/live-vyos/live-vyos-serial/' \
    -e 's/^\tmenu label \^/\tmenu label /' \
    -e '/^\tmenu default/d' \
    -e 's/Live system \((.*vyos)\)/Live system \1 - Serial console/' \
    -e "s/$KVM_CONSOLE/$SERIAL_CONSOLE/g" >> $ISOLINUX_PATH