install.sh

#!/bin/bash

# Debugging mode. 
#set -x

# Stop on errors. 
set -e 

# Enter the label for your USB drive here. 
#disk=/dev/sdb
disk=/dev/loop0

# If writing to an .img file, this is the size of the file
# to be created. 
# size=14G # small image
size=28G # big image

name=dh-usb-0-2-2

if [ "$disk" = "/dev/loop0" ] 
then 
	# Loopback devices have different partition syntax.
	part1="$disk"p1
	part2="$disk"p2
else
	part1="$disk"1
	part2="$disk"2
fi

dir=$PWD

sys="crda dialog gpm grml-zsh-config linux-atm lsscsi mc mtools ndisc6 nfs-utils grub efibootmgr nilfs-utils ntp rp-pppoe sg3_utils smartmontools speedtouch wvdial xl2tpd hdparm gptfdisk exfat-utils dmraid dosfstools hfsprogs gnu-netcat sdparm refind-efi f2fs-tools ntfs-3g testdisk partclone fsarchiver btrfs-progs parted ddrescue partimage wireless_tools wpa_actiond vpnc ppp pptpclient openconnect openssh openvpn dnsmasq dnsutils tcpdump lftp darkhttpd dhclient ethtool b43-fwcutter ipw2100-fw ipw2200-fw zd1211-firmware rsync rfkill usb_modeswitch mesa mesa-vdpau xf86-input-synaptics xf86-video-ati xf86-video-fbdev xf86-video-intel xf86-video-nouveau xf86-video-openchrome xf86-video-vesa xf86-video-vmware xorg-server xorg-xinit xorg-xauth alsa-utils expect git pkgfile powerpill reflector sudo unzip unrar wget xdg-user-dirs-gtk yaourt fakeroot broadcom-wl" 

desktop="gdm baobab chromium chromium-pepper-flash eog espeak evince file-roller gedit gnome-backgrounds gnome-bluetooth gnome-calculator gnome-common gnome-contacts gnome-control-center gnome-disk-utility gnome-documents gnome-keyring gnome-logs gnome-maps gnome-photos gnome-screenshot gnome-shell gnome-shell-extension-dash-to-dock gnome-shell-extensions gnome-sound-recorder gnome-system-monitor gnome-terminal gnome-themes-standard gnome-tweak-tool gnome-weather gnome-boxes gst-libav gst-plugins-bad gst-plugins-base gst-plugins-good gst-plugins-ugly gstreamer-vaapi gvfs-mtp libwnck3 nautilus networkmanager network-manager-applet totem transmission-cli transmission-gtk arc-gtk-theme gnome-software" 

fonts="divehi-fonts ttf-aboriginal-sans ttf-arphic-uming ttf-baekmuk ttf-bitstream-vera ttf-dejavu ttf-freefont ttf-opensans" 

dh="vim python python-pip ruby jupyter jupyter-notebook python-nltk pandoc pandoc-citeproc pandoc-crossref mathjax" 

hacker="gst-libav gst-plugins-bad gst-plugins-base gst-plugins-good gst-plugins-ugly gstreamer-vaapi i3-wm i3status qutebrowser zathura" 

ruby_gems="jekyll"

aur="papirus-icon-theme-git zotero" 

# The files jupyter-notebook.desktop and zotero.desktop don't have 
# icons that look good in the current theme, so this hack replaces them. 
desktop_files="jupyter-notebook zotero" 

function partition() { 
	if [ "$disk" = "/dev/loop0" ]
	then 
		echo "Creating disk image." 
		fallocate -l $size disk.img
		losetup /dev/loop0 disk.img
	fi

	echo "Partitioning $disk!"

	# Make a GPT partition table for the disk. 
	parted $disk mktable gpt

	# Make an EFI boot partition. 
	parted $disk mkpart primary fat32 1MiB 513MiB
	parted $disk set 1 boot on

	# Make the main partition
	parted $disk mkpart primary ext4 513MiB 100%
	parted $disk name 2 dh-usb

	mkfs.fat -F32 $part1
	# Disable journaling to lengthen life of USB disk by minimizing writes. 
	mkfs.ext4 -O "^has_journal" $part2 -L dh-usb
}

function mount { 
	echo "Mounting $disk"
	/usr/sbin/mount $part2 /mnt 
	mkdir -p /mnt/boot 
	/usr/sbin/mount $part1 /mnt/boot
} 

function unmount { 
	echo "Unmounting $disk"
	umount /mnt/boot
	umount /mnt

	if [ "$disk" = "/dev/loop0" ]
	then 
		losetup -d /dev/loop0
	fi
} 

function install_r { 
	# Optional, for installing the R language and R-Studio. 
	# Must be run before main installation. 
	dh="$dh r"
	aur="$aur rstudio-desktop-bin" 
	desktop_files="$desktop_files r rstudio"
} 

function install { 
	echo "Installing system." 
	pacstrap /mnt base base-devel
	pacstrap /mnt $sys $fonts $desktop $dh
}

function install_hacker { 
	echo "Installing hacker-mode system." 
	pacstrap /mnt base base-devel
	pacstrap /mnt $sys $fonts $hacker $dh
}

function config_init { 
	echo "Starting initial configuration of system." 
	#arch-chroot /mnt ln -s /usr/share/zoneinfo/America/New_York /etc/localtime

	# Copy /etc files
	for file in locale.gen locale.conf hostname fstab makepkg.conf
	do
		cp config/$file /mnt/etc/$file
		arch-chroot /mnt chown root:root /etc/$file
	done
		
	# Generate locale
	arch-chroot /mnt locale-gen

	# Set hardware clock
	arch-chroot /mnt hwclock --systohc

	# Generate boot image
	arch-chroot /mnt mkinitcpio -p linux

	# Add default user
	arch-chroot /mnt useradd -m -p "" -g users -G "adm,audio,floppy,log,network,rfkill,scanner,storage,optical,power,wheel" -s /usr/bin/zsh dh-usb

	# Start services
	if [ "$hacker_mode" != true ]
	then 
		arch-chroot /mnt systemctl enable gdm
		arch-chroot /mnt systemctl enable NetworkManager
	fi

	# Install bootloader
	#arch-chroot /mnt grub-install --target=x86_64-efi --efi-directory=/boot --bootloader-id=grub
	#arch-chroot /mnt grub-mkconfig -o /boot/grub/grub.cfg

	# Install systemd-boot
	arch-chroot /mnt bootctl --path=/boot install

	# Copy systemd-boot config
	cp config/loader.conf /mnt/boot/loader/
	cp config/arch.conf /mnt/boot/loader/entries/
	arch-chroot /mnt chown root:root /boot/loader/loader.conf
	arch-chroot /mnt chown root:root /boot/loader/entries/arch.conf 
	arch-chroot /mnt chmod +x /boot/loader/loader.conf
	arch-chroot /mnt chmod +x /boot/loader/entries/arch.conf
}


function install_extra {
	echo "Installing extra packages from the AUR" 
	# Install packages from AUR
	# It'd be great if this worked with `yaourt`,  but there are permissions
	# issues, so we have to do this the long way. 
	for package in $aur
	do
		sudo -u $SUDO_USER git clone https://aur.archlinux.org/$package.git /tmp/$package
		cd /tmp/$package
		sudo -u $SUDO_USER makepkg -src 
		pkgfile=`ls $package*.pkg.tar`
		mv /tmp/$package/$pkgfile /mnt/home/dh-usb/
		arch-chroot /mnt chown root:root /home/dh-usb/$pkgfile
		arch-chroot /mnt pacman -U --noconfirm /home/dh-usb/$pkgfile

		#Clean up
		cd $dir
		rm -rf /tmp/$package
		arch-chroot /mnt rm -rf /home/dh-usb/$pkgfile
	done

	# Install Ruby gems
	echo "Installing Ruby gems." 
	arch-chroot /mnt sudo -u dh-usb gem install -u $ruby_gems

} 

function install_big { 
	# NLTK "Book" Data is that which is needed to follow along with nltk.org/book
	arch-chroot /mnt python -m nltk.downloader -d /usr/local/share/nltk_data book

	# Install Pandas, Numpy, Scikit-learn, Etc. 
	pacstrap /mnt python-pandas python-numpy python-scikit-learn python-virtualenv python-virtualenvwrapper

	# Install spaCy
	arch-chroot /mnt pip install spacy

	# Install spaCy data
	arch-chroot /mnt python -m spacy.en.download all

	# Install textacy
	# Textacy needs cld2-cffi, but that won't compile with GCC 6, so 
	# we have to supply a special CFLAGS argument. See 
	# https://github.com/GregBowyer/cld2-cffi/issues/12
	# for the status on this bug. 
	arch-chroot /mnt env CFLAGS="-Wno-narrowing" pip install -U cld2-cffi
	arch-chroot /mnt pip install textacy
} 

function files { 
	# Copy dotfiles
	DEST=/home/dh-usb/
	for file in xinitrc zshrc setup.sh
	do 
		cp config/$file /mnt$DEST.$file
		arch-chroot /mnt chown dh-usb:users $DEST.$file 
	done

	# Give members of group Wheel access to sudo. 
	cp config/sudo /mnt/etc/sudoers.d/wheel
	arch-chroot /mnt chown root:root /etc/sudoers.d/wheel
	arch-chroot /mnt chmod -c 0440 /etc/sudoers.d/wheel

	# Copy over desktop files. 
	DEST=/home/dh-usb/.local/share/applications/
	arch-chroot /mnt sudo -u dh-usb mkdir -p $DEST
	for file in $desktop_files
	do 
		cp config/$file.desktop /mnt$DEST
		arch-chroot /mnt chown dh-usb:users $DEST$file.desktop
	done
}

function config_post { 
	# Run setup script on the USB system. 
	arch-chroot /mnt sudo -u dh-usb /home/dh-usb/.setup.sh
} 

function clean { 
	# Remove arguably unnecessarily desktop files so that the list of applications 
	# looks cleaner and more friendly.  
	for file in avahi-discover bssh bvnc qv4l2
	do 
		rm /mnt/usr/share/applications/$file.desktop
	done
} 

function package { 
	# Renames the resulting disk.img, bundles the installation
	# README, and compresses the result into a .zip file for distribution
	# over the Internet. 
	echo "Zipping .img file." 
	if [ $size == "28G" ] 
	then 
		name=$name-32G
	else
		name=$name-16G
	fi
	mkdir -p $name
	cp README-installation.md $name/README.md
	mv disk.img $name/$name.img
	zip -r $name.zip $name/ 
} 

function min { 
	export size=14G
	partition
	mount 
	install
	config_init
	install_extra
	files
	config_post
	clean
	unmount
}

function big { 
	export size=28G
	partition
	mount 
	install
	config_init
	install_extra
	install_big
	files
	config_post
	clean
	unmount
}

function hacker { 
	export hacker_mode=true
	partition
	mount 
	install_hacker
	config_init
	install_extra
	install_big
	files
	unmount
}

$1