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

### define name of LiveCD install script

LIVECD_INSTALL=livecd-install

### title and script name

TITLE="LiveCD Install GUI version $VERSION"

MENU_TITLE="LiveCD Install GUI version $VERSION"

SCRIPTNAME=$( basename $0 )

### tmpdir

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

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

  $SCRIPTNAME\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

}

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

help_message() {

    HELP_MESSAGE=$( cat <<EOF

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

     $SCRIPTNAME\n

     $TITLE\n

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

\n

This is the frontend of the $LIVECD_INSTALL script.

Use it to install the LiveCD/DVD on your hardisk.

\n\n

The LiveCD/DVD can be installed on a Linux

partition, which you may have to create first using tools

like qtparted, gparted, fdisk or parted. All data on the selected

Linux partition will be deleted. 

It is recommended to

create in addition a Swap partition that will be later

used by the installed Linux system.

\n\n

In order that you can boot into the installed Linux

system, it is recommended that you choose to install

the bootloader GRUB into the Master Boot Record (MBR)

of your primary boot disk.

\n\n

More options are available, if you run the script 

$LIVECD_INSTALL directly from the command line.

\n

EOF)

    $DIALOG --title "$TITLE" --infobox "$HELP_MESSAGE" 30 60 12

}

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

test_for_livecd_install_script() {

    LIVECD_INSTALL_SCRIPT=$( which $LIVECD_INSTALL 2>/dev/null )

    if [ ! "$LIVECD_INSTALL_SCRIPT" ]; 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" --default-no --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; \

	      $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="Help"

    MENU_4="About"

    MENU_5="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" \

	    5 "$MENU_5" \

	    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)

           help_message ;;

         4)

           about_message ;;

         5)

           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

export LIVECD_INSTALL=$LIVECD_INSTALL

### run $LIVECD_INSTALL

if [ $INSTALL_PART ]; then

    run_livecd_install

fi