#!/bin/sh
#
# $Id: label-partitions-as-necessary 2801 2011-04-29 13:38:41Z bruno $
#
############################################


read_partition_line() {
    local label mountpt command format

	label=`echo "$1" | awk '{print $1}' | cut -d'=' -f2`
	format=`echo "$1" | awk '{print $3}'`
	mountpt=`awk '{print $1,$5}' $mountlist | grep " $label$" | awk '{print $1}'`

	if [ "`echo "$1" | grep -E 'LABEL='`" != "" ] ; then
		[ "$format" = "reiserfs" ] && opttun="-l" || opttun="-L"
	elif [ "`echo "$1" | grep -E 'UUID='`" != "" ] ; then
		[ "$format" = "reiserfs" ] && opttun="-u" || opttun="-U"
	else
		LogIt "Nothing to do on $1"
		return
	fi

	if [ ! "$mountpt" ] ; then
		LogIt "Not labeling anything as ($label) because ($mountpt) is not a mountpoint"
	elif [ ! "$label" ] ; then
		LogIt "Not labeling ($mountpt) as anything because ($label) is not a label"
	else
		if [ "$format" = "ext2" ] || [ "$format" = "ext3" ] || [ "$format" = "ext4" ]; then
			command="tune2fs $opttun $label $mountpt"
			LogIt "Running $command"
			$command
		elif [ "$format" = "reiserfs" ]; then
			command="reiserfstune $opttun $label $mountpt"
			LogIt "Running $command"
			$command
		elif [ "$format" = "swap" ] ; then
			if [ "$opttun" = "-U" ]; then
				LogIt "Creating uuid $label on swap partition $mountpt"
				echo -n "$label" | perl -ne 's/-//g;chomp;print pack "H*",$_' | dd conv=notrunc "of=$mountpt" obs=1 seek=1036
			else
				command="mkswap $opttun $label $mountpt"
				LogIt "Running $command"
				$command
			fi
		else
			LogIt "I am NOT going to run tune2fs/reiserfstune: the partition is format '$format', which doesn't like tune2fs/reiserfstune anyway"
		fi
	fi
}


# ---------------------------------------------

LogIt "Identifying your drives with tune2fs"
if [ "$#" -ne "1" ] ; then
    LogIt "label-partitions-as-necessary /tmp/mountlist.txt < /tmp/fstab.new" 1
    exit 1
fi
mountlist=$1
noof_blank_lines=0
read line
# We end after 50 empty lines, which hopefully means we reach the end of the file
while [ "$noof_blank_lines" -le "50" ] ; do
    if [ "$line" = "" ] ; then
		noof_blank_lines=$(($noof_blank_lines+1))
    else
		noof_blank_lines=0
		read_partition_line "$line"
    fi
    read line
done
LogIt "Labeling complete."
exit 0
