#!/bin/bash
# /etc/init.d/minecraft
# version 0.3.2 2011-01-27 (YYYY-MM-DD)
# Cusom Edits for Multiple Servers by Brandon 2011-02-19
# http://www.minecraftwiki.net/wiki/Tutorials/Server_startup_script

  ### BEGIN INIT INFO
  # Provides:   minecraft
  # Required-Start: $local_fs $remote_fs
  # Required-Stop:  $local_fs $remote_fs
  # Should-Start:   $network
  # Should-Stop:    $network
  # Default-Start:  2 3 4 5
  # Default-Stop:   0 1 6
  # Short-Description:    Minecraft server
  # Description:    Starts the minecraft server
  ### END INIT INFO

#Settings
SERVICE='minecraft_server.jar'
SCREENID='minecraft'
WORLDNAME='MyMCWorld'
USERNAME="minecraft"
MCPATH='/home/minecraft/server'
INVOCATION="java -Xincgc -Xmx1G -jar $SERVICE"
BACKUPPATH='/home/minecraft/backups'

MESSAGE=$2

ME=`whoami`
as_user() {
  if [ "$ME" == "$USERNAME" ] ; then
    bash -c "$1"
  else
    su - $USERNAME -c "$1"
  fi
}

mc_start() {
  if ps ax | grep -v grep | grep -v -i SCREEN | grep $SERVICE > /dev/null
  then
    echo "Tried to start but $SERVICE was already running!"
  else
    echo "$SERVICE was not running... starting."
    cd $MCPATH
    as_user "cd $MCPATH && screen -dmS $SCREENID $INVOCATION"
    sleep 5
    if ps ax | grep -v grep | grep -v -i SCREEN | grep $SERVICE > /dev/null
    then
      echo "$SERVICE is now running."
    else
      echo "Could not start $SERVICE."
    fi
  fi
}

mc_saveoff() {
        if ps ax | grep -v grep | grep -v -i SCREEN | grep $SERVICE > /dev/null
	then
		echo "$SERVICE is running... suspending saves"
		#as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"say SERVER ENTERING READ-ONLY MODE\"\015'"
		as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"save-off\"\015'"
		as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"save-all\"\015'"
		sync
		sleep 10
	else
                echo "$SERVICE was not running. Not suspending saves."
	fi
}

mc_saveon() {
        if ps ax | grep -v grep | grep -v -i SCREEN | grep $SERVICE > /dev/null
	then
		echo "$SERVICE is running... re-enabling saves"
                as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"save-on\"\015'"
                #as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"say SERVER ENTERING READ-WRITE MODE\"\015'"
	else
                echo "$SERVICE was not running. Not resuming saves."
	fi
}

mc_stop() {
        if ps ax | grep -v grep | grep -v -i SCREEN | grep $SERVICE > /dev/null
        then
                echo "$SERVICE is running... stopping."
                as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"say SERVER SHUTTING DOWN.\"\015'"
				as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"save-off\"\015'"
				as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"save-all\"\015'"
                as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"stop\"\015'"
                sleep 5
        else
                echo "$SERVICE was not running."
        fi
        if ps ax | grep -v grep | grep -v -i SCREEN | grep $SERVICE > /dev/null
        then
                echo "$SERVICE could not be shut down... still running."
        else
                echo "$SERVICE is shut down."
        fi
}

# http://ci.bukkit.org/job/dev-CraftBukkit/Recommended/artifact/target/craftbukkit-0.0.1-SNAPSHOT.jar
mc_update() {
  if ps ax | grep -v grep | grep -v -i SCREEN | grep $SERVICE > /dev/null
  then
    echo "$SERVICE is running! Will not start update."
  else
    MC_SERVER_URL=http://minecraft.net/`wget -q -O - http://www.minecraft.net/download.jsp | grep minecraft_server.jar\</a\> | cut -d \" -f 2`
    as_user "cd $MCPATH && wget -q -O $MCPATH/minecraft_server.jar.update $MC_SERVER_URL"
    if [ -f $MCPATH/minecraft_server.jar.update ]
    then
      if `diff $MCPATH/minecraft_server.jar $MCPATH/minecraft_server.jar.update >/dev/null`
        then 
          echo "You are already running the latest version of $SERVICE."
        else
          as_user "mv $MCPATH/minecraft_server.jar.update $MCPATH/minecraft_server.jar"
          echo "Minecraft successfully updated."
      fi
    else
      echo "Minecraft update could not be downloaded."
    fi
  fi
}

mc_backup() {
   echo "Backing up minecraft world: $WORLDNAME"
   if [ -f $BACKUPPATH/${WORLDNAME}_`date "+%m.%d.%Y"`.tar.gz ]
   then
     for i in 1 2 3 4 5 6
     do
       if [ -f $BACKUPPATH/${WORLDNAME}_`date "+%m.%d.%Y"`-$i.tar.gz ]
       then
         continue
       else
         as_user "cd $MCPATH && cp -r $WORLDNAME $BACKUPPATH/${WORLDNAME}_`date "+%m.%d.%Y"`-$i"
		 as_user "cd $BACKUPPATH; tar cvf ${WORLDNAME}_`date "+%m.%d.%Y"`-$i.tar ${WORLDNAME}_`date "+%m.%d.%Y"`-$i/*"
		 as_user "gzip $BACKUPPATH/${WORLDNAME}_`date "+%m.%d.%Y"`-$i.tar"
         as_user "rm -rf $BACKUPPATH/${WORLDNAME}_`date "+%m.%d.%Y"`-$i"
		 
         break
       fi
     done
   else
     as_user "cd $MCPATH && cp -r $WORLDNAME $BACKUPPATH/${WORLDNAME}_`date "+%m.%d.%Y"`"
	 as_user "cd $BACKUPPATH; tar cvf ${WORLDNAME}_`date "+%m.%d.%Y"`.tar ${WORLDNAME}_`date "+%m.%d.%Y"`/*"
     as_user "gzip $BACKUPPATH/${WORLDNAME}_`date "+%m.%d.%Y"`.tar"
     as_user "rm -rf $BACKUPPATH/${WORLDNAME}_`date "+%m.%d.%Y"`"
     echo "Backed up world: $WORLDNAME"
   fi
   echo "Backing up the minecraft server executable"
   if [ -f "$BACKUPPATH/minecraft_server_`date "+%m.%d.%Y"`.jar" ]
   then
     for i in 1 2 3 4 5 6
     do
       if [ -f "$BACKUPPATH/minecraft_server_`date "+%m.%d.%Y"`-$i.jar" ]
       then
         continue
       else
         as_user "cd $MCPATH && cp $SERVICE \"$BACKUPPATH/minecraft_server_`date "+%m.%d.%Y"`-$i.jar\""
         break
       fi
     done
   else
     as_user "cd $MCPATH && cp $SERVICE \"$BACKUPPATH/minecraft_server_`date "+%m.%d.%Y"`.jar\""
   fi
   echo "Backup complete"
}

mc_savemap() {
        if ps ax | grep -v grep | grep -v -i SCREEN | grep $SERVICE > /dev/null
	then
		echo "$SERVICE is running... saving map"
		as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"save-all\"\015'"
	else
		echo "$SERVICE was not running. Map not saved."
	fi
}

mc_sendmessage() {
        if ps ax | grep -v grep | grep -v -i SCREEN | grep $SERVICE > /dev/null
	then
		echo "$SERVICE is running... sending message"
		as_user "screen -p 0 -S $SCREENID -X eval 'stuff \"say $MESSAGE\"\015'"
	else
		echo "$SERVICE was not running. no message sent."
	fi
}


#Start-Stop here
case "$1" in
  start)
    mc_start
    ;;
  stop)
    mc_stop
    ;;
  restart)
    mc_stop
    mc_start
    ;;
  update)
    mc_stop
    mc_backup
    mc_update
    mc_start
    ;;
  backup)
    mc_saveoff
    mc_backup
    mc_saveon
    ;;
  saveoff)
    mc_saveoff
    ;;
  saveon)
    mc_saveon
    ;;
  save)
    mc_savemap
    ;;
  say)
    mc_sendmessage
    ;;
  status)
    if ps ax | grep -v grep | grep -v -i SCREEN | grep $SERVICE > /dev/null
    then
      echo "$SERVICE is running."
    else
      echo "$SERVICE is not running."
    fi
    ;;

  *)
  echo "Usage: /etc/init.d/minecraft {start|stop|update|backup|status|say|restart}"
  exit 1
  ;;
esac

exit 0