#!/bin/sh

help() {
  echo "
  This script unbundles Parted Magic from The Ultimate Boot CD.
 
  Usage: $(basename "$0") [path-to-extracted-UBCD-iso]
  "
}

unbundle() {
# syslinux
sed -i -e '/pmagic[[:digit:]].hlp/s|^F|#F|g' -e '/MENU LABEL Parted Magic/aMENU HIDE' \
  ./ubcd/menus/{iso,sys}linux/main.cfg

sed -i -e '/MENU LABEL partimage/aMENU HIDE' \
  ./ubcd/menus/syslinux/hdd/cloning.cfg
 
sed -i -e '/MENU LABEL GParted/aMENU HIDE' \
  ./ubcd/menus/syslinux/hdd/partmgmt.cfg
 
sed -i -e '/MENU LABEL PhotoRec/aMENU HIDE' -e '/MENU LABEL TestDisk/aMENU HIDE' \
  ./ubcd/menus/syslinux/hdd/recovery.cfg

# grub4dos
sed -i -e '/title Parted Magic/,/configfile/s|^|#|g' \
  ./ubcd/menus/grub4dos/main.lst

sed -i -e '/title partimage/,/configfile/s|^|#|g' \
  ./ubcd/menus/grub4dos/hdd/cloning.lst

sed -i -e '/title GParted/,/configfile/s|^|#|g' \
  ./ubcd/menus/grub4dos/hdd/partmgmt.lst

sed -i -e '/title PhotoRec/,/configfile/s|^|#|g' -e '/title TestDisk/,/configfile/s|^|#|g' \
  ./ubcd/menus/grub4dos/hdd/recovery.lst

# cleanup
rm ./ubcd/menus/syslinux/pmagic*.hlp
rm -rf ./pmagic
rm ./ubcd/tools/{linux,win32}/antivirus/{clamav,fprot}.*
}

ubcddir="$1"

if [ -z "$1" ]; then
  help
  exit
elif [ -f "$ubcddir/ubcd/ubcd.ico" ]; then
  cd -- "$ubcddir"
else
  echo "$(basename "$0"): \`$ubcddir' directory does not contain UBCD files"
  exit 1
fi

unbundle

if [ "$?" != "0" ]; then
  echo "$(basename "$0"): An error occurred, please try again"
  exit 1
else
  echo "$(basename "$0"): All processes completed successfully"
  exit
fi