#!/usr/bin/env bash

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you 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.


# The Tajo command script
#
# Environment Variables
#
#   JAVA_HOME        The java implementation to use.  Overrides JAVA_HOME.
#
#   TAJO_CLASSPATH Extra Java CLASSPATH entries.
#
#   TAJO_USER_CLASSPATH_FIRST      When defined, the TAJO_CLASSPATH is 
#                                    added in the beginning of the global
#                                    classpath. Can be defined, for example,
#                                    by doing 
#                                    export TAJO_USER_CLASSPATH_FIRST=true
#
#   TAJO_MASTER_HEAPSIZE  The maximum amount of heap to use, in MB. 
#                         Default is 1000.
#
#   TAJO_WORKER_HEAPSIZE  The maximum amount of heap to use, in MB. 
#                         Default is 5000.
#
#   TAJO_OPTS      Extra Java runtime options.

#   TAJO_{COMMAND}_OPTS etc  TAJO_JT_OPTS applies to JobTracker 
#                              for e.g.  TAJO_CLIENT_OPTS applies to 
#                              more than one command (fs, dfs, fsck, 
#                              dfsadmin etc)  
#
#   TAJO_CONF_DIR  Alternate conf dir. Default is ${TAJO_HOME}/conf.
#
#

bin=`dirname "$0"`
bin=`cd "$bin"; pwd`

. "$bin"/tajo-config.sh

cygwin=false
case "`uname`" in
CYGWIN*) cygwin=true;;
esac

function print_usage() {
  echo "Usage: tajo [--config confdir] COMMAND"
  echo "where COMMAND is one of:"
  echo "  master               run the Master Server"
  echo "  worker               run the Worker Server"
  echo "  pullserver           run the Pull Server"
  echo "  catalog              run the Catalog server"
  echo "  catutil              catalog utility"
  echo "  cli                  run the tajo cli"
  echo "  admin                run the tajo admin util"
  echo "  haadmin              run the tajo master HA admin util"
  echo "  getconf              print tajo configuration"
  echo "  jar <jar>            run a jar file"
  echo "  benchmark            run the benchmark driver"
  echo "  version              print the version"
  echo " or"
  echo "  CLASSNAME            run the class named CLASSNAME"
  echo "Most commands print help when invoked w/o parameters."
}

# if no args specified, show usage
if [ $# = 0 ]; then
  print_usage
  exit 1
fi

# get arguments
COMMAND=$1
shift

if [ "$COMMAND" == "--help" ] || [ "$COMMAND" == "-help" ] || [ "$COMMAND" == "-h" ] ; then
  print_usage
  exit
fi

if [ -f "${TAJO_CONF_DIR}/tajo-env.sh" ]; then
  . "${TAJO_CONF_DIR}/tajo-env.sh"
fi

# Determine if we're starting a secure datanode, and if so, redefine appropriate variables
if [ "$COMMAND" == "datanode" ] && [ "$EUID" -eq 0 ] && [ -n "$TAJO_SECURE_DN_USER" ]; then
  TAJO_PID_DIR=$TAJO_SECURE_DN_PID_DIR
  TAJO_LOG_DIR=$TAJO_SECURE_DN_LOG_DIR
  TAJO_IDENT_STRING=$TAJO_SECURE_DN_USER
  starting_secure_dn="true"
fi


# This is borrowed from hadoop-config.sh in hadoop
# Newer versions of glibc use an arena memory allocator that causes virtual
# memory usage to explode. This interacts badly with the many threads that
# we use in Hadoop. Tune the variable down to prevent vmem explosion.
export MALLOC_ARENA_MAX=${MALLOC_ARENA_MAX:-4}

# Attempt to set JAVA_HOME if it is not set
if [[ -z $JAVA_HOME ]]; then
  # On OSX use java_home (or /Library for older versions)
  if [ "Darwin" == "$(uname -s)" ]; then
    if [ -x /usr/libexec/java_home ]; then
      export JAVA_HOME=($(/usr/libexec/java_home))
    else
      export JAVA_HOME=(/Library/Java/Home)
    fi
  fi

  # Bail if we did not detect it
  if [[ -z $JAVA_HOME ]]; then
    echo "Error: JAVA_HOME is not set and could not be found." 1>&2
    exit 1
  fi
fi

JAVA=$JAVA_HOME/bin/java

JAVA_TAJO_MASTER_HEAP_MAX=-Xmx1000m 
JAVA_WORKER_HEAP_MAX=-Xmx5000m
JAVA_QUERYMASTER_HEAP_MAX=-Xmx1000m
JAVA_PULLSERVER_HEAP_MAX=-Xmx1000m

# check envvars which might override default args
if [ "$TAJO_MASTER_HEAPSIZE" != "" ]; then
  #echo "run with heapsize $TAJO_MASTER_HEAPSIZE"
  JAVA_TAJO_MASTER_HEAP_MAX="-Xmx""$TAJO_MASTER_HEAPSIZE""m"
  #echo $JAVA_MASTER_HEAP_MAX
fi
if [ "$TAJO_WORKER_HEAPSIZE" != "" ]; then
  #echo "run with heapsize $TAJO_WORKER_HEAPSIZE"
  JAVA_WORKER_HEAP_MAX="-Xmx""$TAJO_WORKER_HEAPSIZE""m"
  #echo $JAVA_WORKER_HEAP_MAX
fi
if [ "$TAJO_PULLSERVER_HEAPSIZE" != "" ]; then
  #echo "run with heapsize $TAJO_PULLSERVER_HEAPSIZE"
  JAVA_PULLSERVER_HEAP_MAX="-Xmx""$TAJO_PULLSERVER_HEAPSIZE""m"
  #echo $JAVA_PULLSERVER_HEAP_MAX
fi
if [ "$TAJO_QUERYMASTER_HEAPSIZE" != "" ]; then
  #echo "run with heapsize $TAJO_QUERYMASTER_HEAPSIZE"
  JAVA_QUERYMASTER_HEAP_MAX="-Xmx""$TAJO_QUERYMASTER_HEAPSIZE""m"
  #echo $JAVA_QUERYMASTER_HEAP_MAX
fi


##############################################################################
# Hadoop Version Checking Section Start
##############################################################################

# check for hadoop in the path
HADOOP_IN_PATH=`which hadoop 2>/dev/null`
if [ -f ${HADOOP_IN_PATH} ]; then
  HADOOP_DIR=`dirname "$HADOOP_IN_PATH"`/..
fi
# HADOOP_HOME env variable overrides hadoop in the path
HADOOP_HOME=${HADOOP_HOME:-$HADOOP_DIR}
if [ "$HADOOP_HOME" == "" ]; then
  echo "Cannot find hadoop installation: \$HADOOP_HOME must be set or hadoop must be in the path";
  exit 4;
fi

HADOOP=$HADOOP_HOME/bin/hadoop
if [ ! -f ${HADOOP} ]; then
  echo "Cannot find hadoop installation: \$HADOOP_HOME must be set or hadoop must be in the path";
  exit 4;
fi

# Make sure we're using a compatible version of Hadoop
if [ "x$HADOOP_VERSION" == "x" ]; then
    HADOOP_VERSION=$($HADOOP version | awk '{if (NR == 1) {print $2;}}');
fi

# Save the regex to a var to workaround quoting incompatibilities
# between Bash 3.1 and 3.2
hadoop_version_re="^([[:digit:]]+)\.([[:digit:]]+)(\.([[:digit:]]+))?.*$"

if [[ "$HADOOP_VERSION" =~ $hadoop_version_re ]]; then
    hadoop_major_ver=${BASH_REMATCH[1]}
    hadoop_minor_ver=${BASH_REMATCH[2]}
    hadoop_patch_ver=${BASH_REMATCH[4]}
else
    echo "Unable to determine Hadoop version information."
    echo "'hadoop version' returned:"
    echo `$HADOOP version`
    exit 5
fi

if [ "$hadoop_major_ver" -lt "1" -a  "$hadoop_minor_ver$hadoop_patch_ver" -lt "201" ]; then
    echo "Tajo requires Hadoop 0.20.x (x >= 1)."
    echo "'hadoop version' returned:"
    echo `$HADOOP version`
    exit 6
fi

# Allow alternate conf dir location.
HADOOP_CONF_DIR="${HADOOP_CONF_DIR:-$HADOOP_HOME/etc/hadoop}"

# CLASSPATH initially contains $HADOOP_CONF_DIR and tools.jar
CLASSPATH="${HADOOP_CONF_DIR}:${JAVA_HOME}/lib/tools.jar"

##############################################################################
# Hadoop Version Checking Section End
##############################################################################

##############################################################################
# Find and Set Hadoop CLASSPATH
##############################################################################

# HADOOP JAR DIRS
HADOOP_MODULE_DIRS="$HADOOP_HOME/share/hadoop/common/lib
$HADOOP_HOME/share/hadoop/common
$HADOOP_HOME/share/hadoop/hdfs
$HADOOP_HOME/share/hadoop/hdfs/lib
$HADOOP_HOME/share/hadoop/yarn
$HADOOP_HOME/share/hadoop/mapreduce
$HADOOP_HOME/share/hadoop/tools/lib"

for d in $HADOOP_MODULE_DIRS; do
  CLASSPATH=${CLASSPATH}:$d/*
done;

HDFS_LIBRARY_PATH="${HADOOP_HOME}/lib/native/"
##############################################################################
# Hadoop Home Configuration End
##############################################################################

##############################################################################
# HBase Configuration Start
##############################################################################

HBASE_CONF=$HBASE_HOME/conf

if [ ! -z ${HBASE_HOME} ] && [ -d ${HBASE_HOME} ] && [ -d ${HBASE_CONF} ]; then
  CLASSPATH=${HBASE_CONF}:${CLASSPATH}
fi

HBASE_LIB=$HBASE_HOME/lib

if [ ! -z ${HBASE_HOME} ] && [ -d ${HBASE_HOME} ] && [ -d ${HBASE_LIB} ]; then

  for f in ${HBASE_LIB}/hbase-server-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HBASE_LIB}/hbase-client-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HBASE_LIB}/hbase-common-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HBASE_LIB}/hbase-protocol-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HBASE_LIB}/htrace-core-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HBASE_LIB}/metrics-core-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

fi

##############################################################################
# HBase Configuration End
##############################################################################

##############################################################################
# Find and Set Hive CLASSPATH
##############################################################################

HIVE_CONF=$HIVE_HOME/conf

if [ ! -z ${HIVE_HOME} ] && [ -d ${HIVE_HOME} ] && [ -d ${HIVE_CONF} ]; then
  CLASSPATH=${HIVE_CONF}:${CLASSPATH}
fi

HIVE_LIB=$HIVE_HOME/lib

if [ ! -z ${HIVE_HOME} ] && [ -d ${HIVE_HOME} ] && [ -d ${HIVE_LIB} ]; then

  for f in ${HIVE_LIB}/hive-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HIVE_LIB}/libthrift-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HIVE_LIB}/libfb*.jar; do
      CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HIVE_LIB}/jdo*-api-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HIVE_LIB}/javax.jdo-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HIVE_LIB}/log4j-core-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done

  for f in ${HIVE_LIB}/datanucleus-*.jar; do
    CLASSPATH=${CLASSPATH}:$f;
  done
fi

if [ "${HIVE_JDBC_DRIVER_DIR}" != "" ]; then
  CLASSPATH=${CLASSPATH}:$HIVE_JDBC_DRIVER_DIR;
fi

##############################################################################
# Hive Home Configuration End
##############################################################################


##############################################################################
# Find and Set Tajo CLASSPATH
##############################################################################

# TAJO_BASE_CLASSPATH initially contains $TAJO_CONF_DIR
TAJO_BASE_CLASSPATH="${TAJO_CONF_DIR}"

# for developers, add Tajo classes to TAJO_BASE_CLASSPATH
if [ -d "$TAJO_HOME/target/classes" ]; then
  TAJO_BASE_CLASSPATH=${TAJO_BASE_CLASSPATH}:$TAJO_HOME/target/classes
fi
if [ -d "$TAJO_HOME/target/test/classes" ]; then
  TAJO_BASE_CLASSPATH=${TAJO_BASE_CLASSPATH}:$TAJO_HOME/target/test/classes
fi

# so that filenames w/ spaces are handled correctly in loops below
IFS=$'\n'

# Tajo Jar Directory
TAJO_JAR_DIRS="$TAJO_HOME
${TAJO_HOME}/lib"

# We need to avoid the inclusion of slf4 twice.
for d in $TAJO_JAR_DIRS; do
  for j in `find $d/*.jar ! -name "*test*" | grep -v slf4`; do
    TAJO_BASE_CLASSPATH=$TAJO_BASE_CLASSPATH:$j
  done
done

# add CLASSPATH for common UDFs and third-party libraries like JDBC.
TAJO_BASE_CLASSPATH=${TAJO_BASE_CLASSPATH}:$TAJO_HOME/extlib

# add user-specified CLASSPATH last
if [ "$TAJO_CLASSPATH" != "" ]; then
  if [ "$TAJO_USER_CLASSPATH_FIRST" != "" ]; then
    TAJO_BASE_CLASSPATH=${TAJO_CLASSPATH}:${TAJO_BASE_CLASSPATH}
  else
    TAJO_BASE_CLASSPATH=${TAJO_BASE_CLASSPATH}:${TAJO_CLASSPATH}
  fi
fi

export TAJO_BASE_CLASSPATH

CLASSPATH="${TAJO_BASE_CLASSPATH}:${CLASSPATH}"

##############################################################################
# Tajo Home Configuration End
##############################################################################

# restore ordinary behaviour
unset IFS

# default log directory & file
if [ "$TAJO_LOG_DIR" = "" ]; then
  TAJO_LOG_DIR="$TAJO_HOME/logs"
fi
if [ "$TAJO_LOGFILE" = "" ]; then
  TAJO_LOGFILE='tajo.log'
fi

# default policy file for service-level authorization
if [ "$TAJO_POLICYFILE" = "" ]; then
  TAJO_POLICYFILE="tajo-policy.xml"
fi

# Disable IPv6 Support for network performance
TAJO_OPTS="$TAJO_OPTS -Djava.net.preferIPv4Stack=true"

# figure out which class to run
if [ "$COMMAND" = "classpath" ] ; then
  if $cygwin; then
    CLASSPATH=`cygpath -p -w "$CLASSPATH"`
  fi
  echo $CLASSPATH
  exit
elif [ "$COMMAND" = "master" ] ; then
  CLASS='org.apache.tajo.master.TajoMaster'
  TAJO_OPTS="$TAJO_OPTS $JAVA_TAJO_MASTER_HEAP_MAX $TAJO_MASTER_OPTS"
elif [ "$COMMAND" = "worker" ] ; then
  CLASS='org.apache.tajo.worker.TajoWorker'
  TAJO_OPTS="$TAJO_OPTS $JAVA_WORKER_HEAP_MAX $TAJO_WORKER_OPTS"
elif [ "$COMMAND" = "pullserver" ] ; then
  CLASS='org.apache.tajo.pullserver.TajoPullServer'
  TAJO_OPTS="$TAJO_OPTS $JAVA_PULLSERVER_HEAP_MAX $TAJO_PULLSERVER_OPTS"
elif [ "$COMMAND" = "catalog" ] ; then
  CLASS='org.apache.tajo.catalog.CatalogServer'
  TAJO_OPTS="$TAJO_OPTS $TAJO_CATALOG_OPTS"
elif [ "$COMMAND" = "cli" ] ; then
  CLASS='org.apache.tajo.cli.tsql.TajoCli'
  TAJO_ROOT_LOGGER_APPENDER="${TAJO_ROOT_LOGGER_APPENDER:-NullAppender}"
  TAJO_OPTS="$TAJO_OPTS $TAJO_CLI_OPTS"
elif [ "$COMMAND" = "admin" ] ; then
  CLASS='org.apache.tajo.cli.tools.TajoAdmin'
  TAJO_ROOT_LOGGER_APPENDER="${TAJO_ROOT_LOGGER_APPENDER:-NullAppender}"
  TAJO_OPTS="$TAJO_OPTS $TAJO_CLI_OPTS"
elif [ "$COMMAND" = "haadmin" ] ; then
  CLASS='org.apache.tajo.cli.tools.TajoHAAdmin'
  TAJO_ROOT_LOGGER_APPENDER="${TAJO_ROOT_LOGGER_APPENDER:-NullAppender}"
  TAJO_OPTS="$TAJO_OPTS $TAJO_CLI_OPTS"
elif [ "$COMMAND" = "getconf" ] ; then
  CLASS='org.apache.tajo.cli.tools.TajoGetConf'
  TAJO_ROOT_LOGGER_APPENDER="${TAJO_ROOT_LOGGER_APPENDER:-NullAppender}"
  TAJO_OPTS="$TAJO_OPTS $TAJO_CLI_OPTS"
elif [ "$COMMAND" = "dump" ] ; then
  CLASS='org.apache.tajo.cli.tools.TajoDump'
  TAJO_ROOT_LOGGER_APPENDER="${TAJO_ROOT_LOGGER_APPENDER:-NullAppender}"
  TAJO_OPTS="$TAJO_OPTS $TAJO_DUMP_OPTS"
elif [ "$COMMAND" = "version" ] ; then
  CLASS='org.apache.tajo.util.VersionInfo'
  TAJO_ROOT_LOGGER_APPENDER="${TAJO_ROOT_LOGGER_APPENDER:-NullAppender}"
  TAJO_OPTS="$TAJO_OPTS $TAJO_CLI_OPTS"
else
  CLASS=$COMMAND
fi

# cygwin path translation
if $cygwin; then
  CLASSPATH=`cygpath -p -w "$CLASSPATH"`
  TAJO_HOME=`cygpath -w "$TAJO_HOME"`
  TAJO_LOG_DIR=`cygpath -w "$TAJO_LOG_DIR"`
  TOOL_PATH=`cygpath -p -w "$TOOL_PATH"`
fi
# setup 'java.library.path' for native-tajo code if necessary
JAVA_LIBRARY_PATH=''
if [ -d "${TAJO_HOME}/build/native" -o -d "${TAJO_HOME}/lib/native" ]; then
  JAVA_PLATFORM=`CLASSPATH=${CLASSPATH} ${JAVA} -Xmx32m ${TAJO_JAVA_PLATFORM_OPTS} org.apache.tajo.util.PlatformName | sed -e "s/ /_/g"`
  
  if [ -d "$TAJO_HOME/build/native" ]; then
    JAVA_LIBRARY_PATH=${TAJO_HOME}/build/native/${JAVA_PLATFORM}/lib
  fi
  
  if [ -d "${TAJO_HOME}/lib/native" ]; then
    if [ "x$JAVA_LIBRARY_PATH" != "x" ]; then
      JAVA_LIBRARY_PATH=${JAVA_LIBRARY_PATH}:${TAJO_HOME}/lib/native/${JAVA_PLATFORM}
    else
      JAVA_LIBRARY_PATH=${TAJO_HOME}/lib/native/${JAVA_PLATFORM}
    fi
  fi
fi

if [ "x$JAVA_LIBRARY_PATH" != "x" ]; then
    JAVA_LIBRARY_PATH=${JAVA_LIBRARY_PATH}:${HDFS_LIBRARY_PATH}
  else
    JAVA_LIBRARY_PATH=${HDFS_LIBRARY_PATH}
fi

# cygwin path translation
if $cygwin; then
  JAVA_LIBRARY_PATH=`cygpath -p "$JAVA_LIBRARY_PATH"`
fi

TAJO_OPTS="$TAJO_OPTS -Dtajo.log.dir=$TAJO_LOG_DIR"
TAJO_OPTS="$TAJO_OPTS -Dtajo.log.file=$TAJO_LOGFILE"
TAJO_OPTS="$TAJO_OPTS -Dtajo.home.dir=$TAJO_HOME"
TAJO_OPTS="$TAJO_OPTS -Dtajo.id.str=$TAJO_IDENT_STRING"
TAJO_OPTS="$TAJO_OPTS -Dtajo.root.logger.appender=${TAJO_ROOT_LOGGER_APPENDER:-console}"
if [ "x$TAJO_ROOT_LOGGER_LEVEL" != "x" ]; then
  TAJO_OPTS="$TAJO_OPTS -Dtajo.root.logger.level=${TAJO_ROOT_LOGGER_LEVEL}"
fi

if [ "x$JAVA_LIBRARY_PATH" != "x" ]; then
  TAJO_OPTS="$TAJO_OPTS -Djava.library.path=$JAVA_LIBRARY_PATH"
fi  
TAJO_OPTS="$TAJO_OPTS -Dtajo.policy.file=$TAJO_POLICYFILE"

# run it
export CLASSPATH=$CLASSPATH
exec "$JAVA" -Dproc_$COMMAND $TAJO_OPTS $CLASS $TAJO_DAEMON_MODE "$@"