#!/usr/bin/env bash

# Copyright © 2023 OpenIM. 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.

set -o errexit
set +o nounset
set -o pipefail

OPENIM_ROOT=$(cd "$(dirname "${BASH_SOURCE[0]}")"/../.. && pwd -P)
[[ -z ${COMMON_SOURCED} ]] && source ${OPENIM_ROOT}/scripts/install/common.sh

SERVER_NAME="openim-rpc"

readonly OPENIM_RPC_ALL_NAME_TARGETS=(
  "${OPENIM_API_SERVICE_TARGETS[@]}"
  "${OPENIM_RPC_SERVICE_TARGETS[@]}"
)
readonly OPENIM_RPC_ALL_NAME_LISTARIES=("${OPENIM_RPC_ALL_NAME_TARGETS[@]##*/}")

# Make sure the environment is only called via common to avoid too much nesting
openim::rpc::service_port() {
  local targets=(
    ${OPENIM_USER_PORT}            # User service 10110
    ${OPENIM_FRIEND_PORT}          # Friend service 10120
    ${OPENIM_MESSAGE_PORT}         # Message service 10130
    # ${OPENIM_MESSAGE_GATEWAY_PORT} # Message gateway 10140
    ${OPENIM_GROUP_PORT}           # Group service 10150
    ${OPENIM_AUTH_PORT}            # Authorization service 10160
    # ${OPENIM_PUSH_PORT}            # Push service 10170
    ${OPENIM_CONVERSATION_PORT}    # Conversation service 10180
    ${OPENIM_THIRD_PORT}           # Third-party service 10190
  )
  echo "${targets[@]}"
}
IFS=" " read -ra OPENIM_RPC_PORT_TARGETS <<< "$(openim::rpc::service_port)"
readonly OPENIM_RPC_PORT_TARGETS
readonly OPENIM_RPC_PORT_LISTARIES=("${OPENIM_RPC_PORT_TARGETS[@]##*/}")

readonly OPENIM_RPC_ALL_PORT_TARGETS=(
  "${OPENIM_API_PORT_TARGETS[@]}"
  "${OPENIM_RPC_PORT_TARGETS[@]}"
)
readonly OPENIM_RPC_ALL_PORT_LISTARIES=("${OPENIM_RPC_ALL_PORT_TARGETS[@]##*/}")

openim::rpc::prometheus_port() {
  # Declare an array to hold all the Prometheus ports for different services
  local targets=(
    ${USER_PROM_PORT}               # Prometheus port for user service
    ${FRIEND_PROM_PORT}             # Prometheus port for friend service
    ${MESSAGE_PROM_PORT}            # Prometheus port for message service
    ${GROUP_PROM_PORT}              # Prometheus port for group service
    ${AUTH_PROM_PORT}               # Prometheus port for authentication service
    ${CONVERSATION_PROM_PORT}       # Prometheus port for conversation service
    ${THIRD_PROM_PORT}              # Prometheus port for third-party integrations service
  )
  # Print the list of ports
  echo "${targets[@]}"
}
IFS=" " read -ra OPENIM_RPC_PROM_PORT_TARGETS <<< "$(openim::rpc::prometheus_port)"
readonly OPENIM_RPC_PROM_PORT_TARGETS
readonly OPENIM_RPC_PROM_PORT_LISTARIES=("${OPENIM_RPC_PROM_PORT_TARGETS[@]##*/}")

readonly OPENIM_RPC_ALL_PROM_PORT_TARGETS=(
  "${OPENIM_API_PORT_TARGETS[@]}"
  "${OPENIM_RPC_PROM_PORT_TARGETS[@]}"
)
readonly OPENIM_RPC_ALL_PROM_PORT_LISTARIES=("${OPENIM_RPC_ALL_PROM_PORT_TARGETS[@]##*/}")

openim::log::info "Starting ${SERVER_NAME} ..."

printf "+------------------------+-------+-----------------+\n"
printf "| Service Name           | Port  | Prometheus Port |\n"
printf "+------------------------+-------+-----------------+\n"

length=${#OPENIM_RPC_ALL_NAME_LISTARIES[@]}

for ((i=0; i<$length; i++)); do
  printf "| %-22s | %-5s | %-15s |\n" "${OPENIM_RPC_ALL_NAME_LISTARIES[$i]}" "${OPENIM_RPC_ALL_PORT_LISTARIES[$i]}" "${OPENIM_RPC_ALL_PROM_PORT_LISTARIES[$i]}"
  printf "+------------------------+-------+-----------------+\n"
done

# #service filename
# service_filename=(
#   #api
#   openim-api
#   #rpc
#   openim-rpc-user
#   openim-rpc-friend
#   openim-rpc-msg
#   openim-rpc-group
#   openim-rpc-auth
#   openim-rpc-conversation
#   openim-rpc-third
# )

# #service config port name
# service_port_name=(
#   #api port name
#   openImApiPort
#   #rpc port name

#   openImUserPort 10110
#   openImFriendPort 10120
#   openImMessagePort 10130
#   openImGroupPort 10150
#   openImAuthPort 10160
#   openImConversationPort 10180
#   openImThirdPort 10190
# )

# service_prometheus_port_name=(
#   #api port name
#   openImApiPort //${API_OPENIM_PORT}
#   #rpc port name
#   userPrometheusPort
#   friendPrometheusPort
#   groupPrometheusPort
#   authPrometheusPort
#   messagePrometheusPort
#   conversationPrometheusPort
#   thirdPrometheusPort
# )


# This function starts a service based on the provided filename, port, and optional Prometheus port.
start_service() {
  local cmd="./$1 --port $2 -c ${configfile_path}"
  if [ -n "$3" ]; then
    cmd="${cmd} --prometheus_port $3"
  fi
  echo "=====================start $1======================" >>$OPENIM_ROOT/logs/openIM.log
  nohup $cmd >>$OPENIM_ROOT/logs/openIM.log 2>&1 &
  sleep 1
  local pid="netstat -ntlp|grep $2 |awk '{printf \$7}'|cut -d/ -f1"
  echo -e "${GREEN_PREFIX}$1 start success, port number:$2 pid:$(eval $pid)$COLOR_SUFFIX"
}

cd $OPENIM_ROOT
cd $BIN_DIR

echo "xxxxxxxxxxxxxxxxxxxxxxxx ${#OPENIM_RPC_ALL_NAME_TARGETS[*]}"

openim::util::stop_services_with_name ${#OPENIM_RPC_ALL_NAME_TARGETS[*]}

# Assuming OPENIM_RPC_ALL_NAME_TARGETS and other variables have been defined from the previous refactor.
for ((i = 0; i < ${#OPENIM_RPC_ALL_NAME_TARGETS[*]}; i++)); do
  openim::util::stop_services_with_name "${OPENIM_RPC_ALL_NAME_TARGETS[$i]}"

  # Get the service and Prometheus ports.
  service_ports=($(cat $config_path | grep ${OPENIM_RPC_ALL_PORT_TARGETS[$i]} | awk -F '[:]' '{print $NF}'))
  prome_ports=($(cat $config_path | grep ${OPENIM_RPC_ALL_PROM_PORT_TARGETS[$i]} | awk -F '[:]' '{print $NF}'))

  for ((j = 0; j < ${#service_ports[*]}; j++)); do
    start_service "${OPENIM_RPC_ALL_NAME_TARGETS[$i]}" "${service_ports[$j]}" "${prome_ports[$j]}"
  done
done

# ---
for ((i = 0; i < ${#service_filename[*]}; i++)); do
  #Check whether the service exists
  service_name="ps |grep -w ${service_filename[$i]} |grep -v grep"
  count="${service_name}| wc -l"

  if [ $(eval ${count}) -gt 0 ]; then
    pid="${service_name}| awk '{print \$2}'"
    echo  "${service_filename[$i]} service has been started,pid:$(eval $pid)"
    echo  "killing the service ${service_filename[$i]} pid:$(eval $pid)"
    #kill the service that existed
    kill -9 $(eval $pid)
    sleep 0.5
  fi
  cd $OPENIM_ROOT
  cd $BIN_DIR
  # Get the rpc port in the configuration file
  portList=$(cat $config_path | grep ${service_port_name[$i]} | awk -F '[:]' '{print $NF}')
  openim::util::list-to-string ${portList}
  service_ports=($ports_array)

  portList2=$(cat $config_path | grep ${service_prometheus_port_name[$i]} | awk -F '[:]' '{print $NF}')
  openim::util::list-to-string $portList2
  prome_ports=($ports_array)
  #Start related rpc services based on the number of ports
  for ((j = 0; j < ${#service_ports[*]}; j++)); do
    #Start the service in the background
    if [ -z "${prome_ports[$j]}" ]; then
      cmd="./${service_filename[$i]} --port ${service_ports[$j]} -c ${configfile_path} "
    else
      cmd="./${service_filename[$i]} --port ${service_ports[$j]} --prometheus_port ${prome_ports[$j]}  -c ${configfile_path} "
    fi
    if [ $i -eq 0 -o $i -eq 1 ]; then
      cmd="./${service_filename[$i]} --port ${service_ports[$j]}"
    fi
    echo "=====================start ${service_filename[$i]}======================">>$OPENIM_ROOT/logs/openIM.log
    nohup $cmd >>$OPENIM_ROOT/logs/openIM.log 2>&1 &
    sleep 1
    pid="netstat -ntlp|grep $j |awk '{printf \$7}'|cut -d/ -f1"
    echo -e "${GREEN_PREFIX}${service_filename[$i]} start success,port number:${service_ports[$j]} pid:$(eval $pid)$COLOR_SUFFIX"
  done
done