#!/bin/bash

################################################################################
#                                                                              #
#  Copyright (C) 2012 Jack-Benny Persson <jack-benny@cyberinfo.se>             #
#                                                                              #
#   This program is free software; you can redistribute it and/or modify       #
#   it under the terms of the GNU General Public License as published by       #
#   the Free Software Foundation; either version 2 of the License, or          #
#   (at your option) any later version.                                        #
#                                                                              #
#   This program is distributed in the hope that it will be useful,            #
#   but WITHOUT ANY WARRANTY; without even the implied warranty of             #
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the              #
#   GNU General Public License for more details.                               #
#                                                                              #
#   You should have received a copy of the GNU General Public License          #
#   along with this program; if not, write to the Free Software                #
#   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA  #
#                                                                              #
################################################################################

###############################################################################
#                                                                             #
# A simple tape backup script to be run in a cronjob. The script check if     #
# there is a tapedrive connected to the machine, if it has a tape inserted    #
# and if one if these fails, it will recheck over and over again until both a #
# tape and a tapedrive is present.                                            #
# See the README for more information!                                        #
#                                                                             #
###############################################################################

VERSION="Version 1.0"
AUTHOR="(c) 2012 Jack-Benny Persson (jack-benny@cyberinfo.se)"

### Config options ###
INCREMENTAL="yes" # yes/no (yes for incremental backup, no for full backup)
BACKUP_DIRS="/mnt/raid1/backups/weekly"
DRIVE="/dev/st0"
SNAPSHOT="/mnt/raid1/backups/weekly.snar" # If using incremental backups
RECHECK_WAIT=1200
MT="/bin/mt"
TAR="/bin/tar"
CP="/bin/cp"


### Functions ###

Do_full_backup()
{
  sleep 15 # Wait for the tape to get ready...
  echo "Doing backup..."
  ${TAR} -cf ${DRIVE} ${BACKUP_DIRS}
  echo "Backup done!"
  sleep 120 #Just in case wait for the tape drive
  ${MT} -f ${DRIVE} offline
  exit 0
}

Do_inc_backup()
{
  ${CP} ${SNAPSHOT} ${SNAPSHOT}\-1
  sleep 15 # Wait for the tape to get ready...
  echo "Doing backup..."
  ${TAR} -cf ${DRIVE} -g ${SNAPSHOT}\-1 ${BACKUP_DIRS}
  echo "Backup done!"
  sleep 120 #Just in case wait for the tape drive
  ${MT} -f ${DRIVE} offline
  exit 0
}

Check_drive()
{
  if [ ! -e ${DRIVE} ]; then
    echo "No drive connected" >&2
    EXIT_STATE=1

  elif [ -e ${DRIVE} ]; then   
       echo "Drive connected, continuing"
    EXIT_STATE=0 
  
  else
       echo "Unknown error" >&2
    EXIT_STATE=3
  fi
}

Check_tape()
{
TESTTAPE=`${MT} -f ${DRIVE} status | grep DR_OPEN`
  if [ $? == 0 ]; then
    echo "No tape in drive" >&2
    EXIT_STATE=1

  else
    echo "Tape seems to be in the drive, continuing"
    EXIT_STATE=0   
  fi
}

Check_utils()
{
if [ ! -x ${MT} ]; then
  echo "Program ${MT} dosen't seem to exist..." >&2
  exit 1
fi

if [ ! -x ${TAR} ]; then
  echo "Program ${TAR} doesn't seem to exist..." >&2
  exit 1
fi
}

Check_backupdir()
{
for dirtest in ${BACKUP_DIRS}; do
  if [ ! -d $dirtest ]; then
  echo "${dirtest} does not exist" >&2
  exit 1
  fi
done
}

Check_snapshot()
{
if [ ! -e ${SNAPSHOT} ]; then
  echo "${SNAPSHOT} does not exist" >&2
  exit 1
fi
}
  

### Main routine ###

# Check if the utils/programs exists
Check_utils

# Check if the dirs we want to backup exists
Check_backupdir

# Check if the snapshot file exist (if running incremental backup)
if [ ${INCREMENTAL} == "yes" ]; then
Check_snapshot
fi

# Check and repeat until the tape drive is on/connected
EXIT_STATE=3
while [ ${EXIT_STATE} != 0 ]; do
  Check_drive
    if [ ${EXIT_STATE} != 0 ]; then
      sleep ${RECHECK_WAIT}
    fi
done

# Check and repeat until a tape is inserted
EXIT_STATE=3
while [ ${EXIT_STATE} != 0 ]; do
  Check_tape
    if [ ${EXIT_STATE} != 0 ]; then
      sleep ${RECHECK_WAIT}
    fi
done

# Finally, begin with the backup!
if [ ${INCREMENTAL} == "yes" ]; then
  Do_inc_backup

  elif [ ${INCREMENTAL} == "no" ]; then
      Do_full_backup

  else
      echo "I don't know what to do, enter yes or no in $INCREMENTAL" >&2
      exit 1
fi