#!/bin/bash

# -------------------------------------------------------------------------- #
# Copyright 2002-2015, OpenNebula Project (OpenNebula.org), C12G Labs        #
#                                                                            #
# Licensed under the Apache License, Version 2.0 (the "License"); you may    #
# not use this file except in compliance with the License. You may obtain    #
# a copy of the License at                                                   #
#                                                                            #
# http://www.apache.org/licenses/LICENSE-2.0                                 #
#                                                                            #
# Unless required by applicable law or agreed to in writing, software        #
# distributed under the License is distributed on an "AS IS" BASIS,          #
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   #
# See the License for the specific language governing permissions and        #
# limitations under the License.                                             #
#--------------------------------------------------------------------------- #


if [ -z "$ONE_LOCATION" ]; then
    SUNSTONE_PID=/var/run/one/sunstone.pid
    SUNSTONE_SERVER=/usr/lib/one/sunstone/sunstone-server.rb
    SUNSTONE_LOCK_FILE=/var/lock/one/.sunstone.lock
    SUNSTONE_LOG=/var/log/one/sunstone.log
    SUNSTONE_LOG_ERROR=/var/log/one/sunstone.error
    SUNSTONE_CONF=/etc/one/sunstone-server.conf
    NOVNC_SERVER=/usr/bin/novnc-server
else
    SUNSTONE_PID=$ONE_LOCATION/var/sunstone.pid
    SUNSTONE_SERVER=$ONE_LOCATION/lib/sunstone/sunstone-server.rb
    SUNSTONE_LOCK_FILE=$ONE_LOCATION/var/.sunstone.lock
    SUNSTONE_LOG=$ONE_LOCATION/var/sunstone.log
    SUNSTONE_LOG_ERROR=$ONE_LOCATION/var/sunstone.error
    SUNSTONE_CONF=$ONE_LOCATION/etc/sunstone-server.conf
    NOVNC_SERVER=$ONE_LOCATION/bin/novnc-server
fi

setup()
{
  TMPDIR=$(grep ^:tmpdir: $SUNSTONE_CONF|awk '{print $2}')

  if [ -n "$TMPDIR" ]; then
    export TMPDIR
    mkdir -p $TMPDIR
  fi

  if [ -f $SUNSTONE_LOCK_FILE ]; then
    if [ -f  $SUNSTONE_PID ]; then
      SUNSTONEPID=`cat $SUNSTONE_PID`
      ps $SUNSTONEPID &> /dev/null
      if [ $? -eq 0 ]; then
        echo -n "Sunstone Server is still running (PID:$SUNSTONEPID). Please "
        echo "try 'sunstone-server stop' first."
        exit 1
      fi
    fi
    echo "Stale .lock detected. Erasing it."
    rm $SUNSTONE_LOCK_FILE
  fi
}


start()
{
  if [ "$1" = "-n" ]; then
    novnc=1
  fi

  if [ ! -f "$SUNSTONE_SERVER" ]; then
    echo "Cannot find $SUNSTONE_SERVER."
    exit 1
  fi

  if [ "$novnc" = "1" ]; then
    # Start novnc server
    $NOVNC_SERVER start

    if [ "$?" != "0" ]; then
      echo "Could not start novnc server" 1>&2
    fi
  fi

  # Start the sunstone daemon
  touch $SUNSTONE_LOCK_FILE

  if [ "$?" != "0" ]; then
    echo "Error creating lock file" 1>&2

    $NOVNC_SERVER stop
    exit 1
  fi

  ruby $SUNSTONE_SERVER > $SUNSTONE_LOG 2>$SUNSTONE_LOG_ERROR &
  LASTPID=$!

  if [ $? -ne 0 ]; then
    echo "Error executing sunstone-server."
    echo "Check $SUNSTONE_LOG_ERROR and $SUNSTONE_LOG for more information"
    [ "$novnc" = "1" ] && $NOVNC_SERVER stop
    exit 1
  else
    echo $LASTPID > $SUNSTONE_PID
  fi

  sleep 1
  ps $LASTPID &> /dev/null

  if [ $? -ne 0 ]; then
    echo "Error executing sunstone-server."
    echo "Check $SUNSTONE_LOG_ERROR and $SUNSTONE_LOG for more information"
    [ "$novnc" = "1" ] && $NOVNC_SERVER stop
    exit 1
  fi

  echo "sunstone-server started"
}

#
# Function that stops the daemon/service
#
stop()
{
  if [ "$1" = "-n" ]; then
    novnc=1
  fi

  # Stop novnc server
  [ "$novnc" = "1" ] && $NOVNC_SERVER stop

  if [ ! -f $SUNSTONE_PID ]; then
    echo "Couldn't find sunstone-server process pid."
    exit 1
  fi

  # Kill the sunstone daemon
  kill -INT `cat $SUNSTONE_PID` &> /dev/null

  # Remove pid files
  rm -f $SUNSTONE_LOCK_FILE &> /dev/null
  rm -f $SUNSTONE_PID &> /dev/null

  echo "sunstone-server stopped"
}

case "$1" in
  start)
    setup
    start -n
    ;;
  stop)
    stop -n
    ;;
  restart)
    stop -n
    setup
    start -n
    ;;
  start-sunstone)
    setup
    start
    ;;
  stop-sunstone)
    stop
    ;;
  restart-sunstone)
    stop
    setup
    start
    ;;
  *)
    echo "Usage: sunstone-server {start|stop|restart|start-sunstone|stop-sunstone|restart-sunstone}" >&2
    exit 3
    ;;
esac
