#!/bin/bash

# Simple script to provision AIRFLOW_CONNECTIONS !

export SQLALCHEMY_SILENCE_UBER_WARNING=1
#export AIRFLOW_CONN_AIRFLOW_DB="${connection}"
export AIRFLOW__DATABASE__SQL_ALCHEMY_CONN="${connection}"
export AIRFLOW__CORE__SQL_ALCHEMY_CONN="${connection}"

echo "====================================="
conns=$(compgen -v  -X '!*AIRFLOW_CONN_*')

if [ -z "${conns}" ];then
   echo " Nothing to do"
   echo "====================================="
else
   for conn in ${conns}
   do
      name="${conn#"AIRFLOW_CONN_"}"
      value=$(eval "echo -e ${!conn}")
      echo "Creating ${name}: ${value}"
      ex=$(airflow connections add "${name}" --conn-uri ${value} 2>&1)
      if [ "${?}" -ne 0 ]; then
         echo "${conn}: Bad connection definition"
         echo "= Error =========================="
         echo "${ex}"
         echo "= End error======================="
      else
         echo "= Ok ================================"
      fi
   done
fi