#!/bin/bash
#Copyright Richard Heck 2006-07
#
#This script is released under the General Public License
#
#Version 0.2
#Added options for unattended operation
#
#TODO:
#Need to add an option for setting raw read or raw-subchan read

CDRDAO=`which cdrdao 2>/dev/null`;
if [ -z "$CDRDAO" ]; then 
  echo "This script uses the cdrdao utility. It does not appear to be in your path.";
  exit 1;
fi

READ_DEVICE=$(lsscsi | grep PIONEER | cut -c54-);
WRITE_DEVICE=$(lsscsi | grep PLEXTOR | cut -c54-);
DRIVER="generic-mmc";

PROG=`basename $0`;

function printUsage {
cat <<EOF
#######################################################################
USAGE: $PROG [-h] [-r READDEVICE] [-w WRITEDEVICE] [FILENAME]
$PROG copies the cd in READDEVICE to one in WRITEDEVICE. The user can
choose the speed at which the copy is made, and multiple copies can be
made from a single read.

Argument:
FILENAME: The filename to use for storage of the CD image, toc, etc. 
Defaults to "cd".

Options:
-d: Driver to use with cdrdao [defaults to generic-mmc]
-D: Show default drivers and exit.
-h: Print this message
-p: Set cdrdao --paranoia-mode [defaults to 0]
-r: Set the read device [defauts to $READ_DEVICE]
-R: --read-raw
-s: Set write speed [defaults to 1x]
-w: Set the write device [defauts to $WRITE_DEVICE]
-y: Assume yes (skip prompts).
The mentioned defaults may not make sense for your system. They can be
changed in $0.
EOF
}

#Option processing
OPTIONS="";
VERBOSE="";
DEBUG="";
SPEED="";
YES="";
PARANOIA="0";
RAW="";

while getopts ":d:Dp:r:Rs:w:y" opt; do
  case $opt in
    d )   DRIVER="$OPTARG";;
    D )   echo "Read: $READ_DEVICE";
          echo "Write: $WRITE_DEVICE";
          exit 0;;
    p )   PARANOIA="$OPTARG";;
    r )	  READ_DEVICE="$OPTARG";;
    R )   RAW="--read-raw";;
    s )   SPEED="$OPTARG";;
    w )   WRITE_DEVICE="$OPTARG";;
    y )   YES="YES";;
    \? )  printUsage; exit 1;;
  esac
done

shift $(($OPTIND - 1));

if [ -z $1 ]; then
 filename=cd;
else
 filename=$1;
fi

if [ -z "$YES" ]; then
  echo "Insert CD into DVD drive...";
  read TEMP;
fi

if [ -f /tmp/$filename.toc ]; then rm -f /tmp/$filename.toc; fi
if [ -f /tmp/$filename.bin ]; then rm /tmp/$filename.bin; fi

echo "Reading from $READ_DEVICE, mode $PARANOIA";

cdrdao read-cd --paranoia-mode $PARANOIA $RAW --datafile /tmp/$filename.bin --device $READ_DEVICE --driver $DRIVER /tmp/$filename.toc

if [ "$?" != "0" ]; then
	echo "Disk read failed!!";
	exit 1;
fi

if [ -z "$SPEED" ]; then 
	if [ -z "$YES" ]; then
		echo -n "CD Copied. Insert blank disk into Plextor drive and ";
		echo "Select speed for write..."
		SPEED="";
		while [ -z "$SPEED" ]; do
			select SPEED in 1 4 8 16; do
				if [ $SPEED ]; then
					break;
				else
					SPEED=1;
					break;
				fi
			done
		done
	else
		SPEED="1";
	fi 
fi

WRITE="Yes";
while [ "$WRITE" = "Yes" ]; do
	echo "Writing to $WRITE_DEVICE at speed $SPEED";
	
	cdrdao write -n --eject --device $WRITE_DEVICE --speed $SPEED --driver $DRIVER /tmp/$filename.toc
	
	if [ "$?" != "0" ]; then
		echo "Disk write failed!!";
		exit 1;
	fi

	echo "Write another?"
	select WRITE in "Yes" "No"; do
		break;
	done
done

rm /tmp/$filename.bin /tmp/$filename.toc;