#!/bin/bash
#
# System installer.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public
# License along with this program.  If not, see
# <http://www.gnu.org/licenses/>.

# Load.
source $APP_BASE/lib/hydra/functions || exit 1
hydra_config_load

# Setup.
hydra_user_input device /dev/sdb "Destination device"
hydra_user_input root_size 20G "Size of root partition"
hydra_user_input swap y "Use swap? (y/n)"
hydra_user_input encrypt y "Encrypt system and storage volumes? (y/n)"
hydra_user_input garbage y "Pre-fill volumes with garbage? (y/n)"
hydra_user_input hostname $HOSTNAME "Hostname"
hydra_user_input domain example.com "Domain"
hydra_user_input arch amd64 "System arch"
hydra_user_input version squeeze "Distro version"
hydra_user_input vg vg "Temporary install vg"
hydra_user_input grub n "Setup GRUB? (y/n)"
hydra_user_input mirror http://cdn.debian.net/debian/ "Debian mirror"

# Check for requirements.
for req in debootstrap cryptsetup grub-pc lvm2 parted; do
  hydra_install_package $req
done

# Warning.
cat <<-EOF
WARNING: about to partition $device!
Press ENTER to continue, Ctrl-C to abort."
EOF
read answer

# Disk partitioning.
if [ "$swap" == "y" ]; then
  parted -s -- $device mklabel msdos
  parted -s -- $device unit MB mkpart primary linux-swap 0     2000
  parted -s -- $device unit MB mkpart primary ext2       2000  2200
  parted -s -- $device unit MB mkpart primary ext2       2200 -1
  parted -s -- $device set 2 boot on
  parted -s -- $device set 3 lvm  on

  # Change devices to absolute path names.
  swap_device="$device"1
  boot_device="$device"2
  syst_device="$device"3
else
  parted -s -- $device mklabel msdos
  parted -s -- $device unit MB mkpart primary ext2 0    200
  parted -s -- $device unit MB mkpart primary ext2 200 -1
  parted -s -- $device set 1 boot on
  parted -s -- $device set 2 lvm  on

  # Change devices to absolute path names.
  boot_device="$device"1
  syst_device="$device"2
fi

# Create volumes.
echo "Creating the needed disk volumes..."

if ! pvdisplay $syst_device &> /dev/null; then
  echo "Creating physical volume..."
  pvcreate $syst_device
fi

if ! vgdisplay $vg &> /dev/null; then
  echo "Creating volume group..."
  vgcreate $vg $syst_device
fi

if ! lvdisplay $vg/root &> /dev/null; then
  echo "Creating logical volume..."
  lvcreate -L$root_size -n root $vg
fi

vgchange -a y $vg

# Garbage.
if [ "$garbage" == "y" ]; then
  echo "Filling volumes with garbage..."
  dd if=/dev/urandom of=/dev/$vg/root
  if [ "$swap" == "y" ]; then
    dd if=/dev/urandom of=$swap_device
  fi
fi

# Setup mountpoint and make sure it's not mounted due to a failed install.
mkdir -p /tmp/debootstrap
umount /tmp/debootstrap/proc &> /dev/null
umount /tmp/debootstrap/dev  &> /dev/null

# Create root device.
if [ "$encrypt" == "y" ]; then
  echo "Creating encrypted root device..."
  cryptsetup -h sha256 -c aes-cbc-essiv:sha256 -s 256 luksFormat /dev/$vg/root
  cryptsetup luksOpen /dev/$vg/root debootstrap
  mkfs.ext3 /dev/mapper/debootstrap
  install_device="/dev/mapper/debootstrap"
else
  echo "Creating root device..."
  mkfs.ext3 /dev/vg/root
  install_device="/dev/vg/root"
fi

# Initial system install.
echo "Installing base system..."
mount $install_device /tmp/debootstrap/
debootstrap --arch=$arch $version /tmp/debootstrap/ $mirror

# Initial configuration.
echo "Applying initial configuration..."
mount none -t proc /tmp/debootstrap/proc/
mount -o bind /dev/ /tmp/debootstrap/dev
echo LANG=C > /tmp/debootstrap/etc/default/locale

# Resolver configuration.
echo "domain $domain" > /tmp/debootstrap/etc/resolv.conf
echo "search $hostname.$domain" >> /tmp/debootstrap/etc/resolv.conf
grep nameserver /etc/resolv.conf >> /tmp/debootstrap/etc/resolv.conf

# Hostname configuration.
echo $hostname.$domain > /tmp/debootstrap/etc/hostname
echo "127.0.0.1 $hostname $hostname.$domain" >> /tmp/debootstrap/etc/hosts
echo "127.0.0.1 localhost" >> /tmp/debootstrap/etc/hosts

# Invert hostname contents to avoid http://projects.puppetlabs.com/issues/2533
tac /tmp/debootstrap/etc/hosts > /tmp/debootstrap/etc/hosts.new
mv /tmp/debootstrap/etc/hosts.new /tmp/debootstrap/etc/hosts

# Initial upgrade.
echo "Applying initial upgrades..."
chroot /tmp/debootstrap/ apt-get update
chroot /tmp/debootstrap/ apt-get upgrade -y
chroot /tmp/debootstrap/ apt-get install locales cryptsetup lvm2 initramfs-tools -y

# Crypttab.
echo "Configuring crypttab..."
echo "" > /tmp/debootstrap/etc/crypttab

if [ "$encrypt" == "y" ]; then
  cat > /tmp/debootstrap/etc/crypttab <<-EOF
# <target name> <source device>   <key file>  <options>
root            /dev/mapper/vg-root     none            luks,cipher=aes-cbc-essiv:sha256
EOF
fi

if [ "$swap" == "y" ]; then
  cat >> /tmp/debootstrap/etc/crypttab <<-EOF
# <target name> <source device>   <key file>  <options>
cswap           $swap_device               /dev/random     swap,cipher=aes-cbc-essiv:sha256
EOF
fi

# Fstab.
echo "Configuring fstab..."
echo "" > /tmp/debootstrap/etc/fstab
if [ "$encrypt" == "y" ]; then
  cat > /tmp/debootstrap/etc/fstab <<-EOF
/dev/mapper/root  /    ext3 defaults,errors=remount-ro 0 1
EOF
fi

if [ "$swap" == "y" ]; then
  cat >> /tmp/debootstrap/etc/fstab <<-EOF
/dev/mapper/cswap none swap sw 0 0
EOF
fi

# Grub.
if [ "$grub" == "y" ]; then
  echo "Boot device setup..."
  mkfs.ext3 $boot_device
  mount $boot_device /tmp/debootstrap/boot
  echo "$boot_device /boot ext3 defaults,errors=remount-ro 0 2" >> /tmp/debootstrap/etc/fstab

  echo "Setting up GRUB..."
  chroot /tmp/debootstrap/ apt-get install grub-pc -y
  grub-install --no-floppy "$device"
  mkdir -p /tmp/debootstrap/boot/grub/
  cat /tmp/debootstrap/boot/grub/menu.lst <<-EOF
title $hostname (hd0)
root  (hd0,1)
kernel  /vmlinuz-2.6.26-2-vserver-amd64 root=/dev/mapper/root ro quiet rootdelay=10
initrd  /initrd.img-2.6.26-2-vserver-amd64

title $hostname (hd0) (single)
root  (hd0,1)
kernel  /vmlinuz-2.6.26-2-vserver-amd64 root=/dev/mapper/root ro single rootdelay=10
initrd  /initrd.img-2.6.26-2-vserver-amd64
EOF
fi

# Kernel.
echo "Installing kernel..."
cat > /tmp/debootstrap/etc/initramfs-tools/modules <<-EOF
dm-mod
dm-crypt
aes
twofish
sha256
EOF

cat > /tmp/debootstrap/etc/kernel-img.conf <<-EOF
do_initrd = Yes
EOF

if [ "$arch" == "i386" ]; then
  kernel_arch="686"
else
  kernel_arch="$arch"
fi

chroot /tmp/debootstrap apt-get install linux-image-2.6-vserver-$kernel_arch -y

# Initramfs.
echo "Creating initramfs..."
chroot /tmp/debootstrap update-initramfs -v -u

# Utils.
echo "Installing basic utilities..."
chroot /tmp/debootstrap apt-get install screen cron lsb-release openssl -y

# Ssh.
echo "Installing OpenSSH daemon..."
chroot /tmp/debootstrap apt-get install openssh-server -y

echo "OpenSSH fingerprints:"
chroot /tmp/debootstrap ssh-keygen -l -f /etc/ssh/ssh_host_dsa_key.pub
chroot /tmp/debootstrap ssh-keygen -l -f /etc/ssh/ssh_host_rsa_key.pub

# Accounts.
echo "Installing sudo..."
chroot /tmp/debootstrap apt-get install sudo -y

echo "Choose a root password."
chroot /tmp/debootstrap passwd root

cat <<-EOF

Now proceeed with final steps:

  - Create an user account with sudo privileges.
  - Network setup.

See http://padrao.sarava.org/install for more information.
EOF