#!/bin/sh

get_list()
{
    for file in /mnt/usb/*
    do
        if [ -f $file ]; then
            name=`echo $file|cut -b10-`
    	    size=`du -b $file|cut -f1`
            printf "%s %d \n" "$name" "$size"
        fi
    done
}

find_usb()
{
if [ ! -e /mnt/usb ]; then
	mkdir /mnt/usb
fi

name=`ls -l /dev/disk/by-id/usb*` 
retval=$?
ch=0
while [ $ch -eq 0 ]; do
case $retval in
	0)
	name=`echo $name | cut -d "/" -f13 | cut -d " " -f1`
	mount /dev/$name /mnt/usb 
	ch=1
	;;
	2)
	dialog --backtitle "WELCOME TO ERIE STRAYER COMPANY" --colors --title "\Zb\Z1 ERROR! " --msgbox "\Z1\Zr\Zb PLEASE TRY AGAIN  Put USB Flash Drive In. \n \n Hit ENTER to Continue or ESC to Exit. " 7 50
	res=$?
	case $res in
	0)
	name=`ls -l /dev/disk/by-id/usb*` 
	retval=$?
	;;
	255)
	hit_esc
	;;
esac	
esac	
done
}

install_update()
{
dialog --backtitle "WELCOME TO ERIE STRAYER COMPANY" --title "INSTALL PROGRAM UPDATE" --nocancel --menu "Select File:" 10 30 4 `get_list` 2>menu.choice
retval=$? 
choice=`cat menu.choice`

case $retval in
	0)
	cp /mnt/usb/$choice /usr/local/sentry/
	cd /usr/local/sentry/
	tar -zxvvf $choice
	;;
	255)
	hit_esc
	;;
esac
}

backup_data()
{
dialog --backtitle "WELCOME TO ERIE STRAYER COMPANY" --title "BACKUP DATA" --nocancel --inputbox "Type Filename:" 10 30 2>tmp
retval=$?
inp=`cat tmp`

case $retval in
	0)
	cd /usr/local/sentry
	tar -zcvvf $inp.tgz DATALIB/*
	cp -avf $inp.tgz /mnt/usb 
	;;
	255)
	hit_esc
	;;
esac
}

restore_data()
{
dialog --backtitle "WELCOME TO ERIE STRAYER COMPANY" --title "RESTORE DATA" --nocancel --menu "Select File:" 10 30 4 `get_list` 2> menu.choice
retval=$? 
choice=`cat menu.choice`

case $retval in
	0)
	cp /mnt/usb/$choice /usr/local/sentry/
	cd /usr/local/sentry/
	tar -zxvvf ./$choice
	;;
	255)
	hit_esc
	;;
esac
}

hit_esc()
{
umount /mnt/usb
dialog --backtitle "WELCOME TO ERIE STRAYER COMPANY" --colors --title "\Zb\Z1 ERROR! " --infobox "\n\Z1\Zr\Zb YOU HIT ESC KEY !!! " 7 50
`killall webupdate`
}

dialog --backtitle "WELCOME TO ERIE STRAYER COMPANY" --colors --title "\Zr\Z1 Please insert USB Flash Drive now! " --nocancel --msgbox "\nIn 5 seconds \nHit ENTER to Continue or ESC for Exit." 7 50
retval=$?
if [ "$retval" -eq "255" ];then
hit_esc
fi

find_usb

while [ -z "$choise" ] ; do
dialog --backtitle "WELCOME TO ERIE STRAYER COMPANY" --title "SELECT ACTIONS:" --separate-output --nocancel --colors --checklist "Arrow Up or Down, \n\Zb\Z3Press Space Bar to SELECT " 10 40 3  \
1 "INSTALL PROGRAM" off \
2 "BACKUP DATA" off \
3 "RESTORE DATA"  off  2>tmp

resp=$?
if [ "$resp" -eq "255" ];then
hit_esc
fi

choise=`cat tmp`

if [ -z "$choise" ]; then
 dialog --backtitle "WELCOME TO ERIE STRAYER COMPANY" --colors --title "\Zb\Z1 ERROR! " --msgbox "\Z1\Zr\Zb YOU DID NOT MAKE ANY SELECTIONS! \n \n Hit ENTER to Continue. ESC to Exit" 7 50
ch=$?
	if [ "$ch" -eq "255" ]; then
	hit_esc
	fi	
fi
done

case $choise in
	1)
	install_update
	;;
	2)
	backup_data
	;; 
	3)
	restore_data
	;;
	255)
	hit_esc
	;;
esac

umount /mnt/usb

dialog --backtitle "WELCOME TO ERIE STRAYER COMPANY" --title "Procedure Complete!" --infobox "\nYou may take out USB Flash Drive now. " 7 50
