#!/bin/bash
BLUE='\033[0;34m'  
YELLOW='\033[1;33m'
RED='\033[0;31m'
NC='\033[0m' # No Color

# echo the ascii art banner
echo -e "${YELLOW}"
echo "    ____                   ______           __             "
echo "   / __ \____  ___  ____  / ____/___  _____/ /____  _  __  "
echo "  / / / / __ \/ _ \/ __ \/ /   / __ \/ ___/ __/ _ \| |/_/  "
echo " / /_/ / /_/ /  __/ / / / /___/ /_/ / /  / /_/  __/>  <    "
echo " \____/ .___/\___/_/ /_/\____/\____/_/   \__/\___/_/|_|    "
echo "     /_/                       CorOS emulation environment "
echo -e "${NC}"

echo -e "${BLUE}========= Initializing CorOS emulation environment =========${NC}"

echo ""
echo -e "${YELLOW}[!]${NC} Mounting the CorOS update, this can take a couple of seconds..."
# Decompress archive and mount rootfs.ext3
tar -xvf /qc-fs/$UPDATE_FILE -C /qc-fs-uncompressed
if [ $? -ne 0 ]
then
    # Check if the file exists
    if [ ! -f /qc-fs/$UPDATE_FILE ]; then
        echo -e "${RED}Error: $UPDATE_FILE does not exist${NC}"
    else
        echo -e "${RED}Error: Failed to extract $UPDATE_FILE${NC}"
    fi
    exit 1
fi

echo "Mounting rootfs.ext3"
mount -t ext4 /qc-fs-uncompressed/rootfs.ext3 $QEMU_LD_PREFIX
echo -e "${YELLOW}Mounting rootfs.ext3 finished!${NC}"
echo ""

echo -e "${YELLOW}[!]${NC} Setting up the CorOS file system..."
echo "Mounting /etc/resolv.conf"
rm $QEMU_LD_PREFIX/etc/resolv.conf && touch $QEMU_LD_PREFIX/etc/resolv.conf
mount --bind /etc/resolv.conf $QEMU_LD_PREFIX/etc/resolv.conf

echo "Creating /media/p4"
mkdir -p $QEMU_LD_PREFIX/media/p4
echo "Creating /media/host and mounting host /mnt on it"
mkdir -p $QEMU_LD_PREFIX/media/host && mount --bind /mnt $QEMU_LD_PREFIX/media/host
echo ""

# Promt the user to install custom QT
# if the user types "y" or "Y" then install custom QT
read -p "Do you want to install the custom compiled QT? (y/n) " -r

if [[ $REPLY =~ ^[Yy]$ ]]
then
    echo -e "${YELLOW}[!]${NC} Installing QT"
    cd /qt_src && ./install_qt_compiled.sh
    echo -e "${YELLOW}Installing QT finished${NC}"
fi
echo ""

# Prompt user to chroot into the $QEMU_LD_PREFIX directory
# if the user types "y" or "Y" then chroot into the $QEMU_LD_PREFIX directory
# otherwise exit the script
read -p "Do you want to chroot into the created QC-filesystem? (y/n) " -r

if [[ $REPLY =~ ^[Yy]$ ]]
then
    echo ""
    echo "${YELLOW}[->]${NC} Chrooting into $QEMU_LD_PREFIX"
    chroot $QEMU_LD_PREFIX sh -c "uname -a" && chroot $QEMU_LD_PREFIX/
    echo ""
fi