#!/bin/env bash

## author: si9ma
## my blog: si9ma.com
##
## use this script to install smallest archlinux
##

#  log
exec 2> >(tee -a "installer.log")

HEADER="Archlinux Installer"
CHROOT="arch-chroot /mnt"
EXIT_MSG="You have left from Archlinux Installer!" 
KEY_MAP=


### 1.Verify the boot mode

# if `sys/firware/efi/efivars` directory exist, the 
# boot mode is UEFI. otherwise,the boot mode is BIOS
if [ -d "/sys/firmware/efi/efivars" ]
then
    BOOT_MODE="UEFI"
else
    BOOT_MODE="BIOS"
fi


### 2.Update the system clock
timedatectl set-ntp true


### 3.Check the network connection
echo "Checking network connection."
echo "Please wait..."
ping archlinux.org -c 1 >/dev/null 2>&1

if [ $? != "0" ]
then
    echo "Network error, please check network connection"
    echo "exiting..."
    exit
fi


### 4.Keymap selection

# set the keymap to default(us)
loadkeys us
KEY_MAP="us"

# window
title="Keymap Selection"
msg="The default console keymap is US. Would you want to set a \
non-default keymap for your keyboard?  [ESC] to exit installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
dialog --ascii-lines --default-button "No" --title "$title" --backtitle "$HEADER" --yesno "$msg" 10 45

# never add a command between command `dialog -ascii-lines...` and command retval_out=$?
retval_out=$?
echo

case $retval_out in
    # 0) Yes -- reset keymap
        0)

        # get all keymap
        function list_all_keymap
        {
            for keymap in $(find /usr/share/kbd/keymaps/ -name "*.map.gz") # should enclose the pattern in quotes
            do
                # get basename
                file=${keymap##*/}
                printf "${file%%.*} ...... "
            done
        }

        # window
        title="Keymap Selection"
        backtitle="Archlinux Installer"
        msg="The default console keymap is US.Other keymaps can be chosen below.\
        \n\n[Arrow keys] to move,[ENTER] to select.	[ESC] to exit installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
        keymap_list="US default $(list_all_keymap)"

        dialog --no-cancel --ascii-lines --ok-label "Select" --title "$title" \
            --backtitle "$HEADER" --menu "$msg" 18 59 18 $keymap_list 2>tempfile

        retval_inner=$?
        echo
        case $retval_inner in
            0) # Yes
                choice=$(cat tempfile)

                #reset keymap temporary
                if [ "$choice" != "US" ]
                then
                    loadkeys "$choice"
                    KEY_MAP="$choice"
                else
                    loadkeys us
                    KEY_MAP="us"
                fi
                ;;
            255) # ESC
                echo $EXIT_MSG
                exit 255;;
        esac
        ;;

    # 255) ESC -- exit the installer
        255)
        echo $EXIT_MSG
        exit 255;;
esac


### 5.Partition the disks

# window
title="Partition the disks"
msg="Before you install Archlinux, you need to partition your hard disk.\n\n[ESC] to exit installer"
dialog --no-cancel --ascii-lines --title "$title" --backtitle "$HEADER" --msgbox "$msg" 9 45

retval=$?
echo

case $retval in
    # 0) Yes -- partition the disks
        0)
        # if the boot mode is uefi,we should create a efi system partition
        if [ "$BOOT_MODE" = "UEFI" ]
        then
            #window
            title="Partition the disks"
            msg="Hey! Your boot mode is UEFI, so you must create an ESP partition \
(EFI system partition). If, \n\n(1)You have installed Windows on your computer, and its boot \
mode is also UEFI, if so, you will see a partition of about 200MiB later. You should not delete \
it or format it, you will use it as your ESP.\n\n(2)You don't have an ESP partition on your computer.\
You need to create an ESP partition of about 200MiB.
\n\nRember, you must ensure that there are one \
and just only one ESP on your disks.\n\n[ESC] to exit the installer"
            dialog --ok-button "I Know" --ascii-lines --title "$title" --backtitle "$HEADER" --msgbox "$msg" 18 70

            retval=$?
            case $retval in
                255) #ESC,exit
                    echo 
                    echo $EXIT_MSG
                    exit 255
                    ;;
            esac
        else
            #window
            title="Partition the disks"
            msg="Hey! Your boot mode is Bios. In order to ensure everything goes well, I think you should ensure your disk is MBR(dos) partition structures\n\n\n[ESC] to exit the installer." 
            dialog --ok-button "I Know" --ascii-lines --title "$title" --backtitle "$HEADER" --msgbox "$msg" 18 70

            retval=$?
            case $retval in
                255) #ESC,exit
                    echo 
                    echo $EXIT_MSG
                    exit 255
                    ;;
            esac
        fi

        # windows
        title="Partition the disks"
        msg="Please select the disk you want to partition. Select nothing to skip.	[ESC] to exit the installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
        disk_list="$(fdisk -l | grep "Disk /dev/" | cut -d "," -f 1)"
        disk_list="${disk_list//Disk /}" # remove Disk
        disk_list="${disk_list// /}"
        disk_list="${disk_list//:/ }"
        disk_list=$(echo $disk_list | xargs)

        ## default to OFF
        count=0
        judge=0
        temp=
        for disk in $disk_list
        do
            count=$[ $count + 1 ]
            judge=$[ $count % 2 ]

            if [ "$judge" = "0" ]
            then
                # set /dev/sda default to ON
                if [ "$disk_temp" = "/dev/sda" ]
                then
                    temp="$temp $disk ON" 
                elif [[ "$disk_temp" = /dev/loop* ]] 
                then
                    continue # ignore /dev/loop*
                else
                    temp="$temp $disk OFF" 
                fi
            else
                if [[ "$disk" = /dev/loop* ]]
                then
                    disk_temp=$disk # ignore /dev/loop*
                else
                    temp="$temp $disk"
                    disk_temp=$disk
                fi
            fi
        done
        disk_list=$temp
        temp=

        dialog --no-cancel --ascii-lines --title "$title" --backtitle "$HEADER" --checklist "$msg" 18 65 18 $disk_list 2>tempfile

        retval=$?
        choice=$(cat tempfile)

        # if ESC,exit
        case $retval in 
            255) # ESC
                echo
                echo $EXIT_MSG
                exit 255;;
        esac

        # use cfdisk to partition disks
        for disk in $choice
        do
            cfdisk $disk
        done

        # info disk table change
        partprobe

        ;;

    ## 255) ESE -- exit the installer
        255)
        echo $EXIT_MSG
        exit 255;;
esac


### 6.Mount the file systems

# firstly,unmount all partition have mounted at /mnt
# swapoff
swap=$(swapon -s | tail -1 | cut -d " " -f 1)
[ -z "$swap" ] || swapoff "$swap"

umount -R /mnt

# select the mount point which you want mount as separated partition
title="Mount the file systems"

# if the boot mode is BIOS,we shouldn't create a separated partition for /boot
if [ "$BOOT_MODE" = "BIOS" ]
then
    msg="Select the mount point which you want mount as separated partition.	[ESC] to exit the installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
    dialog --no-cancel --ok-label "Continue" --ascii-lines --title "$title" --backtitle "$HEADER" --checklist "$msg" 18 75 18 \
        "/" "Entire system's root directory" "ON"\
        "/boot" "Boot loader files" "OFF"\
        "/bin" "Essential command binaries" "OFF"\
        "/home" "Users' home directories" "OFF"\
        "/var" "Variable files" "OFF"\
        "/etc" "Host-specific system-wide configuration files" "OFF"\
        "/lib" "Libraries essential for the binaries in /bin/ and /sbin/" "OFF"\
        "/opt" "Optional application software packages" "OFF"\
        "/usr" "Secondary hierarchy for read-only user data" "OFF"\
        "/run" "Run-time variable data" "OFF"\
        "swap" "swap partition" "OFF" 2>tempfile
else
    msg="Select the mount point which you want mount as separated partition. Your boot mode is UEFI,so \"/\" and \"/boot\" must be selected.		[ESC] to exit the installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
    dialog --no-cancel --ok-label "Continue" --ascii-lines --title "$title" --backtitle "$HEADER" --checklist "$msg" 18 75 18 \
        "/" "Entire system's root directory" "ON"\
        "/boot" "Boot loader files" "ON"\
        "/bin" "Essential command binaries" "OFF"\
        "/home" "Users' home directories" "OFF"\
        "/var" "Variable files" "OFF"\
        "/etc" "Host-specific system-wide configuration files" "OFF"\
        "/lib" "Libraries essential for the binaries in /bin/ and /sbin/" "OFF"\
        "/opt" "Optional application software packages" "OFF"\
        "/usr" "Secondary hierarchy for read-only user data" "OFF"\
        "/run" "Run-time variable data" "OFF"\
        "swap" "swap partition" "OFF" 2>tempfile
fi

retval=$?
choice=$(cat tempfile)

case $retval in 
    0) # continue
        # usable partition
        partition_list=$(fdisk -l -o Device | grep -P "^/dev/[a-z]+[0-9]+")

        # format partition and mount to suitable mount point use for loop
        for mount_point in $choice
        do

            # window
            title="Mount for $mount_point"
            msg="Please select a partition for $mount_point. the second column is partition info, format is:\n   [Current File System]/[Size]-->[File System Will Be Format To]\n\n[ESC] to exit the installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."

            # for swap partition
            if [ "$mount_point" = "swap" ]
            then
                title="Create swap partition"
                msg="Please select a partition as swap partition. the second column is partition info, format is:\n     [File System]/[Size]\n\nOR,you may be want to use a swap file but not a swap partition. if so,please select \"swapfile\".\n\n[ESC] to exit the installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
            fi

            # reset mount_point_list
            available_partition_list=
            info=
            for partition in $partition_list
            do
                # partition info
                info=$(df $partition -T -h | tail -1 | tr -s " " | cut -d " " -f 2)
                size=$(lsblk | grep "${partition/\/dev\//}" | head -1 | tr -s " " | cut -d " " -f 4)
                info="$info/$size"

                # will how to format file system
                if [ "$mount_point" = "/boot" ]
                then
                    info=${info}--">"vfat
                    available_partition_list="${available_partition_list} $partition(vfat) $info "
                elif [ "$mount_point" = "swap" ]
                then
                    available_partition_list="${available_partition_list} $partition $info"
                else
                    format_list="ext2 ext3 ext4" 
                    for format in $format_list
                    do
                        info_1=${info}--">"$format
                        available_partition_list="${available_partition_list} $partition($format) $info_1 "
                    done
                fi
            done

            # for swap file
            if [ "$mount_point" = "swap" ]
            then
                available_partition_list="${available_partition_list} swapfile swapfile"
            fi


            dialog --no-cancel --ok-button "Select" --ascii-lines --title "$title" --backtitle "$HEADER" --menu "$msg" 18 75 18 $available_partition_list 2>tempfile

            retval_1=$?
            selected_partition=$(cat tempfile)
            format=$(echo $selected_partition | cut -d "(" -f 2 | cut -d ")" -f 1 )
            selected_partition=${selected_partition%%(*}

            # get the disk name / on
            if [ "$mount_point" = "/" ]
            then
                # / on which disk 
                ROOT_DISK=$(echo $selected_partition | sed 's/[0-9]//g')
            fi

            # if ESC,exit
            case $retval_1 in
                255) # ESC
                    echo 
                    echo $EXIT_MSG
                    exit 255;
            esac

            # for swap partition
            if [ "$mount_point" = "swap" ]
            then
                # create swap file
                if [ "$selected_partition" = "swapfile" ]
                then
                    title="Create swap file"
                    msg="Please input the size(MB) of swap file.	[ESC] to exit installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
                    dialog --ascii-lines --no-cancel --title "$title" --backtitle "$HEADER" --inputbox "$msg" 10 50 512 2>tempfile

                    retval=$?
                    size=$(cat tempfile)

                    case $retval in
                        0) # create swap file
                            dd if=/dev/zero of=/mnt/swap bs=1M count=$size
                            mkswap /mnt/swap >/dev/null 2>&1
                            swapon /mnt/swap >/dev/null 2>&1
                            ;;
                        255) # ESC
                            echo
                            echo $EXIT_MSG
                            exit 255 ;;
                    esac
                else
                    mkswap $selected_partition >/dev/null 2>&1
                    swapon $selected_partition >/dev/null 2>&1
                fi

                # next mount point
                continue
            fi

            # format the partition
            # you may don't want't to format ESP
            if [ "$mount_point" = "/boot" ]
            then
                title="Mount for /boot -- Confirm ?"
                msg="$selected_partition may be already a ESP, so you may don't want to format it. Do you want to format it?    [ESC] to exit installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
                dialog --ascii-lines --default-button "Yes" --title "$title" --backtitle "$HEADER" --yesno "$msg" 10 45

                retval=$?
                echo

                case $retval in
                        0)
                        mkfs.$format $selected_partition >/dev/null 2>&1
                        ;;

                    255) # ESC
                        echo $EXIT_MSG
                        exit 255;;
                esac

            else
                if [ "$format" = "vfat" ]
                then
                    mkfs.$format $selected_partition >/dev/null 2>&1
                else
                    mkfs.$format -F $selected_partition >/dev/null 2>&1
                fi
            fi

            # mount the partition
            echo
            mkdir -p /mnt$mount_point

            # firstly, try to umount the partition
            # umount -l $selected_partition >/dev/null 2>&1 | true
            mount $selected_partition /mnt$mount_point

            # delete selected partition from partition list
            for partition in $partition_list
            do
                if [ "$partition" != "$selected_partition" ]
                then
                    temp="$temp $partition"
                fi
            done
            partition_list=$temp
            # reset temp
            temp=
        done
        ;;

    255) #ESC
        echo
        echo $EXIT_MSG
        exit 255;;
esac


### 7.Select the mirrors
# if there are mirrors add by Archlinux Installer,delete it
line=$(grep -n "Add by Archlinux Installer" /etc/pacman.d/mirrorlist | cut -d ":" -f 1)
if [ -z "$line" ]
then
    # do nothing
    echo
else
    line_begin=$[ $(echo "$line" | head -1) - 1]
    line_end=$[ $(wc -l /etc/pacman.d/mirrorlist | cut -d " " -f 1) - $(echo "$line" | tail -1) - 1]
    head -$line_begin /etc/pacman.d/mirrorlist >tempfile
    tail -$line_end /etc/pacman.d/mirrorlist >>tempfile
    mv tempfile /etc/pacman.d/mirrorlist
fi

function get_region_list # from /etc/pacman.d/mirrorlist
{
    # skip the file header
    first_usable_line=$[ $(grep -n "Server" /etc/pacman.d/mirrorlist |head -1| cut -d ":" -f 1) - 1 ]
    line_amount=$(cat /etc/pacman.d/mirrorlist | wc -l)
    # line will be used later
    work_line=$[ $line_amount - $first_usable_line + 1 ]

    # which column the region name is at
    first_word_column=$[ $(grep "China" /etc/pacman.d/mirrorlist | head -1 | sort -u | grep -o " " | wc -l) +1 ]

    # init region list
    region_list=$(tail -$work_line /etc/pacman.d/mirrorlist | grep "##" | cut -d " " -f $first_word_column | sort -u | xargs)

    # replace " " with "-"
    second_word_list=$(tail -$work_line /etc/pacman.d/mirrorlist | grep "##" | cut -d " " -f $[$first_word_column + 1] | sort -u | xargs)
    for word in $second_word_list
    do
        first_word=$(grep " $word" /etc/pacman.d/mirrorlist | cut -d " " -f $first_word_column | sort -u)
        echo $region_list | grep "$first_word-" >/dev/null
        if [ "$?" != "0" ]
        then
            region_list=${region_list/$first_word/}
        fi

        region_list="$region_list ${first_word}-${word}"
    done

    third_word_list=$(tail -$work_line /etc/pacman.d/mirrorlist | grep "##" | cut -d " " -f $[ $first_word_column + 2] | sort -u | xargs)
    for word in $third_word_list
    do
        second_word=$(grep " $word" /etc/pacman.d/mirrorlist | cut -d " " -f $[ $first_word_column + 1] | sort -u)
        region_list=${region_list/-$second_word/-${second_word}-$word}
    done
    echo $region_list | tr ' ' '\n' | sort -u | xargs
}

# get region list from /etc/pacman.d/mirrorlist
region_list=$(get_region_list)

# duplicate region,used for dialog
for region in $region_list
do
    temp="${temp} $region $region"
done
region_list=$temp

# select region
title="Select the mirrors"
msg="Before select the mirrors,Please select the region.	[ESC] to exit the installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
dialog --no-cancel --ok-button "Select" --ascii-lines --title "$title" --backtitle "$HEADER" --menu "$msg" 18 75 18 $region_list 2>tempfile

retval=$?
choice=$(cat tempfile)

# if ESC,exit
case $retval in 
    255) # ESC
        echo
        echo $EXIT_MSG
        exit 255;;
esac

mirrorlist=$(awk "/${choice/-/ }/{getline; print}" /etc/pacman.d/mirrorlist | cut -d " " -f 3 | xargs)

# without description
mirror_list=$mirrorlist
mirrorlist=${mirrorlist// / $choice } 
mirrorlist="${mirrorlist} $choice"
region=$choice

# select best mirrors
title="Select the mirrors"
msg="Select best mirrors for you.	[ESC] to exit the installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
dialog --no-cancel --ok-button "Select" --ascii-lines --title "$title" --backtitle "$HEADER" --menu "$msg" 18 75 18 $mirrorlist 2>tempfile

retval=$?
choice=$(cat tempfile)

# if ESC,exit
case $retval in 
    255) # ESC
        echo
        echo $EXIT_MSG
        exit 255;;
esac

first_usable_line=$[ $(grep -n "Server" /etc/pacman.d/mirrorlist |head -1| cut -d ":" -f 1) - 1 ]
pattern=$(head -$first_usable_line /etc/pacman.d/mirrorlist | tail -1)

# add comment
sed -i "0,/$pattern/s/$pattern/## Add by Archlinux Installer,for $region\n&/" /etc/pacman.d/mirrorlist

# insert best mirrors as first mirrors
sed -i "0,/$pattern/s/$pattern/Server = ${choice//\//\\\/}\n&/" /etc/pacman.d/mirrorlist

# add loop
for mirrors in $mirror_list
do
    if [ "$mirrors" != "$choice" ]
    then
        sed -i "0,/$pattern/s/$pattern/Server = ${mirrors//\//\\\/}\n&/" /etc/pacman.d/mirrorlist
    fi
done

# end
sed -i "0,/$pattern/s/$pattern/## Add by Archlinux Installer,for $region\n&/" /etc/pacman.d/mirrorlist
# newline
sed -i "0,/$pattern/s/$pattern/\n\n&/" /etc/pacman.d/mirrorlist


### 8.Install the base and base-devel packages
echo

pacstrap /mnt base base-devel

# if error
if [ "$?" != "0" ]
then
    pacstrap -i /mnt base base-devel
fi


### 9.Configure the system

## Fstab
genfstab -U -p /mnt >> /mnt/etc/fstab
fstab="$(cat /mnt/etc/fstab)"
fstab="${fstab/\/mnt/}"
echo "$fstab" >/mnt/etc/fstab

# clear
region_list=
temp=

## Time zone

# region
title="Time Zone"
msg="Please Select the region.		[ESC] to exit the installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
region_list=$(timedatectl list-timezones | cut -d "/" -f 1 | sort -u | xargs)

# duplicate
for region in $region_list
do
    temp="${temp} $region $region"
done
region_list=$temp
temp=

dialog --no-cancel --ok-button "Select" --ascii-lines --title "$title" --backtitle "$HEADER" --menu "$msg" 18 59 18 $region_list 2>tempfile

retval=$?
choice=$(cat tempfile)
region=$choice

# if ESC,exit
case $retval in
    255)
        echo
        echo $EXIT_MSG
        exit 255
        ;;
esac

if [ "$choice" = "UTC" ]
then
    $CHROOT ln -sf /usr/share/zoneinfo/UTC /etc/localtime
    $CHROOT hwclock --systohc
else
    # city
    title="Time Zone"
    msg="Please Select the city.		[ESC] to exit the installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."
    city_list=$(timedatectl list-timezones | grep $choice | cut -d "/" -f 2 | sort -u | xargs)

    # duplicate
    for city in $city_list
    do
        temp="${temp} $city $city"
    done
    city_list=$temp
    temp=

    dialog --no-cancel --ok-button "Select" --ascii-lines --title "$title" --backtitle "$HEADER" --menu "$msg" 18 60 18 $city_list 2>tempfile

    retval=$?
    choice=$(cat tempfile)
    city=$choice

    # if ESC,exit
    case $retval in
        255)
            echo
            echo $EXIT_MSG
            exit 255
            ;;
    esac

    $CHROOT ln -sf /usr/share/zoneinfo/$region/$city /etc/localtime
    $CHROOT hwclock --systohc
fi

## Locale

# function 
function get_locale_list # from /etc/locale.gen
{
    # skip the file header
    first_usable_line=$($CHROOT grep -n "#aa" /etc/locale.gen | head -1 | cut -d ":" -f 1)
    line_amount=$($CHROOT cat /etc/locale.gen | wc -l)
    # line will be used later
    work_line=$[ $line_amount - $first_usable_line + 1]

    locale_list=$($CHROOT tail -$work_line /etc/locale.gen | cut -d "#" -f 2 | sort -u | xargs)
    echo $locale_list
}

# delete old config
last_line=$(grep -n "Add by Archlinux Installer" /mnt/etc/locale.gen | cut -d ":" -f 1 | head -1)

if [ "$last_line" != "" ]
then
    head -$[ $last_line -1 ] /mnt/etc/locale.gen >tempfile
    mv tempfile /mnt/etc/locale.gen
fi

# select locale
title="Locale"
msg="Please select you needed localizations. the first column is locale,the second is charset.<locale> <charset>\n\nPlease confirm before typing the [ENTER], because you can't undo it."
locale_list=$(get_locale_list)

# add status
count=0
for item in $locale_list
do
    count=$[ $count + 1]
    judge=$[ $count % 2]
    if [ "$judge" = "0" ]
    then
        temp="${temp} $item OFF "
    else
        temp="${temp} $item "
    fi
done
locale_list=$temp
temp=

dialog --no-cancel --ascii-lines --title "$title" --backtitle "$HEADER" --checklist "$msg" 18 65 18 $locale_list 2>tempfile

retval=$?
choice=$(cat tempfile)

# if ESC,exit
case $retval in
    255)
        echo
        echo $EXIT_MSG
        exit 255
        ;;
esac

# new line
echo >>/mnt/etc/locale.gen
echo "## Add by Archlinux Installer">>/mnt/etc/locale.gen
for locale in $choice 
do
    item=$($CHROOT grep "$locale " /etc/locale.gen | head -1 |cut -d "#" -f 2)
    system_locale_list="${system_locale_list} $item"
    echo "$item" >>/mnt/etc/locale.gen
done

# generate the locale
$CHROOT locale-gen >/dev/null

# select sysem locale
title="System Locale"
msg="Please select system locale. the first column is locale, the second column is charset. <locale> <charset>. Don't recommend set chinese locale here. If you want to set chinese locale, you can just set for desktop environment later.	[ESC] to exit the installer\n\nPlease confirm before typing the [ENTER], because you can't undo it."

system_locale_list=$(echo $system_locale_list | xargs)

dialog --no-cancel --ok-button "Select" --ascii-lines --title "$title" --backtitle "$HEADER" --menu "$msg" 18 62 18 $system_locale_list 2>tempfile

retval=$?
choice=$(cat tempfile)

# if ESC,exit
case $retval in
    255)
        echo
        echo $EXIT_MSG
        exit 255
        ;;
esac

echo "## Add by Archlinux Installer">/mnt/etc/locale.conf
echo "LANG=$choice">/mnt/etc/locale.conf

## change Key map
echo "KEYMAP=$KEY_MAP" >/mnt/etc/vconsole.conf

##  hostname
title="Set hostname"
msg="Please input the hostname.		[ESC] to exit the installer.\n\nPlease confirm before typing the [ENTER], because you can't undo it."
dialog --no-cancel --ascii-lines --title "$title" --backtitle "$HEADER" --inputbox "$msg" 10 59 "hostname" 2>tempfile

retval=$?
choice=$(cat tempfile)

# if ESC,exit
case $retval in
    255)
        echo
        echo $EXIT_MSG
        exit 255
        ;;
esac

echo $choice >/mnt/etc/hostname

# set hosts
echo "127.0.1.1		$choice.localdomain	$choice" >>/mnt/etc/hosts

### 10.Set root password
title="Root password"
msg="Please input the password.		[ESC] to exit the installer.\n\nPlease confirm before typing the [ENTER], because you can't undo it."
dialog --no-cancel --ascii-lines --title "$title" --backtitle "$HEADER" --inputbox "$msg" 10 59 "password" 2>tempfile

retval=$?
choice=$(cat tempfile)

# if ESC,exit
case $retval in
    255)
        echo
        echo $EXIT_MSG
        exit 255
        ;;
esac

# change passwd
echo "root:$choice" | $CHROOT chpasswd

### 11.Add New User
title="Add user"
msg="Please input the user name.		[ESC] to exit the installer.\n\nPlease confirm before typing the [ENTER], because you can't undo it."
dialog --no-cancel --ascii-lines --title "$title" --backtitle "$HEADER" --inputbox "$msg" 10 59 "username" 2>tempfile

retval=$?
choice=$(cat tempfile)
usrname=$choice
# if ESC,exit
case $retval in
    255)
        echo
        echo $EXIT_MSG
        exit 255
        ;;
esac

# user add
$CHROOT useradd -m -G wheel -s /bin/bash $choice

# init passwd
title="User Passwd"
msg="Please input the password.		[ESC] to exit the installer.\n\nPlease confirm before typing the [ENTER], because you can't undo it."
dialog --no-cancel --ascii-lines --title "$title" --backtitle "$HEADER" --inputbox "$msg" 10 59 "password" 2>tempfile

retval=$?
choice=$(cat tempfile)

# if ESC,exit
case $retval in
    255)
        echo
        echo $EXIT_MSG
        exit 255
        ;;
esac

# init passwd
echo
echo "$usrname:$choice" |$CHROOT chpasswd
# firstly, install sudo
$CHROOT pacman --noconfirm -S sudo

# add user to sudoers
echo >> /mnt/etc/sudoers 
echo "## Add by Archlinux Installer" >> /mnt/etc/sudoers 
echo "## User privilege specification" >>/mnt/etc/sudoers
echo "#" >>/mnt/etc/sudoers
echo "$usrname ALL=(ALL) ALL" >> /mnt/etc/sudoers 

### 12.Configure network
# for wifi-menu
$CHROOT pacman --noconfirm -S dialog
# Nework manager
$CHROOT pacman --noconfirm -S networkmanager
# dhclient
$CHROOT pacman --noconfirm -S dhclient

# Initramfs
$CHROOT mkinitcpio -p linux

### 13.GRUB

# firstly,install grub and efibootmgr
platform=$(uname -m)

# used for probe other os
$CHROOT pacman --noconfirm -S os-prober

# for UEFI,just for x86_64 or i386
if [ "$BOOT_MODE" = "UEFI" ]
then
    $CHROOT pacman --noconfirm -S grub efibootmgr
    $CHROOT grub-install --target=${platform}-efi --efi-directory=/boot --bootloader-id=Grub 
    $CHROOT grub-mkconfig -o /boot/grub/grub.cfg

    # add entry for windows,just for uefi
    # if ther are the windows ,add entry
    if [ -d "/mnt/boot/EFI/Microsoft" ]
    then
        echo menuentry \"Windows\" { >>/mnt/boot/grub/grub.cfg
        echo insmod part_gpt >>/mnt/boot/grub/grub.cfg
        echo insmod fat >>/mnt/boot/grub/grub.cfg
        echo insmod search_fs_uuid>>/mnt/boot/grub/grub.cfg
        echo insmod chain>>/mnt/boot/grub/grub.cfg
        echo -n	search --fs-uuid --set=root\ >>/mnt/boot/grub/grub.cfg
        echo -n `grub-probe --target=hints_string /mnt/boot/EFI/Microsoft/Boot/bootmgfw.efi`\  >>/mnt/boot/grub/grub.cfg
        grub-probe --target=fs_uuid /mnt/boot/EFI/Microsoft/Boot/bootmgfw.efi>>/mnt/boot/grub/grub.cfg
        echo chainloader /EFI/Microsoft/Boot/bootmgfw.efi>>/mnt/boot/grub/grub.cfg
        echo }>>/mnt/boot/grub/grub.cfg
    fi
else # for BIOS
    $CHROOT pacman --noconfirm -S grub
    $CHROOT grub-install --target=i386-pc --recheck $ROOT_DISK
    $CHROOT grub-mkconfig -o /boot/grub/grub.cfg
fi

# enable dhcpcd.service
$CHROOT ln -s /usr/lib/systemd/system/dhcpcd.service /etc/systemd/system/multi-user.target.wants/dhcpcd.service

# download and copy installer script to new user home directory
$CHROOT pacman --noconfirm -S git
$CHROOT git clone https://github.com/si9ma/Archlinux-Installer.git /home/$usrname/Archlinux-Installer
git clone git@github.com:si9ma/Archlinux-Installer.me.git /mnt/home/$usrname/Archlinux-Installer
# mkdir -p /mnt/home/$usrname/Archlinux-Installer
cp * /mnt/home/$usrname/Archlinux-Installer -r

# backup profile
cp /mnt/etc/profile /mnt/etc/profile.backup

# run Archlinux-Installer when user login
echo >>/mnt/etc/profile
echo "## Add by Archlinux, will be deleted later" >>/mnt/etc/profile
echo "cd /home/$usrname/Archlinux-Installer/" >>/mnt/etc/profile
echo "bash ./option_install.sh" >>/mnt/etc/profile

# Done
title="Complete Installation"
msg="Hey! You have installed a smallest Archlinux on your computer. Plesase type [ENTER] to reboot your computer. When your computer boot again, you need login(in order to install successfully, don't login with root) to complete Subsequent installation.	[ESC] to exit installer"
dialog --no-cancel --ascii-lines --title "$title" --backtitle "$HEADER" --msgbox "$msg" 10 60

retval=$?
echo

case $retval in
    255) #ESC,exit
        echo 
        echo $EXIT_MSG
        exit 255
        ;;
esac

# reboot
#umount -R /mnt
reboot