#!/bin/bash
# ---------------------------------------------------------------------------
#  Copyright (c) 2018, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
#
#  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.

# ----------------------------------------------------------------------------
# Startup Script for Gateway Cli
#
# Environment Variable Prerequisites
#
#   BALLERINA_HOME      Home of Ballerina installation.
#
#   JAVA_HOME           Must point at your Java Development Kit installation.
#
#   JAVA_OPTS           (Optional) Java runtime options used when the commands
#                       is executed.
#
# NOTE: Borrowed generously from Apache Tomcat startup scripts.
# -----------------------------------------------------------------------------

#To update the CLI Class Path
function update_cli_classpath {
    CLI_CLASSPATH=""
    if [ -e "$BALLERINA_HOME/bre/lib/bootstrap/tools.jar" ]; then
        CLI_CLASSPATH="$JAVA_HOME/lib/tools.jar"
    fi

    for f in "$BALLERINA_HOME"/bre/lib/bootstrap/*.jar
    do
        if [ "$f" != "$BALLERINA_HOME/bre/lib/bootstrap/*.jar" ];then
            CLI_CLASSPATH="$CLI_CLASSPATH":$f
        fi
    done

    for j in "$BALLERINA_HOME"/bre/lib/*.jar
    do
        CLI_CLASSPATH="$CLI_CLASSPATH":$j
    done

    for j in "$MICROGW_HOME"/target/*.jar
    do
        CLI_CLASSPATH="$CLI_CLASSPATH":$j
    done

    # Ballerina platform is not extracted initially.
    # If the ballerina platform is not available, it will be extracted during init command
    # Therefore handle classpath separately until the ballerina platform become available.
    if [ ! -d "$MICROGW_HOME/lib/platform/" ]; then
        # Add required ballerina jars to classpath
        for j in "$MICROGW_HOME"/lib/gateway/platform/*.jar
        do
            CLI_CLASSPATH="$CLI_CLASSPATH":$j
        done

        # Add required MGW jars to classpath
        for j in "$MICROGW_HOME"/lib/gateway/cli/*.jar
        do
            CLI_CLASSPATH="$CLI_CLASSPATH":$j
        done
    fi
    # For Cygwin, switch paths to Windows format before running java
    if $cygwin; then
        CLI_CLASSPATH=`cygpath --path --windows "$CLI_CLASSPATH"`
    fi
}
# OS specific support.  $var _must_ be set to either true or false.
cygwin=false
os400=false
case "`uname`" in
    CYGWIN*) 
        cygwin=true
        ;;
    OS400*)
        os400=true
        ;;
esac

# resolve links - $0 may be a softlink
PRG="$0"

while [ -h "$PRG" ]
do
    ls=`ls -ld "$PRG"`
    link=`expr "$ls" : '.*-> \(.*\)$'`
    if expr "$link" : '.*/.*' > /dev/null; then
        PRG="$link"
    else
        PRG=`dirname "$PRG"`/"$link"
    fi
done

# Get standard environment variables
PRGDIR=`dirname "$PRG"`

# if MICROGW_HOME environment variable not set then set it relative to the micro-gw binary location
if [[ -z "${MICROGW_HOME}" ]]; then
    export MICROGW_HOME=`cd "$PRGDIR/.." ; pwd`
fi

# set BALLERINA_HOME
BALLERINA_HOME="$MICROGW_HOME/lib/platform"

if [ ! -d "$BALLERINA_HOME" ]; then
    BALLERINA_HOME="$MICROGW_HOME/lib"
fi

export BALLERINA_HOME=$BALLERINA_HOME
export PATH=$BALLERINA_HOME/bin:$PATH

MICRO_GW_PROJECT_DIR=$(pwd)

# if JAVA_HOME is not set we're not happy
if [ -z "$JAVA_HOME" ]; then
    echo "You must set the JAVA_HOME variable before running Micro-Gateway Tooling."
    exit 1
fi

# For Cygwin, ensure paths are in UNIX format before anything is touched
if $cygwin; then
    [ -n "$JAVA_HOME" ] && JAVA_HOME=`cygpath --unix "$JAVA_HOME"`
    [ -n "$BALLERINA_HOME" ] && BALLERINA_HOME=`cygpath --unix "$BALLERINA_HOME"`
    [ -n "$MICROGW_HOME" ] && MICROGW_HOME=`cygpath --unix "$MICROGW_HOME"`
fi

# For OS400
if $os400; then
    # Set job priority to standard for interactive (interactive - 6) by using
    # the interactive priority - 6, the helper threads that respond to requests
    # will be running at the same priority as interactive jobs.
    COMMAND='chgjob job('$JOBNAME') runpty(6)'
    system $COMMAND

    # Enable multi threading
    QIBM_MULTI_THREADED=Y
    export QIBM_MULTI_THREADED
fi

# For Migwn, ensure paths are in UNIX format before anything is touched
if $mingw ; then
    [ -n "$BALLERINA_HOME" ] &&
        BALLERINA_HOME="`(cd "$BALLERINA_HOME"; pwd)`"
    [ -n "$JAVA_HOME" ] &&
        JAVA_HOME="`(cd "$JAVA_HOME"; pwd)`"
    [ -n "$MICROGW_HOME" ] &&
        MICROGW_HOME="`(cd "$MICROGW_HOME"; pwd)`"
fi

if [ -z "$JAVACMD" ] ; then
    if [ -n "$JAVA_HOME"  ] ; then
        if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
            # IBM's JDK on AIX uses strange locations for the executables
            JAVACMD="$JAVA_HOME/jre/sh/java"
        else
            JAVACMD="$JAVA_HOME/bin/java"
        fi
    else
        JAVACMD=java
    fi
fi

if [ ! -x "$JAVACMD" ] ; then
    echo "Error: JAVA_HOME is not defined correctly."
    exit 1
fi

# ----- Process the input command ----------------------------------------------

ALL_ARGS=()
DEFAULT_ARGS=()
while [[ $# -gt 0 ]]
do
    key="$1"
    case $key in
        --java.debug)
            CMD="--java.debug"
            PORT="$2"
            shift # past java.debug argument
            shift # past port value
            ;;
        help)
            ALL_ARGS+=("$1")
            ALL_ARGS+=("$2")
            shift # past help argument
            shift # past next argument after help
            ;;
        build)
            ALL_ARGS+=("$1")
            IS_BUILD_COMMAND=true
            shift # past build argument
            ;;
        *)    # default argument
            DEFAULT_ARGS+=("$1") # save it in DEFAULT_ARGS array
            ALL_ARGS+=("$1")
            shift # past argument
            ;;
    esac
done

# restore all the shifted arguments
set -- "${ALL_ARGS[@]}"

#add system variable VERBOSE_ENABLED
VerboseRemovedArgs="$@"
if [[ " ${VerboseRemovedArgs[*]} " == *" --verbose "* ]]; then
    VERBOSE_ENABLED=true
    #delete '--verbose' if available
    delete=( --verbose )
    VerboseRemovedArgs=${VerboseRemovedArgs/$delete}
    #set modified args as $@
    set -- $VerboseRemovedArgs
else
    VERBOSE_ENABLED=false
fi

# take the first default argument as the project name
CMD_PRO_NAME_VAL=$(echo ${DEFAULT_ARGS[0]}|sed 's#/##g')

if [ "$CMD" = "--java.debug" ]; then
    if [ "$PORT" = "" ]; then
        echo "Please specify the debug port after the --java.debug option"
        exit 1
    fi
    if [ -n "$JAVA_OPTS" ]; then
        echo "Warning !!!. User specified JAVA_OPTS will be ignored, once you give the --java.debug option."
    fi

    JAVA_OPTS="-Xdebug -Xnoagent -Djava.compiler=NONE -Xrunjdwp:transport=dt_socket,server=y,suspend=y,address=$PORT"
    echo "Please start the remote debugging client to continue..."
fi

update_cli_classpath

# For Cygwin, switch paths to Windows format before running java
if $cygwin; then
    JAVA_HOME=`cygpath --absolute --windows "$JAVA_HOME"`
    BALLERINA_HOME=`cygpath --absolute --windows "$BALLERINA_HOME"`
    CLASSPATH=`cygpath --path --windows "$CLASSPATH"`
fi

#execute build command
if [ "$IS_BUILD_COMMAND" = true ] && [ "$CMD_PRO_NAME_VAL" != "" ] && [ "$MICRO_GW_PROJECT_DIR" != "" ]; then

  # ----- Execute The Build Command without --compiled flag (1st phase) -----------------------------------------

  $JAVACMD \
    -Xms256m -Xmx1024m \
    -XX:+HeapDumpOnOutOfMemoryError \
    -XX:HeapDumpPath="$MICROGW_HOME/heap-dump.hprof" \
    $JAVA_OPTS \
    -classpath "$CLI_CLASSPATH" \
    -Djava.security.egd=file:/dev/./urandom \
    -Dballerina.home=$BALLERINA_HOME \
    -Djava.util.logging.config.class="org.wso2.apimgt.gateway.cli.logging.CLILogConfigReader" \
    -Djava.util.logging.manager="org.wso2.apimgt.gateway.cli.logging.CLILogManager" \
    -Dfile.encoding=UTF8 \
    -Dtemplates.dir.path="$MICROGW_HOME/resources/templates" \
    -Dcli.home=$MICROGW_HOME \
    -DVERBOSE_ENABLED=$VERBOSE_ENABLED \
    org.wso2.apimgt.gateway.cli.cmd.Main "$@"

    if [ $? -eq 0 ]
    then
        # Ballerina path is updated as ballerina platform is extracted
        BALLERINA_HOME="$MICROGW_HOME/lib/platform"
        export BALLERINA_HOME=$BALLERINA_HOME
        export PATH=$BALLERINA_HOME/bin:$PATH
        # For Migwn, ensure paths are in UNIX format before anything is touched
        if $mingw ; then
            [ -n "$BALLERINA_HOME" ] &&
                BALLERINA_HOME="`(cd "$BALLERINA_HOME"; pwd)`"
        fi
        # For Cygwin, ensure paths are in UNIX format before anything is touched
        if $cygwin; then
            [ -n "$BALLERINA_HOME" ] && BALLERINA_HOME=`cygpath --unix "$BALLERINA_HOME"`
            BALLERINA_HOME=`cygpath --absolute --windows "$BALLERINA_HOME"`

        fi
        
        update_cli_classpath
        MICRO_GW_LABEL_PROJECT_DIR="$MICRO_GW_PROJECT_DIR/$CMD_PRO_NAME_VAL"
        pushd $MICRO_GW_LABEL_PROJECT_DIR/gen > /dev/null
            # clean the content of target folder
            if [ -d "$MICRO_GW_LABEL_PROJECT_DIR/target" ]; then
                rm -rf $MICRO_GW_LABEL_PROJECT_DIR/target
            fi
            # build the ballerina source code for the label
            ballerina build src/ -o $CMD_PRO_NAME_VAL.balx --offline --experimental --siddhiruntime
        popd > /dev/null

    # ----- Execute The Build Command with --compiled flag (2nd phase) -----------------------------------------
      $JAVACMD \
        -Xms256m -Xmx1024m \
        -XX:+HeapDumpOnOutOfMemoryError \
        -XX:HeapDumpPath="$MICROGW_HOME/heap-dump.hprof" \
        $JAVA_OPTS \
        -classpath "$CLI_CLASSPATH" \
        -Djava.security.egd=file:/dev/./urandom \
        -Dballerina.home=$BALLERINA_HOME \
        -Djava.util.logging.config.class="org.wso2.apimgt.gateway.cli.logging.CLILogConfigReader" \
        -Djava.util.logging.manager="org.wso2.apimgt.gateway.cli.logging.CLILogManager" \
        -Dfile.encoding=UTF8 \
        -Dtemplates.dir.path="$MICROGW_HOME/resources/templates" \
        -Dcli.home=$MICROGW_HOME \
        -DVERBOSE_ENABLED=$VERBOSE_ENABLED \
        org.wso2.apimgt.gateway.cli.cmd.Main "$@" "--compiled"
     fi
else
  $JAVACMD \
      -Xms256m -Xmx1024m \
      -XX:+HeapDumpOnOutOfMemoryError \
      -XX:HeapDumpPath="$MICROGW_HOME/heap-dump.hprof" \
      $JAVA_OPTS \
      -classpath "$CLI_CLASSPATH" \
      -Djava.security.egd=file:/dev/./urandom \
      -Dballerina.home=$BALLERINA_HOME \
      -Djava.util.logging.config.class="org.wso2.apimgt.gateway.cli.logging.CLILogConfigReader" \
      -Djava.util.logging.manager="org.wso2.apimgt.gateway.cli.logging.CLILogManager" \
      -Dfile.encoding=UTF8 \
      -Dtemplates.dir.path="$MICROGW_HOME/resources/templates" \
      -Dcli.home=$MICROGW_HOME \
      -DVERBOSE_ENABLED=$VERBOSE_ENABLED \
      org.wso2.apimgt.gateway.cli.cmd.Main "$@"

fi

