#!/bin/bash
# Copyright 2015-2016 Metaswitch Networks

exec >> /tmp/calico_route_reflector.log 2>&1

set -x

echo "================ `date` ================"
echo "Hi, I'm a route_reflector node!"

# Calculate IP addresses of route reflector peers.
this_node_address=$(python get_node_ip.py `hostname`)
controller_node_addresses=$(python get_node_ips_by_role.py controller)
route_reflector_peers=("${controller_node_addresses[@]/$this_node_address}")

# Get compute host names from etcd.
compute_hosts=$(etcdctl ls /calico/felix/v1/host | xargs -n1 basename)

# Use /etc/hosts to convert each compute host name to an IP address.
client_peers=
for compute in $compute_hosts; do
    compute=$(awk "/$compute/{print \$1;}" /etc/hosts)
    client_peers="$client_peers $compute"
done

# Generate basic config for a BIRD BGP route reflector.
cat > /etc/bird/bird.conf <<EOF
# Configure logging
log syslog { debug, trace, info, remote, warning, error, auth, fatal, bug };
log stderr all;
#log "tmp" all;

# Override router ID
router id $this_node_address;


filter import_kernel {
if ( net != 0.0.0.0/0 ) then {
   accept;
   }
reject;
}

# Turn on global debugging of all protocols
debug protocols all;

# This pseudo-protocol watches all interface up/down events.
protocol device {
  scan time 2;    # Scan interfaces every 10 seconds
}
EOF

# Add a BGP protocol stanza for all peers.
for node in ${client_peers[@]} ${route_reflector_peers[@]}; do
  cat >> /etc/bird/bird.conf <<EOF
protocol bgp {
  local as 64511;
  neighbor $node as 64511;
  multihop;
EOF

  if [[ "${client_peers[@]}" =~ "${node}" ]]; then
    cat >> /etc/bird/bird.conf <<EOF
  description "Client $node";
  rr client;
EOF
  else
    cat >> /etc/bird/bird.conf <<EOF
  description "Route Reflector $node";
EOF
  fi

  cat >> /etc/bird/bird.conf <<EOF
  rr cluster id 1.2.3.4;
  import all;
  export all;
  source address ${this_node_address};
}
EOF
done

# Restart BIRD with the new config.
service bird restart

exit 0