#!/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.
#

. /etc/predictionio/pio-env.sh

# check elasticsearch status
if [ x"$PIO_STORAGE_SOURCES_ELASTICSEARCH_TYPE" != "x" ] ; then
  RET=-1
  COUNT=0
  ES_HOST=`echo $PIO_STORAGE_SOURCES_ELASTICSEARCH_HOSTS | sed -e "s/,.*//"`
  ES_PORT=`echo $PIO_STORAGE_SOURCES_ELASTICSEARCH_PORTS | sed -e "s/,.*//"`
  # Wait for elasticsearch startup
  while [ $RET != 0 -a $COUNT -lt 10 ] ; do
    echo "Waiting for ${ES_HOST}..."
    curl --connect-timeout 60 --retry 10 -s "$ES_HOST:$ES_PORT/_cluster/health?wait_for_status=green&timeout=1m"
    RET=$?
    COUNT=`expr $COUNT + 1`
    sleep 1
  done
fi

# check mysql jar file
if [ x"$PIO_STORAGE_SOURCES_MYSQL_TYPE" != "x" ] ; then
  MYSQL_JAR_FILE=$PIO_HOME/lib/mysql-connector-java-$MYSQL_VERSION.jar
  if [ ! -f $MYSQL_JAR_FILE ] ; then
    curl -o $MYSQL_JAR_FILE http://central.maven.org/maven2/mysql/mysql-connector-java/$MYSQL_VERSION/mysql-connector-java-$MYSQL_VERSION.jar
  fi
fi

# Check PIO status
RET=-1
COUNT=0
while [ $RET != 0 -a $COUNT -lt 10 ] ; do
  echo "Waiting for PredictionIO..."
  $PIO_HOME/bin/pio status
  RET=$?
  COUNT=`expr $COUNT + 1`
  sleep 1
done


if [ x"$PIO_RUN_FILE" != "x" ] ; then
  sh $PIO_RUN_FILE
else
  # Start PIO Event Server
  $PIO_HOME/bin/pio eventserver
fi