#!/bin/bash

# define some colors to use for output
BLACK=$(tput setaf 0)
RED=$(tput setaf 1)
GREEN=$(tput setaf 2)
YELLOW=$(tput setaf 3)
LIME_YELLOW=$(tput setaf 190)
POWDER_BLUE=$(tput setaf 153)
BLUE=$(tput setaf 4)
MAGENTA=$(tput setaf 5)
CYAN=$(tput setaf 6)
WHITE=$(tput setaf 7)
BRIGHT=$(tput bold)
NORMAL=$(tput sgr0)
BLINK=$(tput blink)
REVERSE=$(tput smso)
UNDERLINE=$(tput smul)

function get_ip_for_pod(){
    local pod_name=$1
    until kubectl get pod ${pod_name} -o jsonpath='{.status.phase}' | grep Running > /dev/null ; do sleep 1 ; done

    local pod_ip=`kubectl get pod ${pod_name} --template={{.status.podIP}}`
    echo "${pod_ip}"
}

function wait_for_k8s_job_completion(){
    local job_name=$1
    until kubectl get jobs ${job_name} -o jsonpath='{.status.conditions[?(@.type=="Complete")].status}' | grep True ; do printf "." && sleep 1 ; done
}

function header(){
    local msg=$1
    printf "\n${REVERSE}${msg}${NORMAL}\n"
}

function log(){
    local msg=$1
    printf "${WHITE}${msg}${NORMAL}\n"
}


TESTER_JOB_NAME=iptables-tester
LAB_YAML_FILE=iptables/iptablestest-lab.yaml

header "Deleting any existing objects from previous test runs..."
kubectl delete -f ${LAB_YAML_FILE}
kubectl delete  jobs/${TESTER_JOB_NAME}

header "Building the image used in tests..."
docker build . -f iptables/Dockerfile-tester --tag buoyantio/iptables-tester:v1
sleep 10

header "Creating the test lab..."
kubectl create -f ${LAB_YAML_FILE}

POD_WITH_NO_RULES_IP=$(get_ip_for_pod "pod-with-no-rules")
log "POD_WITH_NO_RULES_IP=${POD_WITH_NO_RULES_IP}"

POD_REDIRECTS_ALL_PORTS_IP=$(get_ip_for_pod "pod-redirects-all-ports")
log "POD_REDIRECTS_ALL_PORTS_IP=${POD_REDIRECTS_ALL_PORTS_IP}"

POD_REDIRECTS_WHITELISTED_IP=$(get_ip_for_pod "pod-redirects-whitelisted")
log "POD_REDIRECTS_WHITELISTED_IP=${POD_REDIRECTS_WHITELISTED_IP}"

POD_DOEST_REDIRECT_BLACKLISTED_IP=$(get_ip_for_pod "pod-doesnt-redirect-blacklisted")
log "POD_DOEST_REDIRECT_BLACKLISTED_IP=${POD_DOEST_REDIRECT_BLACKLISTED_IP}"

header "Running tester..."
cat <<EOF | kubectl create -f -
apiVersion: batch/v1
kind: Job
metadata:
  name: ${TESTER_JOB_NAME}
spec:
  template:
    metadata:
      name: ${TESTER_JOB_NAME}
    spec:
      containers:
      - name: tester
        image: buoyantio/iptables-tester:v1
        env:
          - name: CONDUIT_INTEGRATION_TESTS_ENABLED
            value: "1"
          - name: POD_REDIRECTS_ALL_PORTS_IP
            value: ${POD_REDIRECTS_ALL_PORTS_IP}
          - name: POD_WITH_NO_RULES_IP
            value: ${POD_WITH_NO_RULES_IP}
          - name: POD_REDIRECTS_WHITELISTED_IP
            value: ${POD_REDIRECTS_WHITELISTED_IP}
          - name: POD_DOEST_REDIRECT_BLACKLISTED_IP
            value: ${POD_DOEST_REDIRECT_BLACKLISTED_IP}
      restartPolicy: Never
EOF

wait_for_k8s_job_completion $TESTER_JOB_NAME

header "Test output:"
kubectl logs jobs/${TESTER_JOB_NAME}

# Makes this script return status 0 if the test returned status 0
kubectl logs jobs/${TESTER_JOB_NAME} | grep "status:0" > /dev/null