setupLocalProxy.sh

#!/usr/bin/env bash
#
# Used to run Proxy locally.
#

# Stop the script if any commands fail
set -o errexit
set -o pipefail

# XYZ = ${XYZ:-8000}
# if env variable $XYZ is set, use that for XYZ, otherwise use 8000
LOCAL_IDLER_PORT=${LOCAL_IDLER_PORT:-9001}
LOCAL_TENANT_PORT=${LOCAL_TENANT_PORT:-9002}
LOCAL_POSTGRES_PORT=${LOCAL_POSTGRES_PORT:-5432}

###############################################################################
# Prints help message
# Globals:
#   None
# Arguments:
#   None
# Returns:
#   None
###############################################################################
printHelp() {
    cat << EOF
Usage: ${0##*/} [start|stop|env|unset]

This script is used to run the Jenkins Proxy on localhost.
As a prerequisite OPENSHIFT_API_TOKEN needs to be exported.
In your shell (from the root of fabric8-jenkins-proxy):

> export DSAAS_PREVIEW_TOKEN=<dsaas-preview token>
> ./scripts/${0##*/} start
# Run command below this in a separate terminal so that we can see logs of the above command.
> export DSAAS_PREVIEW_TOKEN=<dsaas-preview token>
> eval \$(./scripts/${0##*/} env)
> fabric8-jenkins-proxy
EOF
}

###############################################################################
# Wraps oc command custom config location
# Globals:
#   None
# Arguments:
#   Passes all arguments to oc command
# Returns:
#   None
###############################################################################
loc() {
    oc --config $(dirname $0)/config $@
}

###############################################################################
# Ensures login to OpenShift
# Globals:
#   None
# Arguments:
#   None
# Returns:
#   None
###############################################################################
login() {
    [ -z "${DSAAS_PREVIEW_TOKEN}" ] && echo "DSAAS_PREVIEW_TOKEN needs to be exported." && printHelp && exit 1

    loc login https://api.dsaas-stg.openshift.com -n dsaas-preview --token=${DSAAS_PREVIEW_TOKEN} >/dev/null
}

generateCertificates() {
    # generate only if needed
    [[ -r server.key ]] && [[ -r server.crt ]] && return

    openssl req -newkey rsa:2048 -nodes -keyout ./server.key -x509 -days 365 -out ./server.crt
}

###############################################################################
# Retrieves the required Auth token
# Globals:
#   JC_AUTH_TOKEN - Auth token for the auth service
# Arguments:
#   None
# Returns:
#   None
###############################################################################
setTokens() {
    pod=$(loc get pods -l deploymentconfig=jenkins-proxy -o json | jq -r '.items[0].metadata.name')
    if [ "${pod}" == "null" ] ; then
        echo "WARN: Unable to determine Proxy pod name"
        return
    fi

    export JC_AUTH_TOKEN=$(loc exec ${pod} env | grep JC_AUTH_TOKEN | sed -e 's/JC_AUTH_TOKEN=//')
}

###############################################################################
# Forwards the jenkins-idler service to localhost
# Globals:
#   LOCAL_IDLER_PORT - local Idler port
# Arguments:
#   None
# Returns:
#   None
###############################################################################
forwardIdler() {
    pod=$(loc get pods -l deploymentconfig=jenkins-idler -o json | jq -r '.items[0].metadata.name')
    if [ "${pod}" == "null" ] ; then
        echo "WARN: Unable to determine Idler pod name"
        return
    fi
    port=$(loc get pods -l deploymentconfig=jenkins-idler -o json | jq -r '.items[0].spec.containers[0].ports[0].containerPort')

    if lsof -Pi :${LOCAL_IDLER_PORT} -sTCP:LISTEN -t >/dev/null ; then
        echo "INFO: Local Idler port ${LOCAL_IDLER_PORT} already listening. Skipping oc port-forward"
        return
    fi

    while :
    do
	    loc port-forward ${pod} ${LOCAL_IDLER_PORT}:${port}
	    echo "Idler port forward stopped with exit code $?.  Respawning.." >&2
	    sleep 1
    done
    echo "Idler port forward stopped." >&2
}

###############################################################################
# Forwards the f8tenant service to localhost
# Globals:
#   LOCAL_TENANT_PORT - local tenant port
# Arguments:
#   None
# Returns:
#   None
###############################################################################
forwardTenant() {
    pod=$(loc get pods -l deploymentconfig=f8tenant -o json | jq -r '.items[0].metadata.name')
    if [ "${pod}" == "null" ] ; then
        echo "WARN: Unable to determine Tenant pod name"
        return
    fi
    port=$(loc get pods -l deploymentconfig=f8tenant -o json | jq -r '.items[0].spec.containers[0].ports[0].containerPort')

    if lsof -Pi :${LOCAL_TENANT_PORT} -sTCP:LISTEN -t >/dev/null ; then
        echo "INFO: Local Tenant port ${LOCAL_TENANT_PORT} already listening. Skipping oc port-forward"
        return
    fi

    while :
    do
	    loc port-forward ${pod} ${LOCAL_TENANT_PORT}:${port}
	    echo "Tenant port forward stopped with exit code $?.  Respawning.." >&2
	    sleep 1
    done
    echo "Tenant port forward stopped." >&2
}

###############################################################################
# Runs a Postgres Docker container on port 5432
# Globals:
#   None
# Arguments:
#   None
# Returns:
#   None
###############################################################################
runPostgres() {
    container=$(docker ps -q --filter "name=postgres")
    if [ -z "${container}" ] ; then
        docker run --name postgres -e POSTGRES_PASSWORD=postgres -d -p ${LOCAL_POSTGRES_PORT}:5432 postgres 1>&2
    fi
}

###############################################################################
# Starts the port forwarding as well as the Postgres Docker container.
# Globals:
#   None
# Arguments:
#   None
# Returns:
#   None
###############################################################################
start() {
    [ -z "${DSAAS_PREVIEW_TOKEN}" ] && echo "DSAAS_PREVIEW_TOKEN needs to be exported." && printHelp && exit 1

    generateCertificates
    login

    loc get pods > /dev/null
    [ "$?" -ne 0 ] && echo "Your OpenShift token is not valid" && exit 1

    forwardIdler &
    forwardTenant &
    runPostgres &
}

###############################################################################
# Displays the required environment settings for evaluation.
# Globals:
#   None
# Arguments:
#   None
# Returns:
#   None
###############################################################################
env() {
    [ -z "${DSAAS_PREVIEW_TOKEN}" ] && echo "DSAAS_PREVIEW_TOKEN needs to be exported." && printHelp && exit 1

    login
    setTokens

    echo export JC_WIT_API_URL=https://api.prod-preview.openshift.io
    echo export JC_REDIRECT_URL=https://localhost:8080
    echo export JC_ENABLE_HTTPS=true
    echo export JC_AUTH_URL=https://auth.prod-preview.openshift.io
    echo export JC_POSTGRES_PORT=${LOCAL_POSTGRES_PORT}
    echo export JC_POSTGRES_HOST=localhost
    echo export JC_POSTGRES_USER=postgres
    echo export JC_POSTGRES_PASSWORD=postgres
    echo export JC_POSTGRES_DATABASE=postgres
    echo export JC_AUTH_TOKEN=${JC_AUTH_TOKEN}
    echo export JC_IDLER_API_URL=http://localhost:${LOCAL_IDLER_PORT}
    echo export JC_F8TENANT_API_URL=http://localhost:${LOCAL_TENANT_PORT}
    echo export JC_OSO_CLUSTERS="'{\"https://api.free-stg.openshift.com/\": \"1b7d.free-stg.openshiftapps.com\"}'"
    echo export JC_INDEX_PATH=static/html/index.html
}

###############################################################################
# Unsets the exported Idler environment variables.
# Globals:
#   None
# Arguments:
#   None
# Returns:
#   None
###############################################################################
unsetEnv() {
    echo unset JC_WIT_API_URL
    echo unset JC_REDIRECT_URL
    echo unset JC_ENABLE_HTTPS
    echo unset JC_AUTH_URL
    echo unset JC_POSTGRES_PORT
    echo unset JC_POSTGRES_HOST
    echo unset JC_POSTGRES_USER
    echo unset JC_POSTGRES_PASSWORD
    echo unset JC_POSTGRES_DATABASE
    echo unset JC_AUTH_TOKEN
    echo unset JC_IDLER_API_URL
    echo unset JC_F8TENANT_API_URL
    echo unset JC_OSO_CLUSTERS
}

###############################################################################
# Stops oc-port forwarding and Docker container
# Globals:
#   None
# Arguments:
#   None
# Returns:
#   None
###############################################################################
stop() {
    pids=$(pgrep -a -f -d " " "setupLocalProxy.sh start")
    pids+=$(pgrep -a -f -d " " "oc --config $(dirname $0)/config")
    kill -9 ${pids}
    docker rm -f postgres
}

case "$1" in
  start)
    start
    ;;
  stop)
    stop
    ;;
  env)
    env
    ;;
  unset)
    unsetEnv
    ;;
  *)
    printHelp
esac