#!/bin/sh
RESET='\033[0m'
GREEN='\033[0;32m'
YELLOW='\033[0;33m'
RED='\033[0;31m'

disk="/dev/sda"
partition_prefix=""
efi_part_size=500
swap_size=8192
hostname="arch"
normal_user="tola"
rootfstype="ext4"
answer=n

check_system() {
    printf "$YELLOW ::$GREEN Cheking system$RESET  "
    if [[ $(uname) != "Linux" ]]; then
        printf "$RED This is working only on Linux systems (only Arch Linux)\n$RESET"
        exit 1
    fi
    printf "[$GREEN OK $RESET]\n"
}

check_root() {
    printf "$YELLOW ::$GREEN Cheking permissions$RESET  "
    if [[ "$(id -u)" != "0" ]]; then
        printf "$RED ERROR:$RESET You must be root\n"
        exit 1
    fi
    printf "[$GREEN OK $RESET]\n"
}

check_disk() {
    printf "$YELLOW Your disk is $disk?$RESET ($(echo $GREEN)y$RESET/$(echo $RED)n$RESET): "
    read answer
    if [[ "$answer" != "y" ]];then
        printf "$RED ERROR:$RESET Please Write your disk name in this script.\n"
        exit 1
    fi
}

partitioning_disk() {
    umount ${disk}*
    wipefs -af $disk
    printf "g\nn\n1\n2048\n$((($efi_part_size+1)*2048))\nn\n2\n$(((($efi_part_size+1)*2048)+2048))\n$(((($efi_part_size+1)*2048)+4096))\nn\n3\n$(((($efi_part_size+1)*2048)+6144))\n$((((($efi_part_size+1)*2048)+6144)+($swap_size*2048)))\nn\n4\n\n\nt\n1\nEFI System\nt\n2\nBIOS boot\nt\n3\nLinux swap\nt\n4\nLinux filesystem\nw\n" | fdisk $disk
}

formatting_partitions() {
    mkfs.vfat -F 32 ${disk}${partition_prefix}1
    mkswap ${disk}${partition_prefix}3
    case "$rootfstype" in
        ext4)
            mkfs.ext4 -L arch ${disk}${partition_prefix}4
            ;;
        btrfs)
            mkfs.btrfs -L arch -f ${disk}${partition_prefix}4
            ;;
        *)
            printf "$RED ERROR:$RESET Unsupported filesystem type: $rootfstype\n"
            exit 1
            ;;
    esac
}

mount_partitions() {
    mount ${disk}${partition_prefix}4 /mnt
    mkdir -p /mnt/boot/EFI
    mount ${disk}${partition_prefix}1 /mnt/boot/EFI
    swapon ${disk}${partition_prefix}3
}

install_base_system() {
    printf "$YELLOW ::$GREEN Installing the system$RESET\n"
    pacstrap /mnt base base-devel linux linux-firmware vim
}

configure_fstab() {
    printf "$YELLOW ::$GREEN Geerating fstab$RESET\n"
    genfstab -U /mnt > /mnt/etc/fstab
}

configure_timezone() {
    printf "$YELLOW ::$GREEN Installing timezone$RESET\n"
    arch-chroot /mnt bash -c 'ln -svf /usr/share/zoneinfo/Asia/Novosibirsk /etc/localtime'
}

configure_system_clock() {
    printf "$YELLOW ::$GREEN Configuring system clock$RESET\n"
    arch-chroot /mnt bash -c 'hwclock --systohc --utc'
}

configure_locale() {
    printf "$YELLOW ::$GREEN Installing locale$RESET\n"
    arch-chroot /mnt bash -c 'printf "en_US.UTF-8 UTF-8\nru_RU.UTF-8 UTF-8" >> /etc/locale.gen;locale-gen'
    arch-chroot /mnt bash -c 'printf "LC_ALL=en_US.UTF-8\nLANG=en_US.UTF-8" > /etc/locale.conf'
}

configure_console() {
    printf "$YELLOW ::$GREEN Configuring console$RESET\n"
    arch-chroot /mnt bash -c 'printf "FONT=Lat2-Terminus16\nKEYMAP=ruwin_alt_sh-UTF-8" > /etc/vconsole.conf'
}

configure_networking() {
    printf "$YELLOW ::$GREEN Installing hostname and hosts$RESET\n"
    arch-chroot /mnt /usr/bin/env -i hostname=$hostname bash -c 'echo $hostname > /etc/hostname'
    arch-chroot /mnt /usr/bin/env -i hostname=$hostname bash -c 'printf "127.0.0.1 localhost\n::1 localhost\n127.0.1.1 ${hostname}.localdomain ${hostname}" >> /etc/hosts'
}

configure_users() {
    printf "$YELLOW ::$GREEN Installing root password and normal user$RESET\n"
    arch-chroot /mnt /usr/bin/env -i arch_user=$normal_user bash -c 'passwd;useradd -m -g users -G wheel -s /bin/bash $arch_user;passwd $arch_user'
}

configure_sudo() {
    printf "$YELLOW ::$GREEN Installing and configuring sudo$RESET\n"
    arch-chroot /mnt bash -c 'pacman -S sudo;echo "%wheel ALL=(ALL:ALL) ALL" >> /etc/sudoers'
}

configure_grub() {
    printf "$YELLOW ::$GREEN Installing and configuring grub$RESET\n"
    arch-chroot /mnt /usr/bin/env -i grubdisk=$disk bash -c 'pacman -S grub efibootmgr os-prober;grub-install $grubdisk;grub-mkconfig -o /boot/grub/grub.cfg'
}

configure_networkmanager() {
    printf "$YELLOW ::$GREEN Installing and configuring NetworkManager$RESET\n"
    arch-chroot /mnt bash -c 'pacman -S netctl dhcpcd networkmanager network-manager-applet;systemctl enable NetworkManager'
}

configure_iphone_deps() {
    printf "$YELLOW ::$GREEN Installing dependencies for connecting iphone$RESET\n"
    arch-chroot /mnt bash -c 'pacman -S libimobiledevice usbmuxd'
}


umount_filesystem() {
    printf "$YELLOW ::$GREEN Unmounting filesystem$RESET\n"
    umount -R /mnt
}

checking_requirements() {
    check_system
    check_root
    check_disk
}

prepare_disk() {
    partitioning_disk
    formatting_partitions
}

configure_system() {
    printf "$YELLOW ::$GREEN System configuration$RESET\n"
    configure_fstab
    configure_timezone
    configure_system_clock
    configure_locale
    configure_console
    configure_networking
    configure_users
    configure_sudo
    configure_grub
    configure_networkmanager
    configure_iphone_deps
}

main() {
    checking_requirements
    prepare_disk
    mount_partitions
    install_base_system
    configure_system
    umount_filesystem
}

main