#!/bin/bash
#
# Do install stuff for x86, including making bootable CDs
# Works with debian-installer
#
# $1 is the CD number
# $2 is the temporary CD build dir

. $BASEDIR/tools/boot/$DI_CODENAME/common.sh

set -e
#set -x

N=$1
CDDIR=$2
BOOTDIR=
if [ "$DI_WWW_HOME" = "default" ];then
	if [ "$ARCH" = i386 ]; then
		DI_WWW_HOME="http://people.debian.org/~joeyh/d-i/images/daily"
	else
		DI_WWW_HOME="http://people.debian.org/~aba/d-i/images/daily"
	fi
	if [ -n "$DI_DIR" ];then
		DI_DIR="$DI_DIR/~${DI_WWW_HOME#*~}"
		DI_WWW_HOME=""                     
		echo "Using images from $DI_DIR"
	fi
fi
if [ ! "$DI_DIST" ]; then
	DI_DIST="$DI_CODENAME"
fi

cd $CDDIR/..

# This script is called with $1 (now $N) as the CD to 
# make bootable.  N may be in the form "n" or "n_NONUS"
# There may be more than 4 disks...support extras.

# Strip NONUS part of disk number
# NN will be 1...N so it can be used as an index
NN=`echo $N | sed -e 's/_NONUS//'`

# List of boot image types for each CD
KTYPE[1]="isolinux"
#KTYPE[2]="cdrom"
KTYPE[2]=""
KTYPE[3]=""
KTYPE[4]=""
KTYPE[5]=""
KTYPE[6]=""
KTYPE[7]=""
KTYPE[8]=""
KTYPE[9]=""
KTYPE[10]=""

THISTYPE=${KTYPE[$NN]}

INSTALLDIR_i386="install.386"
INSTALLDIR_amd64="install.amd"

if [ "$ARCH" = i386 ]; then
	INSTALLDIR="$INSTALLDIR_i386"
else
	INSTALLDIR="$INSTALLDIR_amd64"
fi

BOOT_IMAGES="cdrom/initrd.gz cdrom/vmlinuz cdrom/debian-cd_info.tar.gz"
# Only include disk images on full CDs, not on smaller images.
if [ "$ARCH" = i386 ] && [ "$INSTALLER_CD" != 1 ] && [ "$INSTALLER_CD" != 2 ]; then
	DISK_IMAGES="floppy/cd-drivers.img floppy/boot.img floppy/root.img"
	#EXTRA_DISK_IMAGES="cdrom/boot.img"
fi

# Download boot images.
for image in $BOOT_IMAGES; do
	if [ ! -e "$image" ]; then
		dir=$(dirname $image)
		mkdir -p $dir
		if [ -n "$LOCAL"  -a -f "${LOCALDEBS:-$MIRROR}/dists/$DI_DIST/local/installer-$ARCH/current/images/$image" ]; then
			cp "${LOCALDEBS:-$MIRROR}/dists/$DI_DIST/local/installer-$ARCH/current/images/$image" "$image"
		elif [ ! "$DI_WWW_HOME" ];then
			if [ ! "$DI_DIR" ];then
				DI_DIR="$MIRROR/dists/$DI_DIST/main/installer-$ARCH/current/images"
			fi
#			echo "Copying images from $DI_DIR"
			cp "$DI_DIR/$image" "$image"
		else
			wget "$DI_WWW_HOME/$image" -O "$image"
		fi
	fi
done

# Images that are not really required or may be unavailable.
for image in $DISK_IMAGES $EXTRA_DISK_IMAGES; do
	if [ ! -e "$image" ]; then
		dir=$(dirname $image)
		mkdir -p $dir
		if [ -n "$LOCAL"  -a -f "${LOCALDEBS:-$MIRROR}/dists/$DI_DIST/local/installer-$ARCH/current/images/$image" ]; then
			cp "${LOCALDEBS:-$MIRROR}/dists/$DI_DIST/local/installer-$ARCH/current/images/$image" "$image"
		elif [ ! "$DI_WWW_HOME" ];then
			if [ ! "$DI_DIR" ];then
				DI_DIR="$MIRROR/dists/$DI_DIST/main/installer-$ARCH/current/images"
			fi
			cp "$DI_DIR/$image" "$image" || true
		else
			wget "$DI_WWW_HOME/$image" -O "$image" || rm -f "$image"
		fi
	fi
done

# Only disk one gets the extra files installed
if [ "$NN" = "1" ]; then
	# populate the install directory as well
	for disk in $DISK_IMAGES $EXTRA_DISK_IMAGES; do
		if [ -e "$disk" ]; then
			dir=$(dirname $disk)
			mkdir -p $CDDIR/$INSTALLDIR/$dir
			cp -lf $disk $CDDIR/$INSTALLDIR/$dir
		fi
	done
fi

extra_image () {
	image=$1
	dir=$(dirname "$image")

	mkdir -p $CDDIR/$INSTALLDIR/$dir
	if [ ! -e $CDDIR/$INSTALLDIR/"$image" ] ; then
		if [ ! "$DI_WWW_HOME" ] ; then
			if [ ! "$DI_DIR" ] ; then
				DI_DIR="$MIRROR/dists/$DI_DIST/main/installer-$ARCH/current/images"
			fi
			cp "$DI_DIR/cdrom/$image" $CDDIR/$INSTALLDIR/"$image"
		else
			wget "$DI_WWW_HOME/cdrom/$image" -O $CDDIR/$INSTALLDIR/"$image"
		fi
	fi
}

if [ "$THISTYPE" = "isolinux" ]; then
	echo "Using ISOLINUX boot-disks image on CD$N"
	mkdir -p $CDDIR/$INSTALLDIR
	cp -lf cdrom/vmlinuz $CDDIR/$INSTALLDIR/
	cp -lf cdrom/initrd.gz $CDDIR/$INSTALLDIR/

    SYSLINUXDEB=`$BASEDIR/tools/which_deb $MIRROR $DI_CODENAME syslinux`
    (ar p "${SYSLINUXDEB}" data.tar.gz | \
	tar zxf - -C $CDDIR/../syslinux ./usr/lib)

	add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-J -joliet-long"
	add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-cache-inodes"
	add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-b isolinux/isolinux.bin"
	add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-c isolinux/boot.cat"
	add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-no-emul-boot"
	add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-boot-load-size 4"
	add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-boot-info-table"	
	add_mkisofs_opt $CDDIR/../$N.mkisofs_dirs "boot$N"
	
	if [ "$ARCH" = i386 ]; then
		ISOLINUXDIR=isolinux
		
		# If the isolinux dir already exists, rename it aside for now
		if [ -e boot$N/isolinux/isolinux.cfg ] ; then
			mv boot$N/isolinux boot$N/isolinux-amd64
		fi
	else
		ISOLINUXDIR=isolinux-"$ARCH"
	fi
	mkdir -p boot$N/$ISOLINUXDIR
	# Isolinux setup including config and help files comes from d-i.
	cat cdrom/debian-cd_info.tar.gz | (cd boot$N/$ISOLINUXDIR/; tar zx)
	# Not everything in the tarball is isolinux stuff
	if [ -e boot$N/$ISOLINUXDIR/win32-loader.ini ] ; then
	    mv boot$N/$ISOLINUXDIR/g2ldr* boot$N/$ISOLINUXDIR/setup.exe \
		boot$N/$ISOLINUXDIR/win32-loader.ini boot$N/
	fi
	if [ -e boot$N/$ISOLINUXDIR/f3.txt.withgtk ]; then
		extra_image gtk/initrd.gz
		mv boot$N/$ISOLINUXDIR/f3.txt.withgtk boot$N/$ISOLINUXDIR/f3.txt
		mv boot$N/$ISOLINUXDIR/f4.txt.withgtk boot$N/$ISOLINUXDIR/f4.txt
		if [ -e boot$N/$ISOLINUXDIR/isolinux.cfg.withgtk ]; then
			mv boot$N/$ISOLINUXDIR/isolinux.cfg.withgtk boot$N/$ISOLINUXDIR/isolinux.cfg
		fi
	else
		# Remove gtk isolinux config files.
		rm -f boot$N/$ISOLINUXDIR/gtk.cfg
		rm -f boot$N/$ISOLINUXDIR/amdgtk.cfg
	fi
	rm -f boot$N/$ISOLINUXDIR/isolinux.cfg.with* 2>/dev/null || true

	for f in isolinux.cfg text.cfg gtk.cfg adtext.cfg adgtk.cfg; do
		if [ -e boot$N/$ISOLINUXDIR/$f ]; then
			sed -i "s|/install/|/$INSTALLDIR/|" boot$N/$ISOLINUXDIR/$f
		fi
	done
	for f in amd.cfg amdgtk.cfg adamd.cfg adamdgtk.cfg; do
		if [ -e boot$N/$ISOLINUXDIR/$f ]; then
			sed -i "s|/install/|/$INSTALLDIR_amd64/|" boot$N/$ISOLINUXDIR/$f
		fi
	done

	if [ -e boot$N/win32-loader.ini ] ; then
	    sed -i "s|install/|$INSTALLDIR/|" boot$N/win32-loader.ini
        if [ "${DESKTOP}" ] ; then
            sed -i "s|^\(default_desktop\)=.*|\1=${DESKTOP}|" boot$N/win32-loader.ini
        fi
	fi

    cp -f $CDDIR/../syslinux/usr/lib/syslinux/isolinux.bin boot$N/$ISOLINUXDIR/
    cp -f $CDDIR/../syslinux/usr/lib/syslinux/vesamenu.c32 boot$N/$ISOLINUXDIR/

	if [ -n "$ISOLINUX_DEFAULT" ]; then
		sed -i "s/^DEFAULT .*$/DEFAULT $ISOLINUX_DEFAULT/" boot$N/$ISOLINUXDIR/isolinux.cfg
	fi

	# Handle merging amd64 and i386 isolinux dirs if both are present.
	if [ -e boot$N/isolinux-amd64/isolinux.cfg ] && [ -e boot$N/isolinux/isolinux.cfg ]; then
		for file in boot$N/isolinux/f3.txt* boot$N/isolinux/f4.txt*
		do
			cat $file | awk '
				NR==3 { next }
				NR==5 { printf " (i386 is default; prepend 'amd64-' to use an amd64 kernel instead)\n" }
				/.*/  { print $0 }' > $file.tmp
			mv $file.tmp $file
		done

		if [ -e boot$N/isolinux/amd.cfg ]; then
			# Split isolinux configs exist, so the amd.cfg will
			# be loaded and set things up for amd64. No munging
			# needed.
			:
		else 
			# This is compatability code for old versions of d-i that
			# do not use split isolinux configs.
			cat boot$N/isolinux-amd64/isolinux.cfg | awk '
				/^[Ll][Aa][Bb][Ee][Ll]/ { printf("label amd64-%s\n", $2) }
				/^[Dd][Ee][Ff][Aa][Uu][Ll][Tt]/ { printf("default64 amd64-%s\n", $2) }
				/[Kk][Ee][Rr][Nn][Ee][Ll]/ { print $0 }
				/[Aa][Pp][Pp][Ee][Nn][Dd]/ { print $0 }' >> boot$N/isolinux/isolinux.cfg
		fi

		sed -i -e "/^arch=/d ; /^i386\//p; s/^i386\//amd64\//; s/=$INSTALLDIR_i386\//=$INSTALLDIR_amd64\//g" \
			boot$N/win32-loader.ini

	else
		if [ "$ARCH" = amd64 ]; then
		        mkdir -p boot$N/isolinux
			mv -f boot$N/isolinux-amd64/* boot$N/isolinux
		fi

		if [ -e boot$N/isolinux/amd.cfg ]; then
			# Split isolinux configs exist. Remove the amd.cfg
			# to avoid it being loaded on a disc that does not
			# have both amd64 and i386 dirs.
			rm -f boot$N/isolinux/amd.cfg
			rm -f boot$N/isolinux/amdgtk.cfg
			rm -f boot$N/isolinux/adamd.cfg
			rm -f boot$N/isolinux/adamdgtk.cfg
		fi
	fi
	rm -rf boot$N/isolinux-amd64
	
	if [ "$SPLASHPNG" ] ; then
		# Insert our own splash screen.  Color index 0 is
		# background, and index 7 is foreground.  Set to black and
		# white respecively
		pngtopnm < $SPLASHPNG | ppmquant 16 | \
			ppmtolss16 "#ffffff=7" "#000000=0" > boot$N/isolinux/splash.rle
	fi

	sed -i "s|built on|built $BUILD_DATE; d-i|" boot$N/isolinux/f1.txt

	if [ -n "$KERNEL_PARAMS" ]; then
		# Substitute custom kernel params into the isolinux config
		# file(s)
        sed -i "/^[[:space:]]\+append .*--/ s|append|append $KERNEL_PARAMS|" boot$N/isolinux/*.cfg
	fi

else
	if [ -n "$THISTYPE" ]; then
		echo "Using $THISTYPE boot-disks image on CD$N"
		mkdir -p boot$N/boot
		cp $THISTYPE/boot.img boot$N/boot/

		add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-J -joliet-long"
		add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-cache-inodes"
		add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-b boot/boot.img"
		add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-c boot/boot.catalog"
		add_mkisofs_opt $CDDIR/../$N.mkisofs_dirs "boot$N"
	else
		mkdir boot$N
		add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-J -joliet-long"
		add_mkisofs_opt $CDDIR/../$N.mkisofs_opts "-cache-inodes"
	fi
fi

# Add autorun
if [ -f boot$N/setup.exe ]; then
# when win32-loader is present, use that (it already checks for README.html)
	todos > $CDDIR/autorun.inf <<EOF
[autorun]
open=setup.exe
EOF
elif [ -f $CDDIR/README.html ]; then
	todos > $CDDIR/autorun.inf <<EOF
[autorun]
open=autorun.bat
EOF
	todos > $CDDIR/autorun.bat <<EOF
@echo Starting "README.html"...
@start README.html
@exit
EOF
fi

rm -rf cdrom floppy

# th,th, thats all