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

# Usage: faiserver-setup
#
# 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.

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

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

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

# tried with stretch at one point, but jessie works for everything,
# so no point in changing it. Leaving some code to deal with
# stretch hanging around as it will eventually become stable.
base=jessie
sed="sed -ri --follow-symlinks"


if [[ $base == jessie ]]; then
  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
else
    # if we use stretch, no need for fai-project repo.
    rm -f /etc/apt/sources.list.d/fai.list
fi

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

# for debian:

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

apt-get update
# all the dependencies except the dhcp server
deps="$(apt-cache show fai-quickstart | grep ^Depends: |head -n 1|\
              sed -r 's/^Depends:|,|\|[^,]+|isc-dhcp-server//g')"
to_install=(tar)
for pkg in $deps; do
    dpkg -s $pkg &>/dev/null && continue ||:
    to_install+=($pkg)
    # just so we have a record.
    echo `date` $pkg >>/var/log/fai-manually-installed-packages.log
done
if [[ $to_install ]]; then
  apt-get -y install ${to_install[@]}
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