#! /bin/sh
#
# Copyright (c) 2010-2011, Couchbase, Inc.
# All rights reserved
#
PATH="/opt/couchbase/bin":$PATH
export PATH

LD_LIBRARY_PATH="/opt/couchbase/lib":"/opt/couchbase/lib/memcached":$LD_LIBRARY_PATH
export LD_LIBRARY_PATH

_check_nofile() {
    if [ `ulimit -n` -lt 10240 ]
    then
        cat <<EOF
The maximum number of open files for the couchbase user is set too low.
It must be at least 10240. Normally this can be increased by adding
the following lines to /etc/security/limits.conf:

couchbase              soft    nofile                  <value>
couchbase              hard    nofile                  <value>

Where <value> is greater than 10240.
EOF
    fi
}

_prepare_datadir() {
    datadir="/opt/couchbase/var/lib/couchbase"

    test -d "$datadir" || mkdir -p "$datadir"
    cd "$datadir"
}

_maybe_start_epmd() {
    # Initialize distributed erlang on the system (i.e. epmd)
    erl -noshell -setcookie nocookie -sname init -run init stop 2>&1 > /dev/null
    if [ $? -ne 0 ]
    then
        exit 1
    fi
}

ERL_LIBS="/opt/couchbase/lib/couchdb/erlang/lib:/opt/couchbase/lib/ns_server/erlang/lib"
export ERL_LIBS

DONT_START_COUCH=1
export DONT_START_COUCH

# Set an ENV variable to force C++ STL and string classes to not use its
# default memory pooling allocator.
# For GCC 3.2.2 and later
GLIBCPP_FORCE_NEW=1
export GLIBCPP_FORCE_NEW
# For GCC 3.4 and later
GLIBCXX_FORCE_NEW=1
export GLIBCXX_FORCE_NEW

PIDFILE="/opt/couchbase/var/lib/couchbase/couchbase-server.pid"
NODEFILE="/opt/couchbase/var/lib/couchbase/couchbase-server.node"
COOKIEFILE="/opt/couchbase/var/lib/couchbase/couchbase-server.cookie"

_start() {
    _check_nofile
    _prepare_datadir
    _maybe_start_epmd

    exec erl \
        +A 16 \
        -setcookie nocookie \
        -kernel inet_dist_listen_min 21100 inet_dist_listen_max 21299 \
        $* \
        -run ns_bootstrap -- \
        -ns_server config_path "\"/opt/couchbase/etc/couchbase/static_config\"" \
        -ns_server pidfile "\"$PIDFILE\"" \
        -ns_server nodefile "\"$NODEFILE\"" \
        -ns_server cookiefile "\"$COOKIEFILE\""
}

_stop() {
    [ -f $PIDFILE ] && [ -f $NODEFILE ] && [ -f $COOKIEFILE ] || return 1

    cookie=`cat "$COOKIEFILE"`
    nodename=`cat "$NODEFILE"`

    erl \
        -name executioner@executioner \
        -noshell \
        -hidden \
        -setcookie "$cookie" \
        -eval "ns_bootstrap:remote_stop('$nodename')"

    errcode=$?

    if [ $errcode -eq 0 ]; then
        rm "$PIDFILE"
        rm "$COOKIEFILE"
        rm "$NODEFILE"

        epmd -kill >/dev/null || true
    fi

    return $errcode
}

_parse_options () {
    # set +e
    options=`getopt k $*`
    if [ ! $? -eq 0 ]; then
        return 1
    fi
    # set -e
    eval set -- $options
    if [ "-k" = "$1" ]; then
        KILL=true;
    fi

    shift

    if [ x"$KILL" = "xtrue" ]; then
        _stop
    else
        _start $*
    fi
}

_parse_options $*
