#!/bin/bash
# Copyright (C) 2016 Ian Kelling

# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# 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 General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

# Setup arch pxe boot server from the base image.
#
if [[ -s ~/.bashrc ]];then . ~/.bashrc;fi

usage() {
    cat <<EOF
Usage: ${0##*/}
Setup arch pxe boot server from the base image

Requires manually downloading image. Image path is hardcoded below to
/a/opt/image_name without .iso on the end. adjust the code for a new image. After this, run pxe-server.
HOST|default arch to enable it.

-h|--help  Print help and exit.
EOF
    exit $1
}

case $1 in
    -h|--help) usage ;;
esac


x="$(readlink -f "$BASH_SOURCE")"
script_dir="${x%/*}"
cd /a/opt/roms
iso="parabola-systemd-cli-x86_64-netinstall-2018.06.02.iso"
idir=${iso%.iso}
rm -rf $idir
ex $iso
# should be parabola or arch
n=${iso%%-*}
sfs=$idir/$n/x86_64/*.sfs

sed -i --follow-symlinks -f - $idir/$n/boot/syslinux/${n}iso_pxe64.cfg <<EOF
1itotaltimeout 1
/^LABEL arch64_nfs/a menu default
s/^APPEND .*/\0 script=archlike-iso-init/
EOF
# based on https://blog.chendry.org/2015/02/06/automating-arch-linux-installation.html
# and https://wiki.archlinux.org/index.php/Remastering_the_Install_ISO

sudo rm -rf squashfs-root # remove any existing folder
sudo unsquashfs $sfs
sudo mkdir -p squashfs-root/root/.ssh
sudo chmod 755 squashfs-root/root/.ssh
sudo cp ~/.ssh/home.pub squashfs-root/root/.ssh/authorized_keys

sudo cp $script_dir/archlike-iso-init squashfs-root/root
sudo rm $sfs
sudo mksquashfs squashfs-root $sfs -comp xz
# file transfer to wrt is slow, so remove some useless files
rm -f $idir/$n/i686/airootfs.sfs $idir/$n/boot/i686/${n}iso.img
pushd $(dirname $sfs); md5sum ${sfs##*/} > airootfs.md5; popd

# seems if you've done a pxe boot, mounted the nfs,
# then shutdown, it's still busy.
ssh cmc "/etc/init.d/nfsd stop; \
umount /run/archiso/bootmnt; \
umount /run/parabolaiso/bootmnt; \
rm -rf /mnt/usb/$idir"

scp -r $idir wrt:/mnt/usb
ssh wrt "cd /mnt/usb && rm -f tftpboot && ln -s $idir tftpboot"

# The default settings in the installer expect to find the NFS at /run/archiso/bootmnt

# background: great documentation at
# https://wiki.archlinux.org/index.php/PXE
# arch can do netboot like ubuntu etc, but the docs look a little
# complicated, so fuck it, we use nfs cuz it's easy

rm -rf $idir
sudo rm -rf squashfs-root