prometheus-deploy.sh

#!/usr/bin/env bash
# Deploys prometheus-operator Helm Chart and Forgerock Metrics which include custom
# endpoints, alerting rules and Grafana dashboards.
# ./prometheus-deploy.sh -n namespace - deploy to different namespace than monitoring.
# ./prometheus-deploy.sh -v values file - use different custom values file.
# ./prometheus-deploy.sh -d - delete deployment.
#
# Prerequisite: Helm version 3.04 or higher.

# You can deploy your own custom values file by using the -v <values file> flag.
set -oe pipefail

DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
ADDONS_BASE="${ADDONS_BASE:-${DIR}/../cluster/addons}"
ADDONS_DIR="${ADDONS_BASE}/prometheus"
PROM_VALUES="${ADDONS_DIR}/prometheus-operator.yaml"
NAMESPACE=monitoring

USAGE="Usage: $0 [-n <namespace>] [-v <values file>] [-d]"

# Create namespace
create_ns() {
    ns=$(kubectl get namespace | grep $NAMESPACE | awk '{ print $1 }' || true)

    if [ -z "${ns}" ]; then
        kubectl create namespace $NAMESPACE
    else
        printf "*** $NAMESPACE namespace already exists ***\n"
    fi
}

# deploy Prometheus Operator and forgerock metrics
deploy() {

    # Deploy self-signed cert-manager issuer to generate a certificate for the webhook connection
    kubectl apply -f - <<EOF
apiVersion: cert-manager.io/v1
kind: Issuer
metadata:
    name: selfsigned-issuer
    namespace: $NAMESPACE
spec:
    selfSigned: {}
EOF

    # Add prometheus-community repo to helm
    helm repo add "prometheus-community" "https://prometheus-community.github.io/helm-charts" --force-update
    helm repo update
    helm upgrade -i prometheus-operator prometheus-community/kube-prometheus-stack  -f $PROM_VALUES --namespace=$NAMESPACE

    kubectl -n $NAMESPACE wait --for condition=established --timeout=60s \
        crd/prometheuses.monitoring.coreos.com \
        crd/servicemonitors.monitoring.coreos.com \
        crd/servicemonitors.monitoring.coreos.com \
        crd/podmonitors.monitoring.coreos.com \
        crd/alertmanagers.monitoring.coreos.com \
        crd/alertmanagerconfigs.monitoring.coreos.com \
        crd/probes.monitoring.coreos.com

    kubectl -n $NAMESPACE wait --for condition=Ready --timeout=60s pod --all
    # Install/Upgrade forgerock-servicemonitors
    helm upgrade -i forgerock-metrics ${ADDONS_DIR}/forgerock-metrics --namespace=$NAMESPACE
}

# Delete all
delete() {

    set +e

    # Delete forgerock-metrics Helm chart
    helm delete forgerock-metrics --namespace=$NAMESPACE

    # Delete Prometheus Operator Helm chart
    helm uninstall prometheus-operator --namespace=$NAMESPACE

    # Delete CRDs
    kubectl delete --wait=true crd prometheuses.monitoring.coreos.com
    kubectl delete --wait=true crd prometheusrules.monitoring.coreos.com
    kubectl delete --wait=true crd servicemonitors.monitoring.coreos.com
    kubectl delete --wait=true crd podmonitors.monitoring.coreos.com
    kubectl delete --wait=true crd alertmanagers.monitoring.coreos.com
    kubectl delete --wait=true crd alertmanagerconfigs.monitoring.coreos.com
    kubectl delete --wait=true crd probes.monitoring.coreos.com
    kubectl delete --wait=true crd thanosrulers.monitoring.coreos.com

    # Delete monitoring namespace
    kubectl delete ns $NAMESPACE
    exit 1
}

# Output help if -h is included
if [[ $1 == "-h" ]];then
    echo $USAGE
    echo "-n <namespace>    namespace"
    echo "-v <values file>  add custom values file for Prometheus operator (relative to $ADDONS_DIR)"
    echo "-d delete Prometheus Operator and Forgerock metrics"
    exit
fi

# Read arguments
while getopts :n:v:d option; do
    case "${option}" in
        n) NAMESPACE=${OPTARG};;
        v) VALUES=${OPTARG};;
        \?) echo "Error: Incorrect usage"
            echo $USAGE
            exit 1;;
    esac
done

echo -e "\n**This script requires Helm version 3.04 or later due to changes in the behaviour of 'helm repo add' command.**\n"

## Validate arguments
# set custom yaml file if not provided with the -v arg
[ $VALUES ] && PROM_VALUES="${ADDONS_DIR}/${VALUES}"
# delete chart if -d select
[[ ${1} =~ "-d" ]] && delete

# Create namespace
create_ns

# Deploy to cluster
deploy