#!/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.

set -eE -o pipefail
trap 'echo "$0:$LINENO:error: \"$BASH_COMMAND\" returned $?" >&2' ERR

[[ $EUID == 0 ]] || exec sudo "${BASH_SOURCE}" "$@"

usage() {
    cat <<EOF
usage: ${0##*/} [-h|--help]
install fai-server on the current machine

Initial setup of a fai server on debian. works on localhost.
Set's the current ip as the tftp server. I vaguely remember
that using a hostname does not work.
Separate from running this, faiserver needs to be setup in dns
to point to whatever host this is run on.
EOF
    exit $1
}
case $1 in
    -h|--help) usage ;;
esac


e() { echo "$@"; "$@"; }

# When stretch becomes stable, change this to stretch.
# I've tested this with stretch, it works, but notably,
# the automatic basefile getting will be for stretch
# instead of jessie, so if you install jessie, you need
# to setup the basefile and it's corresponding class.
base=jessie
sed="sed -ri --follow-symlinks"

if grep -xFq 'VERSION="9 (stretch)"' /etc/os-release; then
  # if we use stretch, no need for fai-project repo.
  # this will need to be updated when there is a codename
  # for stretch+1
  rm -f /etc/apt/sources.list.d/fai.list
else
    wget -O - http://fai-project.org/download/074BCDE4.asc | apt-key add -
    cat >/etc/apt/sources.list.d/fai.list <<'EOF'
deb http://fai-project.org/download jessie koeln
EOF
fi

# for ubuntu:
#add-apt-repository -y ppa:fai/ppa

# for debian:


apt-get update

# Relevant packages from fai-quickstart depends and fai-server recommends.
# I especially do not wait isc-dhcp-server or an inetd
apt-get install -y fai-doc nfs-kernel-server tftpd-hpa tar reprepro squashfs-tools binutils
apt-get install --no-install-recommends -y fai-server

r=http://http.us.debian.org/debian
# like default, but scrap httpredir, and nonfree.
# All my systems should be able to get along without nonfree
# for a base working system afaik.
dd of=/etc/fai/apt/sources.list <<EOF
deb $r $base main contrib
deb http://security.debian.org/debian-security $base/updates main contrib
EOF

if [[ $base == jessie ]]; then
  tee -a /etc/fai/apt/sources.list <<'EOF'
# uncommenting this from the defaults. it's got bug fixes.
# repository that may contain newer fai packages for jessie
deb http://fai-project.org/download jessie koeln
# fix tar https://bugs.debian.org/cgi-bin/bugreport.cgi?bug=819978
deb http://ftp.debian.org/debian jessie-backports main
EOF

  # note, fai doesn't look at /etc/fai/apt/preferences.d
  cat >/etc/fai/apt/preferences <<'EOF'
Package: tar
Pin: release a=jessie-backports
Pin-Priority: 500
EOF
fi


# tried out a stretch base, doesn't work yet.
$sed -f - /etc/fai/nfsroot.conf <<EOF
s,^( *FAI_DEBOOTSTRAP=).*,\1"$base $r",
EOF

$sed 's/#LOGUSER/LOGUSER/' /etc/fai/fai.conf
# from man fai-make-nfsroot,
# figured out after partitioning ignored my crypt partition


if ! grep cryptsetup /etc/fai/NFSROOT &>/dev/null; then
  $sed '/^PACKAGES install$/a cryptsetup' /etc/fai/NFSROOT
fi
e fai-setup -vf
{ head -n 1 /srv/fai/nfsroot/root/.ssh/known_hosts | awk '{print $1}' \
    | tr '\n' ' '; ssh-keyscan localhost | grep -o "ecdsa-sha2-nistp256.*"; \
  } >>/srv/fai/nfsroot/root/.ssh/known_hosts

# initially did the basic fai-chboot -Iv $std_arg default
# but found in console that it wanted to mount nfsroot
# to be the same as my dhcp server.
# Figured out to change the root= parameter from googling,
# and seeing fai-chboot -L
# using hostname failed.
# for -f, combined the 2 defaults so it will reboot and print to screen.

# Add debug to -f flag for more verbose output.

# make the faiserver also the apt proxy server
apt-get -y install apt-cacher-ng

# background on choosing apt-cacher-ng:
# googling around a bit finds 2 main solutions:
# http://askubuntu.com/questions/3503/best-way-to-cache-apt-downloads-on-a-lan
# apt-cacher-ng doesn't have zeroconf.
# It touts having minimal dependencies, but I don't care.
# The downside to squid-deb-proxy is that it's config is for specific repos,
# you have to add all the repos you use.
# That is the main reason I use apt-cacher-ng.
# It has a web portal, at http://faiserver:3142/acng-report.html


# random fai note: as far as I can tell, profiles are just for putting
# in a selectable boot menu, which I don't want.

# the logsave prompted because the hostname faiserver was uknown.
# Here it was faiserver.lan when running from a faiserver vm.
# When running from a normal host with faiserver alias, it was the normal hosts name.
$sed 's/(^[^,]+,)\S+/\1faiserver/' /srv/fai/nfsroot/root/.ssh/known_hosts
# ditch the logo banner up top which screws with less.
touch /srv/fai/nfsroot/.nocolorlogo