#!/bin/bash

# Define color variables
LIGHT_BLUE='\033[1;34m'
LIGHT_GREEN='\033[1;32m'
NC='\033[0m' # No Color

# Clear the screen first
clear

# Check if the script is run as root, if not, attempt to switch to root using sudo -i
if [ "$(id -u)" -ne 0 ]; then
    echo -e "${LIGHT_BLUE}This script requires root privileges!${NC}"
    echo -e "${LIGHT_BLUE}Attempting to switch to root using 'sudo -i'...${NC}"
    exec sudo -i "$0" "$@"
    # exit 1
fi

# Ensure the network is active
if ! systemctl is-active --quiet network-online.target; then
    echo -e "${LIGHT_BLUE}Something went wrong!${NC}"
    echo -e "${LIGHT_BLUE}The host Network is not active!${NC}"

    # Check if wireless network is inactive
    if ! systemctl is-active --quiet wpa_supplicant; then
        echo -e "${LIGHT_BLUE}Wireless network is inactive. Are you using ethernet? Checking whether the wired network is activated.${NC}"
        
        # Check if any wired network interface is up
        wired_interface=$(ip link show up | grep -v 'lo\|wlan\|wlp' | awk '{print $2}' | sed 's/://')
        
        if [ -n "$wired_interface" ]; then
            echo -e "${LIGHT_BLUE}Wired network interface $wired_interface is up.${NC}"
        else
            echo -e "${LIGHT_BLUE}Something went wrong! No wired network interface is up.${NC}"
            echo -e "${LIGHT_BLUE}Please check if the wired network cable is plugged in or if the device is in 'down' mode.${NC}"
            echo -e "${LIGHT_BLUE}For more details, visit https://nixos.org/manual/nixos/stable/#sec-installation-manual-networking${NC}"
            exit 1
        fi
    else
        echo -e "${LIGHT_BLUE}Wireless network is active. Continuing the installation process.${NC}"
    fi
fi

clear

# Print title
echo -e "${LIGHT_GREEN}Welcome to Yaksha!${NC}"
echo -e "${LIGHT_GREEN}The Installation Script of LingmoNix."
echo -e "(C) 2025 By Yaksha Project"
echo -e "This script is only for INSTALL LingmoNix AS SINGLE BOOT."
echo -e "Dual-boot is not supported."
echo "---------------------------------"

# Step 1: Automatically detect boot mode (UEFI or BIOS)
echo -e "${LIGHT_GREEN}1. Detecting boot method${NC}"
if [ -f /sys/firmware/efi/fw_platform_size ]; then
    echo -e "${LIGHT_GREEN}UEFI boot mode detected.${NC}"
    boot_type="UEFI"
else
    echo -e "${LIGHT_GREEN}BIOS boot mode detected.${NC}"
    boot_type="BIOS"
fi

# Step 2: Automatically partition the disk
echo -e "${LIGHT_GREEN}2. Auto partitioning${NC}"
echo "Starting auto partitioning..."
echo -e "${LIGHT_BLUE}Please enter the target disk device (e.g., /dev/sda):${NC}"
echo "---------------------------------"
echo -e "Your host device are:"
lsblk
echo "---------------------------------"
while true; do
    read -p "Choose a disk: " disk </dev/tty

    # Check if the input is empty
    if [ -z "$disk" ]; then
        echo -e "${LIGHT_RED}No device entered, please provide a valid disk device!${NC}"
        continue
    fi

    # Check if the input is a valid block device
    if [ ! -b "$disk" ]; then
        echo -e "${LIGHT_RED}Invalid disk device, please check and enter a valid device!${NC}"
        continue
    fi

    # If the input is valid, break out of the loop
    break
done

echo -e "${LIGHT_GREEN}You have selected the disk device: $disk${NC}"
# Continue with the partitioning process

# Perform partitioning based on boot type
if [ "$boot_type" = "UEFI" ]; then
    # UEFI boot: create EFI partition and root partition
    parted "$disk" -- mklabel gpt
    parted "$disk" -- mkpart primary 512MB -8GB
    parted "$disk" -- mkpart primary linux-swap -8GB 100%
    parted "$disk" -- mkpart ESP fat32 1MB 512MB
    quit
elif [ "$boot_type" = "BIOS" ]; then
    # BIOS boot: create BIOS boot partition and root partition
    parted "$disk" -- mklabel msdos
    parted "$disk" -- mkpart primary 1MB -8GB
    parted "$disk" -- mkpart primary linux-swap -8GB 100%
fi

if [ $? -eq 0 ]; then
    echo -e "${LIGHT_GREEN}Partitioning complete.${NC}"
else
    echo -e "${LIGHT_BLUE}Partitioning failed, please check the disk device!${NC}"
    exit 1
fi

# Step 3: Format the partitions
echo -e "${LIGHT_GREEN}3. Formatting partitions${NC}"
if [ "$boot_type" = "UEFI" ]; then
    # Format EFI partition
    mkfs.ext4 "${disk}1"
    # Format root partition
    mkswap "${disk}2" && swapon "${disk}2"
    # Format ESP partition
    mkfs.fat -F 32 "${disk}3"
    # Mount the partitions
    mount "${disk}1" /mnt
    mkdir -p /mnt/boot
    mount "${disk}3" /mnt/boot
    nixos-generate-config --root /mnt
elif [ "$boot_type" = "BIOS" ]; then
    # Format root partition
    mkfs.ext4 "${disk}1"
    # Mount the partition
    mount "${disk}1" /mnt
    nixos-generate-config --root /mnt
fi

if [ $? -eq 0 ]; then
    echo -e "${LIGHT_GREEN}Partitions formatted and mounted to /mnt${NC}"
else
    echo -e "${LIGHT_BLUE}Formatting failed, please check the partitions and then try again!${NC}"
    exit 1
fi

# Step 4: Clone LingmoNix Starter repository and copy configuration files
while true; do
    read -p "Cloning Starter... Enter region (Global or China): " region </dev/tty
    case $region in
        Global)
            repo_url="https://github.com/LingmoOS-Testing/LingmoNix-Starter.git"
            break
            ;;
        China)
            repo_url="https://gitee.com/arkimium_76/LingmoNix-Starter.git"
            break
            ;;
        *)
            echo "Invalid option, please enter 'Global' or 'China'."
            ;;
    esac
done


# Clone the repository based on the region
git clone "$repo_url"
if [ $? -ne 0 ]; then
    echo -e "${LIGHT_BLUE}Failed to clone the repository, please check the network connection!${NC}"
    exit 1
fi

# Offer flavor options
echo "Please choose the flavor version to install:"
select flavor in general studio neon; do
    case $flavor in
        general|studio|neon)
            break
            ;;
        *)
            echo "Invalid option, please choose again."
            ;;
    esac
done

# Offer language options
echo "Please choose the language version:"
echo "1) English"
echo "2) Chinese"
while true; do
    read -p "Enter language (1 or 2): " lang_choice </dev/tty
    case $lang_choice in
        1|English)
            flavor_dir="${flavor,,}" # Convert flavor name to lowercase
            break
            ;;
        2|Chinese)
            flavor_dir="${flavor,,}-cn" # Convert flavor name to lowercase and append "-cn"
            break
            ;;
        *)
            echo "Invalid option, please enter 1 or 2."
            ;;
    esac
done

# Copy the configuration files
cp -r "LingmoNix-Starter/${flavor_dir}/configuration.nix" /mnt/etc/nixos/configuration.nix
if [ $? -eq 0 ]; then
    echo -e "${LIGHT_GREEN}Configuration file copied to /mnt/etc/nixos/${NC}"
else
    echo -e "${LIGHT_BLUE}Failed to copy configuration file, please check the path!${NC}"
    exit 1
fi

# Step 5: Begin NixOS installation
echo -e "${LIGHT_GREEN}Starting NixOS installation${NC}..."
if [ "$region" = "China" ]; then
    echo -e "${LIGHT_GREEN}Using Tsinghua mirror for China...${NC}"
    nixos-install -j 8 --cores 1 --option substituters https://mirrors.tuna.tsinghua.edu.cn/nix-channels/store
else
    nixos-install -j 8 --cores 1
fi
if [ $? -eq 0 ]; then
    echo -e "${LIGHT_GREEN}NixOS installation completed${NC}"
else
    echo -e "${LIGHT_RED}NixOS installation failed!${NC}"
    exit 1
fi


# Step 6: Post-installation configuration
echo -e "${LIGHT_GREEN}6. Post-installation configuration${NC}"

# Ask for the username
read -p "Please enter your username: " username </dev/tty

# Ensure the user exists before setting the password
if id "$username" &>/dev/null; then
    echo "Setting password for user '$username'."
    passwd "$username"
    echo -e "${LIGHT_GREEN}Password for user '$username' has been set.${NC}"
else
    echo -e "${LIGHT_BLUE}User '$username' does not exist! Please try again!${NC}"
    continue
fi

# Step 7: Install Cutefish Desktop (optional)
echo -e "${LIGHT_GREEN}7. Installing Cutefish Desktop (optional)${NC}"
read -p "[For Experimental] Would you like to install the Cutefish Desktop? (y/n): " install_cutefish </dev/tty

if [[ $install_cutefish =~ ^[Yy]$ ]]; then
    git clone https://github.com/p3psi-boo/nix-cutefish.git 
    cd nix-cutefish && nix build
    if [ $? -eq 0 ]; then
        echo -e "${LIGHT_GREEN}Cutefish Desktop installed${NC}"
    else
        echo -e "${LIGHT_BLUE}Cutefish Desktop installation failed${NC}"
    fi
else
    echo -e "${LIGHT_GREEN}Skipping Cutefish installation${NC}"
fi

# Installation complete
logout
echo -e "${LIGHT_GREEN}Thank you for using Yaksha!${NC}"
echo -e "${LIGHT_GREEN}Please reboot and then remove your installation medium!${NC}"
echo -e "${LIGHT_GREEN}ENJOY YOUR SYSTEM, GOODBYE! :)${NC}"