#!/bin/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.
#
# This script starts hdfs cluster and hadoop service that allows to reach cluster
# from developer's machine. Once the cluster is working, scripts waits till
# external cluster endpoint will be available. It prints out configuration lines that
# should be added to /etc/hosts file in order to work with hdfs cluster.

set -e

kubectl create -f hdfs-multi-datanode-cluster.yml
kubectl create -f hdfs-multi-datanode-cluster-for-local-dev.yml

external_ip="$(kubectl get svc hadoop -o jsonpath='{.status.loadBalancer.ingress[0].ip}')"
echo "Waiting for the hadoop service to come up ........"
while [ -z "$external_ip" ]
do
 sleep 10s
 external_ip="$(kubectl get svc hadoop -o jsonpath='{.status.loadBalancer.ingress[0].ip}')"
 echo "."
done

hadoop_master_pod_name="$(kubectl get pods --selector=name=namenode -o jsonpath='{.items[*].metadata.name}')"
hadoop_datanodes_pod_names="$(kubectl get pods --selector=name=datanode -o jsonpath='{.items[*].metadata.name}')"

echo "For local tests please add the following 4 entries to /etc/hosts file"
printf "%s\\t%s\\n" "${external_ip}" "${hadoop_master_pod_name}"
read -ra datanodes_pods <<< "${hadoop_datanodes_pod_names}"

for pod in "${datanodes_pods[@]}"; do
  external_ip="$(kubectl get svc "${pod}" -o jsonpath='{.status.loadBalancer.ingress[0].ip}')"
  while [ -z "$external_ip" ]
  do
  sleep 10s
  external_ip="$(kubectl get svc "${pod}"  -o jsonpath='{.status.loadBalancer.ingress[0].ip}')"
  done
  printf "%s\\t%s.hadoop-datanodes.default.svc.cluster.local\\n" "${external_ip}" "${pod}"
done

echo "Done."