#!/bin/bash

#This is a script that acts a a sleep timer or basic alarm.

#It is written for crunchbang linux installs--but should move

#easily to most any linux distributions.

#For most users you'll want to edit the series of commands

#below to use whatever programs/utilities you favor

######DEFAULT COMMANDS#####

#The path to the alarm noise (e.g. a beep, song, etc.)

NOISE=~/music/WhatIGot.m4a

#Program for playing the alarm

#Screen locking command

SCREENLOCK="xscreensaver-command -lock"

#The suspending command

SUSPEND="pm-suspend"

#The hibernation command

HIBERNATE="pm-hibernate"

#Command or series of commands for shutting down

SHUTDOWN="gdm-control --shutdown && openbox --exit"

#Common use scripts (1st a backup script) You can

#add more here and then duplicate lines 147-9, and edit accordingly.

BACKUPSCRIPT="/home/jake/bin/remotebackup.sh"

########SCRIPT GENERAL INPUTS#####

#Select action to take

PS3="Choose:"

echo ""

echo "What action would you like to take at the end?"

select act in "suspend" "hibernate" "lock screen" "shutdown" "play alarm"

do

	break

done

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

#Translating the "acts" to  proper commands

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

if [ "$act" = "suspend" ]; then

	actcommand=$SUSPEND

fi

if [ "$act" = "hibernate" ]; then

	actcommand=$HIBERNATE

fi

if [ "$act" = "lock screen" ]; then

	actcommand=$SCREENLOCK

fi

if [ "$act" = "shutdown" ]; then

	actcommand=$SHUTDOWN

fi

if [ "$act" = "play alarm" ]; then

	actcommand=$ALARM

fi

##Asking for passwords for those commands which need root privileges

if [ "$actcommand" = "$SUSPEND" ] || [ "$actcommand" = "$HIBERNATE" ]; then

	echo "Selected action requires root privileges."

	stty_orig=`stty -g`

	stty -echo

	read -p "Enter sudo password:" pass

	stty $stty_orig

fi

#Select when you would like to do it

PS3="Choose: "

echo ""

echo "When would you like to do it?"

select when in "after time" "after task" "whichever is later"

do

	break

done

echo ""

#Show the current alarm path

if [ "$act" = "play alarm" ]; then

	echo "Current path of alarm is set to $NOISE"

	echo ""

fi

#For 'whens' that require a time input

if [ "$when" = "after time" ] || [ "$when" = "whichever is later" ]; then

	read -p "How long of a delay (in minutes): " minutes

	echo ""

	sectime=`date +%s`

	secoff=$(($sectime+($minutes*60)))

	dateoff=`date --date=@$secoff`

fi

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

#####The  After Time Routine

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

if [ "$when" = "after time" ]; then

	#The Main Timer

	elapsedtime=1

	echo ""

	echo ""

	echo "Computer will $act at $dateoff."

	echo -n "Minutes left..."

	echo -n "$minutes..."

	while [ $elapsedtime -lt $minutes ]

		do

		    sleep 1m

		    echo -n "$(($minutes-$elapsedtime)).."

    		elapsedtime=$(($elapsedtime+1))

		done

	#The last 60 seconds....

	echo ""

	echo ""

	if [ $minutes -gt 0 ]; then

		echo "Will $act in (seconds)..."

		lastsixty=1

		while [ $lastsixty -le 60 ]

			do

	    		sleep 1s

	   			 echo -n "$((60-$lastsixty))..."

	    		lastsixty=$(($lastsixty+1))

			done

	fi

	#The Action

	if [ "$actcommand" = "$SUSPEND" ] || [ "$actcommand" = "$HIBERNATE" ]; then

		echo "$pass" | sudo -S $actcommand

		exit 0

		else

			$actcommand

			exit 0

	fi

fi

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

#Inputing the task to follow

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

if [ "$when" = "after task" ] || [ "$when"  = "whichever is later" ]; then

	PS3="Choose: "

	echo ""

	echo "Which task do you want to follow?"

	select task in "Already running task (pid)" "A new task" "Backup script"

	do

		break

	done

	if [ "$task" = "Already running task (pid)" ]; then

		read -p "Enter pid of task" taskpid

	fi

	if [ "$task" = "A new task" ]; then

		taskpid=0

		read -p "Enter (terminal) command to run: " taskcommand

	fi

	if [ "$task" = "Backup script" ]; then

		taskpid=0

		taskcommand=$BACKUPSCRIPT

	fi

fi

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

#####The After Task Routine

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

if [ "$when" = "after task" ]; then

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

		terminator --command="$taskcommand" &

		clear

		taskpid=$!

	fi

	RUNNING=1

	while [ $RUNNING -ge 1 ]

		do

			echo "Task still running"

			sleep 5s

			RUNNING=`ps $taskpid | grep -c $taskpid`

		done

	echo "Task complete...will now $act ."

		if [ "$actcommand" = "$SUSPEND" ] || [ "$actcommand" = "$HIBERNATE" ]; then

		echo "$pass" | sudo -S $actcommand

		exit 0

		else

			$actcommand

			exit 0

		fi

fi

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

#####The Whichever Is Later Routine

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

if [ "$when"  = "whichever is later" ]; then

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

		terminator --command="$taskcommand" &

		taskpid=$!

	fi

	RUNNING=1

	while [ $RUNNING -ge 1 ]

		do

			echo "Task still running..."

			sleep 5s

			RUNNING=`ps $taskpid | grep -c $taskpid`

			clear

		done

	echo ""

	echo "Task complete...comparing with set $dateoff ."

	postbackuptime=`date +%s`

	if [ $secoff -gt $postbackuptime ]; then

		timedifference=$(($secoff-($postbackuptime)))

		echo ""

		echo "Sleeping for $timedifference seconds"

		sleep $timedifference

	fi

	if [ "$actcommand" = "$SUSPEND" ] || [ "$actcommand" = "$HIBERNATE" ]; then

		echo "$pass" | sudo -S $actcommand

		exit 0

		else

			$actcommand

			exit 0

	fi

fi

exit

#end