#!/bin/bash
# Filename:      grml-live
# Purpose:       build process script for generating a (grml based) Linux Live-ISO
# Authors:       grml-team (grml.org), (c) Michael Prokop <mika@grml.org>
# Bug-Reports:   see http://grml.org/bugs/
# License:       This file is licensed under the GPL v2 or any later version.
# Latest change: Mon Sep 17 18:31:35 CEST 2007 [mika]
################################################################################

# read configuration files, set some misc variables {{{

export LANG=C
export LC_ALL=C

# exit on any error:
set -e

# we need root permissions for the build-process:
if [ "$(id -u 2>/dev/null)" != 0 ] ; then
   echo "Error: please run this script with uid 0 (root)." >&2
   exit 1
fi

# make sure they are not set by default
VERBOSE=''
FORCE=''

# source main configuration file:
. /etc/grml/grml-live.conf

PN=$(basename $0)
# TMPFILE=$(mktemp)
# }}}

# clean exit {{{
bailout() {
  # rm -f "$TMPFILE"
  [ -n "$1" ] && EXIT="$1" || EXIT="1"
  [ -n "$2" ] && echo "$2">&2
  exit "$EXIT"
}
trap bailout 1 2 3 15
# }}}

# check for important variables {{{
[ -n "$GRML_FAI_CONFIG" ] || GRML_FAI_CONFIG=/etc/grml/fai
[ -n "$HOSTNAME" ] || HOSTNAME=grml
[ -n "$USERNAME" ] || USERNAME=grml
[ -n "$CLASSES" ]  || CLASSES="GRML,I386"
[ -n "$TARGET" ] || bailout 1 "${PN}: \$TARGET not specified. Please adjust /etc/grml/grml-live.conf. Exiting."
# }}}

# usage information {{{
usage()
{
  echo "
$PN - build process script for generating a (grml based) Linux Live-ISO

Usage: $PN [-c <classe[s]>] [-t <target_directory>] [-s <suite> [-Fvh]

Usage examples:

    $PN
    $PN -c GRMLBASE,I386 -t /dev/shm/grml
    $PN -c GRMLBASE,GRML_X,I386 -t /grml/
    $PN -c GRMLBASE,GRML_SMALL,I386
    $PN -s sid -c GRMLBASE,I386

More details: man grml-live
              /usr/share/doc/grml-live/grml-live.html

Please send your bug reports, feedback,.. to the grml-team.
http://grml.org/bugs/
"
}
# }}}

# command line parsing {{{

while getopts "c:s:t:Fhv" opt; do
  case "$opt" in
    c) CLASSES="$OPTARG" ;;
    # f) FLAVOUR="$OPTARG" ;; # TODO
    F) FORCE=1 ;;
    h) usage ; bailout 0 ;;
    s) SUITE="$OPTARG" ;;
    t) TARGET="$OPTARG"
       CHROOT_TARGET="$TARGET/grml_chroot"
       BUILD_TARGET="$TARGET/grml_cd"
       ISO_TARGET="$TARGET/grml_isos"
       ;;
    v) VERBOSE="-v" ;;
    ?) echo "invalid option -$OPTARG" >&2; bailout 1 ;;
  esac
done
shift $(($OPTIND - 1))  # set ARGV to the first not parsed commandline parameter

# }}}

# some misc checks before executing FAI {{{
[ -n "$CLASSES" ] || bailout 1 "Error: \$CLASSES unset, please set it in /etc/grml/grml-live.conf or
specify it on the command line using the -c|--classes option."
[ -n "$TARGET" ] || bailout 1 "Error: \$TARGET unset, please set it in /etc/grml/grml-live.conf or
specify it on the command line using the -t|--target option."
# }}}

# ask user whether the setup is ok {{{
if [ -z "$FORCE" ] ; then
   echo
   echo "$PN - check your configuration (or invoke using -F to force execution without prompting)"
   echo
   echo "  FAI classes:       $CLASSES"
   echo "  output directory:  $TARGET"
   [ -n "$CHROOT_TARGET" ] && echo "  chroot target:     $CHROOT_TARGET"
   [ -n "$BUILD_TARGET" ]  && echo "  build target:      $BUILD_TARGET"
   [ -n "$ISO_TARGET" ]    && echo "  ISO target:        $ISO_TARGET"
   [ -n "$SUITE" ]         && echo "  Debian suite:      $SUITE"
   [ -n "$FAI_ARGS" ]      && echo "  additional arguments for FAI: $FAI_ARGS"
   [ -n "$VERBOSE" ]       && echo "  Using VERBOSE mode."
   echo
   echo -n "Is this ok for you? [y/N] "
   read a
   if ! [ "$a" = 'y' -o "$a" = 'Y' ] ; then
      bailout 1 "Exiting as requested."
   fi
   echo
fi
# }}}

# on-the-fly configuration {{{
if [ -n "$GRML_LIVE_SOURCES" ] ; then
   echo "$GRML_LIVE_SOURCES" > /etc/grml/fai/apt/sources.list
fi

# this heuristic is broken as it works exactly *once* by default :-/
# TODO / FIXME
if [ -n "$SUITE" ] ; then
   sed -i "s#etch #$SUITE #"   /etc/grml/fai/make-fai-nfsroot.conf
   sed -i "s#etch #$SUITE #"   /etc/grml/grml-live.conf
   sed -i "s# etch # $SUITE #" /etc/grml/grml-live.conf
   sed -i "s# etch # $SUITE #" /etc/grml/fai/apt/sources.list
fi

if [ -n "$FAI_DEBOOTSTRAP" ] ; then
   sed -i "s#^FAI_DEBOOTSTRAP#FAI_DEBOOTSTRAP=$FAI_DEBOOTSTRAP#" /etc/grml/fai/make-fai-nfsroot.conf
fi
# }}}

# CHROOT_TARGET - execute FAI {{{
[ -n "$CHROOT_TARGET" ] || CHROOT_TARGET="$TARGET/grml_chroot"

if [ -d "$CHROOT_TARGET" ] ; then
   echo "  [x] $CHROOT_TARGET exists already, skipping the stage 'fai dirnstall'"
else
   mkdir "$CHROOT_TARGET" || bailout 5 "Problem with creating $CHROOT_TARGET for FAI"
   fai $VERBOSE -C "$GRML_FAI_CONFIG" -c"$CLASSES" dirinstall "$CHROOT_TARGET" $FAI_ARGS
   umount $CHROOT_TARGET/proc 2>/dev/null || /bin/true
   umount $CHROOT_TARGET/sys  2>/dev/null || /bin/true
   # notice: 'fai dirinstall' does not seem to exit appropriate ->
   # we need better error handling
   if [ -r "$CHROOT_TARGET"/etc/grml_cd ] ; then
      echo "  [*] Finished execution of stage 'fai dirinstall'"
   else
      echo "  [!] There was an error during execution of stage 'fai dirinstall'"
      exit 1
   fi
fi
# }}}

# BUILD_TARGET - execute arch specific stuff and squashfs {{{
[ -n "$BUILD_TARGET" ] || BUILD_TARGET="$TARGET/grml_cd"
mkdir -p "$BUILD_TARGET" || bailout 6 "Problem with creating $BUILD_TARGET for stage ARCH"

# i386:
[ -n "$ARCH" ] || ARCH="$(dpkg --print-architecture)"
if [ "$ARCH" = i386 ] ; then
   if [ -d "$BUILD_TARGET"/boot/isolinux ] ; then
      echo "  [x] $BUILD_TARGET/boot/isolinux exists already - skipping stage 'boot/isolinux'"
      continue
   else
      # booting stuff:
      mkdir -p "$BUILD_TARGET"/boot/isolinux
      mkdir "$BUILD_TARGET"/GRML
      cp /boot/memtest86+.bin                                        "$BUILD_TARGET"/boot/isolinux/memtest
      cp "$CHROOT_TARGET"/boot/initrd*                               "$BUILD_TARGET"/boot/isolinux/initrd.gz
      cp "$CHROOT_TARGET"/boot/vmlinuz*                              "$BUILD_TARGET"/boot/isolinux/linux26
      cp /usr/lib/syslinux/chain.c32                                 "$BUILD_TARGET"/boot/isolinux/
      cp /usr/lib/syslinux/isolinux.bin                              "$BUILD_TARGET"/boot/isolinux/
      cp /usr/lib/syslinux/memdisk                                   "$BUILD_TARGET"/boot/isolinux/
      cp /usr/lib/syslinux/menu.c32                                  "$BUILD_TARGET"/boot/isolinux/
      cp /usr/share/grml-live/i386_files/boot/isolinux/allinone.img  "$BUILD_TARGET"/boot/isolinux/
      cp /usr/share/grml-live/i386_files/boot/isolinux/balder10.imz  "$BUILD_TARGET"/boot/isolinux/
      cp /usr/share/grml-live/i386_files/boot/isolinux/boot-beep.msg "$BUILD_TARGET"/boot/isolinux/
      cp /usr/share/grml-live/i386_files/boot/isolinux/boot.msg      "$BUILD_TARGET"/boot/isolinux/
      cp /usr/share/grml-live/i386_files/boot/isolinux/f*            "$BUILD_TARGET"/boot/isolinux/
      cp /usr/share/grml-live/i386_files/boot/isolinux/isolinux.cfg  "$BUILD_TARGET"/boot/isolinux/
      cp /usr/share/grml-live/i386_files/boot/isolinux/logo.16       "$BUILD_TARGET"/boot/isolinux/
      cp /usr/share/grml-live/i386_files/boot/isolinux/syslinux.cfg  "$BUILD_TARGET"/boot/isolinux/
      # autostart for Windows:
      cp /usr/share/grml-live/windows/autostart/autorun.bat          "$BUILD_TARGET"/
      cp /usr/share/grml-live/windows/autostart/autorun.inf          "$BUILD_TARGET"/
      cp /usr/share/grml-live/windows/autostart/autorun.pif          "$BUILD_TARGET"/
      cp /usr/share/grml-live/windows/autostart/cdrom.ico            "$BUILD_TARGET"/
      # windows-binaries:
      if [ -n "$WINDOWS_BINARIES" ] ; then
         if [ -d "$BUILD_TARGET"/windows ] ; then
            echo "  [x] $BUILD_TARGET/windows exists already - skipping stage 'WINDOWS_BINARIES'"
            return 0
         else
            mkdir "$BUILD_TARGET"/windows
            ( cd "$BUILD_TARGET"/windows
              for file in pageant plink pscp psftp putty puttygen ; do
                  wget ${WINDOWS_BINARIES}/${file}.exe
              done )
         fi
      echo "  [*] Finished execution of stage 'WINDOWS_BINARIES'"
      fi
   echo "  [*] Finished execution of stage 'boot/isolinux'"
   fi
# amd64:
elif [ "$ARCH" = amd64 ] ; then
    echo 'Warning: gebi, it is your turn. :)'>&2
# ppc:
elif [ "$ARCH" = powerpc ] ; then
    echo 'Warning: formorer, it is your turn. :)'>&2
# unsuported:
else
   echo 'Warning: Unsupported ARCH, sorry. Want to support it? Contribute!'>&2
fi

if [ -d "$BUILD_TARGET"/live ] ; then
   echo "  [x] $BUILD_TARGET/live exists already, skipping stage 'squashfs'"
else
   mkdir "$BUILD_TARGET"/live
   mksquashfs $CHROOT_TARGET/* $BUILD_TARGET/live/grml.squashfs -noappend
fi
echo "  [*] Finished execution of stage 'squashfs'"
# }}}

# ISO_TARGET - mkisofs {{{
[ -n "$ISO_TARGET" ] || ISO_TARGET="$TARGET/grml_isos"
if [ -d "$ISO_TARGET" ] ; then
   echo "  [x] $ISO_TARGET exists already, skipping the stage 'iso build'"
else
   mkdir -p "$ISO_TARGET" || bailout 6 "Problem with creating $ISO_TARGET for stage 'iso build'"
   ( cd "$BUILD_TARGET" &&
     mkisofs -V "Debian/etch grml" -publisher 'Michael Prokop <mika@grml.org>' \
             -l -r -J -no-emul-boot -boot-load-size 4 -boot-info-table    \
             -c boot/isolinux/boot.cat -b boot/isolinux/isolinux.bin      \
             -o "$ISO_TARGET"/grml_0.0-1.iso .
   )
   echo "  [*] Finished execution of stage 'iso build'"
fi
# }}}

# finalize {{{
echo "  [*] Sucessfully finished execution of $PN"
bailout 0
# }}}

## END OF FILE #################################################################
# vim:foldmethod=marker ts=2 ft=sh ai expandtab tw=80 sw=3