#!/bin/ash
#
# Author: Magnus Uppman, magnus_uppman@hotmail.com
# License: GPL

# Uses: lilo, mktemp, cfdisk, blkid, extlinux, mbr.bin, tar, mkfs (ext2,ext4)
# Changelog:
# 2009-11-23 No longer copies system_13.0.tar.gz to the root partition.
# v0.04, 2009-12-02 Added backup option. Removed set -e

VERSION=0.04

# Create temp dir
TMPDIR=`mktemp -d /tmp/slackdeploy.XXXXXX`
#echo $TMPDIR
mkdir $TMPDIR/install
mkdir $TMPDIR/boot
mkdir $TMPDIR/backup
mkdir $TMPDIR/backup-store

show_info()
{
cat /proc/partitions
/sbin/blkid -c /dev/null
}

edit_disk()
{
echo "Enter disk device name:"
echo -n "(/dev/XXX) ->"
read ASK
case $ASK in
  /dev/???)
	  /sbin/cfdisk $ASK
	  ;;
  *)	
	  echo "Wrong parameter, please try again!"
esac
}


reinstall_lilo_mbr()
{
echo "Enter root partition device name:"
echo -n "(/dev/XXXy) ->"
read ASK
case $ASK in
  /dev/????)
	  # Install lilo to MBR?
	  ROOT_PART=$ASK
	  DISK=`echo $ROOT_PART | cut -c -8`
	  echo "Install lilo to the MBR on disk $DISK? Warning, the MBR on disk $DISK will be overwritten, make sure no other operating systems are installed on this disk!"
	  echo -n "(y/N)"
	  read ASK
	  case $ASK in
	    y)
		    echo "Installing lilo to MBR in $DISK"
		    mount $ROOT_PART $TMPDIR/install
		    /sbin/lilo -v -b $DISK -r $TMPDIR/install
		    umount $TMPDIR/install
		    ;;
	    *)	
		    echo "No changes made to the MBR!"
	  esac
	  ;;
  *)	
	  echo "Wrong parameter, please try again!"
esac
}

reinstall_lilo_superblock()
{
echo "Enter root partition device name:"
echo -n "(/dev/XXXy) ->"
read ASK
case $ASK in
  /dev/????)
	  ROOT_PART=$ASK
	  # Install lilo to superblock 
	  echo "Installing lilo to superblock in $ROOT_PART"
	  mount $ROOT_PART $TMPDIR/install
	  /sbin/lilo -v -b $ROOT_PART -r $TMPDIR/install
	  umount $TMPDIR/install
	  ;;
  *)	
	  echo "Wrong parameter, please try again!"
esac
}

reinstall_plain_mbr()
{
echo "Enter disk device name:"
echo -n "(/dev/XXX) ->"
read ASK
case $ASK in
  /dev/???)
	  DISK=$ASK
	  echo "Install new MBR on disk ${DISK}? Warning, the MBR on disk $DISK will be overwritten, make sure no other operating systems are installed on this disk!"
	  echo -n "(y/N)"
	  read ASK
	  case $ASK in
	    y)
	      echo "Installing new MBR on $DISK"
	      cat /usr/lib/syslinux/mbr.bin > $DISK
	      ;;
	    *)	
	      echo "No changes made to the MBR!"
	  esac
	  ;;
  *)	
	  echo "Wrong parameter, please try again!"
esac
}


install_root_part()
{
CONTINUE=0
while `test $CONTINUE -eq 0` 
do
echo "Enter root partition device name:"
echo "(c)ancel. Show disk (i)nfo"
echo -n "(/dev/XXXy, c, i) ->"
read ASK
case $ASK in
  /dev/????)
	  echo "Install root to ${ASK}? (Warning, ${ASK} will be completely erased!!!)"
	  echo -n "(yes,No) ->"
	  read WARN
	  case $WARN in
	    yes)
	      ROOT_PART=$ASK
	      CONTINUE=1
	      install_root
	      ;;
	    *)
	      echo "Selection aborted!"
	  esac
	  ;;
  i)
	    show_info
	    ;;
  c)
	    CONTINUE=1
	    ;;
  *)	
	  echo "Wrong parameter, please try again!"
esac
done
}

install_root_and_boot_part()
{
CONTINUE=0
while `test $CONTINUE -eq 0` 
do
echo "Enter root partition device name:"
echo "(c)ancel. Show disk (i)nfo"
echo -n "(/dev/XXXy, c, i) ->"
read ASK
case $ASK in
  /dev/????)
	  echo "Install root to ${ASK}? (Warning, ${ASK} will be completely erased!!!)"
	  echo -n "(yes,No) ->"
	  read WARN
	  case $WARN in
	    yes)
	      ROOT_PART=$ASK
	      CONTINUE=1
	      install_root_and_boot_part2
	      ;;
	    *)
	      echo "Selection aborted!"
	  esac
	  ;;
  i)
	    show_info
	    ;;
  c)
	    CONTINUE=1
	    ;;
  *)	
	  echo "Wrong parameter, please try again!"
esac
done
}

install_root_and_boot_part2()
{
CONTINUE=0
while `test $CONTINUE -eq 0` 
do
echo "Enter boot partition device name:"
echo "(c)ancel. Show disk (i)nfo"
echo -n "(/dev/XXXy, c, i) ->"
read ASK
case $ASK in
  /dev/????)
	  echo "Install extlinux to ${ASK}? (Warning, ${ASK} will be completely erased!!!)"
	  echo -n "(yes,No) ->"
	  read WARN
	  case $WARN in
	    yes)
	      BOOT_PART=$ASK
	      CONTINUE=1
	      install_root
	      install_boot
	      ;;
	    *)
	      echo "Selection aborted!"
	  esac
	  ;;
  i)
	    show_info
	  ;;
  c)
	    CONTINUE=1
	    ;;
  *)	
	  echo "Wrong parameter, please try again!"
esac
done
}

install_root()
{
  echo "Installing filesystem to root partition: $ROOT_PART.."
  echo "Note: The disk image system_13.0.tar.gz must be in the current working directory!" 
  ls -l ./system_13.0.tar.gz || exit
  /sbin/mkfs -t ext4 $ROOT_PART || exit
  mount $ROOT_PART $TMPDIR/install || exit
  echo "Copying system to $ROOT_PART.."
  tar --checkpoint=.10000 -xpzf ./system_13.0.tar.gz -C $TMPDIR/install || exit
#  cp ./system_13.0.tar.gz $TMPDIR/install

# Install lilo to superblock 
   echo -n "Installing lilo to superblock in $ROOT_PART"
   /sbin/lilo -v -b $ROOT_PART -r $TMPDIR/install || exit

  umount $TMPDIR/install || exit
  /sbin/e2label $ROOT_PART SW_13.0 || exit
  echo "Root installed to partition: $ROOT_PART"

  sync
}

install_boot()
{
  echo "Installing extlinux to $BOOT_PART"
  mount $ROOT_PART $TMPDIR/install || exit
  /sbin/mkfs -t ext2 $BOOT_PART || exit
  mount $BOOT_PART $TMPDIR/boot || exit
  cp -a  $TMPDIR/install/extlinux $TMPDIR/boot || exit
  /sbin/extlinux --install $TMPDIR/boot/extlinux || exit
  umount $TMPDIR/boot || exit
  umount $TMPDIR/install || exit
  echo "Extlinux installed to: $BOOT_PART"
  DISK=`echo $BOOT_PART | cut -c -8`
  echo "Install new MBR on disk ${DISK}? Warning, the MBR on disk $DISK will be overwritten, make sure no other operating systems are installed on this disk!"
	  echo -n "(y/N)"
	  read ASK
	  case $ASK in
	    y)
	      echo "Installing new MBR on $DISK, make sure that $BOOT_PART is active"
	      cat /usr/lib/syslinux/mbr.bin > $DISK 
	      ;;
	    *)	
	      echo "No changes made to the MBR!"
	  esac

  sync

}

backup_part()
{
CONTINUE=0
while `test $CONTINUE -eq 0` 
do
echo "Enter device name for partition to backup:"
echo "(c)ancel. Show disk (i)nfo"
echo -n "(/dev/XXXy, c, i) ->"
read ASK
case $ASK in
  /dev/????)
	  BACKUP_PART=$ASK
	  CONTINUE=1
	  backup_part2
	  ;;
  i)
	  show_info
	  ;;
  c)
	  CONTINUE=1
	  ;;
  *)	
	  echo "Wrong parameter, please try again!"
esac
done
}

backup_part2()
{
CONTINUE=0
while `test $CONTINUE -eq 0` 
do
echo "Enter device name for partition to store the backup on:"
echo "(c)ancel. Show disk (i)nfo"
echo -n "(/dev/XXXy, c, i) ->"
read ASK
case $ASK in
  /dev/????)
	  BACKUP_DATE=`date -I`
	  DISK_NAME=`basename $BACKUP_PART`
	  BACKUP_NAME=${DISK_NAME}-${BACKUP_DATE}.tar.gz
	  echo "Store backup of partition $BACKUP_PART on partition ${ASK}? The backup will be stored as $BACKUP_NAME"
	  echo -n "(yes,No) ->"
	  read WARN
	  case $WARN in
	    yes)
	      BACKUP_STORE_PART=$ASK
	      CONTINUE=1
	      mount $BACKUP_PART $TMPDIR/backup || exit
	      mount $BACKUP_STORE_PART $TMPDIR/backup-store || exit
	      cd $TMPDIR/backup || exit
	      tar -cvpszf $TMPDIR/backup-store/${BACKUP_NAME} . || exit
	      cd - || exit
	      umount $TMPDIR/backup || exit
	      umount $TMPDIR/backup-store || exit
	      ;;
	    *)
	      echo "Aborted!"
	  esac
	  ;;
  i)
	    show_info
	  ;;
  c)
	    CONTINUE=1
	    ;;
  *)	
	  echo "Wrong parameter, please try again!"
esac
done
}

while true 
do
echo
echo "*** Welcome to slackdeploy! ***"
echo "Author: Magnus Uppman <magnus_uppman@hotmail.com>"
echo
  echo "Install with two partitions (b)oot and root, or only (r)oot partition?"
  echo "Show disk (i)nfo, Edit (d)isk"
  echo "Reinstall (l)ilo to MBR, Reinstall lilo to (s)uperblock"
  echo "Reinstall (p)lain MBR to boot the active partition"
  echo "Backup par(t)ition"
  echo "(q)uit, s(h)ell, (v)ersion"
  echo -n "(i,d,l,s,p,b,r,t,q,h,v) ->"
  read ASK
  case $ASK in
    i)
	    show_info
	    ;;
    d)
	    edit_disk
	    ;;
    l)
	    reinstall_lilo_mbr
	    ;;
    s)
	    reinstall_lilo_superblock
	    ;;
    p)
	    reinstall_plain_mbr
	    ;;
    b)
	    install_root_and_boot_part
	    ;;
    r)
	    install_root_part
	    ;;
    t)
	    backup_part
	    ;;
    v)
	    echo "Version: "$VERSION
	    ;;
    q)
	    exit
	    ;;
    h) 	    
	    /bin/sh
	    ;;
    *)			  
	  echo "Wrong parameter, please try again!"
  esac

  echo -n "Press Enter to continue.."
  read ASK

done