#!/bin/sh

#set -e

# initramfs-tools header

PREREQ=""

prereqs()
{
	echo "${PREREQ}"
}

case "${1}" in
	prereqs)
		prereqs
		exit 0
		;;
esac

# live-initramfs header

if [ -n "${NOSWAP}" ] || [ -n "${NOFSTAB}" ]
then
	exit 0
fi

. /scripts/live-functions

log_begin_msg "Setting up swap..."

# live-initramfs script

FSTAB=/root/etc/fstab
devices=""

for device in /dev/[hs]d[a-z][0-9]*
do
	if ! [ -b "${device}" ]
	then
		continue
	fi

	magic=$(/bin/dd if="${device}" bs=4086 skip=1 count=1 2>/dev/null | /bin/dd bs=10 count=1 2>/dev/null) || continue

	if [ "${magic}" = "SWAPSPACE2" -o "${magic}" = "SWAP-SPACE" ]
	then
		#log "Found ${device}"
		devices="${devices} ${device}"
	fi
done

for device in ${devices}
do

cat >> ${FSTAB} << EOF
${device} swap swap defaults 0 0
EOF

done

log_end_msg
