refractainstaller-yad

#!/usr/bin/env bash
version="Refracta Installer (Yad) 9.0.6-2 (201201208)"
# Copyright 2011, 2012 fsmithred@gmail.com
# Based on refractainstaller-8.0.3 by Dean Linkous
# License: GPL-3
# This is free software with NO WARRANTY. Use at your own risk!


# DESCRIPTION
# This script is used for installing a live system to a hard drive. User
# input is via popup windows created by yad. It should be run from
# a terminal; if it's started from a menu item or a panel launcher, it
# should be run in a persistent terminal, so that progress messages can
# be seen and for user input in a few places.
#
# There are two modes for installation - Simple or Expert
#   Simple Mode:
# Create rsync excludes file if default file is missing.
# User can run partitioner inside the installer or skip it.
# User selects partition for installation. Bootloader goes to /dev/sda.
# Summary window asks to proceed with installation.
# Stuff happens without interaction.
#
#  Expert Mode:
# User selects installation options - change username, select up to three
# partitions (/, /boot, /home), select filesystem type for each partition, 
# choose whether to encrypt partitions or not, choose whether to write 
# random data or zeros to partitions.
# User has option to exit and use custom excludes file.
# User can run partitioner inside the installer.
# Summary window asks to proceed with installation.
# Stuff happens with some interaction (passwords, username, edit /etc/sudoers)
#
#  Stuff:
# Cleanup (in case of previous aborted run)
# Create encrypted volumes *(Expert mode only)
# Write random data or zeros *
# Mount partition(s) and create filesystem(s)
# Copy system with rsync
# Create swapfile
# Copy update-initramfs
# Set up fstab
# Set up crypttab *
# Install bootloader
# Cleanup
# Change username and passwords, edit /etc/sudoers *
# Re-enable update-db and freshclam, disable ssh root login.


# If you want to change any defaults, change them in the configfile.
# Default is /etc/refractainstaller.conf
# If you want to use a different config file for testing, change this
# variable. Normally, users should not edit anything in this script.
configfile="/etc/refractainstaller.conf"


show_help () {
	printf "$help_text"
	exit 0
}

help_text="
	Usage:  $0  [option]
	
	Run refractainstaller-yad from a terminal with no options  
	or select Refracta Installer  from the System menu to install 
	a running live-CD or live-usb-hdd to a hard drive. 
	
	
	valid options:
		-h, --help		show this help text
		-v, --version	display the version information

"

while [[ $1 == -* ]]; do
	case "$1" in
	
		-h|--help)
			show_help ;;
		
		-v|--version)
			printf "\n$version\n\n" 
			exit 0 ;;
		
		*) 
			printf "\t invalid option: $1 \n\n"
			printf "\t Try:  $0 -h for full help. \n\n"
			exit 1 ;;
    esac
done


# Greeting window
yad --title="$version" --button="Simple installation":0 \
	--button="Expert installation":1 --button=Exit:2 \
	--text="  This utility will install a running live-CD or live-USB to your hard drive. 

  This is free software that comes with no warranty or guarantee of any type,
  including but not limited to express, implied, merchantability or fitness of purpose.
  Copyright 2011 fsmithred@gmail.com, based on refractainstaller-8.0.3 by Dean Linkous. "
mode="$?"
	case $mode in
		0) install="simple" ;;
		1) install="expert" ;;
		2) exit 0 ;;
	esac


# Check that xserver is running and user is root.
[[ $DISPLAY ]] || { echo "There is no xserver running. Exiting..." ; exit 1 ; }
if [[ $(id -u) -ne 0 ]] ; then
	yad --title=Error --window-icon=error --text="
	You need to be root!
"
	exit 1
fi


refractainstaller_configuration () {
if [[ -f $configfile ]]; then
    source $configfile
else
	yad --title=Warning --window-icon=error \
	--button=Continue:0 --button=Exit:1 \
	--text="  Config file $configfile is missing  
  Continue with default settings?  "
	if [[ $? = 1 ]] ; then
		exit 0
	fi
fi
# Check for values in $configfile and use them.
# If any are unset, these defaults will be used.
error_log=${error_log:="/var/log/refractainstaller_error.log"}
rsync_excludes=${rsync_excludes:="/usr/lib/refractainstaller/installer_exclude.list"}
home_boot_excludes=${home_boot_excludes:="/usr/lib/refractainstaller/home_boot_exclude.list"}
swapfile_blocksize=${swapfile_blocksize:="1024"}
swapfile_count=${swapfile_count:="262144"}
pmount_fixed=${pmount_fixed:="no"}
enable_updatedb=${enable_updatedb:="yes"}
enable_freshclam=${enable_freshclam:="yes"}
root_ssh=${root_ssh:="no"}
}

refractainstaller_configuration


# Record errors in a logfile.
exec 2>"$error_log"


# determine grub version now, it gets used for installing the bootloader and
# preventing simple install from using ext4 with grub-legacy or grub-gfx.
	grubversion=$(dpkg -l | grep ii | grep -v doc | awk '$2 ~ "grub-[glp]" { print $2}')
#	grubversion="grub-legacy"   # for testing, comment out the above line and uncomment this one


# function to exit the script if there are errors
check_exit () {
if [[ $? -ne 0 ]] ; then
	yad --title="Error" --window-icon=error \
	--text="Exit due to error: $?
See $error_log for details." 
	exit 1
fi
}


copy_excludes () {
        cat > "$rsync_excludes" <<EOF
# It is safe to delete this file after installation.

- /dev/*
- /cdrom/*
- /media/*
- /target
- /swapfile
- /mnt/*
- /sys/*
- /proc/*
- /tmp/*
- /live
- /boot/grub/grub.cfg
- /boot/grub/menu.lst
- /boot/grub/device.map
- /etc/udev/rules.d/70-persistent-cd.rules
- /etc/udev/rules.d/70-persistent-net.rules
- /etc/fstab
- /etc/mtab
- /home/snapshot/
- /home/*/.gvfs
EOF

    chmod 666 "$rsync_excludes"
}



# Check that rsync excludes file exists, or create one.
if ! [[ -f  $rsync_excludes ]] ; then
    yad --title=Warning --window-icon=error \
    --button=Continue:0 --button=Exit:1 \
    --text="  There is no rsync excludes file, or its name does not match what this script expects.  
  You should continue and let the script create one, or if you have a custom excludes file,
  and you know what you're doing, you can exit the script and edit the 
  rsync_excludes variable in $configfile so that it matches the name 
  and path of your custom file.
    
  If you have any other drives or partitions mounted that you don't want
  to be copied, unmount them or edit the excludes file to list them."
    if [[ $? = 0 ]] ; then
		rsync_excludes="$(pwd)/installer_exclude.list"
        copy_excludes
    else
        exit 0
    fi
fi


# Select expert installation options
if [[ $install = "expert" ]]; then
    opts=$(yad --list --title="Installation Options" \
     --text="Check the options you want for the installation" \
     --checklist --column "Choose" --column "Num" --column "Option" \
     --width=590 --height=400  \
  FALSE 01 "Change user name" \
  FALSE 02 "Create a separate /home partition" \
  FALSE 03 "Create a separate /boot partition" \
  FALSE 04 "Encrypt the root filesystem (separate /boot required)" \
  FALSE 05 "Encrypt the /home partition (separate /home required)" \
  FALSE 06 "Write random data to encrypted partitions (more secure)" \
  FALSE 07 "Write zeroes to all partitions (to erase previous data)" \
  FALSE 08 "Do not install bootloader. I'll handle it myself." \
  FALSE 09 "Use UUID in /etc/fstab. (Useful if drive order changes.)" \
  FALSE 10 "Use filesystem labels (disk labels) in /etc/fstab." \
  FALSE 11 "Change hostname." \
  FALSE 12 "Disable automatic login to desktop." \
  FALSE 13 "Disable automatic login to console. (Use stock Debian inittab)" \
  FALSE xx "Exit the installation now.")
fi


if $(echo $opts | grep -q 01); then 
    change_user="yes"
fi
if $(echo $opts | grep -q 02); then
    sep_home="yes"
fi
if $(echo $opts | grep -q 03); then
    sep_boot="yes"
fi
if $(echo $opts | grep -q 04); then
    encrypt_os="yes"
fi
if $(echo $opts | grep -q 05); then
    encrypt_home="yes"
fi
if $(echo $opts | grep -q 06); then
    write_random="yes"
fi
if $(echo $opts | grep -q 07); then
    write_zero="yes"               
fi
if $(echo $opts | grep -q 08); then
    bootloader="no"
else
    bootloader="yes"
fi
if $(echo $opts | grep -q 09); then
	if [[ $encrypt_os = "yes" ]] || [[ $encrypt_home = "yes" ]]; then
		uuid_message="--> UUIDs in fstab won't work with encrypted filesystems and
 will not be used. Edit fstab manually after the installation."
	else
		use_uuid="yes"
	fi
fi
if $(echo $opts |grep -q 10); then
	if [[ $encrypt_os = "yes" ]] || [[ $encrypt_home = "yes" ]]; then
		disklabel_message="--> Disk labels in fstab won't work with encrypted filesystems and
 will not be used. Edit fstab manually after the installation."
	elif [[ $use_uuid = "yes" ]]; then
		disklabel_message="--> This script can't do both UUIDs and disk labels for fstab.
 UUIDs will be used, and you can add disk labels manually, after the install."
	else
		use_labels="yes"
	fi
fi
if $(echo $opts | grep -q 11); then
	change_hostname="yes"
fi
if $(echo $opts | grep -q 12); then
	disable_auto_desktop="yes"
fi
if $(echo $opts | grep -q 13); then
	disable_auto_console="yes"
fi
if $(echo $opts | grep -q xx); then
    exit 0
fi


if [[ $encrypt_os = "yes" ]] || [[ $encrypt_home = "yes" ]]; then
    # test for cryptsetup
    if ! [[ -f /sbin/cryptsetup ]] ; then
        yad --title=Error --window-icon=error \
         --button="Proceed without encrypting partitions":0 \
         --button=Exit:1 --text="You need to install cryptsetup and run the command, 'sudo modprobe dm-mod' before you can use encryption."
        if [[ $? = 0 ]] ;  then
            encrypt_os="no"
            encrypt_home="no"
        else
            exit 1
        fi
    fi
    # end test for cryptsetup
fi    


## Partition a disk ##### Simple install now does get to partition the disk - uncomment the conditional below to change it back.
#if [[ $install = "expert" ]]; then
yad --title=Partitioning --button="Run GParted":0 --button="Run cfdisk":1 \
    --button="Skip this step":2 --button=Exit:3 \
    --text="  You need to have at least one partition ready for the installation, plus one for each separate  
  partition that you chose. If you already have the partition(s) ready, you can skip this step.

  Run the partitioner now?"
	
	ans="$?"
	case $ans in
		0) gparted ;;
		1)  xterm -fa monaco -fs 12 -geometry 80x20+0+0 -e cfdisk ;;
		2)  ;;
		3) exit 0 ;;
	esac
#fi

#    # test to make sure there's a separate /boot partition
if [[ $sep_boot = "no" ]]; then
    if [[ $encrypt_os = "yes" ]]; then
        yad --window-icon=error --title=Error \
         --button="Proceed without encrypting partition":0 \
         --button=Exit:1 --text="You MUST have a separate, unencrypted /boot partition if you intend to boot an encrypted operating system. You can proceed without encrypting the root filesystem, or you can exit and start over."
        if [[ $? = 0 ]] ;  then
            encrypt_os="no"
        else
            exit 1
        fi
    fi
fi


# Find hard drives, and choose one for grub
choose_grub () {
yad --title="Install GRUB bootloader" --text="  Choose a location to install the bootloader. The usual choice is to  
  put it in the master boot record of the first hard drive (/dev/sda).  

		Choose MBR to install to the mbr of any hard disk. 
		Choose Partition to install to a partition. 
		Choose No Bootloader to proceed without a bootloader. 
		Choose Exit to exit this program.
		" \
--button=MBR:0 --button=Partition:1 --button="No Bootloader":2 --button=Exit:3
answer="$?"

if [[ $answer = 0 ]] ; then
grub_dev=$(find /dev -mindepth 1 -maxdepth 1  -name "*[sh]d[a-z]" \
  | sort | awk '{print "\n" $0 }' \
  | yad --list --separator="" --title=Bootloader --text="Choose a location to install the bootloader.
" \
  --column ' ' --column 'Hard Drives' --height=200)

	if [[ -z $grub_dev ]] ; then
		yad --title=Error --window-icon=error --button="Yes, I'm sure.":0 --button="Go back":1 \
	--text="No bootloader will be installed. Are you sure you want this?"
			if [[ $? = 1 ]] ; then
				choose_grub
			fi
	elif ! [[ -b $grub_dev ]] ; then
		yad --title=Error  --window-icon=error --button=Exit:0 --button="Go back":1 \
	--text="Something is wrong.  $grub_dev is not a block device."
			if [[ $? = 0 ]] ; then
				exit 1
			else
				choose_grub
			fi
	fi

elif [[ $answer = 1 ]] ; then
grub_partition=$(find /dev -mindepth 1 -maxdepth 1  -name "*[sh]d[a-z][1-9]*" \
  | sort | awk '{print "\n" $0 }' \
  | yad --list   --title="Bootloader" --text="Select a partition for the bootloader (GRUB)." \
  --separator="" --column ' ' --column 'Partitions' --height=380 --width=150)

	if [[ -z $grub_partition ]] ; then
		yad --title=Error --window-icon=error --button="Yes, I'm sure.":0 --button="Go back":1 \
	--text="No bootloader will be installed. Are you sure you want this?"
			if [[ $? = 1 ]] ; then
				choose_grub
			fi
	elif ! [[ -b $grub_partition ]] ; then
		yad --title=Error  --window-icon=error --button=Exit:0 --button="Go back":1 \
	--text="Something is wrong. Maybe you checked more than one box. You said you want to install the bootloader in $grub_dev"
			if [[ $? = 0 ]] ; then
				exit 1
			else
				choose_grub
			fi
	fi


elif [[ $answer = 2 ]] ; then
	yad  --title=Bootloader --text="  Proceeding without a bootloader. 
  You will need to do special things to boot your operating system. Be sure  
  that you know what you're doing." \
	--button=Proceed:0 --button=Exit:1
		if [[ $? = 1 ]] ; then
			exit 0
		fi
elif [[ $answer = 3 ]] ; then
	exit 0
fi
}

### Simple install gets default grub bootloader in /dev/sda
if [[ $install = "expert" ]]; then
	if [[ $bootloader = "yes" ]]; then
		choose_grub
	fi
fi

if [[ $install = "simple" ]]; then
    grub_dev="/dev/sda"
fi


# Show output of blkid for reference.
xterm  -fa monaco -fs 12 -geometry 90x20+0+0 -hold -e 'echo "Partition list (for reference.)  You may need this later." && blkid' &


# Show the partition list in a menu, and choose one for /boot
choose_boot () {
boot_dev=$(find /dev -mindepth 1 -maxdepth 1  -name "*[sh]d[a-z][1-9]*" \
  | sort | awk '{print "\n" $0 }' \
  | yad --list   --title="/boot partition" --text="Select a partition for /boot." \
  --separator="" --column ' ' --column 'Partitions' --height=380 --width=150)
}

if [[ $sep_boot = "yes" ]]; then
    choose_boot
fi

# Choose filesystem type for /boot
choose_fs_boot () {
if [[ -n $boot_dev ]]; then
     fs_type_boot=$(yad --list --title="/boot filesystem" --text="What type of filesystem would you like on $boot_dev?" \
  --separator="" --column "Format" --height=200\
  "ext2" \
  "ext3" \
  "ext4")
fi

if [[ -z $fs_type_boot ]]; then
    yad --window-icon=error --title=Error --button="Go back":0 --button=Exit:1 \
      --text="You must choose a file system type for /boot"
    if [[ $? = 0 ]]; then
        choose_fs_boot
    else
        exit 1
    fi
fi
}

if [[ -n $boot_dev ]]; then
    choose_fs_boot
fi


# Show the partition list in a menu, and choose one for the OS
choose_root () {
install_dev=$(find /dev -mindepth 1 -maxdepth 1  -name "*[sh]d[a-z][1-9]*" \
  | sort | awk '{print "\n" $0 }' \
  | yad --list --title="Root Partition" --text="Choose a partition to use for the installation of the operating system." \
    --separator="" --column ' ' --column 'Partitions' --height 380 --width 150)

if [[ -z $install_dev ]] ; then
    yad --window-icon=error --title=Error  --button="Go back":0 --button=Exit:1 \
  --text="Nothing was selected. You must select a partition for the installation. What would you like to do?"
        if [[ $? = 0 ]] ; then
            choose_root
        else
            exit 1
        fi
elif ! [[ -b $install_dev ]] ; then
    yad --window-icon=error --title=Error  --button="Go back":0 --button=Exit:1 \
  --text="    Something is wrong. Maybe you checked
    more than one box. You said you want to install
    the system to $install_dev"
        if [[ $? = 0 ]] ; then
            choose_root
        else
            exit 1
        fi
elif
    [[ $install_dev = $boot_dev ]] ; then
    yad --window-icon=error --title=Error --text="You chose the same partition for the operating system as the one for /boot. Try again."
        choose_root
fi
}

choose_root


# Choose filesystem type for OS.
choose_fs_os () {
fs_type_os=$(yad --list --title="Root Filesystem" --text="What type of filesystem would you like on $install_dev?" \
  --separator="" --column "Format" --height=200\
  "ext2" \
  "ext3" \
  "ext4")
if [[ -z $fs_type_os ]]; then
        yad --window-icon=error --title=Error --button="Go back":0 --button=Exit:1 \
      --text="You must choose a file system type
for the operating system"
    if [[ $? = 0 ]]; then
        choose_fs_os
    else
        exit 1
    fi
fi  
}


### Simple install gets default ext4 filesystem (or ext3 with older grub)
if [[ $install = "expert" ]]; then
    choose_fs_os
else
    if [[ $grubversion = "grub-pc" ]] ; then
        fs_type_os="ext4"
    else
		fs_type_os="ext3"
	fi
fi



# Show the partition list in a menu, and choose one for /home
choose_home () {
home_dev=$(find /dev -mindepth 1 -maxdepth 1  -name "*[sh]d[a-z][1-9]*" \
  | sort | awk '{print "\n" $0 }' \
  | yad --list --title="/home partition" --text="Select a partition for /home" \
  --separator="" --column ' ' --column 'Partitions' --height=380 --width=150)
if [[ -n $home_dev ]] ; then
    if ! [[ -b $home_dev ]] ; then
        yad --info --title=Error --button="Go back":0 --button=Exit:1 \
        --text="    Something is wrong. Maybe you checked
    more than one box. You said you want to install
    the system to $home_dev"
        if [[ $? = 0 ]] ; then
            choose_home
        else
            exit 1
        fi
    elif
        [[ $install_dev = $home_dev ]] ; then
        yad --window-icon=error --title=Error --text="You chose the same partition for /home as the one for the operating system. If you don't want a separate /home partition, then click OK without selecting one."
        choose_home
    elif
        [[ $boot_dev = $home_dev ]] ; then
        yad --window-icon=error --title=Error --text="You chose the same partition for /home as the one for /boot. Try again."
        choose_home
    fi
fi
}

if [[ $sep_home = "yes" ]]; then
    choose_home
fi


# Choose filesystem type for /home
choose_fs_home () {
if [[ -n $home_dev ]]; then
     fs_type_home=$(yad --list --title="/home filesystem" --text="What type of filesystem would you like on $home_dev?" \
  --separator="" --column "Format" --height=200\
  "ext2" \
  "ext3" \
  "ext4")
fi

if [[ -z $fs_type_home ]]; then
    yad --window-icon=error --title=Error --button="Go back":0 --button=Exit:1 \
      --text="You must choose a file system type for /home"
    if [[ $? = 0 ]]; then
        choose_fs_home
    else
        exit 1
    fi
fi
}

if [[ -n $home_dev ]]; then
    choose_fs_home
fi


# Enter new hostname (or use the old hostname as the new one)
if [[ $change_hostname = "yes" ]]; then
	new_hostname=$(yad --entry --title="Change hostname" \
	--text="Enter new hostname for installed system." \
	--entry-text="$HOSTNAME")
fi
# In case null was entered above as hostname, then set it to $HOSTNAME
new_hostname=${new_hostname:="$HOSTNAME"}


# Show a summary of what will be done
if [[ $change_user = "yes" ]]; then
    user_message="--> User name will be changed."
fi

if [[ -n $grub_dev ]] ; then
    grub_dev_message="--> Bootloader will be installed in $grub_dev"
elif [[ -n $grub_partition ]] ; then
	grub_dev_message="--> Bootloader will be installed in $grub_partition"
else
    grub_dev_message="--> Bootloader will not be installed."
fi

if [[ $encrypt_os = yes ]] ; then
    os_enc_message=", and will be encrypted."
fi

if [[ -z $home_dev ]] ; then
    home_dev_message="--> /home will not be on a separate partition."
else
    home_dev_message="--> /home will be installed on $home_dev and formatted as $fs_type_home"
fi

if [[ -n $home_dev ]] && [[ $encrypt_home = yes ]] ; then
    home_enc_message=", and will be encrypted."
fi

if [[ -n $boot_dev ]] ; then
    boot_dev_message="--> /boot will be installed on $boot_dev and formatted as $fs_type_boot."
fi

if [[ $encrypt_os = yes ]] || [[ $encrypt_home = yes ]] ; then
    proceed_message="***  IF YOU PROCEED, YOU WILL NEED TO RESPOND TO SOME QUESTIONS IN THE TERMINAL.   Be prepared to create 
    passphrases for any encrypted partitions (several times each.) When you see the progress bar come up, you can take a break."
fi

if [[ $disable_auto_desktop = "yes" ]]; then
	desktop_message="Desktop autologin will be disabled."
fi

if [[ $disable_auto_console = "yes" ]]; then
	console_message="Console autologin will be disabled."
fi

yad --info --title=Summary --button="Proceed with the installation.":0 --button="Exit":1 \
    --text="Here is a summary of what will be done. THIS IS YOUR LAST CHANCE TO EXIT before any changes are made to the disk.

$grub_dev_message
--> Operating system will be installed on $install_dev and formatted as $fs_type_os$os_enc_message
$home_dev_message$home_enc_message
$boot_dev_message
$user_message
$desktop_message
$console_message
$uuid_message
$disklabel_message
$uuid_message
Hostname: $new_hostname

$proceed_message"
    if [[ $? != 0 ]] ; then
        exit 0
    fi


# Actual installation begins here

# Unmount or close anything that might need unmounting or closing
cleanup () {
if $(df | grep -q /target/proc/) ; then
    umount /target/proc/
fi

if $(df | grep -q /target/dev/) ; then
    umount /target/dev/
fi

if $(df | grep -q /target/sys/) ; then
    umount /target/sys/
fi

# grep gives an error if $boot_dev is null
if $(df | grep -q $boot_dev) ; then
    umount -l $boot_dev
fi

if $(df | grep -q /target_boot) ; then
    umount -l /target_boot/
fi

if $(df | grep -q /target_home) ; then
    umount -l /target_home/
fi

# grep gives an error if $home is null
if $(df | grep -q $home_dev) ; then
    umount $home_dev
fi

if $(df | grep -q "\/dev\/mapper\/home-fs") ; then
    umount /dev/mapper/home-fs
fi

if [[ -h /dev/mapper/home-fs ]] ; then
    cryptsetup luksClose home-fs
fi

if $(df | grep -q /target) ; then
    umount -l /target/
fi

if $(df | grep -q $install_dev) ; then
    umount $install_dev
fi    

if $(df | grep "\/dev\/mapper\/root-fs") ; then
    umount /dev/mapper/root-fs
fi

if [[ -h /dev/mapper/root-fs ]] ; then
    cryptsetup luksClose /dev/mapper/root-fs
fi


# These next ones might be unnecessary
if [[ -d /target ]] ; then
    rm -rf /target
fi

if [[ -d /target_home ]] ; then
    rm -rf /target_home
fi

if [[ -d /target_boot ]] ; then
    rm -rf /target_boot
fi
}

cleanup


# Write random data to OS partition
if [[ $write_random = "yes" ]]; then
    if [[ $encrypt_os = "yes" ]]; then
        xterm -fa monaco -fs 12 -geometry 80x20+0+0 -e dd if=/dev/urandom of="$install_dev"
    fi  
fi

# Write random data to /home partition
if [[ $write_random = "yes" ]]; then
    if [[ $encrypt_home = "yes" ]]; then
        xterm -fa monaco -fs 12 -geometry 80x20+0+0 -e dd if=/dev/urandom of="$home_dev"
    fi  
fi


# Write zeros to partitions
if [[ $write_zero = "yes" ]]; then
    xterm -fa monaco -fs 12 -geometry 80x20+0+0 -e dd if=/dev/zero of="$install_dev"
    if [[ $sep_home = "yes" ]]; then
        xterm -fa monaco -fs 12 -geometry 80x20+0+0 -e dd if=/dev/zero of="$home_dev"
    fi
    if [[ $sep_boot = "yes" ]]; then
        xterm -fa monaco -fs 12 -geometry 80x20+0+0 -e dd if=/dev/zero of="$boot_dev"
    fi
fi


# make mount point, format, adjust reserve and mount
# install_dev must maintain the device name for cryptsetup
# install_part will be either device name or /dev/mapper name as needed.
mkdir /target ;  check_exit
if [[ $encrypt_os = yes ]] ; then
    xterm -fa monaco -fs 12 -geometry 80x20+0+0 -e cryptsetup luksFormat "$install_dev" ; check_exit
    xterm -fa monaco -fs 12 -geometry 80x20+0+0 -e cryptsetup luksOpen "$install_dev" root-fs ; check_exit
    install_part="/dev/mapper/root-fs"
else
    install_part="$install_dev"
fi 
mke2fs -t $fs_type_os "$install_part" ; check_exit 
tune2fs -r 10000 "$install_part" ; check_exit 
mount "$install_part" /target ; check_exit 


# make mount point for separate home if needed
# and add /home/* to the excludes list if it's not already there
if [[ -n $home_dev ]] ; then
    mkdir /target_home ; check_exit
    if [[ $encrypt_home = yes ]]; then
        xterm -fa monaco -fs 12 -geometry 80x20+0+0 -e cryptsetup luksFormat "$home_dev"
        check_exit
        xterm -fa monaco -fs 12 -geometry 80x20+0+0 -e cryptsetup luksOpen "$home_dev" home-fs
        check_exit
        home_part="/dev/mapper/home-fs"
    else
        home_part=$home_dev
fi
    mke2fs -t $fs_type_home "$home_part" ; check_exit
    tune2fs -r 10000 "$home_part" ; check_exit
    mount "$home_part" /target_home ; check_exit
    if ! $(grep -q "\/home\/\*" "$rsync_excludes"); then
        echo "- /home/*" >> "$rsync_excludes"
    fi
fi


# make mount point for separate /boot if needed
# and add /boot/* to the excludes list if it's not already there
# allow default for reserved blocks (don't need tune2fs here)
if [[ -n $boot_dev ]] ; then
    mkdir /target_boot ; check_exit
    mke2fs -t $fs_type_boot $boot_dev ; check_exit
    mount $boot_dev /target_boot
    if ! $(grep -q "\/boot\/\*" "$rsync_excludes"); then
        echo "- /boot/*" >> "$rsync_excludes"
    fi
fi


# make sure there's not a leftover entry in excludes list for /home/*
# or /boot/* from a previous run if not needed this time.
if [[ -z $boot_dev ]] ; then
    sed -i 's:- /boot/\*::' "$rsync_excludes"
fi

if [[ -z $home_dev ]] ; then
    sed -i 's:- /home/\*::' "$rsync_excludes"
fi


# copy everything over except the things listed in the exclude list
rsync -av / /target/ --exclude-from="$rsync_excludes"  | \
tee >(yad --progress --pulsate --width=350 --auto-close --title="Copying system to new partition.")


# copy separate /home if needed
if ! [[ -z $home_dev ]] ; then
    rsync -av /home/ /target_home/ --exclude-from="$home_boot_excludes" | \
tee >(yad --progress --pulsate --width=350 --auto-close --title="Copying home folders to new partition.")
fi

# copy separate /boot if needed
if [[ -n $boot_dev ]] ; then
    rsync -av /boot/ /target_boot/ --exclude-from="$home_boot_excludes" | \
tee >(yad --progress --pulsate --width=350 --auto-close --title="Copying files to boot partition.")
fi


# create swap
dd if=/dev/zero of=/target/swapfile bs="$swapfile_blocksize" count="$swapfile_count" | \
tee >(yad --progress --pulsate --width=350 --auto-close --title="Making a swap file...")
check_exit 
mkswap /target/swapfile ; check_exit 


# copy the real update-initramfs back in place
if [[ -f /target/usr/sbin/update-initramfs.distrib ]] ; then
    cp /target/usr/sbin/update-initramfs.distrib /target/usr/sbin/update-initramfs
fi
if [[ -f /target/usr/sbin/update-initramfs.debian ]] ; then
    cp /target/usr/sbin/update-initramfs.debian /target/usr/sbin/update-initramfs
fi


# Disallow mounting of all fixed drives with pmount
if [[ -f /target/etc/pmount.allow ]] ; then
	if [[ $pmount_fixed = "no" ]] ; then
		sed -i 's:/dev/sd\[a-z\]:#/dev/sd\[a-z\]:' /target/etc/pmount.allow
	fi
fi

# Re-enable updatedb if it was disabled by snapshot
if [[ -e /target/usr/bin/updatedb.mlocate ]] ; then
	if [[  $enable_updatedb = "yes" ]] ; then
		chmod +x /target/usr/bin/updatedb.mlocate
	fi
fi



# Disable autologin
if [[ $disable_auto_desktop = "yes" ]]; then

	#gdm
    if [[ -f /target/etc/gdm/gdm.conf ]]; then
        sed -i 's/^AutomaticLogin/#AutomaticLogin/' /target/etc/gdm/gdm.conf
    fi

	#gdm3
    if [[ -f /target/etc/gdm3/daemon.conf ]]; then
        sed -i 's/^AutomaticLogin/#AutomaticLogin/' /target/etc/gdm3/daemon.conf
    fi

	#lightdm
	if [[ -f /target/etc/lightdm/lightdm.conf ]]; then
		sed -i 's/^autologin/#autologin/g' /target/etc/lightdm/lightdm.conf
	fi

	#kdm
	if [ -f /target/etc/default/kdm.d/live-autologin ]; then
	rm -f /target/etc/default/kdm.d/live-autologin
	fi

	if [ -f /target/etc/kde3/kdm/kdmrc ]; then
	sed -i -e 's/^AutoLogin/#AutoLogin/g' /target/etc/kde3/kdm/kdmrc
	sed -i -e 's/^AutoReLogin/#AutoReLogin/g' /target/etc/kde3/kdm/kdmrc
	fi

	if [ -f /target/etc/kde4/kdm/kdmrc ]; then
	sed -i -e 's/^AutoLogin/#AutoLogin/g' /target/etc/kde4/kdm/kdmrc
	sed -i -e 's/^AutoReLogin/#AutoReLogin/g' /target/etc/kde4/kdm/kdmrc
	fi

	#trinity
	if [[ -f /target/etc/default/kdm-trinity.d/live-autologin ]]; then
		sed -i 's/^AUTOLOGIN/#AUTOLOGIN/g' /target/etc/default/kdm-trinity.d/live-autologin
	fi
	
	if [ -f /target/etc/trinity/kdm/kdmrc ]; then
	sed -i -e 's/^AutoLogin/#AutoLogin/g' /target/etc/trinity/kdm/kdmrc
	sed -i -e 's/^AutoReLogin/#AutoReLogin/g' /target/etc/trinity/kdm/kdmrc
	fi

fi	


# Disable console autologin
if [[ $disable_auto_console = "yes" ]]; then
	if grep -q "respawn:/bin/login -f" /target/etc/inittab ; then
		mv /target/etc/inittab /target/etc/inittab.$(date +%Y%m%d_%H%M)
		cp /usr/lib/refractainstaller/inittab.debian /target/etc/inittab
	fi
fi


# Change hostname
if ! [[ $new_hostname = $HOSTNAME ]]; then
	chroot /target hostname "$new_hostname"
	if [[ $? -eq 0 ]]; then
		sed -i "s/$HOSTNAME/$new_hostname/" /target/etc/hostname
		sed -i "s/$HOSTNAME/$new_hostname/g" /target/etc/hosts
	fi
fi


# setup fstab

# add entry for root filesystem
if [[ $use_uuid = yes ]]; then
	install_part="$(blkid -s UUID $install_dev | awk '{ print $2 }' | sed 's/\"//g')"
elif [[ $use_labels = yes ]]; then
	rootfslabel=$(yad  --entry --title="Filesystem Label"  --text="Enter a disk label for $install_dev")
	e2label $install_dev  $rootfslabel
	install_part="LABEL=$rootfslabel"
else
	install_part="$install_dev"
fi

echo -e "proc\t\t/proc\tproc\tdefaults\t0\t0
/swapfile\tswap\tswap\tdefaults\t0\t0
$install_part\t/\t$fs_type_os\tdefaults,noatime\t0\t1" >> /target/etc/fstab
check_exit


# add entry for /home to fstab if needed
if ! [[ -z $home_dev ]] ; then
	if [[ $use_uuid = yes ]]; then
		home_part="$(blkid -s UUID $home_dev | awk '{ print $2 }' | sed 's/\"//g')"
	elif [[ $use_labels = yes ]]; then
		homefslabel=$(yad  --entry --title="Filesystem Label"  --text="Enter a disk label for $home_dev")
		e2label $home_dev  $homefslabel
		home_part="LABEL=$homefslabel"
	else
		home_part="$home_dev"
	fi

    echo -e "$home_part\t/home\t$fs_type_home\tdefaults,noatime\t0\t2" >> /target/etc/fstab
    check_exit
fi


# add entry for /boot to fstab if needed
if [[ -n $boot_dev ]] ; then
	if [[ $use_uuid = yes ]]; then
		boot_part="$(blkid -s UUID $boot_dev | awk '{ print $2 }' | sed 's/\"//g')"
	elif [[ $use_labels = "yes" ]]; then
		bootfslabel=$(yad  --entry --title="Filesystem Label"  --text="Enter a disk label for $boot_dev")
		e2label $boot_dev  $bootfslabel
		boot_part="LABEL=$bootfslabel"
	else
		boot_part="$boot_dev"
	fi
    echo -e "$boot_part\t/boot\t$fs_type_boot\tdefaults,noatime,\t0\t2" >> /target/etc/fstab
    check_exit
fi


# Add entry for root filesystem to crypttab if needed
if [[ $encrypt_os = yes ]] ; then
    echo -e "root-fs\t\t$install_dev\t\tnone\t\tluks" >> /target/etc/crypttab
fi


# Add entry for /home to crypttab if needed
if [[ $encrypt_home = yes ]] ; then
    echo -e "home-fs\t\t$home_dev\t\tnone\t\tluks" >> /target/etc/crypttab
fi


# mount stuff so grub will behave (so chroot will work)
mount -t tmpfs --bind /dev/ /target/dev/ ; check_exit 
mount -t proc --bind /proc/ /target/proc/ ; check_exit 
mount -t sysfs --bind /sys/ /target/sys/ ; check_exit 


# Re-enable freshclam if it was disabled by snapshot 		##### This ain't perfect, but it works!
if type -p freshclam ; then
	if [[ $enable_freshclam = "yes" ]] ; then
		if ! [[ -h /target/etc/rc2.d/S02clamav-freshclam ]] ; then
			chroot /target update-rc.d clamav-freshclam defaults
		fi
	fi
fi


# Disable root login through ssh for the installed system
if [[ -f /etc/ssh/sshd_config ]] ; then
	if [[ $root_ssh = "no" ]] ; then
		sed -i~ 's/PermitRootLogin yes/PermitRootLogin no/' /target/etc/ssh/sshd_config
	fi
fi

# Setup GRUB 


# If /boot is separate partition, need to mount it in chroot for grub
if [[ -n $boot_dev ]] ; then
    chroot /target mount $boot_dev /boot
fi


# If grub is installed to a partition, we need to know if it's grub-pc
# or grub-legacy/grub-gfx to handle it properly.
if [[ -n $grub_partition ]] ; then
	if ! [[ $grubversion = "grub-pc" ]] ; then

		# isolate the device (sdx) letter then use tr like this to translate to the right number for grub
		GRUBDEVICENUM=$(echo $grub_partition |sed 's:/dev/sd::' |sed 's:[0-9]::'g |tr '[a-j]' '[0-9]')

		# isolate the partition number
		INSTALLPARTNUM=$(echo $grub_partition |sed 's:/dev/sd::'|sed 's:[a-z]::')

		# and reduce it by 1 for grub
		GRUBPARTNUM=$(expr $INSTALLPARTNUM - 1)

		# finally get the finished grub root syntax
		GRUBROOT="(hd$GRUBDEVICENUM,$GRUBPARTNUM)"


		chroot /target /bin/bash grub-install $grub_partition
		grub --batch <<EOF
		root $GRUBROOT
		setup $GRUBROOT
		quit
EOF

	else
		chroot /target /bin/bash grub-install --root-directory=/target --recheck --no-floppy --force $grub_partition
	fi
fi


if [[ -n $grub_dev ]]; then
    echo -e "\n Installing the boot loader...\n" >> "$error_log"
    chroot /target grub-install $grub_dev >> "$error_log"
fi


# Run update-initramfs to include dm-mod if using encryption
if [[ $encrypt_os = yes ]] || [[ $encrypt_home = yes ]] ; then
    chroot /target update-initramfs -u
fi

if [[ -n $grub_dev ]] || [[ -n $grub_partition ]] ; then
    chroot /target update-grub ; check_exit 
fi


# INSTALLATION FINISHED - BEGIN CHANGE USERNAME

# Need to mount the target home partition under the target root partition
# so the commands can find it (for changing user configs gksu)
if [[ $sep_home = "yes" ]]; then
    mount $home_part /target/home
fi


# Change the username
if [[ $change_user = "yes" ]]; then
	oldname=$(awk -F: '/1000:1000/ { print $1 }' /target/etc/passwd)
	newname=$(yad --entry --entry-text="$oldname" --title="Change login name" --text="Enter the new login name you want to use.")

	chroot /target usermod -l $newname $oldname ; check_exit
	chroot /target groupmod -n $newname $oldname ; check_exit
	chroot /target usermod -d /home/$newname -m $newname ; check_exit

	for i in $(grep -r "/home/$oldname" /target/home/$newname/.config | awk -F":" '{ print $1 }'); do
		sed -i "s/\/home\/$oldname/\/home\/$newname/g" "$i"
	done

	for i in $(grep -r "/home/$oldname" /target/home/$newname/.local | awk -F":" '{ print $1 }'); do
		sed -i "s/\/home\/$oldname/\/home\/$newname/g" "$i"
	done
fi

#### Add user's real name in /etc/passwd (for Refracta)
if [[ $change_user = "yes" ]]; then
    live_user=$(awk -v pattern="$newname" -F: '$1 ~ pattern { print $5 }' /target/etc/passwd)
    real_name=$(yad --entry --title="Change real name" --text="The user's real name is currently $live_user.
Enter the real name you want to use \(without 
the trailing commas\).")
    sed -i~ "s/$live_user/$real_name,,,/" /target/etc/passwd
fi


# Edit /etc/sudoers
if [[ $install = "expert" ]]; then
    yad --title="Edit /etc/sudoers" --button=Yes:0 --button=No:1 \
     --text="You should edit /etc/sudoers to comment out the last line, which gives \"user\" absolute power, 
or replace \"user\" with the new user name. A simple text-editor will open in the terminal if you do this now.
 
Edit /etc/sudoers now?" 
	ans="$?"
    if [[ $ans = "0" ]]; then
        edit_sudoers="yes"
        xterm  -fa monaco -fs 12 -geometry 80x20+0+0 -e chroot /target visudo
    fi
fi


# Disable sudo-mode for gksu
if [[ $edit_sudoers = "yes" ]]; then
    yad --title="Disable sudo mode for gksu" --button="Yes, I commented out the line.":0 --button="No, I replaced \"user\" with the new username.":1 \
     --text="If you commented out the last line in /etc/sudoers in the last step, one more file will be changed for you so that gksu will work properly." 
		ans="$?"
        if [[ $ans = "0" ]]; then
            if [[ -n $newname ]]; then
                sed -i~ '/sudo-mode/s/true/false/' /target/home/"$newname"/.gconf/apps/gksu/%gconf.xml
            else
                sed -i~ '/sudo-mode/s/true/false/' /target/home/user/.gconf/apps/gksu/%gconf.xml
            fi
        fi
fi



# Change/create root password
if [[ $install = "expert" ]]; then
    yad title="Change/create root password" --button=Yes:0 --button=No:1 \
     --text="Would you like to change the root password? (Recommended)
 
You'll need to go to the terminal again..."
		ans="$?"
        if [[ $ans = 0 ]]; then
            xterm  -fa monaco -fs 12 -geometry 80x20+0+0 -e chroot /target passwd
        fi 
fi


# Change user password
if [[ $change_user = "yes" ]]; then
    yad --title="Change user password" --button=Yes:0 --button=No:1 \
     --text="Would you like to change the user's password? The new user still has 
the old user's password. You'll need to go to the terminal again to do this." 
		ans="$?"
        if [[ $ans = 0 ]]; then
            xterm  -fa monaco -fs 12 -geometry 80x20+0+0 -e chroot /target passwd "$newname"
        fi
fi


# call cleanup function
cleanup


yad --info --text="Done!\n\n You may now reboot into the new system.\n\nRemember to remove your installation media.\n"

exit 0