#!/usr/bin/bash

# Modeled closely after flask portion of amazon/aws-eb-python:3.4.2-onbuild-3.5.1 entrypoint script

UWSGI_NUM_PROCESSES='4'
UWSGI_NUM_THREADS='15'
UWSGI_LISTEN_BACKLOG='1024'
UWSGI_UID='uwsgi'
UWSGI_GID='uwsgi'
# increase header buffer size if one is seeing 502s & 'invalid request block size' errors in uwsgi log
UWSGI_BUFFER_SIZE='12288'
UWSGI_TIMEOUT='90'
UWSGI_HEADERS='Connection:Keep-Alive'

set -xv

cd /var/app

# Flask support
if cat Pipfile.lock | grep -q -i Flask && [ -z "${WSGI_MODULE}" ]; then
  UWSGI_MODULE='--module application:app'
fi

# defaulting to application.py if not explicitly set
[ -z "${WSGI_PATH}" ] && WSGI_PATH=application.py

DDTRACE_BINARY=$(command -v ddtrace-run)

uwsgi --version

if [ -x ${DDTRACE_BINARY} ]; then
  ${DDTRACE_BINARY} uwsgi \
    --cache2 name=uwsgi_cache,items=10 \
    --http11-socket :${UWSGI_PORT:-8080} \
    --need-plugin python3  \
    --chdir /var/app \
    --wsgi-file ${WSGI_PATH} ${UWSGI_MODULE} \
    --master \
    --processes ${UWSGI_NUM_PROCESSES} \
    --threads ${UWSGI_NUM_THREADS} \
    --listen ${UWSGI_LISTEN_BACKLOG} \
    --uid ${UWSGI_UID} \
    --gid ${UWSGI_GID} \
    -t ${UWSGI_TIMEOUT} \
    --so-keepalive \
    --add-header ${UWSGI_HEADERS} \
    --buffer-size ${UWSGI_BUFFER_SIZE} \
    --virtualenv /var/app/.venv \
    --plugin python3
else
  uwsgi \
    --cache2 name=uwsgi_cache,items=10 \
    --http11-socket :${UWSGI_PORT:-8080} \
    --need-plugin python3  \
    --chdir /var/app \
    --wsgi-file ${WSGI_PATH} ${UWSGI_MODULE} \
    --master \
    --processes ${UWSGI_NUM_PROCESSES} \
    --threads ${UWSGI_NUM_THREADS} \
    --listen ${UWSGI_LISTEN_BACKLOG} \
    --uid ${UWSGI_UID} \
    --gid ${UWSGI_GID} \
    -t ${UWSGI_TIMEOUT} \
    --so-keepalive \
    --add-header ${UWSGI_HEADERS} \
    --buffer-size ${UWSGI_BUFFER_SIZE} \
    --virtualenv /var/app/.venv \
    --plugin python3
fi
