archlabs-installer

   1 #!/bin/bash
   2
   3 # This program is free software, provided under the GNU GPL
   4 # Written by Nathaniel Maia for use in Archlabs
   5 # Some ideas and code reworked from other resources
   6 # AIF, Cnichi, Calamares, Arch Wiki.. Credit where credit is due
   7
   8 VER=2.0.79
   9
  10 # bulk default values {
  11
  12 : ${DIST=ArchLabs}                 # distro name if not set
  13 MNT=/mnt                           # install mountpoint
  14 ANS=/tmp/ans                       # dialog answer file
  15 BOOTDIR=boot                       # location to mount boot partition
  16 FONT=ter-i16n                      # font used for the linux console
  17 HOOKS=shutdown                     # list of additional mkinitcpio HOOKS
  18 SEL=0                              # currently selected menu item
  19 SYS=Unknown                        # bios type to be determined: UEFI/BIOS
  20 ERR=/tmp/errlog                    # error log used internally
  21 DBG=/tmp/debuglog                  # debug log when passed -d
  22 RUN=/run/archiso/bootmnt/arch/boot # path for live system /boot
  23 VM="$(dmesg | grep -i hypervisor)" # system running in a virtual machine
  24 export DIALOGOPTS="--cr-wrap"      # see `man dialog`
  25
  26 BASE_PKGS="base base-devel xorg xorg-drivers sudo git gvfs gtk3 libmad libmatroska tumbler "
  27 BASE_PKGS+="playerctl pulseaudio pulseaudio-alsa pavucontrol pamixer scrot xdg-user-dirs "
  28 BASE_PKGS+="ffmpeg gstreamer gst-libav gst-plugins-base gst-plugins-good bash-completion "
  29
  30 AL_BASE_PKGS="archlabs-keyring archlabs-icons archlabs-fonts archlabs-themes "
  31 AL_BASE_PKGS+="archlabs-baph archlabs-wallpapers archlabs-scripts archlabs-skel-base"
  32
  33 WM_BASE_PKGS="arandr nitrogen polkit-gnome network-manager-applet "
  34 WM_BASE_PKGS+="volumeicon xclip exo laptop-detect xdotool compton wmctrl feh "
  35 WM_BASE_PKGS+="gnome-keyring dunst gsimplecal xfce4-power-manager xfce4-settings"
  36
  37 SYS_MEM="$(awk '/MemTotal/ {print int($2 / 1024) "M"}' /proc/meminfo)"
  38 LOCALES="$(awk '/\.UTF-8/ {gsub(/# .*|#/, ""); if ($1) {print $1 " - "}}' /etc/locale.gen)"
  39 CMAPS="$(find /usr/share/kbd/keymaps -name '*.map.gz' | awk '{gsub(/\.map\.gz|.*\//, ""); print $1 " - "}' | sort)"
  40
  41 [[ $LINES ]] || LINES=$(tput lines)
  42 [[ $COLUMNS ]] || COLUMNS=$(tput cols)
  43
  44 # }
  45
  46 # commands used to install each bootloader, however most get modified during runtime {
  47 declare -A BCMDS=(
  48 [refind-efi]='refind-install'
  49 [grub]='grub-install --recheck --force' [syslinux]='syslinux-install_update -i -a -m'
  50 [efistub]='efibootmgr -v -d /dev/sda -p 1 -c -l' [systemd-boot]='bootctl --path=/boot install'
  51 ) # }
  52
  53 # executable name for each wm/de used in ~/.xinitrc {
  54 declare -A WM_SESSIONS=(
  55 [dwm]='dwm' [i3-gaps]='i3' [bspwm]='bspwm' [awesome]='awesome' [plasma]='startkde' [xfce4]='startxfce4'
  56 [gnome]='gnome-session' [fluxbox]='startfluxbox' [openbox]='openbox-session' [cinnamon]='cinnamon-session'
  57 ) # }
  58
  59 # packages installed for each wm/de, most are depends of the skel packages {
  60 declare -A WM_EXT=(
  61 [dwm]='' [gnome]='' [cinnamon]='gnome-terminal' [plasma]='kdebase-meta'
  62 [awesome]='archlabs-skel-awesome' [bspwm]='archlabs-skel-bspwm' [fluxbox]='archlabs-skel-fluxbox'
  63 [i3-gaps]='archlabs-skel-i3-gaps' [openbox]='archlabs-skel-openbox archlabs-skel-xfce4' [xfce4]='archlabs-skel-xfce4 xfce4-goodies'
  64 ) # }
  65
  66 # files offered for editing after install is complete {
  67 declare -A EDIT_FILES=(
  68 [login]='' # login is populated once we know the username and shell
  69 [fstab]='/etc/fstab' [sudoers]='/etc/sudoers' [crypttab]='/etc/crypttab' [pacman]='/etc/pacman.conf'
  70 [console]='/etc/vconsole.conf' [mkinitcpio]='/etc/mkinitcpio.conf' [hostname]='/etc/hostname /etc/hosts'
  71 [bootloader]="/boot/loader/entries/$DIST.conf"  # ** based on bootloader
  72 [locale]='/etc/locale.conf /etc/default/locale' [keyboard]='/etc/X11/xorg.conf.d/00-keyboard.conf /etc/default/keyboard'
  73 ) # }
  74
  75 # mkfs command flags for filesystem formatting {
  76 declare -A FS_CMD_FLAGS=(
  77 [f2fs]='' [jfs]='-q' [xfs]='-f' [ntfs]='-q' [ext2]='-q' [ext3]='-q' [ext4]='-q' [vfat]='-F32' [nilfs2]='-q' [reiserfs]='-q'
  78 ) # }
  79
  80 # mount options for each filesystem {
  81 declare -A FS_OPTS=(
  82 [vfat]='' [ntfs]='' [ext2]='' [ext3]=''  # NA
  83 [jfs]='discard errors=continue errors=panic nointegrity'
  84 [reiserfs]='acl nolog notail replayonly user_xattr off'
  85 [ext4]='discard dealloc nofail noacl relatime noatime nobarrier nodelalloc'
  86 [xfs]='discard filestreams ikeep largeio noalign nobarrier norecovery noquota wsync'
  87 [nilfs2]='discard nobarrier errors=continue errors=panic order=relaxed order=strict norecovery'
  88 [f2fs]='discard fastboot flush_merge data_flush inline_xattr inline_data noinline_data inline_dentry no_heap noacl nobarrier norecovery noextent_cache disable_roll_forward disable_ext_identify'
  89 ) # }
  90
  91 # packages installed for each login option {
  92 declare -A LOGIN_PKGS=(
  93 [xinit]='xorg-xinit' [ly]='archlabs-ly' [gdm]='gdm' [sddm]='sddm'
  94 [lightdm]='lightdm lightdm-gtk-greeter lightdm-gtk-greeter-settings accountsservice'
  95 ) # }
  96
  97 # extras installed for user selected packages {
  98 # if a package requires additional packages that aren't already dependencies
  99 # they can be added here eg. [package]="extra"
 100 declare -A PKG_EXT=(
 101 [vlc]='qt4' [mpd]='mpc' [mupdf]='mupdf-tools'
 102 [rxvt-unicode]='urxvt-pearls' [zathura]='zathura-pdf-poppler' [noto-fonts]='noto-fonts-emoji' [cairo-dock]='cairo-dock-plug-ins' [qt5ct]='qt5-styleplugins'
 103 [vlc]='qt5ct qt5-styleplugins' [qutebrowser]='qt5ct qt5-styleplugins' [qbittorrent]='qt5ct qt5-styleplugins' [transmission-qt]='qt5ct qt5-styleplugins'
 104 [bluez]='bluez-libs bluez-utils bluez-tools bluez-plugins bluez-hid2hci' [kdenlive]='kdebase-meta dvdauthor frei0r-plugins breeze breeze-gtk qt5ct qt5-styleplugins'
 105 ) # }
 106
 107 # dialog text variables {
 108 # Basics (somewhat in order)
 109 _welcome="\nThis will help you get $DIST installed and setup on your system.\n\nIf you are unsure about a section the default option will be listed or\nthe first selected item will be the default.\n\n\nMenu Navigation:\n\n - Select items with the arrow keys or the option number.\n - Use [Space] to toggle check boxes and [Enter] to accept.\n - Switch between fields using [Tab] or the arrow keys.\n - Use [Page Up] and [Page Down] to jump whole pages\n - Press the highlighted key of an option to select it.\n"
 110 _keymap="\nPick which keymap to use for the system from the list below\n\nThis is used once a graphical environment is running (Xorg).\n\nSystem default: us"
 111 _vconsole="\nSelect the console keymap, the console is the tty shell you reach before starting a graphical environment (Xorg).\n\nIts keymap is seperate from the one used by the graphical environments, though many do use the same such as 'us' English.\n\nSystem default: us"
 112 _device="\nSelect a device to use from the list below.\n\nDevices (/dev) are the available drives on the system. /sda, /sdb, /sdc ..."
 113 _resize="\nSelect a new filesystem size in MB, a new partition will be created from the free space but will be left unformatted.\nThe lowest size is just enough to fit the currently in use space on the partition while the default is set to split the free space evenly.\n\nUse Tab or the arrow keys move the cursor between the buttons and the value, when the cursor is on the value, you can edit it by:\n\n - left/right cursor movement to select a digit to modify\n - +/-  characters to increment/decrement the digit by one\n - 0 through 9 to set the digit to the given value\n\nSome keys are also recognized in all cursor positions:\n\n - Home/End set the value to its maximum or minimum\n - Pageup/Pagedown increment the value so that the slider moves by one column."
 114 _mount="\nUse [Space] to toggle mount options from below, press [Enter] when done to confirm selection.\n\nNot selecting any and confirming will run an automatic mount."
 115 _warn="\nIMPORTANT:\n\nChoose carefully when editing, formatting, and mounting partitions or your DATA MAY BE LOST.\n\nTo mount a partition without formatting it, select 'skip' when prompted to choose a filesystem during the mounting stage.\nThis can only be used for partitions that already contain a filesystem and cannot be the root (/) partition, it needs to be formatted before install.\n"
 116 _part="\nFull device auto partitioning is available for beginners otherwise cfdisk is recommended.\n\n  - All systems will require a root partition (8G or greater).\n  - UEFI or BIOS using LUKS without LVM require a separate boot partition (100-512M)."
 117 _uefi="\nSelect the EFI boot partition (/boot), required for UEFI boot.\n\nIt's usually the first partition on the device, 100-512M, and will be formatted as vfat/fat32 if not already."
 118 _bios="\nDo you want to use a separate boot partition? (optional)\n\nIt's usually the first partition on the device, 100-512M, and will be formatted as ext3/4 if not already."
 119 _biosluks="\nSelect the boot partition (/boot), required for LUKS.\n\nIt's usually the first partition on the device, 100-512M, and will be formatted as ext3/4 if not already."
 120 _format="is already formatted correctly.\n\nFor a clean install, previously existing partitions should be reformatted, however this removes ALL data (bootloaders) on the partition so choose carefully.\n\nDo you want to reformat the partition?\n"
 121 _swapsize="\nEnter the size of the swapfile in megabytes (M) or gigabytes (G).\n\neg. 100M will create a 100 megabyte swapfile, while 10G will create a 10 gigabyte swapfile.\n\nFor ease of use and as an example it is filled in to match the size of your system memory (RAM).\n\nMust be greater than 1, contain only whole numbers, and end with either M or G."
 122 _expart="\nYou can now choose any additional partitions you want mounted, you'll be asked for a mountpoint after.\n\nSelect 'done' to finish the mounting step and begin unpacking the base system in the background."
 123 _exmnt="\nWhere do you want the partition mounted?\n\nEnsure the name begins with a slash (/).\nExamples include: /usr, /home, /var, etc."
 124 _user="\nEnter a name and password for the new user account.\n\nThe name must not use capital letters, contain any periods (.), end with a hyphen (-), or include any colons (:)\n\nNOTE: Use [Up], [Down], or [Tab] to switch between fields, and [Enter] to accept."
 125 _hostname="\nEnter a hostname for the new system.\n\nA hostname is used to identify systems on the network.\n\nIt's restricted to alphanumeric characters (a-z, A-Z, 0-9).\nIt can contain hyphens (-) BUT NOT at the beggining or end."
 126 _locale="\nLocale determines the system language and currency formats.\n\nThe format for locale names is languagecode_COUNTRYCODE\n\neg. en_US is: english United States\n    en_GB is: english Great Britain"
 127 _timez="\nThe time zone is used to set the system clock.\n\nSelect your country or continent from the list below"
 128 _timesubz="\nSelect the nearest city to you or one with the same time zone.\n\nTIP: Pressing the first letter of the city name repeatedly will navigate between entries beggining with that letter."
 129 _sessions="\nUse [Space] to toggle available sessions, use [Enter] to accept the selection and continue.\n\nA basic package set will be installed for compatibility and functionality."
 130 _login="\nSelect which of your session choices to use for the initial login.\n\nYou can be change this later by editing your ~/.xinitrc"
 131 _autologin="\nDo you want autologin enabled for USER?\n\nIf so the following two files will be created (disable autologin by removing them):\n\n - /home/USER/RC (run startx when logging in on tty1)\n - /etc/systemd/system/getty@tty1.service.d/autologin.conf (login USER without password)\n"
 132 _packages="\nUse [Space] to move a package into the selected area and press [Enter] to accept the selection.\n\nPackages may be installed by your DE/WM (if any), or for the packages you select."
 133 _edit="\nBefore exiting you can select configuration files to review/change.\n\nIf you need to make other changes with the drives still mounted, use Ctrl-z to pause the installer, when finished type 'fg' and [Enter] to resume the installer, if you want to avoid the automatic reboot using Ctrl-c will cleanly exit."
 134
 135 # LUKS
 136 _luksnew="Basic LUKS Encryption"
 137 _luksadv="Advanced LUKS Encryption"
 138 _luksopen="Open Existing LUKS Partition"
 139 _luksmenu="\nA seperate boot partition without encryption or logical volume management (LVM) is required (except BIOS systems using grub).\n\nBasic uses the default encryption settings, and is recommended for beginners. Advanced allows cypher and key size parameters to be entered manually."
 140 _luksomenu="\nEnter a name and password for the encrypted device.\n\nIt is not necessary to prefix the name with /dev/mapper/,an example has been provided."
 141 _lukskey="Once the specified flags have been amended, they will automatically be used with the 'cryptsetup -q luksFormat /dev/...' command.\n\nNOTE: Do not specify any additional flags such as -v (--verbose) or -y (--verify-passphrase)."
 142
 143 # LVM
 144 _lvmmenu="\nLogical volume management (LVM) allows 'virtual' hard drives (volume groups) and partitions (logical volumes) to be created from existing device partitions.\n\nA volume group must be created first, then one or more logical volumes within it.\n\nLVM can also be used with an encrypted partition to create multiple logical volumes (e.g. root and home) within it."
 145 _lvmnew="Create Volume Group and Volume(s)"
 146 _lvmdel="Delete an Existing Volume Group"
 147 _lvmdelall="Delete ALL Volume Group(s) and Volume(s)"
 148 _lvmvgname="\nEnter a name for the volume group (VG) being created from the partition(s) selected."
 149 _lvmlvname="\nEnter a name for the logical volume (LV) being created.\n\nThis is similar to setting a label for a partition."
 150 _lvmlvsize="\nEnter what size you want the logical volume (LV) to be in megabytes (M) or gigabytes (G).\n\neg. 100M will create a 100 megabyte volume, 10G will create a 10 gigabyte volume."
 151 _lvmdelask="\nConfirm deletion of volume group(s) and logical volume(s).\n\nDeleting a volume group, will delete all logical volumes within it.\n"
 152
 153 # Errors
 154 _errexpart="\nCannot mount partition due to a problem with the mountpoint.\n\nEnsure it begins with a slash (/) followed by atleast one character.\n"
 155 _errpart="\nYou need create the partiton(s) first.\n\n\nBIOS systems require at least one partition (ROOT).\n\nUEFI systems require at least two (ROOT and EFI).\n"
 156 _lukserr="\nA minimum of two partitions are required for encryption:\n\n 1. root (/) - standard or LVM.\n 2. boot (/boot) - standard (unless using LVM on BIOS systems).\n"
 157 _lvmerr="\nThere are no viable partitions available to use for LVM, a minimum of one is required.\n\nIf LVM is already in use, deactivating it will allow the partition(s) to be used again.\n"
 158 _lvmerrvgname="\nInvalid name entered.\n\nThe volume group name may be alpha-numeric, but may not contain spaces, start with a '/', or already be in use.\n"
 159 _lvmerlvname="\nInvalid name entered.\n\nThe logical volume (LV) name may be alpha-numeric, but may not contain spaces or be preceded with a '/'\n"
 160 _lvmerrlvsize="\nInvalid value Entered.\n\nMust be a numeric value with 'M' (megabytes) or 'G' (gigabytes) at the end.\n\neg. 400M, 10G, 250G, etc...\n\nThe value may also not be equal to or greater than the remaining size of the volume group.\n"
 161
 162 # }
 163
 164 ###############################################################################
 165 # selection menus
 166 # main is the entry point which calls functions including outside of its block
 167 # once those functions finished they always are returned here with the
 168 # exception of install_main(), it exits upon completion
 169
 170 main()
 171 {
 172         (( SEL < 12 )) && (( SEL++ ))
 173         tput civis
 174         dialog --backtitle "$DIST Installer - $SYS - v$VER" --title " Prepare " --default-item $SEL --cancel-label 'Exit' --menu "$_prep" 0 0 0 \
 175                 1 "Device tree (optional)" \
 176                 2 "Partitioning (optional)" \
 177                 3 "LUKS setup (optional)" \
 178                 4 "LVM setup (optional)" \
 179                 5 "Mount partitions" \
 180                 6 "System bootloader" \
 181                 7 "User and password" \
 182                 8 "System configuration" \
 183                 9 "Select WM/DE (optional)" \
 184                 10 "Select Packages (optional)" \
 185                 11 "View configuration (optional)" \
 186                 12 "Start the installation" 2>"$ANS"
 187
 188         read -r SEL < "$ANS"
 189         [[ -z $WARN && $SEL =~ (2|5) ]] && { msg "Data Warning" "$_warn"; WARN=true; }
 190         case $SEL in
 191                 1) part_show ;;
 192                 2) part_menu || (( SEL-- )) ;;
 193                 3) luks_menu || (( SEL-- )) ;;
 194                 4) lvm_menu || (( SEL-- )) ;;
 195                 5) mount_menu || (( SEL-- )) ;;
 196                 6) prechecks 0 && { select_boot || (( SEL-- )); } ;;
 197                 7) prechecks 1 && { select_mkuser || (( SEL-- )); } ;;
 198                 8) prechecks 2 && { select_config || (( SEL-- )); } ;;
 199                 9) prechecks 3 && { select_sessions || (( SEL-- )); } ;;
 200                 10) prechecks 3 && { select_packages || (( SEL-- )); } ;;
 201                 11) prechecks 3 && select_show ;;
 202                 12) prechecks 3 && install_main ;;
 203                 *) yesno "Exit" "\nUnmount partitions (if any) and exit the installer?\n" && die 0
 204         esac
 205 }
 206
 207 select_boot()
 208 {
 209         if [[ $SYS == 'BIOS' ]]; then
 210                 dlg BOOTLDR menu "BIOS Bootloader" "\nSelect which bootloader to use." \
 211                         "grub"     "The Grand Unified Bootloader, standard among many Linux distributions" \
 212                         "syslinux" "A collection of boot loaders for booting drives, CDs, or over the network" || return 1
 213         else
 214                 dlg BOOTLDR menu "UEFI Bootloader" "\nSelect which bootloader to use." \
 215                         "systemd-boot" "A simple UEFI boot manager which executes configured EFI images" \
 216                         "grub"         "The Grand Unified Bootloader, standard among many Linux distributions" \
 217                         "refind-efi"   "A UEFI boot manager that aims to be platform neutral and simplify multi-boot" \
 218                         "efistub"      "Boot the kernel image directly (no chainloading support)" \
 219                         "syslinux"     "A collection of boot loaders for booting drives, CDs, or over the network (no chainloading support)" || return 1
 220         fi
 221         setup_${BOOTLDR}
 222 }
 223
 224 select_show()
 225 {
 226         local pkgs="${USER_PKGS//  / } ${PACKAGES//  / }"
 227         [[ $BOOT_PART ]] && local mnt="/$BOOTDIR"
 228         [[ $INSTALL_WMS == *dwm* ]] && pkgs="dwm st dmenu $pkgs"
 229         pkgs="${pkgs//  / }" pkgs="${pkgs# }"
 230         msg "Show Configuration" "
 231 ---------- PARTITION CONFIGURATION ------------
 232
 233   Root:  ${ROOT_PART:-none}
 234   Boot:  ${BOOT_PART:-${BOOT_DEV:-none}}
 235   Mount: ${mnt:-none}
 236   Swap:  ${SWAP_PART:-none}
 237   Size:  ${SWAP_SIZE:-none}
 238   Extra: ${EXMNTS:-${EXMNT:-none}}
 239   Hooks: ${HOOKS:-none}
 240
 241   LVM:   ${LVM:-none}
 242   LUKS:  ${LUKS:-none}
 243
 244
 245 ------------ SYSTEM CONFIGURATION -------------
 246
 247   Locale:   ${MYLOCALE:-none}
 248   Keymap:   ${KEYMAP:-none}
 249   Hostname: ${MYHOST:-none}
 250   Timezone: ${ZONE:-none}/${SUBZ:-none}
 251
 252
 253 ------------ USER CONFIGURATION ---------------
 254
 255   Username:      ${NEWUSER:-none}
 256   Login Shell:   ${MYSHELL:-none}
 257   Login Session: ${LOGIN_WM:-none}
 258   Autologin:     ${AUTOLOGIN:-none}
 259   Login Type:    ${LOGIN_TYPE:-none}
 260
 261
 262 ----------- PACKAGE CONFIGURATION -------------
 263
 264   Kernel:     ${KERNEL:-none}
 265   Bootloader: ${BOOTLDR:-none}
 266   Packages:   ${pkgs:-none}
 267 "
 268 }
 269
 270 select_login()
 271 {
 272         [[ $INSTALL_WMS ]] || return 0
 273
 274         AUTOLOGIN='' # no autologin unless using xinit
 275
 276         dlg LOGIN_TYPE menu "Login Management" "\nSelect what kind of login management to use." \
 277                 "xinit"   "Console login without a display manager" \
 278                 "ly"      "TUI display manager with a ncurses-like interface" \
 279                 "lightdm" "Lightweight display manager with a gtk greeter" \
 280                 "gdm"     "Gnome display manager" \
 281                 "sddm"    "Simple desktop display manager" || return 1
 282
 283         case $LOGIN_TYPE in
 284                 ly) EDIT_FILES[login]="/etc/ly/config.ini" ;;
 285                 gdm|sddm) EDIT_FILES[login]="" ;;
 286                 lightdm) EDIT_FILES[login]="/etc/lightdm/lightdm.conf /etc/lightdm/lightdm-gtk-greeter.conf" ;;
 287                 xinit) EDIT_FILES[login]="/home/$NEWUSER/.xinitrc /home/$NEWUSER/.xprofile"
 288                         if (( $(wc -w <<< "$INSTALL_WMS") > 1 )); then
 289                                 dlg LOGIN_WM menu "Login Management" "$_login" $LOGIN_CHOICES || return 1
 290                                 LOGIN_WM="${WM_SESSIONS[$LOGIN_WM]}"
 291                         fi
 292                         [[ -z $LOGIN_WM ]] && LOGIN_WM="${WM_SESSIONS[${INSTALL_WMS%% *}]}"
 293                         yesno "Autologin" "$(sed "s|USER|$NEWUSER|g; s|RC|$LOGINRC|g" <<< "$_autologin")" && AUTOLOGIN=true || AUTOLOGIN=''
 294                         ;;
 295         esac
 296 }
 297
 298 select_config()
 299 {
 300         typeset -i i=0
 301         CONFIG_DONE=''
 302
 303         until [[ $CONFIG_DONE ]]; do
 304                 case $i in
 305                         0) dlg MYSHELL menu "Shell" "\nChoose which shell to use." \
 306                                         zsh  'A very advanced and programmable command interpreter (shell) for UNIX' \
 307                                         bash 'The GNU Bourne Again shell, standard in many GNU/Linux distributions' \
 308                                         mksh 'The MirBSD Korn Shell - an enhanced version of the public domain ksh' || return 1
 309                                 ;;
 310                         1) dlg MYHOST input "Hostname" "$_hostname" "${DIST,,}" limit || { i=0; continue; } ;;
 311                         2) dlg MYLOCALE menu "Locale" "$_locale" $LOCALES || { i=1; continue; } ;;
 312                         3) ZONE='' SUBZ=''
 313                                 until [[ $ZONE && $SUBZ ]]; do
 314                                         dlg ZONE menu "Timezone" "$_timez" America - Australia - Asia - Atlantic - Africa - Europe - Indian - Pacific - Arctic - Antarctica - || break
 315                                         dlg SUBZ menu "Timezone" "$_timesubz" $(awk '/'"$ZONE"'\// {gsub(/'"$ZONE"'\//, ""); print $3 " - "}' /usr/share/zoneinfo/zone.tab | sort) || continue
 316                                 done
 317                                 [[ $ZONE && $SUBZ ]] || { i=2; continue; } ;;
 318                         4) dlg KERNEL menu "Kernel" "\nChoose which kernel to use." \
 319                                         linux          'Vanilla linux kernel and modules, with a few patches applied' \
 320                                         linux-lts      'Long-term support (LTS) linux kernel and modules' \
 321                                         linux-zen      'A effort of kernel hackers to provide the best kernel for everyday systems' \
 322                                         linux-hardened 'A security-focused linux kernel with hardening patches to mitigate exploits' || { i=3; continue; }
 323                                 CONFIG_DONE=true
 324                                 ;;
 325                 esac
 326                 (( i++ )) # progress through to the next choice
 327         done
 328
 329         case $MYSHELL in
 330                 bash) LOGINRC='.bash_profile' ;;
 331                 zsh) LOGINRC='.zprofile' ;;
 332                 mksh) LOGINRC='.profile' ;;
 333         esac
 334
 335         return 0
 336 }
 337
 338 select_mkuser()
 339 {
 340         NEWUSER=''
 341         typeset -a ans
 342
 343         until [[ $NEWUSER ]]; do
 344                 tput cnorm
 345                 dialog --insecure --backtitle "$DIST Installer - $SYS - v$VER" --separator $'\n' --title " User " --mixedform "$_user" 0 0 0 \
 346                         "Username:"  1 1 "${ans[0]}" 1 11 "$COLUMNS" 0 0 \
 347                         "Password:"  2 1 ''          2 11 "$COLUMNS" 0 1 \
 348                         "Password2:" 3 1 ''          3 12 "$COLUMNS" 0 1 \
 349                         "--- Root password, if left empty the user password will be used ---" 6 1 '' 6 68 "$COLUMNS" 0 2 \
 350                         "Password:"  8 1 ''          8 11 "$COLUMNS" 0 1 \
 351                         "Password2:" 9 1 ''          9 12 "$COLUMNS" 0 1 2>"$ANS" || return 1
 352
 353                 mapfile -t ans <"$ANS"
 354
 355                 # root passwords empty, so use the user passwords
 356                 if [[ -z "${ans[4]}" && -z "${ans[5]}" ]]; then
 357                         ans[4]="${ans[1]}"
 358                         ans[5]="${ans[2]}"
 359                 fi
 360
 361                 # make sure a username was entered and that the passwords match
 362                 if [[ -z ${ans[0]} || ${ans[0]} =~ \ |\' || ${ans[0]} =~ [^a-z0-9] ]]; then
 363                         msg "Invalid Username" "\nInvalid user name.\n\nPlease try again.\n"; u=''
 364                 elif [[ -z "${ans[1]}" || "${ans[1]}" != "${ans[2]}" ]]; then
 365                         msg "Password Mismatch" "\nThe user passwords do not match.\n\nPlease try again.\n"
 366                 elif [[ "${ans[4]}" != "${ans[5]}" ]]; then
 367                         msg "Password Mismatch" "\nThe root passwords do not match.\n\nPlease try again.\n"
 368                 else
 369                         NEWUSER="${ans[0]}"
 370                         USER_PASS="${ans[1]}"
 371                         ROOT_PASS="${ans[4]}"
 372                 fi
 373         done
 374         return 0
 375 }
 376
 377 select_keymap()
 378 {
 379         dlg KEYMAP menu "Keyboard Layout" "$_keymap" \
 380                 us English    cm    English     gb English    au English    gh English \
 381                 za English    ng    English     ca French    'cd' French    gn French \
 382                 tg French     fr    French      de German     at German     ch German \
 383                 es Spanish    latam Spanish     br Portuguese pt Portuguese ma Arabic \
 384                 sy Arabic     ara   Arabic      ua Ukrainian  cz Czech      ru Russian \
 385                 sk Slovak     nl    Dutch       it Italian    hu Hungarian  cn Chinese \
 386                 tw Taiwanese  vn    Vietnamese  kr Korean     jp Japanese   th Thai \
 387                 la Lao        pl    Polish      se Swedish    is Icelandic 'fi' Finnish \
 388                 dk Danish     be    Belgian     in Indian     al Albanian   am Armenian \
 389                 bd Bangla     ba    Bosnian    'bg' Bulgarian dz Berber     mm Burmese \
 390                 hr Croatian   gr    Greek       il Hebrew     ir Persian    iq Iraqi \
 391                 af Afghani    fo    Faroese     ge Georgian   ee Estonian   kg Kyrgyz \
 392                 kz Kazakh     lt    Lithuanian  mt Maltese    mn Mongolian  ro Romanian \
 393                 no Norwegian  rs    Serbian     si Slovenian  tj Tajik      lk Sinhala \
 394                 tr Turkish    uz    Uzbek       ie Irish      pk Urdu      'mv' Dhivehi \
 395                 np Nepali     et    Amharic     sn Wolof      ml Bambara    tz Swahili \
 396                 ke Swahili    bw    Tswana      ph Filipino   my Malay      tm Turkmen \
 397                 id Indonesian bt    Dzongkha    lv Latvian    md Moldavian mao Maori \
 398                 by Belarusian az    Azerbaijani mk Macedonian kh Khmer     epo Esperanto \
 399                 me Montenegrin || return 1
 400
 401         if [[ $CMAPS == *"$KEYMAP"* ]]; then
 402                 CMAP="$KEYMAP"
 403         else
 404                 dlg CMAP menu "Console Keymap" "$_vconsole" $CMAPS || return 1
 405         fi
 406
 407         if [[ $TERM == 'linux' ]]; then
 408                 loadkeys "$CMAP" >/dev/null 2>&1
 409         else
 410                 setxkbmap "$KEYMAP" >/dev/null 2>&1
 411         fi
 412
 413         return 0
 414 }
 415
 416 select_sessions()
 417 {
 418         LOGIN_CHOICES=''
 419         dlg INSTALL_WMS check "Sessions" "$_sessions\n" \
 420                 openbox "A lightweight, powerful, and highly configurable stacking wm" "$(ofn openbox "${INSTALL_WMS[*]}")" \
 421                 i3-gaps "A fork of i3wm with more features including gaps" "$(ofn i3-gaps "${INSTALL_WMS[*]}")" \
 422                 dwm "A dynamic WM for X that manages windows in tiled, floating, or monocle layouts" "$(ofn dwm "${INSTALL_WMS[*]}")" \
 423                 bspwm "A tiling wm that represents windows as the leaves of a binary tree" "$(ofn bspwm "${INSTALL_WMS[*]}")" \
 424                 xfce4 "A lightweight and modular desktop environment based on gtk+2/3" "$(ofn xfce4 "${INSTALL_WMS[*]}")" \
 425                 awesome "A customized Awesome WM session created by @elanapan" "$(ofn awesome "${INSTALL_WMS[*]}")" \
 426                 fluxbox "A lightweight and highly-configurable window manager" "$(ofn fluxbox "${INSTALL_WMS[*]}")" \
 427                 plasma "A kde software project currently comprising a full desktop environment" "$(ofn plasma "${INSTALL_WMS[*]}")" \
 428                 gnome "A desktop environment that aims to be simple and easy to use" "$(ofn gnome "${INSTALL_WMS[*]}")" \
 429                 cinnamon "A desktop environment combining traditional desktop with modern effects" "$(ofn cinnamon "${INSTALL_WMS[*]}")"
 430
 431         [[ $INSTALL_WMS ]] || return 0
 432
 433         WM_PKGS="${INSTALL_WMS/dwm/}" # remove dwm from package list
 434         WM_PKGS="${WM_PKGS//  / }"    # remove double spaces
 435         WM_PKGS="${WM_PKGS# }"        # remove leading space
 436
 437         for i in $INSTALL_WMS; do
 438                 LOGIN_CHOICES+="$i - "
 439                 [[ ${WM_EXT[$i]} && $WM_PKGS != *"${WM_EXT[$i]}"* ]] && WM_PKGS+=" ${WM_EXT[$i]}"
 440         done
 441
 442         select_login || return 1
 443
 444         while IFS=' ' read -r pkg; do
 445                 [[ $PACKAGES != *"$pkg"* ]] && PACKAGES+=" $pkg"
 446         done <<< "$WM_PKGS"
 447
 448         return 0
 449 }
 450
 451 select_packages()
 452 {
 453         dlg USER_PKGS check " Packages " "$_packages" \
 454                 abiword "A Fully-featured word processor" "$(ofn abiword "${USER_PKGS[*]}")" \
 455                 alacritty "A cross-platform, GPU-accelerated terminal emulator" "$(ofn alacritty "${USER_PKGS[*]}")" \
 456                 atom "An open-source text editor developed by GitHub" "$(ofn atom "${USER_PKGS[*]}")" \
 457                 audacious "A free and advanced audio player based on GTK+" "$(ofn audacious "${USER_PKGS[*]}")" \
 458                 audacity "A program that lets you manipulate digital audio waveforms" "$(ofn audacity "${USER_PKGS[*]}")" \
 459                 blueman "GUI bluetooth device manager" "$(ofn blueman "${USER_PKGS[*]}")" \
 460                 bluez "Simple CLI based bluetooth support" "$(ofn bluez "${USER_PKGS[*]}")" \
 461                 cairo-dock "Light eye-candy fully themable animated dock" "$(ofn cairo-dock "${USER_PKGS[*]}")" \
 462                 calligra "A set of applications for productivity" "$(ofn calligra "${USER_PKGS[*]}")" \
 463                 chromium "An open-source web browser based on the Blink rendering engine" "$(ofn chromium "${USER_PKGS[*]}")" \
 464                 clementine "A modern music player and library organizer" "$(ofn clementine "${USER_PKGS[*]}")" \
 465                 cmus "A small, fast and powerful console music player" "$(ofn cmus "${USER_PKGS[*]}")" \
 466                 deadbeef "A GTK+ audio player for GNU/Linux" "$(ofn deadbeef "${USER_PKGS[*]}")" \
 467                 deluge "A BitTorrent client written in python" "$(ofn deluge "${USER_PKGS[*]}")" \
 468                 emacs "An extensible, customizable, self-documenting real-time display editor" "$(ofn emacs "${USER_PKGS[*]}")" \
 469                 epiphany "A GNOME web browser based on the WebKit rendering engine" "$(ofn epiphany "${USER_PKGS[*]}")" \
 470                 evince "A document viewer" "$(ofn evince "${USER_PKGS[*]}")" \
 471                 evolution "Manage your email, contacts and schedule" "$(ofn evolution "${USER_PKGS[*]}")" \
 472                 file-roller "Create and modify archives" "$(ofn file-roller "${USER_PKGS[*]}")" \
 473                 firefox "A popular open-source web browser from Mozilla" "$(ofn firefox "${USER_PKGS[*]}")" \
 474                 gcolor2 "A simple GTK+2 color selector" "$(ofn gcolor2 "${USER_PKGS[*]}")" \
 475                 geany "A fast and lightweight IDE" "$(ofn geany "${USER_PKGS[*]}")" \
 476                 geary "A lightweight email client for the GNOME desktop" "$(ofn geary "${USER_PKGS[*]}")" \
 477                 gimp "GNU Image Manipulation Program" "$(ofn gimp "${USER_PKGS[*]}")" \
 478                 gnome-calculator "GNOME Scientific calculator" "$(ofn gnome-calculator "${USER_PKGS[*]}")" \
 479                 gnome-disk-utility "Disk Management Utility" "$(ofn gnome-disk-utility "${USER_PKGS[*]}")" \
 480                 gnome-system-monitor "View current processes and monitor system state" "$(ofn gnome-system-monitor "${USER_PKGS[*]}")" \
 481                 gparted "A GUI frontend for creating and manipulating partition tables" "$(ofn gparted "${USER_PKGS[*]}")" \
 482                 gpick "Advanced color picker using GTK+ toolkit" "$(ofn gpick "${USER_PKGS[*]}")" \
 483                 gpicview "Lightweight image viewer" "$(ofn gpicview "${USER_PKGS[*]}")" \
 484                 guvcview "Capture video from camera devices" "$(ofn guvcview "${USER_PKGS[*]}")" \
 485                 hexchat "A popular and easy to use graphical IRC client" "$(ofn hexchat "${USER_PKGS[*]}")" \
 486                 inkscape "Professional vector graphics editor" "$(ofn inkscape "${USER_PKGS[*]}")" \
 487                 irssi "Modular text mode IRC client" "$(ofn irssi "${USER_PKGS[*]}")" \
 488                 kdenlive "A popular non-linear video editor for Linux" "$(ofn kdenlive "${USER_PKGS[*]}")" \
 489                 krita "Edit and paint images" "$(ofn krita "${USER_PKGS[*]}")" \
 490                 libreoffice-fresh "Full featured office suite" "$(ofn libreoffice-fresh "${USER_PKGS[*]}")" \
 491                 lollypop "A new music playing application" "$(ofn lollypop "${USER_PKGS[*]}")" \
 492                 mousepad "A simple text editor" "$(ofn mousepad "${USER_PKGS[*]}")" \
 493                 mpd "A flexible, powerful, server-side application for playing music" "$(ofn mpd "${USER_PKGS[*]}")" \
 494                 mpv "A media player based on mplayer" "$(ofn mpv "${USER_PKGS[*]}")" \
 495                 mupdf "Lightweight PDF and XPS viewer" "$(ofn mupdf "${USER_PKGS[*]}")" \
 496                 mutt "Small but very powerful text-based mail client" "$(ofn mutt "${USER_PKGS[*]}")" \
 497                 nautilus "The default file manager for Gnome" "$(ofn nautilus "${USER_PKGS[*]}")" \
 498                 ncmpcpp "A mpd client and almost exact clone of ncmpc with some new features" "$(ofn ncmpcpp "${USER_PKGS[*]}")" \
 499                 neovim "A fork of Vim aiming to improve user experience, plugins, and GUIs." "$(ofn neovim "${USER_PKGS[*]}")" \
 500                 noto-fonts "Google Noto fonts" "$(ofn noto-fonts "${USER_PKGS[*]}")" \
 501                 noto-fonts-cjk "Google Noto CJK fonts (Chinese, Japanese, Korean)" "$(ofn noto-fonts-cjk "${USER_PKGS[*]}")" \
 502                 obs-studio "Free opensource streaming/recording software" "$(ofn obs-studio "${USER_PKGS[*]}")" \
 503                 openshot "An open-source, non-linear video editor for Linux" "$(ofn openshot "${USER_PKGS[*]}")" \
 504                 opera "A Fast and secure, free of charge web browser from Opera Software" "$(ofn opera "${USER_PKGS[*]}")" \
 505                 pcmanfm "A fast and lightweight file manager based in Lxde" "$(ofn pcmanfm "${USER_PKGS[*]}")" \
 506                 pidgin "Multi-protocol instant messaging client" "$(ofn pidgin "${USER_PKGS[*]}")" \
 507                 plank "An elegant, simple, and clean dock" "$(ofn plank "${USER_PKGS[*]}")" \
 508                 qbittorrent "An advanced BitTorrent client" "$(ofn qbittorrent "${USER_PKGS[*]}")" \
 509                 qpdfview "A tabbed PDF viewer" "$(ofn qpdfview "${USER_PKGS[*]}")" \
 510                 qt5ct "GUI for managing Qt based application themes, icons, and fonts" "$(ofn qt5ct "${USER_PKGS[*]}")" \
 511                 qutebrowser "A keyboard-focused vim-like web browser based on Python and PyQt5" "$(ofn qutebrowser "${USER_PKGS[*]}")" \
 512                 rhythmbox "A Music playback and management application" "$(ofn rhythmbox "${USER_PKGS[*]}")" \
 513                 rxvt-unicode "A unicode enabled rxvt-clone terminal emulator" "$(ofn rxvt-unicode "${USER_PKGS[*]}")" \
 514                 sakura "A terminal emulator based on GTK and VTE" "$(ofn sakura "${USER_PKGS[*]}")" \
 515                 simple-scan "Simple scanning utility" "$(ofn simple-scan "${USER_PKGS[*]}")" \
 516                 simplescreenrecorder "A feature-rich screen recorder" "$(ofn simplescreenrecorder "${USER_PKGS[*]}")" \
 517                 steam "A popular game distribution platform by Valve" "$(ofn steam "${USER_PKGS[*]}")" \
 518                 surf "A simple web browser based on WebKit2/GTK+" "$(ofn surf "${USER_PKGS[*]}")" \
 519                 terminator "Terminal emulator that supports tabs and grids" "$(ofn terminator "${USER_PKGS[*]}")" \
 520                 termite "A minimal VTE-based terminal emulator" "$(ofn termite "${USER_PKGS[*]}")" \
 521                 thunar "A modern file manager for the Xfce Desktop Environment" "$(ofn thunar "${USER_PKGS[*]}")" \
 522                 thunderbird "Standalone mail and news reader from mozilla" "$(ofn thunderbird "${USER_PKGS[*]}")" \
 523                 tilda "A GTK based drop down terminal for Linux and Unix" "$(ofn tilda "${USER_PKGS[*]}")" \
 524                 tilix "A tiling terminal emulator for Linux using GTK+ 3" "$(ofn tilix "${USER_PKGS[*]}")" \
 525                 transmission-cli "Free BitTorrent client CLI" "$(ofn transmission-cli "${USER_PKGS[*]}")" \
 526                 transmission-gtk "GTK+ Front end for transmission" "$(ofn transmission-gtk "${USER_PKGS[*]}")" \
 527                 transmission-qt "Qt Front end for transmission" "$(ofn transmission-qt "${USER_PKGS[*]}")" \
 528                 ttf-anonymous-pro "A family fixed-width fonts designed with code in mind" "$(ofn ttf-anonymous-pro "${USER_PKGS[*]}")" \
 529                 ttf-fira-code "Monospaced font with programming ligatures" "$(ofn ttf-fira-code "${USER_PKGS[*]}")" \
 530                 ttf-font-awesome "Iconic font designed for Bootstrap" "$(ofn ttf-font-awesome "${USER_PKGS[*]}")" \
 531                 ttf-hack "A hand groomed typeface based on Bitstream Vera Mono" "$(ofn ttf-hack "${USER_PKGS[*]}")" \
 532                 vlc "A free and open source cross-platform multimedia player" "$(ofn vlc "${USER_PKGS[*]}")" \
 533                 weechat "Fast, light and extensible IRC client" "$(ofn weechat "${USER_PKGS[*]}")" \
 534                 xapps "Common library for X-Apps project" "$(ofn xapps "${USER_PKGS[*]}")" \
 535                 xarchiver "A GTK+ frontend to various command line archivers" "$(ofn xarchiver "${USER_PKGS[*]}")" \
 536                 xed "A small and lightweight text editor. X-Apps Project." "$(ofn xed "${USER_PKGS[*]}")" \
 537                 xfce4-terminal "A terminal emulator based in the Xfce Desktop Environment" "$(ofn xfce4-terminal "${USER_PKGS[*]}")" \
 538                 xreader "Document viewer for files like PDF and Postscript. X-Apps Project." "$(ofn xed "${USER_PKGS[*]}")" \
 539                 xterm "The standard terminal emulator for the X window system" "$(ofn xterm "${USER_PKGS[*]}")" \
 540                 zathura "Minimalistic document viewer" "$(ofn zathura "${USER_PKGS[*]}")"
 541
 542         if [[ $USER_PKGS ]]; then    # add any needed PKG_EXT to the list
 543                 for i in $USER_PKGS; do
 544                         [[ ${PKG_EXT[$i]} && $USER_PKGS != *"${PKG_EXT[$i]}"* ]] && USER_PKGS+=" ${PKG_EXT[$i]}"
 545                 done
 546         fi
 547
 548         return 0
 549 }
 550
 551 ###############################################################################
 552 # partitioning menus
 553 # non-essential partitioning helpers called by the user when using the optional
 554 # partition menu and selecting a device to edit
 555
 556 part_menu()
 557 {
 558         no_bg_install || return 0
 559         local device choice devhash
 560         devhash="$(lsblk -f | base64)"
 561         umount_dir $MNT
 562         part_device || return 1
 563         device="$DEVICE"
 564
 565         while :; do
 566                 choice=""
 567                 dlg choice menu "Edit Partitions" "$_part\n\n$(lsblk -no NAME,MODEL,SIZE,TYPE,FSTYPE $device)" \
 568                         "auto"   "Whole device automatic partitioning" \
 569                         "shrink" "Shrink an existing ext or ntfs partition" \
 570                         "cfdisk" "Curses based variant of fdisk" \
 571                         "parted" "GNU partition editor" \
 572                         "fdisk"  "Dialog-driven creation and manipulation of partitions" \
 573                         "done"   "Return to the main menu"
 574
 575                 if [[ -z $choice || $choice == 'done' ]]; then
 576                         return 0
 577                 elif [[ $choice == 'shrink' ]]; then
 578                         part_shrink "$device"
 579                 elif [[ $choice == 'auto' ]]; then
 580                         local root_size txt table boot_fs
 581                         root_size=$(lsblk -lno SIZE "$device" | awk 'NR == 1 {
 582                                 if ($1 ~ "G") {
 583                                         sub(/G/, "")
 584                                         print ($1 * 1000 - 512) / 1000 "G"
 585                                 } else {
 586                                         sub(/M/, "")
 587                                         print ($1 - 512) "M"
 588                                 }
 589                         }')
 590                         txt="\nWARNING:\n\nALL data on $device will be destroyed and the following partitions will be created\n\n- "
 591                         if [[ $SYS == 'BIOS' ]]; then
 592                                 table="msdos" boot_fs="ext4"
 593                                 txt+="An $boot_fs boot partition with the boot flag enabled (512M)\n- "
 594                         else
 595                                 table="gpt" boot_fs="fat32"
 596                                 txt+="A $boot_fs efi boot partition (512M)\n- "
 597                         fi
 598                         txt+="An ext4 partition using all remaining space ($root_size)\n\nDo you want to continue?\n"
 599                         yesno "Auto Partition" "$txt" && part_auto "$device" "$table" "$boot_fs" "$root_size"
 600                 else
 601                         clear
 602                         tput cnorm
 603                         $choice "$device"
 604                 fi
 605                 if [[ $devhash != "$(lsblk -f | base64)" ]]; then
 606                         msg "Probing Partitions" "\nInforming kernel of partition changes using partprobe\n" 0
 607                         partprobe >/dev/null 2>&1
 608                         [[ $choice == 'auto' ]] && return
 609                 fi
 610         done
 611 }
 612
 613 part_show()
 614 {
 615         local txt
 616         if [[ $IGNORE_DEV ]]; then
 617                 txt="$(lsblk -o NAME,MODEL,SIZE,TYPE,FSTYPE,MOUNTPOINT | awk "!/$IGNORE_DEV/"' && /disk|part|lvm|crypt|NAME/')"
 618         else
 619                 txt="$(lsblk -o NAME,MODEL,SIZE,TYPE,FSTYPE,MOUNTPOINT | awk '/disk|part|lvm|crypt|NAME/')"
 620         fi
 621         msg "Device Tree" "\n\n$txt\n\n"
 622 }
 623
 624 part_auto()
 625 {
 626         local device="$1" table="$2" boot_fs="$3" size="$4" dev_info=""
 627         dev_info="$(parted -s "$device" print)"
 628
 629         msg "Auto Partition" "\nRemoving partitions on $device and setting table to $table\n" 1
 630
 631         swapoff -a
 632         while read -r PART; do
 633                 parted -s "$device" rm "$PART" >/dev/null 2>&1
 634         done <<< "$(awk '/^ [1-9][0-9]?/ {print $1}' <<< "$dev_info" | sort -r)"
 635
 636         [[ $(awk '/Table:/ {print $3}' <<< "$dev_info") != "$table" ]] && parted -s "$device" mklabel "$table" >/dev/null 2>&1
 637
 638         msg "Auto Partition" "\nCreating a 512M $boot_fs boot partition.\n" 1
 639         if [[ $SYS == "BIOS" ]]; then
 640                 parted -s "$device" mkpart primary "$boot_fs" 1MiB 513MiB >/dev/null 2>&1
 641         else
 642                 parted -s "$device" mkpart ESP "$boot_fs" 1MiB 513MiB >/dev/null 2>&1
 643         fi
 644
 645         sleep 0.5
 646         BOOT_DEV="$device"
 647         AUTO_BOOT_PART=$(lsblk -lno NAME,TYPE "$device" | awk 'NR==2 {print "/dev/" $1}')
 648
 649         if [[ $SYS == "BIOS" ]]; then
 650                 mkfs.ext4 -q "$AUTO_BOOT_PART" >/dev/null 2>&1
 651         else
 652                 mkfs.vfat -F32 "$AUTO_BOOT_PART" >/dev/null 2>&1
 653         fi
 654
 655         msg "Auto Partition" "\nCreating a $size ext4 root partition.\n" 0
 656         parted -s "$device" mkpart primary ext4 513MiB 100% >/dev/null 2>&1
 657         sleep 0.5
 658         AUTO_ROOT_PART="$(lsblk -lno NAME,TYPE "$device" | awk 'NR==3 {print "/dev/" $1}')"
 659         mkfs.ext4 -q "$AUTO_ROOT_PART" >/dev/null 2>&1
 660         sleep 0.5
 661         msg "Auto Partition" "\nProcess complete.\n\n$(lsblk -o NAME,MODEL,SIZE,TYPE,FSTYPE "$device")\n"
 662 }
 663
 664 part_shrink()
 665 {
 666         part=""
 667         typeset -i size num
 668         local device="$1" fs=""
 669
 670         part_find "${device##*/}[^ ]" || return 1
 671         (( COUNT == 1 )) && part="$(awk '{print $1}' <<< "${PARTS[@]}" )"
 672
 673         if (( COUNT == 1 )) || dlg part menu "Resize" "\nWhich partition on $device do you want to resize?" $PARTS; then
 674                 fs=$(lsblk -lno FSTYPE "$part")
 675                 case "$fs" in
 676                         ext*|ntfs)
 677                                 msg "Resize" "\nGathering device size info.\n" 0
 678                                 num="${part: -1}"
 679                                 end=$(parted -s "$device" unit KiB print | awk '/^\s*'"$num"'/ {print $3}')                    # part size in KiB
 680                                 devsize=$(parted -s "$device" unit KiB print | awk '/Disk '"${device//\//\\/}"':/ {print $3}') # whole device size in KiB
 681                                 mount "$part" $MNT >/dev/null 2>&1; sleep 0.5
 682                                 min=$(df --output=used --block-size=MiB "$part" | awk 'NR == 2 {print int($1) + 256}')
 683                                 max=$(df --output=avail --block-size=MiB "$part" | awk 'NR == 2 {print int($1)}')
 684                                 umount_dir $MNT
 685                                 tput cnorm
 686                                 if dialog --backtitle "$DIST Installer - $SYS - v$VER" --title " Resize: $part " --rangebox "$_resize" 17 "$COLUMNS" "$min" "$max" $((max / 2)) 2>$ANS; then
 687                                         size=$(< "$ANS")
 688                                         size=$((size * 1024))
 689                                 else
 690                                         return 1
 691                                 fi
 692                                 clear
 693                                 case "$fs" in
 694                                         ntfs)
 695                                                 if ntfsresize -fc "$part"; then
 696                                                         ntfsresize -ff --size $(( (size * 1024) / 1000 ))k "$part" 2>$ERR # k=10^3 bytes
 697                                                         errshow "ntfsresize -f -s $(( (size * 1024) / 1000 ))k $part" || return 1
 698                                                 else
 699                                                         msg "Resize" "\nThe ntfs partition $part cannot be resized because it is scheduled for a consistency check.\n\nTo do a consistency check in windows open command prompt as admin and run:\n\n\tchkdsk /f /r /x\n"
 700                                                         return 1
 701                                                 fi
 702                                                 ;;
 703                                         *)
 704                                                 e2fsck -f "$part"; sleep 0.5
 705                                                 resize2fs -f "$part" ${size}K 2>$ERR # K=2^10 bytes
 706                                                 errshow "resize2fs -f $part ${size}K" || return 1
 707                                                 ;;
 708                                 esac
 709                                 sleep 0.5
 710                                 parted "$device" resizepart "$num" ${size}KiB || return 1
 711                                 (( size++ ))
 712                                 sleep 0.5
 713                                 if [[ $devsize == "$end" ]]; then
 714                                         parted -s "$device" mkpart primary ext4 ${size}KiB 100% 2>$ERR
 715                                         errshow "parted -s $device mkpart primary ext4 ${size}KiB 100%" || return 1
 716                                 else
 717                                         parted -s "$device" mkpart primary ext4 ${size}KiB ${end}KiB 2>$ERR
 718                                         errshow "parted -s $device mkpart primary ext4 ${size}KiB ${end}KiB" || return 1
 719                                 fi
 720                                 msg "Resize Complete" "\n$part has been successfully resized to $((size / 1024))M.\n" 1
 721                                 ;;
 722                         "") msg "No Filesystem" "\nFor unformatted partitions, cfdisk can be used in the partition menu.\n" ;;
 723                         *) msg "Invalid Filesystem: $fs" "\nResizing only supports ext and ntfs.\n" ;;
 724                 esac
 725         fi
 726 }
 727
 728 ###############################################################################
 729 # partition management functions
 730 # these are helpers for use by other functions to do essential setup/teardown
 731
 732 part_find()
 733 {
 734         local regexp="$1" err=''
 735
 736         # string of partitions as /TYPE/PART SIZE
 737         if [[ $IGNORE_DEV ]]; then
 738                 PARTS="$(lsblk -lno TYPE,NAME,SIZE |
 739                         awk "/$regexp/"' && !'"/$IGNORE_DEV/"' {
 740                                 sub(/^part/, "/dev/")
 741                                 sub(/^lvm|^crypt/, "/dev/mapper/")
 742                                 print $1$2, $3
 743                         }')"
 744         else
 745                 PARTS="$(lsblk -lno TYPE,NAME,SIZE |
 746                         awk "/$regexp/"' {
 747                                 sub(/^part/, "/dev/")
 748                                 sub(/^lvm|^crypt/, "/dev/mapper/")
 749                                 print $1$2 " " $3
 750                         }')"
 751         fi
 752
 753         # number of partitions total
 754         COUNT=0
 755         while read -r line; do
 756                 (( COUNT++ ))
 757         done <<< "$PARTS"
 758
 759         # ensure we have enough partitions for the system and action type
 760         case "$str" in
 761                 'part|lvm|crypt') [[ $COUNT -lt 1 || ($SYS == 'UEFI' && $COUNT -lt 2) ]] && err="$_errpart" ;;
 762                 'part|crypt') (( COUNT < 1 )) && err="$_lvmerr" ;;
 763                 'part|lvm') (( COUNT < 2 )) && err="$_lukserr" ;;
 764         esac
 765
 766         # if there aren't enough partitions show the relevant error message
 767         [[ $err ]] && { msg "Not Enough Partitions" "$err" 2; return 1; }
 768
 769         return 0
 770 }
 771
 772 part_swap()
 773 {
 774         if [[ $1 == "$MNT/swapfile" && $SWAP_SIZE ]]; then
 775                 fallocate -l $SWAP_SIZE "$1" 2>$ERR
 776                 errshow "fallocate -l $SWAP_SIZE $1"
 777                 chmod 600 "$1" 2>$ERR
 778                 errshow "chmod 600 $1"
 779         fi
 780         mkswap "$1" >/dev/null 2>$ERR
 781         errshow "mkswap $1"
 782         swapon "$1" >/dev/null 2>$ERR
 783         errshow "swapon $1"
 784         return 0
 785 }
 786
 787 part_mount()
 788 {
 789         local part="$1" mountp="${MNT}$2" fs=""
 790         fs="$(lsblk -lno FSTYPE "$part")"
 791         mkdir -p "$mountp"
 792
 793         if [[ $fs && ${FS_OPTS[$fs]} && $part != "$BOOT_PART" && $part != "$AUTO_ROOT_PART" ]] && select_mntopts "$fs"; then
 794                 mount -o "$MNT_OPTS" "$part" "$mountp" >/dev/null 2>&1
 795         else
 796                 mount "$part" "$mountp" >/dev/null 2>&1
 797         fi
 798
 799         part_mountconf "$part" "$mountp" || return 1
 800         part_cryptlv "$part"
 801
 802         return 0
 803 }
 804
 805 part_format()
 806 {
 807         local part="$1" fs="$2" delay="$3"
 808
 809         msg "Format" "\nFormatting $part as $fs\n" 0
 810         mkfs.$fs ${FS_CMD_FLAGS[$fs]} "$part" >/dev/null 2>$ERR
 811         errshow "mkfs.$fs ${FS_CMD_FLAGS[$fs]} "$part"" || return 1
 812         FORMATTED+="$part "
 813         sleep "${delay:-0}"
 814 }
 815
 816 part_device()
 817 {
 818         if [[ $DEV_COUNT -eq 1 && $SYS_DEVS ]]; then
 819                 DEVICE="$(awk '{print $1}' <<< "$SYS_DEVS")"
 820         elif (( DEV_COUNT > 1 )); then
 821                 if [[ $1 ]]; then
 822                         dlg DEVICE menu "Boot Device" "\nSelect the device to use for bootloader install." $SYS_DEVS
 823                 else
 824                         dlg DEVICE menu "Select Device" "$_device" $SYS_DEVS
 825                 fi
 826                 [[ $DEVICE ]] || return 1
 827         elif [[ $DEV_COUNT -lt 1 && ! $1 ]]; then
 828                 msg "Device Error" "\nNo available devices.\n\nExiting..\n" 2
 829                 die 1
 830         fi
 831
 832         [[ $1 ]] && BOOT_DEV="$DEVICE"
 833
 834         return 0
 835 }
 836
 837 part_bootdev()
 838 {
 839         BOOT_DEV="${BOOT_PART%[1-9]}"
 840         BOOT_PART_NUM="${BOOT_PART: -1}"
 841         [[ $BOOT_PART = /dev/nvme* ]] && BOOT_DEV="${BOOT_PART%p[1-9]}"
 842         if [[ $SYS == 'UEFI' ]]; then
 843                 parted -s $BOOT_DEV set $BOOT_PART_NUM esp on >/dev/null 2>&1
 844         else
 845                 parted -s $BOOT_DEV set $BOOT_PART_NUM boot on >/dev/null 2>&1
 846         fi
 847         return 0
 848 }
 849
 850 part_cryptlv()
 851 {
 852         local part="$1" devs=""
 853         devs="$(lsblk -lno NAME,FSTYPE,TYPE)"
 854
 855         # Identify if $part is LUKS+LVM, LVM+LUKS, LVM alone, or LUKS alone
 856         if lsblk -lno TYPE "$part" | grep -q 'crypt'; then
 857                 LUKS='encrypted'
 858                 LUKS_NAME="${part#/dev/mapper/}"
 859                 for dev in $(awk '/lvm/ && /crypto_LUKS/ {print "/dev/mapper/"$1}' <<< "$devs" | uniq); do
 860                         if lsblk -lno NAME "$dev" | grep -q "$LUKS_NAME"; then
 861                                 LUKS_DEV="$LUKS_DEV cryptdevice=$dev:$LUKS_NAME"
 862                                 LVM='logical volume'
 863                                 break
 864                         fi
 865                 done
 866                 for dev in $(awk '/part/ && /crypto_LUKS/ {print "/dev/"$1}' <<< "$devs" | uniq); do
 867                         if lsblk -lno NAME "$dev" | grep -q "$LUKS_NAME"; then
 868                                 LUKS_UUID="$(lsblk -lno UUID,TYPE,FSTYPE "$dev" | awk '/part/ && /crypto_LUKS/ {print $1}')"
 869                                 LUKS_DEV="$LUKS_DEV cryptdevice=UUID=$LUKS_UUID:$LUKS_NAME"
 870                                 break
 871                         fi
 872                 done
 873         elif lsblk -lno TYPE "$part" | grep -q 'lvm'; then
 874                 LVM='logical volume'
 875                 VNAME="${part#/dev/mapper/}"
 876                 for dev in $(awk '/crypt/ && /lvm2_member/ {print "/dev/mapper/"$1}' <<< "$devs" | uniq); do
 877                         if lsblk -lno NAME "$dev" | grep -q "$VNAME"; then
 878                                 LUKS_NAME="${dev/\/dev\/mapper\//}"
 879                                 break
 880                         fi
 881                 done
 882                 for dev in $(awk '/part/ && /crypto_LUKS/ {print "/dev/"$1}' <<< "$devs" | uniq); do
 883                         if lsblk -lno NAME "$dev" | grep -q "$LUKS_NAME"; then
 884                                 LUKS_UUID="$(lsblk -lno UUID,TYPE,FSTYPE "$dev" | awk '/part/ && /crypto_LUKS/ {print $1}')"
 885                                 LUKS_DEV="$LUKS_DEV cryptdevice=UUID=$LUKS_UUID:$LUKS_NAME"
 886                                 LUKS='encrypted'
 887                                 break
 888                         fi
 889                 done
 890         fi
 891 }
 892
 893 part_countdec()
 894 {
 895         for pt; do
 896                 if (( COUNT > 0 )); then
 897                         PARTS="$(sed "/${pt//\//\\/}/d" <<< "$PARTS")"
 898                         (( COUNT-- ))
 899                 fi
 900         done
 901 }
 902
 903 part_mountconf()
 904 {
 905         if grep -qw "$1" /proc/mounts; then
 906                 msg "Mount Success" "\nPartition $1 mounted at $2\n" 1
 907                 part_countdec "$1"
 908                 return 0
 909         else
 910                 msg "Mount Fail" "\nPartition $1 failed to mount at $2\n" 2
 911                 return 1
 912         fi
 913 }
 914
 915 ###############################################################################
 916 # mounting menus
 917 # mount_menu is the entry point which calls all other functions
 918 # once finished it returns to the main menu: main()
 919
 920 mount_menu()
 921 {
 922         msg "Info" "\nGathering device info.\n" 0
 923         no_bg_install || return 0
 924         lvm_detect
 925         umount_dir $MNT
 926         part_find 'part|lvm|crypt' || { SEL=2; return 1; }
 927
 928         [[ $LUKS && $LUKS_PART ]] && part_countdec $LUKS_PART
 929         [[ $LVM && $LVM_PARTS ]] && part_countdec $LVM_PARTS
 930
 931         select_root_partition || return 1
 932
 933         if [[ $SYS == 'UEFI' ]]; then
 934                 select_efi_partition || { BOOT_PART=''; return 1; }
 935         elif (( COUNT > 0 )); then
 936                 select_boot_partition || { BOOT_PART=''; return 1; }
 937         fi
 938
 939         if [[ $BOOT_PART ]]; then
 940                 part_mount "$BOOT_PART" "/$BOOTDIR" && SEP_BOOT=true || return 1
 941                 part_bootdev
 942         fi
 943
 944         select_swap || return 1
 945         select_extra_partitions || return 1
 946         install_background
 947
 948         return 0
 949 }
 950
 951 select_swap()
 952 {
 953         dlg SWAP_PART menu "Swap Setup" "\nSelect whether to use a swapfile, swap partition, or none." \
 954                 "none" "Don't allocate any swap space" \
 955                 "swapfile" "Allocate $SYS_MEM at /swapfile" \
 956                 $PARTS
 957
 958         if [[ -z $SWAP_PART || $SWAP_PART == "none" ]]; then
 959                 SWAP_PART=''
 960                 return 0
 961         elif [[ $SWAP_PART == "swapfile" ]]; then
 962                 local i=0
 963                 until [[ ${SWAP_SIZE:0:1} =~ [1-9] && ${SWAP_SIZE: -1} =~ (M|G) ]]; do
 964                         (( i > 0 )) && msg "Swap Size Error" "\nSwap size must be 1(M|G) or greater, and can only contain whole numbers\n\nSize entered: $SWAP_SIZE\n" 2
 965                         dlg SWAP_SIZE input "Swap Setup" "$_swapsize" "$SYS_MEM" || { SWAP_PART=''; SWAP_SIZE=''; return 1; }
 966                         (( i++ ))
 967                 done
 968                 part_swap "$MNT/$SWAP_PART"
 969                 SWAP_PART="/$SWAP_PART"
 970         elif [[ $PARTS == *"$SWAP_PART"* ]]; then
 971                 part_swap $SWAP_PART
 972                 part_countdec $SWAP_PART
 973                 SWAP_SIZE="$(lsblk -lno SIZE $SWAP_PART)"
 974         else
 975                 return 1
 976         fi
 977
 978         return 0
 979 }
 980
 981 select_mntopts()
 982 {
 983         local fs="$1" opts=''
 984         local title="${fs^} Mount Options"
 985
 986         for i in ${FS_OPTS[$fs]}; do
 987                 opts+="$i - off "
 988         done
 989
 990         until [[ $MNT_OPTS ]]; do
 991                 dlg MNT_OPTS check "$title" "$_mount" $opts
 992                 [[ $MNT_OPTS ]] || return 1
 993                 MNT_OPTS="${MNT_OPTS// /,}"
 994                 yesno "$title" "\nConfirm the following options: $MNT_OPTS\n" || MNT_OPTS=''
 995         done
 996
 997         return 0
 998 }
 999
1000 select_mountpoint()
1001 {
1002         EXMNT=''
1003         until [[ $EXMNT ]]; do
1004                 dlg EXMNT input "Extra Mount $part" "$_exmnt" "/" || return 1
1005                 if [[ ${EXMNT:0:1} != "/" || ${#EXMNT} -le 1 || $EXMNT =~ \ |\' || $EXMNTS == *"$EXMNT"* ]]; then
1006                         msg "Mountpoint Error" "$_errexpart"
1007                         EXMNT=''
1008                 fi
1009         done
1010         return 0
1011 }
1012
1013 select_filesystem()
1014 {
1015         local part="$1" fs='' cur=''
1016         local txt="\nSelect which filesystem to use for: $part\n\nDefault:  ext4"
1017         cur="$(lsblk -lno FSTYPE "$part" 2>/dev/null)"
1018
1019         # bail early if the partition was created in part_auto()
1020         [[ $cur && $part == "$AUTO_ROOT_PART" ]] && return 0
1021
1022         until [[ $fs ]]; do
1023                 if [[ $cur && $FORMATTED == *"$part"* ]]; then
1024                         dlg fs menu "Filesystem" "$txt\nCurrent:  $cur" skip - ext4 - ext3 - ext2 - vfat - ntfs - f2fs - jfs - xfs - nilfs2 - reiserfs - || return 1
1025                 else
1026                         dlg fs menu "Filesystem" "$txt" ext4 - ext3 - ext2 - vfat - ntfs - f2fs - jfs - xfs - nilfs2 - reiserfs - || return 1
1027                 fi
1028                 [[ $fs == 'skip' ]] && return 0
1029                 yesno "Filesystem" "\nFormat $part as $fs?\n" || fs=''
1030         done
1031         part_format "$part" "$fs"
1032 }
1033
1034 select_efi_partition()
1035 {
1036         if [[ $AUTO_BOOT_PART ]]; then
1037                 BOOT_PART="$AUTO_BOOT_PART"
1038                 return 0 # were done here
1039         else
1040                 local pts size dev isize bsize ptcount=0
1041
1042                 # walk partition list and skip ones that are too small/big for boot
1043                 while read -r dev size; do
1044                         size_t="${size: -1:1}"  # size type eg. K, M, G, T
1045                         isize=${size:0:-1}      # remove trailing size type character
1046                         isize=${isize%.*}       # remove any decimal (round down)
1047                         [[ $size_t =~ [KT] || ($size_t == 'G' && $isize -gt 2) || ($size_t == 'M' && $isize -lt 100) ]] || { pts+="$dev $size "; (( ptcount++ )); }
1048                 done <<< "$PARTS"
1049
1050                 if (( ptcount == 1 )); then
1051                         msg "EFI Boot Partition" "\nOnly one partition available that meets size requirements.\n" 1
1052                         BOOT_PART="$(awk 'NF > 0 {print $1}' <<< "$pts")"
1053                 else
1054                         dlg BOOT_PART menu "EFI Partition" "$_uefi" $pts
1055                 fi
1056         fi
1057
1058         if [[ -z $BOOT_PART ]]; then
1059                 return 1
1060         elif grep -q 'fat' <<< "$(fsck -N "$BOOT_PART")"; then
1061                 local txt="\nIMPORTANT:\n\nThe EFI partition $BOOT_PART $_format"
1062                 if yesno "Format EFI Partition" "$txt" "Format $BOOT_PART" "Skip Formatting" 1; then
1063                         part_format "$BOOT_PART" "vfat" 2
1064                 fi
1065         else
1066                 part_format "$BOOT_PART" "vfat" 2
1067         fi
1068
1069         return 0
1070 }
1071
1072 select_boot_partition()
1073 {
1074         if [[ $AUTO_BOOT_PART && ! $LVM ]]; then
1075                 BOOT_PART="$AUTO_BOOT_PART"
1076                 return 0 # were done here
1077         else
1078                 local pts size dev isize bsize ptcount=0
1079
1080                 # walk partition list and skip ones that are too small/big for boot
1081                 while read -r dev size; do
1082                         size_t="${size: -1:1}"  # size type eg. K, M, G, T
1083                         isize=${size:0:-1}      # remove trailing size type character
1084                         isize=${isize%.*}       # remove any decimal (round down)
1085                         [[ $size_t =~ [KT] || ($size_t == 'G' && $isize -gt 2) || ($size_t == 'M' && $isize -lt 100) ]] || { pts+="$dev $size "; (( ptcount++ )); }
1086                 done <<< "$PARTS"
1087
1088                 if [[ $LUKS && ! $LVM ]]; then
1089                         dlg BOOT_PART menu "Boot Partition" "$_biosluks" $pts
1090                         [[ $BOOT_PART ]] || return 1
1091                 else
1092                         dlg BOOT_PART menu "Boot Partition" "$_bios" "skip" "don't use a separate boot" $pts
1093                         [[ -z $BOOT_PART || $BOOT_PART == "skip" ]] && { BOOT_PART=''; return 0; }
1094                 fi
1095         fi
1096
1097         if grep -q 'ext[34]' <<< "$(fsck -N "$BOOT_PART")"; then
1098                 local txt="\nIMPORTANT:\n\nThe boot partition $BOOT_PART $_format"
1099                 if yesno "Format Boot Partition" "$txt" "Format $BOOT_PART" "Skip Formatting" 1; then
1100                         part_format "$BOOT_PART" "ext4" 2
1101                 fi
1102         else
1103                 part_format "$BOOT_PART" "ext4" 2
1104         fi
1105         return 0
1106 }
1107
1108 select_root_partition()
1109 {
1110         if [[ $AUTO_ROOT_PART && -z $LVM && -z $LUKS ]]; then
1111                 ROOT_PART="$AUTO_ROOT_PART"
1112                 msg "Mount Menu" "\nUsing partitions created during automatic format.\n" 2
1113                 part_mount "$ROOT_PART" || { ROOT_PART=''; return 1; }
1114                 return 0  # we're done here
1115         else
1116                 local pts size dev isize bsize ptcount=0
1117
1118                 # walk partition list and skip ones that are too small for / (root)
1119                 while read -r dev size; do
1120                         size_t="${size: -1:1}"  # size type eg. K, M, G, T
1121                         isize=${size:0:-1}      # remove trailing size type character
1122                         isize=${isize%.*}       # remove any decimal (round down)
1123                         [[ $size_t =~ [MK] || ($size_t == 'G' && $isize -lt 4) ]] || { pts+="$dev $size "; (( ptcount++ )); }
1124                 done <<< "$PARTS"
1125
1126                 if (( ptcount == 1 )); then  # only one available device
1127                         msg "Root Partition (/)" "\nOnly one partition available that meets size requirements.\n" 2
1128                         ROOT_PART="$(awk 'NF > 0 {print $1}' <<< "$pts")"
1129                 else
1130                         dlg ROOT_PART menu "Mount Root" "\nSelect the root (/) partition, this is where $DIST will be installed.\n\nDevices smaller than 8G will not be shown here." $pts
1131                 fi
1132         fi
1133
1134         if [[ -z $ROOT_PART ]] || ! select_filesystem "$ROOT_PART" || ! part_mount "$ROOT_PART"; then
1135                 ROOT_PART=''
1136                 return 1
1137         fi
1138
1139         return 0
1140 }
1141
1142 select_extra_partitions()
1143 {
1144         local part size dev
1145
1146         # walk partition list and skip ones that are too small to be usable
1147         while read -r dev size; do
1148                 [[ ${size: -1:1} =~ [KM] ]] && part_countdec "$dev"
1149         done <<< "$PARTS"
1150
1151         while (( COUNT > 0 )); do
1152                 part=''
1153                 dlg part menu 'Mount Extra' "$_expart" 'done' 'finish mounting step' $PARTS || break
1154                 if [[ $part == 'done' ]]; then
1155                         break
1156                 elif select_filesystem "$part" && select_mountpoint && part_mount "$part" "$EXMNT"; then
1157                         EXMNTS+="$part: $EXMNT "
1158                         [[ $EXMNT == '/usr' && $HOOKS != *usr* ]] && HOOKS="usr $HOOKS"
1159                 else
1160                         return 1
1161                 fi
1162         done
1163         return 0
1164 }
1165
1166 ###############################################################################
1167 # installation
1168 # main is the entry point which calls all other install functions, once
1169 # complete it shows a dialog to edit files on the new system before reboot
1170
1171 install_main()
1172 {
1173         install_base
1174         genfstab -U $MNT >$MNT/etc/fstab 2>$ERR
1175         errshow 1 "genfstab -U $MNT >$MNT/etc/fstab"
1176         [[ -f $MNT/swapfile ]] && sed -i "s~${MNT}~~" $MNT/etc/fstab
1177         install_packages
1178         install_mkinitcpio
1179         install_boot
1180         chrun "hwclock --systohc --utc" || chrun "hwclock --systohc --utc --directisa"
1181         install_user
1182         install_login
1183         chrun "chown -Rf $NEWUSER:users /home/$NEWUSER"
1184
1185         while :; do
1186                 dlg choice menu "Finalization" "$_edit" \
1187                         finished   "exit the installer and reboot" \
1188                         keyboard   "${EDIT_FILES[keyboard]}" \
1189                         console    "${EDIT_FILES[console]}" \
1190                         locale     "${EDIT_FILES[locale]}" \
1191                         hostname   "${EDIT_FILES[hostname]}" \
1192                         sudoers    "${EDIT_FILES[sudoers]}" \
1193                         mkinitcpio "${EDIT_FILES[mkinitcpio]}" \
1194                         fstab      "${EDIT_FILES[fstab]}" \
1195                         crypttab   "${EDIT_FILES[crypttab]}" \
1196                         bootloader "${EDIT_FILES[bootloader]}" \
1197                         pacman     "${EDIT_FILES[pacman]}" \
1198                         login      "${EDIT_FILES[login]}"
1199
1200                 if [[ -z $choice || $choice == 'finished' ]]; then
1201                         [[ $DEBUG == true && -r $DBG ]] && $EDITOR $DBG
1202                         clear && die 127
1203                 else
1204                         local exists=''
1205                         for f in ${EDIT_FILES[$choice]}; do
1206                                 [[ -e ${MNT}$f ]] && exists+=" ${MNT}$f"
1207                         done
1208                         if [[ $exists ]]; then
1209                                 $EDITOR -O $exists
1210                         else
1211                                 msg "File Missing" "\nThe file(s) selected do not exist:\n\n${EDIT_FILES[$choice]}\n"
1212                         fi
1213                 fi
1214         done
1215 }
1216
1217 install_base()
1218 {
1219         clear
1220         tput cnorm
1221         while kill -0 $BG_PID 2>/dev/null; do
1222                 clear; printf "\nA background install process is still running...\n"; sleep 1
1223         done
1224         trap - EXIT
1225         unset BG_PID
1226
1227         rm -rf $MNT/etc/mkinitcpio-archiso.conf
1228         find $MNT/usr/lib/initcpio -name 'archiso*' -type f -delete
1229         sed -i 's/#\(Storage=\)volatile/\1auto/' $MNT/etc/systemd/journald.conf
1230         find $MNT/boot -name '*-ucode.img' -delete
1231
1232         [[ $DIST != "ArchLabs" ]] || sed -i "s/ArchLabs/$DIST/g" $MNT/etc/{lsb-release,os-release}
1233
1234         if [[ $VM ]]; then
1235                 find $MNT/etc/X11/xorg.conf.d/ -name '*.conf' -delete
1236         elif lspci | grep ' VGA ' | grep -q 'Intel'; then
1237                 echo "Creating Intel VGA Tear Free config /etc/X11/xorg.conf.d/20-intel.conf"
1238                 cat > $MNT/etc/X11/xorg.conf.d/20-intel.conf <<- EOF
1239                 Section "Device"
1240                     Identifier  "Intel Graphics"
1241                     Driver      "intel"
1242                     Option      "TearFree" "true"
1243                 EndSection
1244                 EOF
1245         fi
1246
1247         [[ -e /run/archiso/sfs/airootfs && $KERNEL == 'linux' ]] && cp -vf $RUN/x86_64/vmlinuz $MNT/boot/vmlinuz-linux
1248         [[ -d /etc/netctl ]] && cp -rfv /etc/netctl $MNT/etc/
1249         [[ -f /etc/resolv.conf ]] && cp -fv /etc/resolv.conf $MNT/etc/
1250         [[ -e /etc/NetworkManager/system-connections ]] && cp -rvf /etc/NetworkManager/system-connections $MNT/etc/NetworkManager/
1251
1252         echo "LANG=$MYLOCALE" > $MNT/etc/locale.conf
1253         cp -fv $MNT/etc/locale.conf $MNT/etc/default/locale
1254         sed -i "s/#en_US.UTF-8/en_US.UTF-8/g; s/#${MYLOCALE}/${MYLOCALE}/g" $MNT/etc/locale.gen
1255         chrun "locale-gen"
1256         chrun "ln -svf /usr/share/zoneinfo/$ZONE/$SUBZ /etc/localtime"
1257
1258         cat > $MNT/etc/X11/xorg.conf.d/00-keyboard.conf <<- EOF
1259         # Use localectl(1) to instruct systemd-localed to update it.
1260         Section "InputClass"
1261             Identifier      "system-keyboard"
1262             MatchIsKeyboard "on"
1263             Option          "XkbLayout" "$KEYMAP"
1264         EndSection
1265         EOF
1266
1267         cat > $MNT/etc/default/keyboard <<- EOF
1268         # KEYBOARD CONFIGURATION FILE
1269         # Consult the keyboard(5) manual page.
1270         XKBMODEL=""
1271         XKBLAYOUT="$KEYMAP"
1272         XKBVARIANT=""
1273         XKBOPTIONS=""
1274         BACKSPACE="guess"
1275         EOF
1276         printf "KEYMAP=%s\nFONT=%s\n" "$CMAP" "$FONT" > $MNT/etc/vconsole.conf
1277         echo "$MYHOST" > $MNT/etc/hostname
1278         cat > $MNT/etc/hosts <<- EOF
1279         127.0.0.1       localhost
1280         127.0.1.1       $MYHOST
1281         ::1                     localhost ip6-localhost ip6-loopback
1282         ff02::1         ip6-allnodes
1283         ff02::2         ip6-allrouters
1284         EOF
1285 }
1286
1287 install_boot()
1288 {
1289         echo "Installing $BOOTLDR"
1290
1291         if [[ $ROOT_PART == /dev/mapper* ]]; then
1292                 ROOT_PART_ID="$ROOT_PART"
1293         else
1294                 local uuid_type="UUID"
1295                 [[ $BOOTLDR =~ (systemd-boot|refind-efi|efistub) ]] && uuid_type="PARTUUID"
1296                 ROOT_PART_ID="$uuid_type=$(blkid -s $uuid_type -o value $ROOT_PART)"
1297         fi
1298
1299         if [[ $SYS == 'UEFI' ]]; then
1300                 # remove our old install and generic BOOT/ dir
1301                 find $MNT/$BOOTDIR/EFI/ -maxdepth 1 -mindepth 1 -iname "$DIST" -type d -delete
1302                 find $MNT/$BOOTDIR/EFI/ -maxdepth 1 -mindepth 1 -iname 'BOOT' -type d -delete
1303         fi
1304
1305         prerun_$BOOTLDR
1306         chrun "${BCMDS[$BOOTLDR]}" 2>$ERR
1307         errshow 1 "${BCMDS[$BOOTLDR]}"
1308
1309         if [[ -d $MNT/hostrun ]]; then
1310                 # cleanup the bind mounts we made earlier for the grub-probe module
1311                 umount_dir $MNT/hostrun/{udev,lvm}
1312                 rm -rf $MNT/hostrun >/dev/null 2>&1
1313         fi
1314
1315         if [[ $SYS == 'UEFI' ]]; then
1316                 # some UEFI firmware requires a generic esp/BOOT/BOOTX64.EFI
1317                 mkdir -pv $MNT/$BOOTDIR/EFI/BOOT
1318                 case "$BOOTLDR" in
1319                         grub) cp -fv $MNT/$BOOTDIR/EFI/$DIST/grubx64.efi $MNT/$BOOTDIR/EFI/BOOT/BOOTX64.EFI ;;
1320                         syslinux) cp -rf "$MNT/$BOOTDIR/EFI/syslinux/"* $MNT/$BOOTDIR/EFI/BOOT/ && cp -f $MNT/$BOOTDIR/EFI/syslinux/syslinux.efi $MNT/$BOOTDIR/EFI/BOOT/BOOTX64.EFI ;;
1321                         refind-efi) sed -i '/#extra_kernel_version_strings/ c extra_kernel_version_strings linux-hardened,linux-zen,linux-lts,linux' $MNT/$BOOTDIR/EFI/refind/refind.conf
1322                                 cp -fv $MNT/$BOOTDIR/EFI/refind/refind_x64.efi $MNT/$BOOTDIR/EFI/BOOT/BOOTX64.EFI ;;
1323                 esac
1324         fi
1325
1326         return 0
1327 }
1328
1329 install_user()
1330 {
1331         rm -f $MNT/root/.zshrc  # remove welcome message from root zshrc
1332
1333         chrun "chpasswd <<< 'root:$ROOT_PASS'" 2>$ERR
1334         errshow 1 "set root password"
1335         if [[ $MYSHELL != 'zsh' ]]; then # root uses zsh by default
1336                 chrun "usermod -s /bin/$MYSHELL root" 2>$ERR
1337                 errshow 1 "usermod -s /bin/$MYSHELL root"
1338                 # copy the default mkshrc to /root if it was selected
1339                 [[ $MYSHELL == 'mksh' ]] && cp -fv $MNT/etc/skel/.mkshrc $MNT/root/.mkshrc
1340         fi
1341
1342         echo "Creating new user $NEWUSER and setting password"
1343         local groups='audio,floppy,log,network,rfkill,scanner,storage,optical,power,wheel'
1344
1345         chrun "useradd -m -u 1000 -g users -G $groups -s /bin/$MYSHELL $NEWUSER" 2>$ERR
1346         errshow 1 "useradd -m -u 1000 -g users -G $groups -s /bin/$MYSHELL $NEWUSER"
1347         chrun "chpasswd <<< '$NEWUSER:$USER_PASS'" 2>$ERR
1348         errshow 1 "set $NEWUSER password"
1349
1350         if [[ $INSTALL_WMS == *dwm* ]];then
1351                 mkdir -pv "$MNT/home/$NEWUSER/suckless"
1352                 for i in dwm dmenu st; do
1353                         if chrun "git clone https://git.suckless.org/$i /home/$NEWUSER/suckless/$i"; then
1354                                 chrun "cd /home/$NEWUSER/suckless/$i; make prefix=/usr install; make clean"
1355                         else
1356                                 printf "failed to clone %s repo\n" "$i"
1357                         fi
1358                 done
1359         fi
1360
1361         [[ $MYSHELL != 'bash' ]] && rm -rf "$MNT/home/$NEWUSER/.bash"*
1362         [[ $MYSHELL != 'zsh' ]] && rm -rf "$MNT/home/$NEWUSER/.z"*
1363
1364         # remove some commands from ~/.xprofile when using KDE or Gnome as the login session
1365         if [[ $LOGIN_WM =~ (startkde|gnome-session) || ($LOGIN_TYPE != 'xinit' && $WM_PKGS =~ (plasma|gnome)) ]]; then
1366                 sed -i '/super/d; /nitrogen/d; /compton/d' "$MNT/home/$NEWUSER/.xprofile" "$MNT/root/.xprofile"
1367         elif [[ $LOGIN_WM == 'dwm' ]]; then # and dwm
1368                 sed -i '/super/d; /compton/d' "$MNT/home/$NEWUSER/.xprofile" "$MNT/root/.xprofile"
1369         fi
1370
1371         # create user home directories (Music, Documents, Downloads, etc..)
1372         chrun 'xdg-user-dirs-update'
1373
1374         return 0
1375 }
1376
1377 install_login()
1378 {
1379         local serv="$MNT/etc/systemd/system/getty@tty1.service.d"
1380         echo "Setting up $LOGIN_TYPE"
1381         case $LOGIN_TYPE in
1382                 ly|sddm|gdm|lightdm)
1383                         if [[ $LOGIN_WM == *dwm* ]]; then # dwm doesn't include an xsession file for display managers
1384                                 mkdir -p $MNT/usr/share/xsessions
1385                                 cat >$MNT/usr/share/xsessions/dwm.desktop <<- EOF
1386                                 [Desktop Entry]
1387                                 Encoding=UTF-8
1388                                 Name=Dwm
1389                                 Comment=Dynamic Window Manager
1390                                 Exec=dwm
1391                                 Type=XSession
1392                                 EOF
1393                         fi
1394                         rm -rf "$serv" "$MNT/home/$NEWUSER/.xinitrc"
1395                         chrun "systemctl enable $LOGIN_TYPE.service" 2>$ERR
1396                         errshow 1 "systemctl enable $LOGIN_TYPE.service"
1397                         ${LOGIN_TYPE}_config
1398                         ;;
1399                 xinit)
1400                         if [[ $INSTALL_WMS ]]; then
1401                                 sed -i "/exec/ c exec ${LOGIN_WM}" "$MNT/home/$NEWUSER/.xinitrc"
1402                         elif [[ -e $MNT/home/$NEWUSER/.xinitrc ]]; then
1403                                 sed -i '/exec/d' "$MNT/home/$NEWUSER/.xinitrc"
1404                                 return 0
1405                         fi
1406                         if [[ $AUTOLOGIN ]]; then
1407                                 sed -i "s/root/${NEWUSER}/g" $serv/autologin.conf
1408                                 cat > "$MNT/home/$NEWUSER/$LOGINRC" <<- EOF
1409                                 # automatically run startx when logging in on tty1
1410                                 [ -z "\$DISPLAY" ] && [ \$XDG_VTNR -eq 1 ] && startx
1411                                 EOF
1412                         else
1413                                 rm -rf $serv
1414                         fi
1415                         ;;
1416         esac
1417 }
1418
1419 install_packages()
1420 {
1421         local rmpkg="archlabs-installer "
1422         local inpkg="$PACKAGES $USER_PKGS $AL_BASE_PKGS "
1423
1424         if [[ $MYSHELL == 'zsh' ]]; then
1425                 inpkg+="zsh-completions "
1426         else
1427                 rmpkg+="zsh "
1428         fi
1429
1430         [[ $KERNEL != 'linux' ]] && rmpkg+='linux '
1431
1432         if [[ $INSTALL_WMS == 'dwm' ]]; then # dwm only needs a very limited package set
1433                 inpkg+="nitrogen polkit-gnome gnome-keyring dunst "
1434         else
1435                 [[ $INSTALL_WMS =~ (plasma|gnome|cinnamon) ]] || inpkg+="archlabs-ksuperkey "
1436                 [[ $INSTALL_WMS =~ (openbox|bspwm|i3-gaps|fluxbox) ]] && inpkg+="$WM_BASE_PKGS "
1437                 [[ $inpkg =~ (term|plasma|rxvt|tilda|tilix|sakura) || $INSTALL_WMS =~ (gnome|plasma|xfce4) ]] || inpkg+="xterm "
1438         fi
1439
1440         # update and install crucial packages first to avoid issues
1441         chrun "pacman -Syyu $KERNEL $BASE_PKGS ${LOGIN_PKGS[$LOGIN_TYPE]} $MYSHELL --noconfirm --needed" 2>$ERR
1442         errshow 1 "pacman -Syyu $KERNEL $BASE_PKGS ${LOGIN_PKGS[$LOGIN_TYPE]} $MYSHELL --noconfirm --needed"
1443
1444         # remove the packages we don't want on the installed system
1445         chrun "pacman -Rnsc $rmpkg --noconfirm"
1446
1447         # reinstalling iputils fixes the network issue for non-root users
1448         chrun "pacman -S iputils $UCODE --noconfirm"
1449
1450         # install the packages chosen throughout the install
1451         chrun "pacman -S $inpkg --needed --noconfirm" 2>$ERR
1452         errshow 1 "pacman -S $inpkg --needed --noconfirm"
1453
1454         # bootloader packages
1455         if [[ $BOOTLDR == 'grub' ]]; then
1456                 [[ $SYS == 'UEFI' ]] && local efib="efibootmgr"
1457                 chrun "pacman -S os-prober grub $efib --needed --noconfirm" 2>$ERR
1458                 errshow 1 "pacman -S os-prober grub $efib --needed --noconfirm"
1459         elif [[ $BOOTLDR == 'refind-efi' ]]; then
1460                 chrun "pacman -S refind-efi efibootmgr --needed --noconfirm" 2>$ERR
1461                 errshow 1 "pacman -S refind-efi efibootmgr --needed --noconfirm"
1462         elif [[ $SYS == 'UEFI' ]]; then
1463                 chrun "pacman -S efibootmgr --needed --noconfirm" 2>$ERR
1464                 errshow 1 "pacman -S efibootmgr --needed --noconfirm"
1465         fi
1466
1467         # allow members of the wheel group to run commands as root
1468         sed -i "s/# %wheel ALL=(ALL) ALL/%wheel ALL=(ALL) ALL/g" $MNT/etc/sudoers
1469
1470         return 0
1471 }
1472
1473 install_mkinitcpio()
1474 {
1475         local add=''
1476         [[ $LUKS ]] && add="encrypt"
1477         [[ $LVM ]] && { [[ $add ]] && add+=" lvm2" || add+="lvm2"; }
1478         sed -i "s/block filesystems/block ${add} filesystems ${HOOKS}/g" $MNT/etc/mkinitcpio.conf
1479         chrun "mkinitcpio -p $KERNEL" 2>$ERR
1480         errshow 1 "mkinitcpio -p $KERNEL"
1481 }
1482
1483 install_mirrorlist()
1484 {
1485         if hash reflector >/dev/null 2>&1; then
1486                 reflector --verbose --score 80 -l 40 -f 5 --sort rate --save "$1"
1487         elif hash rankmirrors >/dev/null 2>&1; then
1488                 echo "Sorting mirrorlist"
1489                 local key="access_key=5f29642060ab983b31fdf4c2935d8c56"
1490                 ip_add="$(curl -fsSL "http://api.ipstack.com/check&?$key&fields=ip" | python -c "import sys, json; print(json.load(sys.stdin)['ip'])")"
1491                 country="$(curl -fsSL "http://api.ipstack.com/$ip_add?$key&fields=country_code" | python -c "import sys, json; print(json.load(sys.stdin)['country_code'])")"
1492                 if [[ "$country" ]]; then
1493                         if [[ $country =~ (CA|US) ]]; then
1494                                 # use both CA and US mirrors for CA or US countries
1495                                 mirror="https://www.archlinux.org/mirrorlist/?country=US&country=CA&use_mirror_status=on"
1496                         elif [[ $country =~ (AU|NZ) ]]; then
1497                                 # use both AU and NZ mirrors for AU or NZ countries
1498                                 mirror="https://www.archlinux.org/mirrorlist/?country=AU&country=NZ&use_mirror_status=on"
1499                         else
1500                                 mirror="https://www.archlinux.org/mirrorlist/?country=${country}&use_mirror_status=on"
1501                         fi
1502                 else # no country code so just grab all mirrors, will be a very slow sort but we don't have other options
1503                         mirror="https://www.archlinux.org/mirrorlist/?country=all&use_mirror_status=on"
1504                 fi
1505                 curl -fsSL "$mirror" | sed -e 's/^#Server/Server/' -e '/^#/d' | rankmirrors -n 6 - >"$1"
1506         fi
1507
1508         return 0
1509 }
1510
1511 install_background()
1512 {
1513         ( rsync -a /run/archiso/sfs/airootfs/ $MNT/ && install_mirrorlist "$MNT/etc/pacman.d/mirrorlist" >/dev/null 2>&1 ) &
1514         BG_PID=$!
1515         trap "kill $BG_PID 2>/dev/null" EXIT
1516 }
1517
1518 ###############################################################################
1519 # display manager config
1520 # these are called based on which DM is chosen after it is installed
1521 # additional config can be  handled here, for now only lightdm has one
1522
1523 lightdm_config()
1524 {
1525         cat > $MNT/etc/lightdm/lightdm-gtk-greeter.conf <<- EOF
1526         [greeter]
1527         default-user-image=/usr/share/icons/ArchLabs-Dark/64x64/places/distributor-logo-archlabs.png
1528         background=/usr/share/backgrounds/archlabs/archlabs.jpg
1529         theme-name=Adwaita-dark
1530         icon-theme-name=Adwaita
1531         font-name=DejaVu Sans Mono 11
1532         position=30%,end 50%,end
1533         EOF
1534 }
1535
1536 ly_config()
1537 {
1538         :
1539 }
1540
1541 gdm_config()
1542 {
1543         :
1544 }
1545
1546 sddm_config()
1547 {
1548         :
1549 }
1550
1551 ###############################################################################
1552 # bootloader setup
1553 # prerun_* set up the configs needed before actually running the commands
1554 # setup_* are run after selecting a bootloader and build the command used later
1555 # they can also be used for further user input as these run before control is taken away
1556
1557 setup_grub()
1558 {
1559         EDIT_FILES[bootloader]="/etc/default/grub"
1560
1561         if [[ $SYS == 'BIOS' ]]; then
1562                 [[ $BOOT_DEV ]] || { part_device 1 || return 1; }
1563                 BCMDS[grub]="grub-install --recheck --force --target=i386-pc $BOOT_DEV"
1564         else
1565                 BCMDS[grub]="mount -t efivarfs efivarfs /sys/firmware/efi/efivars >/dev/null 2>&1
1566                 grub-install --recheck --force --target=x86_64-efi --efi-directory=/$BOOTDIR --bootloader-id=$DIST"
1567                 grep -q /sys/firmware/efi/efivars /proc/mounts || mount -t efivarfs efivarfs /sys/firmware/efi/efivars >/dev/null 2>&1
1568         fi
1569
1570         BCMDS[grub]="mkdir -p /run/udev /run/lvm &&
1571                 mount --bind /hostrun/udev /run/udev &&
1572                 mount --bind /hostrun/lvm /run/lvm &&
1573                 ${BCMDS[grub]} &&
1574                 grub-mkconfig -o /boot/grub/grub.cfg &&
1575                 sleep 1 && umount /run/udev /run/lvm"
1576
1577         return 0
1578 }
1579
1580 prerun_grub()
1581 {
1582         sed -i "s/GRUB_DISTRIBUTOR=.*/GRUB_DISTRIBUTOR=\"${DIST}\"/g; s/GRUB_CMDLINE_LINUX_DEFAULT=.*/GRUB_CMDLINE_LINUX_DEFAULT=\"\"/g" $MNT/etc/default/grub
1583
1584         if [[ $LUKS_DEV ]]; then
1585                 sed -i "s~#GRUB_ENABLE_CRYPTODISK~GRUB_ENABLE_CRYPTODISK~g; s~GRUB_CMDLINE_LINUX=.*~GRUB_CMDLINE_LINUX=\"${LUKS_DEV}\"~g" $MNT/etc/default/grub 2>$ERR
1586                 errshow 1 "sed -i 's~#GRUB_ENABLE_CRYPTODISK~GRUB_ENABLE_CRYPTODISK~g; s~GRUB_CMDLINE_LINUX=.*~GRUB_CMDLINE_LINUX=\"${LUKS_DEV}\"~g' $MNT/etc/default/grub"
1587         fi
1588
1589         if [[ $SYS == 'BIOS' && $LVM && -z $SEP_BOOT ]]; then
1590                 sed -i "s/GRUB_PRELOAD_MODULES=.*/GRUB_PRELOAD_MODULES=\"lvm\"/g" $MNT/etc/default/grub 2>$ERR
1591                 errshow 1 "sed -i 's/GRUB_PRELOAD_MODULES=.*/GRUB_PRELOAD_MODULES=\"lvm\"/g' $MNT/etc/default/grub"
1592         fi
1593
1594         # setup for os-prober module
1595         mkdir -p /run/{lvm,udev} $MNT/hostrun/{lvm,udev}
1596         mount --bind /run/lvm $MNT/hostrun/lvm
1597         mount --bind /run/udev $MNT/hostrun/udev
1598
1599         return 0
1600 }
1601
1602 setup_efistub()
1603 {
1604         EDIT_FILES[bootloader]=""
1605 }
1606
1607 prerun_efistub()
1608 {
1609         BCMDS[systemd-boot]="mount -t efivarfs efivarfs /sys/firmware/efi/efivars >/dev/null 2>&1
1610                 efibootmgr -v -d $BOOT_DEV -p $BOOT_PART_NUM -c -L '${DIST} Linux' -l /vmlinuz-${KERNEL} \
1611                 -u 'root=$ROOT_PART_ID rw $([[ $UCODE ]] && printf 'initrd=\%s.img ' "$UCODE")initrd=\initramfs-${KERNEL}.img'"
1612 }
1613
1614 setup_syslinux()
1615 {
1616         if [[ $SYS == 'BIOS' ]]; then
1617                 EDIT_FILES[bootloader]="/boot/syslinux/syslinux.cfg"
1618         else
1619                 EDIT_FILES[bootloader]="/boot/EFI/syslinux/syslinux.cfg"
1620                 BCMDS[syslinux]="mount -t efivarfs efivarfs /sys/firmware/efi/efivars >/dev/null 2>&1
1621                 efibootmgr -v -c -d $BOOT_DEV -p $BOOT_PART_NUM -l /EFI/syslinux/syslinux.efi -L $DIST"
1622         fi
1623 }
1624
1625 prerun_syslinux()
1626 {
1627         local c="$MNT/boot/syslinux" s="/usr/lib/syslinux/bios" d=".."
1628         [[ $SYS == 'UEFI' ]] && { c="$MNT/boot/EFI/syslinux"; s="/usr/lib/syslinux/efi64"; d=''; }
1629
1630         mkdir -pv "$c"
1631         cp -rfv "$s/"* "$c/"
1632         cat > "$c/syslinux.cfg" <<- EOF
1633         UI menu.c32
1634         PROMPT 0
1635         MENU TITLE $DIST Boot Menu
1636         TIMEOUT 50
1637         DEFAULT $DIST
1638
1639         LABEL $DIST
1640         MENU LABEL $DIST Linux
1641         LINUX $d/vmlinuz-$KERNEL
1642         APPEND root=$ROOT_PART_ID $([[ $LUKS_DEV ]] && printf "%s " "$LUKS_DEV")rw
1643         INITRD $([[ $UCODE ]] && printf "%s" "$d/$UCODE.img,")$d/initramfs-$KERNEL.img
1644
1645         LABEL ${DIST}fallback
1646         MENU LABEL $DIST Linux Fallback
1647         LINUX $d/vmlinuz-$KERNEL
1648         APPEND root=$ROOT_PART_ID $([[ $LUKS_DEV ]] && printf "%s " "$LUKS_DEV")rw
1649         INITRD $([[ $UCODE ]] && printf "%s" "$d/$UCODE.img,")$d/initramfs-$KERNEL-fallback.img
1650         EOF
1651         return 0
1652 }
1653
1654 setup_refind-efi()
1655 {
1656         EDIT_FILES[bootloader]="/boot/refind_linux.conf"
1657         BCMDS[refind-efi]="mount -t efivarfs efivarfs /sys/firmware/efi/efivars >/dev/null 2>&1; refind-install"
1658 }
1659
1660 prerun_refind-efi()
1661 {
1662         cat > $MNT/boot/refind_linux.conf <<- EOF
1663         "$DIST Linux"          "root=$ROOT_PART_ID $([[ $LUKS_DEV ]] &&
1664                                                 printf "%s " "$LUKS_DEV")rw add_efi_memmap $([[ $UCODE ]] &&
1665                                                 printf "initrd=%s " "/$UCODE.img")initrd=/initramfs-$KERNEL.img"
1666         "$DIST Linux Fallback" "root=$ROOT_PART_ID $([[ $LUKS_DEV ]] &&
1667                                                 printf "%s " "$LUKS_DEV")rw add_efi_memmap $([[ $UCODE ]] &&
1668                                                 printf "initrd=%s " "/$UCODE.img")initrd=/initramfs-$KERNEL-fallback.img"
1669         EOF
1670         mkdir -p $MNT/etc/pacman.d/hooks
1671         cat > $MNT/etc/pacman.d/hooks/refind.hook <<- EOF
1672         [Trigger]
1673         Operation = Upgrade
1674         Type = Package
1675         Target = refind-efi
1676
1677         [Action]
1678         Description = Updating rEFInd on ESP
1679         When = PostTransaction
1680         Exec = /usr/bin/refind-install
1681         EOF
1682 }
1683
1684 setup_systemd-boot()
1685 {
1686         EDIT_FILES[bootloader]="/boot/loader/entries/$DIST.conf"
1687         BCMDS[systemd-boot]="mount -t efivarfs efivarfs /sys/firmware/efi/efivars >/dev/null 2>&1; bootctl --path=/boot install"
1688 }
1689
1690 prerun_systemd-boot()
1691 {
1692         mkdir -p $MNT/boot/loader/entries
1693         cat > $MNT/boot/loader/loader.conf <<- EOF
1694         default  $DIST
1695         timeout  5
1696         editor   no
1697         EOF
1698         cat > $MNT/boot/loader/entries/$DIST.conf <<- EOF
1699         title   $DIST Linux
1700         linux   /vmlinuz-${KERNEL}$([[ $UCODE ]] && printf "\ninitrd  %s" "/$UCODE.img")
1701         initrd  /initramfs-$KERNEL.img
1702         options root=$ROOT_PART_ID $([[ $LUKS_DEV ]] && printf "%s " "$LUKS_DEV")rw
1703         EOF
1704         cat > $MNT/boot/loader/entries/$DIST-fallback.conf <<- EOF
1705         title   $DIST Linux Fallback
1706         linux   /vmlinuz-${KERNEL}$([[ $UCODE ]] && printf "\ninitrd  %s" "/$UCODE.img")
1707         initrd  /initramfs-$KERNEL-fallback.img
1708         options root=$ROOT_PART_ID $([[ $LUKS_DEV ]] && printf "%s " "$LUKS_DEV")rw
1709         EOF
1710         mkdir -p $MNT/etc/pacman.d/hooks
1711         cat > $MNT/etc/pacman.d/hooks/systemd-boot.hook <<- EOF
1712         [Trigger]
1713         Type = Package
1714         Operation = Upgrade
1715         Target = systemd
1716
1717         [Action]
1718         Description = Updating systemd-boot
1719         When = PostTransaction
1720         Exec = /usr/bin/bootctl update
1721         EOF
1722         systemd-machine-id-setup --root="$MNT"
1723         return 0
1724 }
1725
1726 ###############################################################################
1727 # lvm functions
1728
1729 lvm_menu()
1730 {
1731         msg "Info" "\nGathering device info.\n" 0
1732         no_bg_install || return 1
1733         lvm_detect
1734         local choice
1735         while :; do
1736                 dlg choice menu "Logical Volume Management" "$_lvmmenu" \
1737                         "$_lvmnew"    "vgcreate -f, lvcreate -L -n" \
1738                         "$_lvmdel"    "vgremove -f" \
1739                         "$_lvmdelall" "lvrmeove, vgremove, pvremove -f" \
1740                         "Back"        "Return to the main menu"
1741                 case "$choice" in
1742                         "$_lvmnew") lvm_create && break ;;
1743                         "$_lvmdel") lvm_delgroup && yesno "$_lvmdel" "$_lvmdelask" && vgremove -f "$DEL_VG" >/dev/null 2>&1 ;;
1744                         "$_lvmdelall") lvm_del_all ;;
1745                         *) break ;;
1746                 esac
1747         done
1748
1749         return 0
1750 }
1751
1752 lvm_detect()
1753 {
1754         local v pv
1755         pv="$(pvs -o pv_name --noheading 2>/dev/null)"
1756         v="$(lvs -o vg_name,lv_name --noheading --separator - 2>/dev/null)"
1757         VGROUP="$(vgs -o vg_name --noheading 2>/dev/null)"
1758
1759         if [[ $VGROUP && $v && $pv ]]; then
1760                 msg "LVM Setup" "\nActivating existing logical volume management.\n" 0
1761                 modprobe dm-mod >/dev/null 2>$ERR
1762                 errshow 'modprobe dm-mod'
1763                 vgscan >/dev/null 2>&1
1764                 vgchange -ay >/dev/null 2>&1
1765         fi
1766 }
1767
1768 lvm_create()
1769 {
1770         VGROUP='' LVM_PARTS='' VGROUP_MB=0
1771         umount_dir $MNT
1772         lvm_mkgroup || return 1
1773         local txt="\nThe last (or only) logical volume will automatically use all remaining space in the volume group."
1774         dlg VOL_COUNT menu "$_lvmnew" "\nSelect the number of logical volumes (LVs) to create in: $VGROUP\n$txt" 1 - 2 - 3 - 4 - 5 - 6 - 7 - 8 - 9 -
1775         [[ $VOL_COUNT ]] || return 1
1776         lvm_extra_lvs || return 1
1777         lvm_volume_name "$_lvmlvname\nNOTE: This LV will use up all remaining space in the volume group (${VGROUP_MB}MB)" || return 1
1778         msg "$_lvmnew (LV:$VOL_COUNT)" "\nCreating volume $VNAME from remaining space in $VGROUP\n" 0
1779         lvcreate -l +100%FREE "$VGROUP" -n "$VNAME" >/dev/null 2>$ERR
1780         errshow "lvcreate -l +100%FREE $VGROUP -n $VNAME" || return 1
1781         LVM='logical volume'; sleep 0.5
1782         txt="\nDone, volume: $VGROUP-$VNAME (${VOLUME_SIZE:-${VGROUP_MB}MB}) has been created.\n"
1783         msg "$_lvmnew (LV:$VOL_COUNT)" "$txt\n$(lsblk -o NAME,MODEL,TYPE,FSTYPE,SIZE $LVM_PARTS)\n"
1784         return 0
1785 }
1786
1787 lvm_lv_size()
1788 {
1789         local txt="${VGROUP}: ${SIZE}$SIZE_UNIT (${VGROUP_MB}MB remaining).$_lvmlvsize"
1790
1791         while :; do
1792                 ERR_SIZE=0
1793                 dlg VOLUME_SIZE input "$_lvmnew (LV:$VOL_COUNT)" "$txt" ''
1794                 if [[ -z $VOLUME_SIZE ]]; then
1795                         ERR_SIZE=1
1796                         break # allow bailing with escape or an empty choice
1797                 elif (( ${VOLUME_SIZE:0:1} == 0 )); then
1798                         ERR_SIZE=1 # size values can't begin with '0'
1799                 else
1800                         # walk the string and make sure all but the last char are digits
1801                         local lv=$((${#VOLUME_SIZE} - 1))
1802                         for (( i=0; i<lv; i++ )); do
1803                                 [[ ${VOLUME_SIZE:$i:1} =~ [0-9] ]] || { ERR_SIZE=1; break; }
1804                         done
1805                         if (( ERR_SIZE != 1 )); then
1806                                 case ${VOLUME_SIZE:$lv:1} in
1807                                         [mMgG]) local s=${VOLUME_SIZE:0:$lv} m=$((s * 1000))
1808                                                 case ${VOLUME_SIZE:$lv:1} in
1809                                                         [Gg]) (( m >= VGROUP_MB )) && ERR_SIZE=1 || VGROUP_MB=$((VGROUP_MB - m)) ;;
1810                                                         [Mm]) (( ${VOLUME_SIZE:0:$lv} >= VGROUP_MB )) && ERR_SIZE=1 || VGROUP_MB=$((VGROUP_MB - s)) ;;
1811                                                         *) ERR_SIZE=1
1812                                                 esac ;;
1813                                         *) ERR_SIZE=1
1814                                 esac
1815                         fi
1816                 fi
1817                 if (( ERR_SIZE )); then
1818                         msg "Invalid Logical Volume Size" "$_lvmerrlvsize"
1819                 else
1820                         break
1821                 fi
1822         done
1823
1824         return $ERR_SIZE
1825 }
1826
1827 lvm_mkgroup()
1828 {
1829         local named=''
1830
1831         until [[ $named ]]; do
1832                 lvm_partitions || return 1
1833                 lvm_group_name || return 1
1834                 yesno "$_lvmnew" "\nCreate volume group: $VGROUP\n\nusing these partition(s): $LVM_PARTS\n" && named=true
1835         done
1836
1837         msg "$_lvmnew" "\nCreating volume group: $VGROUP\n" 0
1838         vgcreate -f "$VGROUP" $LVM_PARTS >/dev/null 2>$ERR
1839         errshow "vgcreate -f $VGROUP $LVM_PARTS" || return 1
1840
1841         SIZE=$(vgdisplay "$VGROUP" | awk '/VG Size/ { gsub(/[^0-9.]/, ""); print int($0) }')
1842         SIZE_UNIT="$(vgdisplay "$VGROUP" | awk '/VG Size/ { print substr($NF, 0, 1) }')"
1843
1844         if [[ $SIZE_UNIT == 'G' ]]; then
1845                 VGROUP_MB=$((SIZE * 1000))
1846         else
1847                 VGROUP_MB=$SIZE
1848         fi
1849
1850         msg "$_lvmnew" "\nVolume group $VGROUP (${SIZE}$SIZE_UNIT) successfully created\n"
1851 }
1852
1853 lvm_del_all()
1854 {
1855         local v pv
1856         pv="$(pvs -o pv_name --noheading 2>/dev/null)"
1857         v="$(lvs -o vg_name,lv_name --noheading --separator - 2>/dev/null)"
1858         VGROUP="$(vgs -o vg_name --noheading 2>/dev/null)"
1859
1860         if [[ $VGROUP || $v || $pv ]]; then
1861                 if yesno "$_lvmdelall" "$_lvmdelask"; then
1862                         for i in $v; do lvremove -f "/dev/mapper/$i" >/dev/null 2>&1; done
1863                         for i in $VGROUP; do vgremove -f "$i" >/dev/null 2>&1; done
1864                         for i in $pv; do pvremove -f "$i" >/dev/null 2>&1; done
1865                         LVM=''
1866                 fi
1867         else
1868                 msg "Delete LVM" "\nNo LVMs to remove...\n" 2
1869                 LVM=''
1870         fi
1871 }
1872
1873 lvm_delgroup()
1874 {
1875         DEL_VG=''
1876         VOL_GROUP_LIST=''
1877
1878         for i in $(lvs --noheadings | awk '{print $2}' | uniq); do
1879                 VOL_GROUP_LIST+="$i $(vgdisplay "$i" | awk '/VG Size/ {print $3$4}') "
1880         done
1881
1882         [[ $VOL_GROUP_LIST ]] || { msg "No Groups" "\nNo volume groups found."; return 1; }
1883
1884         dlg DEL_VG menu "Logical Volume Management" "\nSelect volume group to delete.\n\nAll logical volumes within will also be deleted." $VOL_GROUP_LIST
1885         [[ $DEL_VG ]]
1886 }
1887
1888 lvm_extra_lvs()
1889 {
1890         while (( VOL_COUNT > 1 )); do
1891                 lvm_volume_name "$_lvmlvname" && lvm_lv_size || return 1
1892                 msg "$_lvmnew (LV:$VOL_COUNT)" "\nCreating a $VOLUME_SIZE volume $VNAME in $VGROUP\n" 0
1893                 lvcreate -L "$VOLUME_SIZE" "$VGROUP" -n "$VNAME" >/dev/null 2>$ERR
1894                 errshow "lvcreate -L $VOLUME_SIZE $VGROUP -n $VNAME" || return 1
1895                 msg "$_lvmnew (LV:$VOL_COUNT)" "\nDone, logical volume (LV) $VNAME ($VOLUME_SIZE) has been created.\n"
1896                 (( VOL_COUNT-- ))
1897         done
1898         return 0
1899 }
1900
1901 lvm_partitions()
1902 {
1903         part_find 'part|crypt' || return 1
1904         PARTS="$(awk 'NF > 0 {print $0 " off"}' <<< "$PARTS")"
1905         dlg LVM_PARTS check "$_lvmnew" "\nSelect the partition(s) to use for the physical volume (PV)." $PARTS
1906         [[ $LVM_PARTS ]]
1907 }
1908
1909 lvm_group_name()
1910 {
1911         VGROUP=''
1912         until [[ $VGROUP ]]; do
1913                 dlg VGROUP input "$_lvmnew" "$_lvmvgname" "lvgroup"
1914                 if [[ -z $VGROUP ]]; then
1915                         return 1
1916                 elif [[ ${VGROUP:0:1} == "/" || $VGROUP =~ \ |\' ]] || vgdisplay | grep -q "$VGROUP"; then
1917                         msg "LVM Name Error" "$_lvmerrvgname"
1918                         VGROUP=''
1919                 fi
1920         done
1921         return 0
1922 }
1923
1924 lvm_volume_name()
1925 {
1926         VNAME=''
1927         local txt="$1" default="mainvolume"
1928         (( VOL_COUNT > 1 )) && default="extvolume$VOL_COUNT"
1929         until [[ $VNAME ]]; do
1930                 dlg VNAME input "$_lvmnew (LV:$VOL_COUNT)" "\n$txt" "$default"
1931                 if [[ -z $VNAME ]]; then
1932                         return 1
1933                 elif [[ ${VNAME:0:1} == "/" || $VNAME =~ \ |\' ]] || lsblk | grep -q "$VNAME"; then
1934                         msg "LVM Name Error" "$_lvmerlvname"
1935                         VNAME=''
1936                 fi
1937         done
1938         return 0
1939 }
1940
1941 ###############################################################################
1942 # luks functions
1943
1944 luks_menu()
1945 {
1946         local choice
1947         no_bg_install || return 1
1948         dlg choice menu "LUKS Encryption" "$_luksmenu" \
1949                 "$_luksnew"  "cryptsetup -q luksFormat" \
1950                 "$_luksopen" "cryptsetup open --type luks" \
1951                 "$_luksadv"  "cryptsetup -q -s -c luksFormat" \
1952                 "Back"       "Return to the main menu"
1953
1954         case "$choice" in
1955                 "$_luksnew") luks_basic || return 1 ;;
1956                 "$_luksopen") luks_open || return 1 ;;
1957                 "$_luksadv") luks_advanced || return 1 ;;
1958         esac
1959
1960         return 0
1961 }
1962
1963 luks_open()
1964 {
1965         modprobe -a dm-mod dm_crypt >/dev/null 2>&1
1966         umount_dir $MNT
1967         part_find 'part|crypt|lvm' || return 1
1968
1969         if (( COUNT == 1 )); then
1970                 LUKS_PART="$(awk 'NF > 0 {print $1}' <<< "$PARTS")"
1971         else
1972                 dlg LUKS_PART menu "$_luksopen" "\nSelect which partition to open." $PARTS
1973         fi
1974
1975         [[ $LUKS_PART ]] || return 1
1976
1977         luks_pass "$_luksopen" || return 1
1978         msg "$_luksopen" "\nOpening encrypted partition: $LUKS_NAME\n\nUsing device/volume: $LUKS_PART\n" 0
1979         cryptsetup open --type luks "$LUKS_PART" "$LUKS_NAME" <<< "$LUKS_PASS" 2>$ERR
1980         errshow "cryptsetup open --type luks $LUKS_PART $LUKS_NAME" || return 1
1981         LUKS='encrypted'; luks_show
1982         return 0
1983 }
1984
1985 luks_pass()
1986 {
1987         LUKS_PASS=''
1988         local t="$1"
1989         typeset -a ans=(cryptroot) # default name to start
1990
1991         until [[ $LUKS_PASS ]]; do
1992                 tput cnorm
1993                 dialog --insecure --backtitle "$DIST Installer - $SYS - v$VER" --separator $'\n' --title " $t " --mixedform "$_luksomenu" 0 0 0 \
1994                         "Name:"      1 1 "${ans[0]}" 1  7 "$COLUMNS" 0 0 \
1995                         "Password:"  2 1 ''          2 11 "$COLUMNS" 0 1 \
1996                         "Password2:" 3 1 ''          3 12 "$COLUMNS" 0 1 2>"$ANS" || return 1
1997
1998                 mapfile -t ans <"$ANS"
1999
2000                 if [[ -z "${ans[0]}" ]]; then
2001                         msg "Name Empty" "\nEncrypted device name cannot be empty.\n\nPlease try again.\n" 2
2002                 elif [[ -z "${ans[1]}" || "${ans[1]}" != "${ans[2]}" ]]; then
2003                         LUKS_NAME="${ans[0]}"
2004                         msg "Password Mismatch" "\nThe passwords entered do not match.\n\nPlease try again.\n" 2
2005                 else
2006                         LUKS_NAME="${ans[0]}"
2007                         LUKS_PASS="${ans[1]}"
2008                 fi
2009         done
2010
2011         return 0
2012 }
2013
2014 luks_show()
2015 {
2016         sleep 0.5
2017         msg "$_luksnew" "\nEncrypted partition ready for mounting.\n\n$(lsblk -o NAME,MODEL,SIZE,TYPE,FSTYPE "$LUKS_PART")\n\n"
2018 }
2019
2020 luks_setup()
2021 {
2022         modprobe -a dm-mod dm_crypt >/dev/null 2>&1
2023         umount_dir $MNT
2024         part_find 'part|lvm' || return 1
2025
2026         if [[ $AUTO_ROOT_PART ]]; then
2027                 LUKS_PART="$AUTO_ROOT_PART"
2028         elif (( COUNT == 1 )); then
2029                 LUKS_PART="$(awk 'NF > 0 {print $1}' <<< "$PARTS")"
2030         else
2031                 dlg LUKS_PART menu "$_luksnew" "\nSelect the partition you want to encrypt." $PARTS
2032         fi
2033
2034         [[ $LUKS_PART ]] || return 1
2035         luks_pass "$_luksnew"
2036 }
2037
2038 luks_basic()
2039 {
2040         luks_setup || return 1
2041         msg "$_luksnew" "\nCreating encrypted partition: $LUKS_NAME\n\nDevice or volume used: $LUKS_PART\n" 0
2042         cryptsetup -q luksFormat "$LUKS_PART" <<< "$LUKS_PASS" 2>$ERR
2043         errshow "cryptsetup -q luksFormat $LUKS_PART" || return 1
2044         cryptsetup open "$LUKS_PART" "$LUKS_NAME" <<< "$LUKS_PASS" 2>$ERR
2045         errshow "cryptsetup open $LUKS_PART $LUKS_NAME" || return 1
2046         LUKS='encrypted'; luks_show
2047         return 0
2048 }
2049
2050 luks_advanced()
2051 {
2052         if luks_setup; then
2053                 local cipher
2054                 dlg cipher input "LUKS Encryption" "$_lukskey" "-s 512 -c aes-xts-plain64"
2055                 [[ $cipher ]] || return 1
2056                 msg "$_luksadv" "\nCreating encrypted partition: $LUKS_NAME\n\nDevice or volume used: $LUKS_PART\n" 0
2057                 cryptsetup -q $cipher luksFormat "$LUKS_PART" <<< "$LUKS_PASS" 2>$ERR
2058                 errshow "cryptsetup -q $cipher luksFormat $LUKS_PART" || return 1
2059                 cryptsetup open "$LUKS_PART" "$LUKS_NAME" <<< "$LUKS_PASS" 2>$ERR
2060                 errshow "cryptsetup open $LUKS_PART $LUKS_NAME" || return 1
2061                 luks_show
2062                 return 0
2063         fi
2064         return 1
2065 }
2066
2067 ###############################################################################
2068 # simple functions
2069 # some help avoid repetition and improve usability of some commands
2070 # others are initial setup functions used before reaching the main loop
2071
2072 ofn()
2073 {
2074         [[ "$2" == *"$1"* ]] && printf "on" || printf "off"
2075 }
2076
2077 die()
2078 {
2079         # cleanup and exit the installer cleanly with exit code $1
2080         local e="$1" # when e is 127 unmount /run/archiso/bootmnt and reboot
2081
2082         trap - INT
2083         tput cnorm
2084         if [[ -d $MNT ]]; then
2085                 umount_dir $MNT
2086                 if (( e == 127 )); then
2087                         umount_dir /run/archiso/bootmnt && sleep 0.5 && reboot -f
2088                 fi
2089         fi
2090         exit $e
2091 }
2092
2093 dlg()
2094 {
2095         local var="$1"   # assign output from dialog to var
2096         local dlg_t="$2" # dialog type (menu, check, input)
2097         local title="$3" # dialog title
2098         local body="$4"  # dialog message
2099         local n=0        # number of items to display for menu and check dialogs
2100
2101         shift 4  # shift off args assigned above
2102
2103         # adjust n when passed a large list
2104         local l=$((LINES - 20))
2105         (( ($# / 2) > l )) && n=$l
2106
2107         tput civis
2108         case "$dlg_t" in
2109                 menu) dialog --backtitle "$DIST Installer - $SYS - v$VER" --title " $title " --menu "$body" 0 0 $n "$@" 2>"$ANS" || return 1 ;;
2110                 check) dialog --backtitle "$DIST Installer - $SYS - v$VER" --title " $title " --checklist "$body" 0 0 $n "$@" 2>"$ANS" || return 1 ;;
2111                 input)
2112                         tput cnorm
2113                         local def="$1" # assign default value for input
2114                         shift
2115                         if [[ $1 == 'limit' ]]; then
2116                                 dialog --backtitle "$DIST Installer - $SYS - v$VER" --max-input 63 --title " $title " --inputbox "$body" 0 0 "$def" 2>"$ANS" || return 1
2117                         else
2118                                 dialog --backtitle "$DIST Installer - $SYS - v$VER" --title " $title " --inputbox "$body" 0 0 "$def" 2>"$ANS" || return 1
2119                         fi
2120                         ;;
2121         esac
2122         # if answer file isn't empty read from it into $var
2123         [[ -s "$ANS" ]] && printf -v "$var" "%s" "$(< "$ANS")"
2124 }
2125
2126 msg()
2127 {
2128         # displays a message dialog
2129         # when more than 2 args the message will disappear after sleep time ($3)
2130         local title="$1"
2131         local body="$2"
2132         shift 2
2133         tput civis
2134         if (( $# )); then
2135                 dialog --backtitle "$DIST Installer - $SYS - v$VER" --sleep "$1" --title " $title " --infobox "$body\n" 0 0
2136         else
2137                 dialog --backtitle "$DIST Installer - $SYS - v$VER" --title " $title " --msgbox "$body\n" 0 0
2138         fi
2139 }
2140
2141 live()
2142 {
2143         local e=0
2144
2145         if (( $# == 0 )); then
2146                 msg "No Session" "\nRunning live requires a session to use.\n\nExiting..\n" 2
2147                 clear
2148                 die 1
2149         elif ! select_keymap; then
2150                 clear
2151                 die 0
2152         elif ! net_connect; then
2153                 msg "Not Connected" "\nRunning live requires an active internet connection.\n\nExiting..\n" 2
2154                 die 1
2155         else
2156                 clear
2157                 pacman -Syyu archlinux-keyring --needed --noconfirm || die 1
2158                 pacman -S $AL_BASE_PKGS xorg-xinit xorg-server --needed --noconfirm || die 1
2159                 for ses; do
2160                         case "$ses" in
2161                                 dwm)
2162                                         pacman -S git --needed --noconfirm || die 1
2163                                         mkdir -pv /root/suckless
2164                                         for i in dwm dmenu st; do
2165                                                 git clone https://git.suckless.org/$i /root/suckless/$i && cd /root/suckless/$i && make PREFIX=/usr install
2166                                         done
2167                                         ;;
2168                                 i3-gaps|oepnbox|fluxbox|bspwm|awesome)
2169                                         pacman -S "$ses" $WM_BASE_PKGS ${WM_EXT[$ses]} xterm --needed --noconfirm || die 1
2170                                         ;;
2171                                 gnome|plasma|cinnamon|xfce4)
2172                                         pacman -S "$ses" ${WM_EXT[$ses]} xterm --needed --noconfirm || die 1
2173                                         ;;
2174                                 *) echo "error: invalid session for -l, --live, see -h, --help"; die 1 ;;
2175                         esac
2176                 done
2177                 pacman -Scc --noconfirm
2178                 rm -rf "/var/cache/pacman/pkg/"*
2179                 cp -rfT /etc/skel /root || die 1
2180                 sed -i "/exec/ c exec ${WM_SESSIONS[$ses]}" /root/.xinitrc
2181                 printf "\n%s has been set as the login session in ~/.xinitrc, to start the session simply run\n\n\tstartx\n\n" "${WM_SESSIONS[$ses]}"
2182                 die 0
2183         fi
2184 }
2185
2186 usage()
2187 {
2188         cat <<-EOF
2189         usage: $1 [-hdl] [session]
2190
2191         options:
2192             -h, --help     print this message and exit
2193             -l, --live     install and setup a live session
2194             -d, --debug    enable debugging and log output to $DBG
2195
2196         sessions:
2197             i3-gaps  - A fork of i3wm with more features including gaps
2198             openbox  - A lightweight, powerful, and highly configurable stacking wm
2199             dwm      - A dynamic WM for X that manages windows in tiled, floating, or monocle layouts
2200             awesome  - A customized Awesome WM session created by @elanapan
2201             bspwm    - A tiling wm that represents windows as the leaves of a binary tree
2202             fluxbox  - A lightweight and highly-configurable window manager
2203             gnome    - A desktop environment that aims to be simple and easy to use
2204             cinnamon - A desktop environment combining traditional desktop with modern effects
2205             plasma   - A kde software project currently comprising a full desktop environment
2206             xfce4    - A lightweight and modular desktop environment based on gtk+2/3
2207
2208         EOF
2209         exit 0
2210 }
2211
2212 yesno()
2213 {
2214         local title="$1" body="$2" yes='Yes' no='No'
2215         (( $# >= 3 )) && yes="$3"
2216         (( $# >= 4 )) && no="$4"
2217         tput civis
2218         if (( $# == 5 )); then
2219                 dialog --backtitle "$DIST Installer - $SYS - v$VER" --defaultno --title " $title " --yes-label "$yes" --no-label "$no" --yesno "$body\n" 0 0
2220         else
2221                 dialog --backtitle "$DIST Installer - $SYS - v$VER" --title " $title " --yes-label "$yes" --no-label "$no" --yesno "$body\n" 0 0
2222         fi
2223 }
2224
2225 chrun()
2226 {
2227         arch-chroot "$MNT" bash -c "$1"
2228 }
2229
2230 debug()
2231 {
2232         export PS4='| ${BASH_SOURCE} LINE:${LINENO} FUNC:${FUNCNAME[0]:+ ${FUNCNAME[0]}()} |>  '
2233         set -x
2234         exec 3>| $DBG
2235         BASH_XTRACEFD=3
2236         DEBUG=true
2237 }
2238
2239 termcol()
2240 {
2241         local colors=(
2242         "\e]P0191919" # #191919
2243         "\e]P1D15355" # #D15355
2244         "\e]P2609960" # #609960
2245         "\e]P3FFCC66" # #FFCC66
2246         "\e]P4255A9B" # #255A9B
2247         "\e]P5AF86C8" # #AF86C8
2248         "\e]P62EC8D3" # #2EC8D3
2249         "\e]P7949494" # #949494
2250         "\e]P8191919" # #191919
2251         "\e]P9D15355" # #D15355
2252         "\e]PA609960" # #609960
2253         "\e]PBFF9157" # #FF9157
2254         "\e]PC4E88CF" # #4E88CF
2255         "\e]PDAF86C8" # #AF86C8
2256         "\e]PE2ec8d3" # #2ec8d3
2257         "\e]PFE1E1E1" # #E1E1E1
2258         )
2259
2260         [[ $TERM == 'linux' ]] && printf "%b" "${colors[@]}" && clear
2261 }
2262
2263 errshow()
2264 {
2265         [ $? -eq 0 ] && return 0
2266
2267         local fatal=0 err=""
2268         err="$(sed 's/[^[:print:]]//g; s/\[[0-9\;:]*\?m//g; s/==> //g; s/] ERROR:/]\nERROR:/g' "$ERR")"
2269         [[ -z $err ]] && err="no error message was found"
2270
2271         (( $1 == 1 )) && { fatal=1; shift; }
2272
2273         local txt="\nCommand: $1\n\n\n\nError: $err\n\n"
2274
2275         if (( fatal )); then
2276                 msg "Install Error" "${txt}Errors at this stage are fatal, the install cannot continue.\n"
2277                 [[ -r $DBG && $TERM == 'linux' ]] && less "$DBG"
2278                 die 1
2279         fi
2280
2281         msg "Install Error" "${txt}Errors at this stage are non-fatal and may be fixed or ignored depending on the error.\n"
2282         return 1
2283 }
2284
2285 prechecks()
2286 {
2287         local i=1
2288
2289         if (( $1 >= 0 )) && ! grep -qw "$MNT" /proc/mounts; then
2290                 msg "Not Mounted" "\nPartition(s) must be mounted first.\n" 2
2291                 SEL=4 i=0
2292         elif [[ $1 -ge 1 && -z $BOOTLDR ]]; then
2293                 msg "No Bootloader" "\nBootloader must be selected first.\n" 2
2294                 SEL=5 i=0
2295         elif [[ $1 -ge 2 && (-z $NEWUSER || -z $USER_PASS) ]]; then
2296                 msg "No User" "\nA user must be created first.\n" 2
2297                 SEL=6 i=0
2298         elif [[ $1 -ge 3 && -z $CONFIG_DONE ]]; then
2299                 msg "Not Configured" "\nSystem configuration must be done first.\n" 2
2300                 SEL=7 i=0
2301         fi
2302         (( i )) # return code
2303 }
2304
2305 umount_dir()
2306 {
2307         mount | grep -q 'swap' && swapoff -a
2308         for dir; do
2309                 if [[ -d $dir ]] && mount | grep -q "on $dir "; then
2310                         if ! umount "$dir" 2>/dev/null; then
2311                                 sleep 0.5
2312                                 umount -f "$dir" 2>/dev/null || umount -l "$dir"
2313                         fi
2314                 fi
2315         done
2316 }
2317
2318 chk_connect()
2319 {
2320         msg "Network Connect" "\nVerifying network connection\n" 0
2321         curl -sIN --connect-timeout 5 'https://www.archlinux.org/' | sed '1q' | grep -q '200'
2322 }
2323
2324 net_connect()
2325 {
2326         if chk_connect; then
2327                 return 0
2328         elif hash nmtui >/dev/null 2>&1; then
2329                 tput civis
2330                 if [[ $TERM == 'linux' ]]; then
2331                         printf "%b" "\e]P1191919" "\e]P4191919"
2332                         nmtui-connect
2333                         printf "%b" "\e]P1D15355" "\e]P4255a9b"
2334                 else
2335                         nmtui-connect
2336                 fi
2337                 chk_connect
2338         elif hash wifi-menu >dev/null 2>&1; then
2339                 wifi-menu
2340                 chk_connect
2341         else
2342                 return 1
2343         fi
2344 }
2345
2346 no_bg_install()
2347 {
2348         [[ $BG_PID ]] || return 0
2349         msg "Install Running" "\nA background install process is currently running.\n" 2
2350         return 1
2351 }
2352
2353 system_devices()
2354 {
2355         IGNORE_DEV="$(lsblk -lno NAME,MOUNTPOINT | awk '/\/run\/archiso\/bootmnt/ {sub(/[1-9]/, ""); print $1}')"
2356
2357         if [[ $IGNORE_DEV ]]; then
2358                 SYS_DEVS="$(lsblk -lno NAME,SIZE,TYPE | awk '/disk/ && !'"/$IGNORE_DEV/"' {print "/dev/" $1 " " $2}')"
2359         else
2360                 SYS_DEVS="$(lsblk -lno NAME,SIZE,TYPE | awk '/disk/ {print "/dev/" $1 " " $2}')"
2361         fi
2362
2363         if [[ -z $SYS_DEVS ]]; then
2364                 msg "Device Error" "\nNo available devices...\n\nExiting..\n" 2
2365                 die 1
2366         fi
2367
2368         DEV_COUNT=0
2369         while read -r line; do
2370                 (( DEV_COUNT++ ))
2371         done <<< "$SYS_DEVS"
2372 }
2373
2374 system_identify()
2375 {
2376         if [[ $VM ]]; then
2377                 UCODE=''
2378         # amd-ucode is not needed it's provided by linux-firmware
2379         # elif grep -q 'AuthenticAMD' /proc/cpuinfo; then
2380         #       UCODE="amd-ucode"
2381         elif grep -q 'GenuineIntel' /proc/cpuinfo; then
2382                 UCODE="intel-ucode"
2383         fi
2384
2385         modprobe -q efivarfs >/dev/null 2>&1
2386
2387         _prep="\nOnce a step is finished a step you will be returned here, if the step was successful the cursor will be advanced to the next step.\nIf a step is unsuccessful the cursor will be placed on the step required to advance (when possible).\n\nTo begin the install you should have:\n\n  - A root (/) partition mounted."
2388         if [[ -d /sys/firmware/efi/efivars ]]; then
2389                 export SYS="UEFI"
2390                 grep -q /sys/firmware/efi/efivars /proc/mounts || mount -t efivarfs efivarfs /sys/firmware/efi/efivars
2391                 _prep+="\n  - An EFI boot partition mounted."
2392         else
2393                 export SYS="BIOS"
2394         fi
2395         _prep+="\n\nOnce finished mounting, a portion of the install can be done in the background while you continue configuring the system:\n"
2396         _prep+="\n  - Choose the system bootloader.\n  - Create a user and password."
2397         _prep+="\n  - Basic system configuration, kernel, shell, login, packages, etc..\n\nOnce you're happy with the choices and the required steps are complete, the main install can be started."
2398 }
2399
2400 ###############################################################################
2401 # entry point
2402
2403 # enable some nicer colours in the linux console
2404 termcol
2405
2406 if (( UID != 0 )); then
2407         msg "Not Root" "\nThis installer must be run as root or using sudo.\n\nExiting..\n" 2
2408         die 1
2409 elif ! grep -qwm 1 'lm' /proc/cpuinfo; then
2410         msg "Not x86_64 Architecture" "\nThis installer only supports x86_64 architectures.\n\nExiting..\n" 2
2411         die 1
2412 else
2413         case "$1" in
2414                 -d|--debug) debug ;;
2415                 -h|--help) usage "$0" ;;
2416                 -l|--live) shift; live "$@" ;;
2417         esac
2418 fi
2419
2420 # trap ^C to perform cleanup
2421 trap 'printf "\n^C\n" && die 1' INT
2422
2423 system_identify
2424 system_devices
2425
2426 msg "Welcome to the $DIST Installer" "$_welcome"
2427
2428 if ! select_keymap; then
2429         clear; die 0
2430 elif ! net_connect; then
2431         msg "Not Connected" "\nThis installer requires an active internet connection.\n\nExiting..\n" 2
2432         die 1
2433 fi
2434
2435 FORMATTED=""
2436
2437 while :; do
2438         main
2439 done
2440 # vim:fdm=marker:fmr={,}