pre.sh

#!/bin/sh

# setup permissions
# setup promgrafana dashboards\sources

TMPPATH=$(mktemp -d -p /tmp)
TMPPATH1=$(mktemp -d -p /tmp)

CREATEDIR() {
    mkdir -p "$INSTALLPATH"/data/promgrafana/etc/provisioning/datasources
    mkdir -p "$INSTALLPATH"/data/promgrafana/etc/provisioning/notifiers
    mkdir -p "$INSTALLPATH"/data/promgrafana/etc/provisioning/dashboards
    mkdir -p "$INSTALLPATH"/data/promgrafana/etc/dashboards
    mkdir -p "$INSTALLPATH"/data/promgrafana/plugins
    mkdir -p "$INSTALLPATH"/data/promgrafana/db
    mkdir -p "$INSTALLPATH"/data/promvm
    mkdir -p "$INSTALLPATH"/data/prometheus/metrics
    mkdir -p "$INSTALLPATH"/data/prometheus/etc/rules.d
    mkdir -p "$INSTALLPATH"/data/consul
    mkdir -p "$INSTALLPATH"/data/clickhouse/data
    mkdir -p "$INSTALLPATH"/data/nsq
    mkdir -p "$INSTALLPATH"/data/mongo
    mkdir -p "$INSTALLPATH"/data/noc/custom
    mkdir -p "$INSTALLPATH"/data/postgres
    mkdir -p "$INSTALLPATH"/data/nginx/ssl
    mkdir -p "$INSTALLPATH"/data/grafana/plugins
    mkdir -p "$INSTALLPATH"/data/sentry/redis
    mkdir -p "$INSTALLPATH"/data/sentry/pg
}

SETPERMISSION() {
    chown 101 -R "$INSTALLPATH"/data/clickhouse/data
    chown 999 -R "$INSTALLPATH"/data/postgres
    chown 999 -R "$INSTALLPATH"/data/mongo
    chown 472 -R "$INSTALLPATH"/data/grafana/
    chown 65534 -R "$INSTALLPATH"/data/prometheus/metrics
    chown 472 -R "$INSTALLPATH"/data/promgrafana/plugins
    chown 999 -R "$INSTALLPATH"/data/sentry/redis
    chown 70 -R "$INSTALLPATH"/data/sentry/pg
}

SETUPPROMGRAFANA() {
    echo "Setup GRAFANA dashboards from code.getnoc.com/noc/grafana-selfmon-dashboards"
    echo "---"
    cd "$TMPPATH" && git clone -q https://code.getnoc.com/noc/grafana-selfmon-dashboards.git .
    cp -f -r "$TMPPATH"/dashboards/* "$INSTALLPATH"/data/promgrafana/etc/dashboards
    cp -f -r "$TMPPATH"/provisioning/* "$INSTALLPATH"/data/promgrafana/etc/provisioning
}

SETUPPROMRULES() {
    echo "Setup PROMETHEUS alert rules from code.getnoc.com/noc/noc-prometheus-alerts.git"
    echo "---"
    cd "$TMPPATH1" && git clone -q https://code.getnoc.com/noc/noc-prometheus-alerts.git .
    cp -f "$TMPPATH1"/*.yml "$INSTALLPATH"/data/prometheus/etc/rules.d
}

SETUPSENTRY() {
    if [ ! -f "$INSTALLPATH"/data/sentry/sentry.env ]
        then
# @TODO
            GENERATE_PASSWORD="$(dd if=/dev/urandom bs=1 count=32 2>/dev/null | base64 -w 0 | rev | cut -b 2- | rev)"

            echo "Setup Sentry env in $INSTALLPATH/data/sentry/sentry.env"
            echo "after firsh start container need run command for make migration and setup admin user\passwd"
            echo "cd $INSTALLPATH && docker-compose exec sentry sentry upgrade"
            echo "---"
            { echo SENTRY_POSTGRES_HOST=sentry-postgres
              echo SENTRY_DB_NAME=sentry
              echo SENTRY_DB_USER=sentry
              echo SENTRY_DB_PASSWORD="$GENERATE_PASSWORD"
              echo SENTRY_SECRET_KEY="$(dd 'if=/dev/random' 'bs=1' 'count=32' 2>/dev/null | base64)"
              echo SENTRY_REDIS_HOST=sentry-redis
              echo SENTRY_METRICS_SAMPLE_RATE=0.9
              echo POSTGRES_USER=sentry
              echo POSTGRES_DBNAME=sentry
              echo POSTGRES_PASSWORD="$GENERATE_PASSWORD"
              echo "#Important!!! POSTGRES_PASSWORD == SENTRY_DB_PASSWORD"
            } >> "$INSTALLPATH"/data/sentry/sentry.env
    fi
}

SETUPNOCCONF() {
    if [ ! -f "$INSTALLPATH"/data/noc/etc/noc.conf ]
        then
            echo "Copy " "$INSTALLPATH"/data/noc/etc/noc.conf.example " to " "$INSTALLPATH"/data/noc/etc/noc.conf
            echo "---"
            # shellcheck disable=SC2086
            cp $INSTALLPATH/data/noc/etc/noc.conf.example $INSTALLPATH/data/noc/etc/noc.conf
    fi
}

# @TODO
# need check $INSTALLPATH == $COMPOSEPATH and make warning if not
SETUPENV() {
    if [ ! -f "$INSTALLPATH"/.env ]
        then
            echo "Setup COMPOSEPATH=$INSTALLPATH in $INSTALLPATH/.env"
            echo "---"
            { echo "COMPOSEPATH=$INSTALLPATH"
              echo "NOC_VERSION_TAG=$PARAM_TAG"
              echo "COMPOSE_HTTP_TIMEOUT=300"
              echo "# logging driver: json-file, local, journald"
              echo "COMPOSE_LOG_DRIVER=json-file"
              echo "COMPOSE_LOG_MAX_SIZE=10m"
              echo "COMPOSE_LOG_MAX_FILE=1"
            } >> "$INSTALLPATH"/.env
    fi
}

while [ -n "$1" ]
do
    case "$1" in
        -t) PARAM_TAG="$2"
            #echo "Found the -t option, with parameter value $PARAM_TAG"
            shift ;;
        -p) PARAM_P="$2"
            #echo "Found the -p option, with parameter value $PARAM_P"
            shift ;;
        -d) INSTALLPATH="$2"
            #echo "Found the -d option, with parameter value $INSTALLPATH"
            shift ;;
        -h) echo "Example: ./pre.sh -p all -d /opt/noc-dc -t stable"
            shift ;;
        --) shift
            break ;;
        *) echo "Example: ./pre.sh -p all -d /opt/noc-dc -t stable";;
    esac
    shift
done

if [ -z "$INSTALLPATH" ]
    then
        INSTALLPATH=/opt/noc-dc
        echo "Setup NOC-DC to: $INSTALLPATH"
        echo "---"
fi

if [ -z "$PARAM_TAG" ]
    then
        PARAM_TAG="stable"
        echo "Setup Docker use image with tag: $PARAM_TAG"
        echo "See all tags in https://code.getnoc.com/noc/noc/container_registry"
        echo "---"
fi

if [ -n "$PARAM_P" ]
    then
        if [ "$PARAM_P" = "all" ]
            then
                CREATEDIR
                SETUPENV
                SETPERMISSION
                SETUPPROMGRAFANA
                SETUPPROMRULES
                SETUPNOCCONF
                SETUPSENTRY
        elif [ "$PARAM_P" = "perm" ]
            then
                CREATEDIR
                SETPERMISSION
        elif [ "$PARAM_P" = "grafana" ]
            then
                CREATEDIR
                SETUPPROMGRAFANA
        elif [ "$PARAM_P" = "promrules" ]
            then
                CREATEDIR
                SETUPPROMRULES
        elif [ "$PARAM_P" = "nocconf" ]
            then
                SETUPNOCCONF
        elif [ "$PARAM_P" = "sentry" ]
            then
                SETUPSENTRY
        elif [ "$PARAM_P" = "env" ]
            then
                SETUPENV
        else
            echo "Unknown parameter for -p"
            echo "Use one of: all,env,perm,grafana,promrules,nocconf,sentry"
        fi
else
    echo "No -p parameters found."
    echo "Use one of: all,env,perm,grafana,promrules,nocconf,sentry"
fi