#!/bin/bash -e

FIRST_START_DONE="/etc/docker-keepalived-first-start-done"

# container first start
if [ ! -e "$FIRST_START_DONE" ]; then

  # config folder is empty use bootstrap config if available
  if [ ! -e /etc/keepalived/keepalived.conf ]; then
    echo "No keepalived.conf provided using image default one"
    if [ ! -e /container/service/keepalived/assets/keepalived.conf ]; then
      echo "Error: No default keepalived.conf found in /container/service/keepalived/assets/keepalived.conf"
      exit 1
    else

      ln -s /container/service/keepalived/assets/keepalived.conf /etc/keepalived/keepalived.conf

      #
      # bootstrap config
      #
      sed -i "s|{{ keepalived_interface }}|$KEEPALIVED_INTERFACE|g" /etc/keepalived/keepalived.conf
      sed -i "s|{{ keepalived_priority }}|$KEEPALIVED_PRIORITY|g" /etc/keepalived/keepalived.conf
      sed -i "s|{{ keepalived_password }}|$KEEPALIVED_PASSWORD|g" /etc/keepalived/keepalived.conf

      if [ -n "$KEEPALIVED_NOTIFY" ]; then
        sed -i "s|{{ keepalived_notify }}|notify \"$KEEPALIVED_NOTIFY\"|g" /etc/keepalived/keepalived.conf
      else
        sed -i "/{{ keepalived_notify }}/d" /etc/keepalived/keepalived.conf
      fi

      # unicast peers
      KEEPALIVED_UNICAST_PEERS=($KEEPALIVED_UNICAST_PEERS)
      for peer in "${KEEPALIVED_UNICAST_PEERS[@]}"
      do
        # it's just a peer
        # stored in a variable
        if [ -n "${!peer}" ]; then
          sed -i "s|{{ keepalived_unicast_peers }}|${!peer}\n    {{ keepalived_unicast_peers }}|g" /etc/keepalived/keepalived.conf
        # directly
        else
          sed -i "s|{{ keepalived_unicast_peers }}|${peer}\n    {{ keepalived_unicast_peers }}|g" /etc/keepalived/keepalived.conf
        fi
      done
      sed -i "/{{ keepalived_unicast_peers }}/d" /etc/keepalived/keepalived.conf

      # virtual ips
      KEEPALIVED_VIRTUAL_IPS=($KEEPALIVED_VIRTUAL_IPS)
      for vip in "${KEEPALIVED_VIRTUAL_IPS[@]}"
      do
        # it's just a peer
        # stored in a variable
        if [ -n "${!vip}" ]; then
          sed -i "s|{{ keepalived_virtual_ips }}|${!vip}\n    {{ keepalived_virtual_ips }}|g" /etc/keepalived/keepalived.conf
        # directly
        else
          sed -i "s|{{ keepalived_virtual_ips }}|${vip}\n    {{ keepalived_virtual_ips }}|g" /etc/keepalived/keepalived.conf
        fi
      done
      sed -i "/{{ keepalived_virtual_ips }}/d" /etc/keepalived/keepalived.conf
    fi

  fi

  touch $FIRST_START_DONE
fi

exit 0