Subversion Repositories livecd

#!/bin/bash

#

################################################################

#

# Simple GUI for livecd-install 

#

# Urs beyerle, PSI

#

################################################################

#

  VERSION=0.1

#

################################################################

clean_exit()

{

  rm -rf $TMPDIR

}

trap "clean_exit" EXIT

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

### Definitions

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

### set path and lang

PATH="/sbin:/bin:/usr/sbin:/usr/bin:/usr/X11R6/bin"

LANG=en_US.UTF-8

### title

TITLE="LiveCD Install GUI version $VERSION"

MENU_TITLE="LiveCD Install GUI version $VERSION"

### tmpdir

TMPDIR=/tmp/livecd-install-.$$

TMP=$TMPDIR/dialog

mkdir -p $TMPDIR

### dialog or xdialog?

DIALOG="dialog"

XDIALOG_HIGH_DIALOG_COMPAT=1

export XDIALOG_HIGH_DIALOG_COMPAT

[ -n "$DISPLAY" ] && [ -x /usr/bin/Xdialog ] && DIALOG="Xdialog"; XDIALOG="yes"

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

### Functions

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

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

about_message() {

    ABOUT_MESSAGE=$( cat <<EOF

-------------------------------------------\n

 $TITLE\n

-------------------------------------------\n

 (C) 2007, Urs Beyerle, PSI\n

\n

 Press OK to continue\n

EOF)

    $DIALOG --title "$TITLE" --infobox "$ABOUT_MESSAGE" 15 60 8

    [ $? -ne 0 ] && exit

}

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

test_for_livecd_install_script() {

    LIVECD_INSTALL=$( which livecd-install 2>/dev/null )

    if [ ! "$LIVECD_INSTALL" ]; then

	MESSAGE0="Script livecd-install not found"

	$DIALOG --title "$TITLE" --msgbox "$MESSAGE0" 0 0

	exit 1

    fi

}

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

test_for_root() {

    if [ "$UID" -ne "0" ]; then

	MESSAGE0="To use this program, you need to be root"

	$DIALOG --title "$TITLE" --msgbox "$MESSAGE0" 0 0

	exit 1

    fi

}

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

select_install_part() {

    PART_LIST=""

    PARTS=$( LANG=C fdisk -l | grep Linux$ | cut -d" " -f1 )

    if [ ! "$PARTS" ]; then

	NOPART_MESSAGE="No Linux partition found.\n\n\

Please create one first" 

	$DIALOG --title "$TITLE" --infobox "$NOPART_MESSAGE" 15 60 8

	exit 1

    fi

    for p in $PARTS; do

	PART_LIST="$PART_LIST $p Linux-Partiton off"

    done

    MESSAGE1="Please select a partition to install the LiveCD"

    $DIALOG --title "$TITLE" --no-cancel --radiolist "$MESSAGE1" 20 60 8 $PART_LIST 2>$TMP

    cat $TMP

}

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

select_swap_part() {

    PART_LIST=""

    PARTS=$( LANG=C fdisk -l | grep "Linux swap" | cut -d" " -f1 )

    if [ ! "$PARTS" ]; then

	NOSWAP_MESSAGE="No Swap partition found.\n\n\

You can install the LiveCD without having\n\

a Swap partition, but it is not recommended.\n\n\

Continue ?" 

	$DIALOG --title "$TITLE" --yesno "$NOSWAP_MESSAGE" 15 60 8

	return $?

    fi

    for p in $PARTS; do

	PART_LIST="$PART_LIST $p Swap-Partiton off"

    done

    MESSAGE2="Please select a SWAP partition"

    $DIALOG --title "$TITLE" --no-cancel --radiolist "$MESSAGE2" 15 60 5 $PART_LIST \

    NONE "Do not use a swap partition (not recommended)" off 2>$TMP

    a=$( cat $TMP )

    [ "$a" != "NONE" ] && echo $a

    return 0

}

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

ok_continue() {

    OK_MESSAGE="\nLiveCD will be installed on $INSTALL_PART.\n\

All data on $INSTALL_PART will be deleted !!\n\n"

    if [ $SWAP_PART ]; then

	OK_MESSAGE="${OK_MESSAGE}$SWAP_PART will be used as Swap partition.\n\n"

    fi

    if [ $MBR_DISK ]; then

	OK_MESSAGE="${OK_MESSAGE}GRUB will be installed as bootloader into the\n\

Master Boot Record (MBR) of disk $MBR_DISK.\n\n"

    fi

    OK_MESSAGE="${OK_MESSAGE}Is this ok ?\n"

    $DIALOG --title "$TITLE" --yesno "$OK_MESSAGE" 15 60 8

    return $?

}

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

select_mbr_disk() {

    DISK_LIST=""

    DISKS=$( LANG=C fdisk -l | grep ^Disk | cut -d":" -f1 | cut -d" " -f2 )

    for p in $DISKS; do

	DISK_LIST="$DISK_LIST $p Disk off"

    done

    MESSAGE3="Please select a disk to install the bootloader GRUB"

    $DIALOG --title "$TITLE" --no-cancel --radiolist "$MESSAGE3" 15 60 5 $DISK_LIST \

    NONE "Do not install GRUB (not recommended)" off 2>$TMP

    a=$( cat $TMP )

    [ "$a" != "NONE" ] && echo $a

}

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

run_qtparted() {

    QTPARTED=$( which qtparted 2>/dev/null )

    if [ ! $QTPARTED ]; then

	NO_QTPARTED_MESSAGE="Sorry, qtparted not found.\n\n\

Please use an other tool like fdisk or parted."

	$DIALOG --title "$TITLE" --infobox "$NO_QTPARTED_MESSAGE" 10 60 8

    else

	$QTPARTED &

    fi

}

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

run_livecd_install() {

    xterm -sb -title "Run: livecd-install -y $SWAP_PART_OPT $MBR_DISK_OPT $INSTALL_PART" \

          -geometry 130x45+100+20 \

          -e 'echo; \

	      echo "Run: livecd-install -y $SWAP_PART_OPT $MBR_DISK_OPT $INSTALL_PART"; \

	      echo; \

	      echo "livecd-install finished.";\

	      echo;\

	      echo "- Press a key to close this window -"; \

	      read -n 1'

}

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

main_menu() {

    MENU_1="Install the LiveCD"

    MENU_2="Create Linux/Swap Partitions with qtparted"

    MENU_3="About"

    MENU_4="Quit"

    while true; do

	$DIALOG --title "$TITLE" \

	    --menu "$MENU_TITLE" 15 60 6\

	    1 "$MENU_1" \

	    2 "$MENU_2" \

	    3 "$MENU_3" \

	    4 "$MENU_4" \

	    2> $TMP

	[ $? -ne 0 ] && break

	CHOICE=$(cat $TMP)

	case "$CHOICE" in

         1)

           INSTALL_PART=$( select_install_part )

	   if [ "$?" = "0" ]; then

	       SWAP_PART=$( select_swap_part )

	       if [ "$?" = "0" ]; then

		   MBR_DISK=$( select_mbr_disk )

		   ok_continue && break 

	       fi

	   fi 

	   INSTALL_PART="" ;;

	 2)

           run_qtparted ;;

         3)

           about_message ;;

         4)

           exit ;;

        esac

done

}

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

### Main Program

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

### requirements

test_for_livecd_install_script

test_for_root

### main menu

main_menu

[ $SWAP_PART ] && export SWAP_PART_OPT="-swap $SWAP_PART"

[ $MBR_DISK ]  && export MBR_DISK_OPT="-mbr $MBR_DISK"

export INSTALL_PART=$INSTALL_PART

### run livecd-install

if [ $INSTALL_PART ]; then

    run_livecd_install

fi