#!/usr/bin/env bash
#
# install.sh -- installs an agent
#
#  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.


usage() {
  printf "Usage: %s: -d [directory to deploy to] -t [routing|storage|computing] -z [zip file] -h [host] -p [pod] -c [data center] -m [expert|novice|setup]\n" $(basename $0) >&2
}

mode=
host=
pod=
zone=

deploydir=
confdir=
zipfile=
typ=

#set -x

while getopts 'd:z:t:x:m:h:p:c:' OPTION
do
  case "$OPTION" in
  d)	deploydir="$OPTARG"
		;;
  z)    zipfile="$OPTARG"
                ;;
  t)    typ="$OPTARG"
                ;;
  m)    mode="$OPTARG"
                ;;
  h)    host="$OPTARG"
                ;;
  p)    pod="$OPTARG"
                ;;
  c)    zone="$OPTARG"
                ;;
  ?)	usage
		exit 2
		;;
  esac
done

printf "NOTE: You must have root privileges to install and run this program.\n"

if [ "$typ" == "" ]; then
    if [ "$mode" != "expert" ]
    then
        printf "Type of agent to install [routing|computing|storage]: "
        read typ
    fi
fi
if [ "$typ" != "computing" ] && [ "$typ" != "routing" ] && [ "$typ" != "storage" ]
then
    printf "ERROR: The choices are computing, routing, or storage.\n"
    exit 4
fi

if [ "$host" == "" ]; then
  if [ "$mode" != "expert" ]
  then
    printf "Host name or ip address of management server [Required]: "
    read host
    if [ "$host" == "" ]; then
      printf "ERROR: Host is required\n"
      exit 23;
    fi
  fi
fi

port=
if [ "$mode" != "expert" ]
then
    printf "Port number of management server [defaults to 8250]: "
    read port
fi
if [ "$port" == "" ]
then
    port=8250
fi

if [ "$zone" == "" ]; then
  if [ "$mode" != "expert" ]; then
    printf "Availability Zone [Required]: "
    read zone
    if [ "$zone" == "" ]; then
	printf "ERROR: Zone is required\n";
	exit 21;
    fi
  fi
fi

if [ "$pod" == "" ]; then
  if [ "$mode" != "expert" ]; then
    printf "Pod [Required]: "
    read pod
    if [ "$pod" == "" ]; then
      printf "ERROR: Pod is required\n";
      exit 22;
    fi
  fi
fi

workers=
if [ "$mode" != "expert" ]; then
    printf "# of workers to start [defaults to 3]: "
    read workers
fi
if [ "$workers" == "" ]; then
    workers=3
fi

if [ "$deploydir" == "" ]; then 
    if [ "$mode" != "expert" ]; then
        printf "Directory to deploy to [defaults to /usr/local/vmops/agent]: "
        read deploydir
    fi
    if [ "$deploydir" == "" ]; then 
        deploydir="/usr/local/vmops/agent"
    fi
fi
if ! mkdir -p $deploydir
then
  printf "ERROR: Unable to create $deploydir\n"
  exit 5
fi

if [ "$zipfile" == "" ]; then 
    if [ "$mode" != "expert" ]; then
        printf "Path of the zip file [defaults to agent.zip]: "
        read zipfile
    fi
    if [ "$zipfile" == "" ]; then
        zipfile="agent.zip"
    fi

fi
if ! unzip -o $zipfile -d $deploydir
then
  printf "ERROR: Unable to unzip $zipfile to $deploydir\n"
  exit 6
fi

#if ! chmod -R +x $deploydir/scripts/*.sh
#then
#  printf "ERROR: Unable to change scripts to executable.\n"
#  exit 7
#fi
#if ! chmod -R +x $deploydir/scripts/iscsi/*.sh
#then
#  printf "ERROR: Unable to change scripts to executable.\n"
#  exit 8
#fi
#if ! chmod -R +x $deploydir/*.sh
#then
#  printf "ERROR: Unable to change scripts to executable.\n"
#  exit 9
#fi

if [ "$mode" == "setup" ]; then
  mode="expert"
  deploydir="/usr/local/vmops/agent"
  confdir="/etc/vmops"
  /bin/cp -f $deploydir/conf/agent.properties $confdir/agent.properties
  if [ $? -gt 0 ]; then
    printf "ERROR: Failed to copy the agent.properties file into the right place."
    exit 10;
  fi
else
  confdir="$deploydir/conf"
fi

if [ "$typ" != "" ]; then
  sed s/@TYPE@/"$typ"/ $confdir/agent.properties > $confdir/tmp
  /bin/mv -f $confdir/tmp $confdir/agent.properties
else
  printf "INFO: Type is not set\n"
fi

if [ "$host" != "" ]; then
  sed s/@HOST@/"$host"/ $confdir/agent.properties > $confdir/tmp
  /bin/mv -f $confdir/tmp $confdir/agent.properties
else
  printf "INFO: host is not set\n"
fi

if [ "$port" != "" ]; then
  sed s/@PORT@/"$port"/ $confdir/agent.properties > $confdir/tmp
  /bin/mv -f $confdir/tmp $confdir/agent.properties
else
  printf "INFO: Port is not set\n"
fi

if [ "$pod" != "" ]; then
  sed s/@POD@/"$pod"/ $confdir/agent.properties > $confdir/tmp
  /bin/mv -f $confdir/tmp $confdir/agent.properties
else
  printf "INFO: Pod is not set\n"
fi

if [ "$zone" != "" ]; then
  sed s/@ZONE@/"$zone"/ $confdir/agent.properties > $confdir/tmp
  /bin/mv -f $confdir/tmp $confdir/agent.properties
else
  printf "INFO: Zone is not set\n"
fi

if [ "$workers" != "" ]; then
  sed s/@WORKERS@/"$workers"/ $confdir/agent.properties > $confdir/tmp
  /bin/mv -f $confdir/tmp $confdir/agent.properties
else
  printf "INFO: Workers is not set\n"
fi

printf "SUCCESS: Installation is now complete. If you like to make changes, edit $confdir/agent.properties\n"
exit 0