#!/bin/sh

## live-build(7) - System Build Scripts
## Copyright (C) 2016-2020 The Debian Live team
## Copyright (C) 2006-2015 Daniel Baumann <mail@daniel-baumann.ch>
##
## This program comes with ABSOLUTELY NO WARRANTY; for details see COPYING.
## This is free software, and you are welcome to redistribute it
## under certain conditions; see COPYING for details.


set -e

# Including common functions
[ -e "${LIVE_BUILD}/scripts/build.sh" ] && . "${LIVE_BUILD}/scripts/build.sh" || . /usr/lib/live/build.sh

# Setting static variables
DESCRIPTION="Install disk information into binary"
USAGE="${PROGRAM} [--force]"

# Processing arguments and configuration files
Init_config_data "${@}"

if ! In_list "${LIVE_IMAGE_TYPE}" iso iso-hybrid hdd; then
	exit 0
fi

Echo_message "Begin installing disk information..."

# Requiring stage file
Require_stagefiles config bootstrap

# Checking stage file
Check_stagefile

# Acquire lock file
Acquire_lockfile

mkdir -p binary/.disk

ARCHITECTURE="$(echo ${LB_ARCHITECTURE} | sed -e 's| |/|g')"
DISTRIBUTION="$(echo ${LB_DISTRIBUTION_BINARY} | cut -b 1 | tr '[a-z]' '[A-Z]')"
DISTRIBUTION="${DISTRIBUTION}$(echo ${LB_DISTRIBUTION_BINARY} | cut -b 2-)"

if [ -e chroot/etc/os-release ]
then
	VERSION="$(. chroot/etc/os-release && echo ${VERSION_ID})"
fi

VERSION="${VERSION:-none}"

case "${LB_MODE}" in
	debian)
		TITLE="Debian GNU/Linux"
		STRING="Official Snapshot ${ARCHITECTURE}"
		TRACE="project/trace/ftp-master.debian.org"
		;;

	*)
		TITLE="Debian GNU/Linux"
		STRING="Snapshot ${ARCHITECTURE}"
		TRACE=""
		;;
esac

if [ -n "${TRACE}" ]
then
	case "${LB_DERIVATIVE}" in
		true)
			echo "$(echo ${LB_PARENT_MIRROR_BOOTSTRAP} | awk -F:// '{ print $2 }'): $(wget -q ${LB_PARENT_MIRROR_BOOTSTRAP}/${TRACE} -O - | head -n 1)" \
			> binary/.disk/archive_trace

			echo "$(echo ${LB_MIRROR_BOOTSTRAP} | awk -F:// '{ print $2 }'): $(wget -q ${LB_MIRROR_BOOTSTRAP}/${TRACE} -O - | head -n 1)" \
			>> binary/.disk/archive_trace
			;;

		false)
			echo "$(wget -q ${LB_PARENT_MIRROR_BOOTSTRAP}/${TRACE} -O - | head -n 1)" \
			> binary/.disk/archive_trace
			;;
	esac
fi

case "${LB_DEBIAN_INSTALLER}" in
	cdrom)
		DISK_LABEL="LIVE/CD"
		DISK_TYPE="full_cd"
		;;
	netboot)
		DISK_LABEL="LIVE/NETBOOT"
		#FIXME: type?
		DISK_TYPE=""
		;;
	netinst)
		DISK_LABEL="LIVE/NETINST"
		DISK_TYPE="not_complete"
		;;
	live)
		DISK_LABEL="LIVE/INSTALL"
		DISK_TYPE="live"
		;;
	businesscard)
		DISK_LABEL="LIVE/BC"
		DISK_TYPE="not_complete"
		;;
	none)
		DISK_LABEL="LIVE"
		DISK_TYPE=""
		;;
esac

echo "${TITLE} ${VERSION} \"${DISTRIBUTION}\" - ${STRING} ${DISK_LABEL} Binary $(date +%Y%m%d-%H:%M)" > binary/.disk/info

if [ -n "${DISK_TYPE}" ]; then
	echo "${DISK_TYPE}" > binary/.disk/cd_type
fi

#FIXME review the netboot case
case "${LB_DEBIAN_INSTALLER}" in
	none|netboot)
		;;
	businesscard)
		echo "main" > binary/.disk/base_components
		;;
	*)
		echo "main" > binary/.disk/base_components
		touch binary/.disk/base_installable
		;;
esac

#FIXME review the netboot case
for LOCATION in "${LIVE_BUILD}/data/debian-cd" /usr/share/live/build/data/debian-cd
do
	if [ -e "${LOCATION}" ]
	then
		case "${LB_DEBIAN_INSTALLER}" in
			cdrom)
				cp "${LOCATION}/${LB_PARENT_DISTRIBUTION_BINARY}/${LB_ARCHITECTURE}_udeb_include" binary/.disk/udeb_include
				;;
			netinst|live)
				cp "${LOCATION}/${LB_PARENT_DISTRIBUTION_BINARY}/${LB_ARCHITECTURE}_netinst_udeb_include" binary/.disk/udeb_include
				;;
			businesscard)
				cp "${LOCATION}/${LB_PARENT_DISTRIBUTION_BINARY}/${LB_ARCHITECTURE}_businesscard_udeb_include" binary/.disk/udeb_include
				;;
			none|netboot)
				;;
		esac
	fi
done

if [ "${LB_DEBIAN_INSTALLER}" = "live" ]
then
	echo "live-installer" >> binary/.disk/udeb_include
fi

# Creating stage file
Create_stagefile