frappe_docker/build/common/worker/docker-entrypoint.sh

#!/bin/bash

function configureEnv() {
  if [ ! -f /home/frappe/frappe-bench/sites/common_site_config.json ]; then

    if [[ -z "$MARIADB_HOST" ]]; then
      if [[ -z "$POSTGRES_HOST" ]]; then
        echo "MARIADB_HOST or POSTGRES_HOST is not set"
        exit 1
      fi
    fi

    if [[ -z "$REDIS_CACHE" ]]; then
      echo "REDIS_CACHE is not set"
      exit 1
    fi

    if [[ -z "$REDIS_QUEUE" ]]; then
      echo "REDIS_QUEUE is not set"
      exit 1
    fi

    if [[ -z "$REDIS_SOCKETIO" ]]; then
      echo "REDIS_SOCKETIO is not set"
      exit 1
    fi

    if [[ -z "$SOCKETIO_PORT" ]]; then
      echo "SOCKETIO_PORT is not set"
      exit 1
    fi

    if [[ -z "$DB_PORT" ]]; then
      export DB_PORT=3306
    fi

    export DB_HOST="${MARIADB_HOST:-$POSTGRES_HOST}"

    envsubst '${DB_HOST}
      ${DB_PORT}
      ${REDIS_CACHE}
      ${REDIS_QUEUE}
      ${REDIS_SOCKETIO}
      ${SOCKETIO_PORT}' < /opt/frappe/common_site_config.json.template > /home/frappe/frappe-bench/sites/common_site_config.json
  fi
}

function checkConnection() {
  . /home/frappe/frappe-bench/env/bin/activate \
    && python /home/frappe/frappe-bench/commands/check_connection.py
}

function checkConfigExists() {
  COUNTER=0
  while [[ ! -e /home/frappe/frappe-bench/sites/common_site_config.json ]] && [[ $COUNTER -le 30 ]] ; do
      sleep 1
      (( COUNTER=COUNTER+1 ))
      echo "config file not created, retry $COUNTER"
  done

  if [[ ! -e /home/frappe/frappe-bench/sites/common_site_config.json ]]; then
    echo "timeout: config file not created"
    exit 1
  fi
}

if [[ ! -e /home/frappe/frappe-bench/sites/apps.txt ]]; then
  find /home/frappe/frappe-bench/apps -mindepth 1 -maxdepth 1 -type d -printf '%f\n' | sort -r > /home/frappe/frappe-bench/sites/apps.txt
fi

# symlink node_modules
ln -sfn /home/frappe/frappe-bench/sites/assets/frappe/node_modules \
  /home/frappe/frappe-bench/apps/frappe/node_modules

if [ "$1" = 'start' ]; then
  configureEnv
  checkConnection

  if [[ -z "$WORKERS" ]]; then
    export WORKERS=2
  fi

  if [[ -z "$FRAPPE_PORT" ]]; then
    export FRAPPE_PORT=8000
  fi

  if [[ -z "$WORKER_CLASS" ]]; then
    export WORKER_CLASS=gthread
  fi

  export LOAD_CONFIG_FILE=""
  if [ "$WORKER_CLASS" = "gevent" ]; then
    export LOAD_CONFIG_FILE="-c /home/frappe/frappe-bench/commands/gevent_patch.py"
  fi

  if [[ ! -z "$AUTO_MIGRATE" ]]; then
    . /home/frappe/frappe-bench/env/bin/activate \
      && python /home/frappe/frappe-bench/commands/auto_migrate.py
  fi

  . /home/frappe/frappe-bench/env/bin/activate
  gunicorn $LOAD_CONFIG_FILE -b 0.0.0.0:$FRAPPE_PORT \
    --worker-tmp-dir /dev/shm \
    --threads=4 \
    --workers $WORKERS \
    --worker-class=$WORKER_CLASS \
    --log-file=- \
    -t 120 frappe.app:application --preload

elif [ "$1" = 'worker' ]; then
  checkConfigExists
  checkConnection
  # default WORKER_TYPE=default

  . /home/frappe/frappe-bench/env/bin/activate
  python /home/frappe/frappe-bench/commands/worker.py

elif [ "$1" = 'schedule' ]; then
  checkConfigExists
  checkConnection

  . /home/frappe/frappe-bench/env/bin/activate
  python /home/frappe/frappe-bench/commands/background.py

elif [ "$1" = 'new' ]; then
  checkConfigExists
  checkConnection

  . /home/frappe/frappe-bench/env/bin/activate
  python /home/frappe/frappe-bench/commands/new.py
  exit

elif [ "$1" = 'drop' ]; then
  checkConfigExists
  checkConnection

  . /home/frappe/frappe-bench/env/bin/activate
  python /home/frappe/frappe-bench/commands/drop.py
  exit

elif [ "$1" = 'migrate' ]; then

  . /home/frappe/frappe-bench/env/bin/activate \
    && python /home/frappe/frappe-bench/commands/migrate.py
  exit

elif [ "$1" = 'doctor' ]; then

  . /home/frappe/frappe-bench/env/bin/activate \
    && python /home/frappe/frappe-bench/commands/doctor.py ${@:2}
  exit

elif [ "$1" = 'backup' ]; then

  . /home/frappe/frappe-bench/env/bin/activate
  python /home/frappe/frappe-bench/commands/backup.py
  exit

elif [ "$1" = 'console' ]; then

  if [[ -z "$2" ]]; then
    echo "Need to specify a sitename with the command:"
    echo "console <sitename>"
    exit 1
  fi

  . /home/frappe/frappe-bench/env/bin/activate
  python /home/frappe/frappe-bench/commands/console.py "$2"
  exit

elif [ "$1" = 'push-backup' ]; then

  . /home/frappe/frappe-bench/env/bin/activate \
    && python /home/frappe/frappe-bench/commands/push_backup.py
  exit

elif [ "$1" = 'restore-backup' ]; then

  . /home/frappe/frappe-bench/env/bin/activate \
    && python /home/frappe/frappe-bench/commands/restore_backup.py
  exit

else

  exec $@

fi