locker

#!/bin/sh

show_help() {
    cat <<EOF
Usage: locker [options]
Options:
    --production    Run in production mode (default: development mode)

EOF
}

locker_mode=development

# Parse command line options
while true; do
    case "$1" in
      -h|--help|-\?) show_help; exit 0;;
      --production) locker_mode=production; shift ;;
      --development) shift ;;
      --) shift; break;;
      -*) echo "invalid option: $1" 1>&2; show_help; exit 1;;
      *)  break;;
    esac
done

# If we were installed via lockerbox, find our dependencies there
lockerbox_env=../lockerbox_environment.sh
if [ -f "$lockerbox_env" ]; then
    . "$lockerbox_env"
fi

# Make sure we have some API keys, otherwise we can't do much
apikeys=Config/apikeys.json
if [ ! -f "$apikeys" ]; then
    echo "No API keys found in $apikeys" >&2
    echo "see https://github.com/LockerProject/Locker/wiki/GettingAPIKeys" >&2
    exit 1
fi

cleanup_mongo() {
    mongodata=Me/mongodata
    mongolock=$mongodata/mongod.lock
    if [ -s $mongolock ]; then
        echo "mongod lock file exists: $mongolock..."
        pid=$(cat "$mongolock")
        if [ "$locker_mode" = "production" ]; then
            # Shut it down
            echo "Shutting down mongod."
            kill "$pid"
            while kill -0 "$pid" >/dev/null 2>&1; do
                echo "Waiting for mongod to exit..."
                sleep 1
            done
            echo "mongod shut down."
        else
            # Automatically check and repair mongo
            if [ -n "$pid" ] && kill -0 "$pid" 2>/dev/null; then
                echo "...and it's still running.  Hopefully that's OK."
            else
                echo "...but it's dead.  Cleaning up..."
                rm -f "$mongolock"
                if ! mongod --dbpath "$mongodata" --repair; then
                    echo "mongod --repair failed!" >&2
                    exit 1
                fi
            fi
        fi
    fi
}

monitor() {
    # Monitor the process and restart it if it dies

    trap shutdown INT TERM
    while true; do
        start_time=$(date +%s)
        # start the command
        $* &
        lockerd_pid=$!

        if wait $lockerd_pid; then
            # Clean shutdown
            exit 0
        fi

        # Abnormal exit
        cleanup_mongo

        end_time=$(date +%s)
        runtime=$(($end_time - $start_time))
        if [ "$runtime" -lt 10 ]; then
            echo "locker was only running for $runtime seconds, not respawning" >&2
            exit 1
        fi

        wait=5
        echo "Locker exited unexpectedly, respawning in $wait seconds..." >&2
        sleep $wait
    done
}

shutdown() {
    kill -TERM $lockerd_pid
    wait $lockerd_pid
    status=$?
    echo "locker exited with status $status"
    cleanup_mongo
    exit $status
}

cleanup_mongo

# Extra paths to search for files : separated
export NODE_PATH=Common/node

# Start the locker
case "$locker_mode" in
    production)
        # Tell node to run in production mode
        NODE_ENV=production
        export NODE_ENV
        monitor node lockerd.js "$@"
        ;;
    development|*)
        # Just run it
        exec node lockerd.js "$@"
        ;;
esac

exit 1