docker-zulip/entrypoint.sh

#!/bin/bash

if [ "$DEBUG" == "true" ]; then
  set -x
fi
set -e

ZULIP_CURRENT_DEPLOY="$ZULIP_DIR/deployments/current"
MANAGE_PY="$ZULIP_CURRENT_DEPLOY/manage.py"
ZULIP_SETTINGS="/etc/zulip/settings.py"

# Some functions were originally taken from the zulip/zulip repo folder scripts
# I modified them to fit the "docker way" of installation ;)
function database-settings-setup(){
  cat <<EOF >> "$ZULIP_SETTINGS"
from zerver.lib.db import TimeTrackingConnection

DATABASES = {"default": {
    'ENGINE': 'django.db.backends.postgresql_psycopg2',
    'NAME': '$DB_NAME',
    'USER': '$DB_USER',
    'PASSWORD': '$DB_PASSWORD', # Authentication done via certificates
    'HOST': '$DB_HOST',
    'SCHEMA': 'zulip',
    'CONN_MAX_AGE': 600,
    'OPTIONS': {
        'connection_factory': TimeTrackingConnection
        },
    },
}

EOF
}
function database-data-setup(){
  if [ -z "$PGPASSWORD" ]; then
    export PGPASSWORD="$DB_PASSWORD"
  fi
  # TODO Shall we change that and really create the database here or are we expecting a ready zulip database?
  psql -h "$DB_HOST" -p "$DB_PORT" -U "$DB_USER" "CREATE USER zulip;
    ALTER ROLE zulip SET search_path TO zulip,public;
    DROP DATABASE IF EXISTS zulip;
    CREATE DATABASE zulip OWNER=zulip;" || :
  psql -h "$DB_HOST" -p "$DB_PORT" -U "$DB_USER" "zulip" "CREATE SCHEMA zulip AUTHORIZATION zulip;
    CREATE EXTENSION tsearch_extras SCHEMA zulip;" || :
}
function database-initiation(){
  su zulip -c "$MANAGE_PY checkconfig"
  su zulip -c "$MANAGE_PY migrate --noinput"
  su zulip -c "$MANAGE_PY createcachetable third_party_api_results"
  su zulip -c "$MANAGE_PY initialize_voyager_db"
}
function zulip-add-custom-secrets(){
  ZULIP_SECRETS="/etc/zulip/zulip-secrets.conf"
  POSSIBLE_SECRETS=(
    "s3_key" "s3_secret_key" "android_gcm_api_key" "google_oauth2_client_secret"
    "dropbox_app_key" "mailchimp_api_key" "mandrill_api_key" "twitter_consumer_key" "twitter_consumer_secret"
    "twitter_access_token_key" "twitter_access_token_secret" "email_password" "rabbitmq_password"
  )
  for SECRET_KEY in "${POSSIBLE_SECRETS[@]}"; do
    KEY="ZULIP_SECRETS_$SECRET_KEY"
    SECRET_VAR="${!KEY}"
    if [ -z "$SECRET_VAR" ]; then
      echo "No settings env var for key \"$SECRET_KEY\". Continuing."
      continue
    fi
    echo "Setting secret \"$SECRET_KEY\"."
    echo "$SECRET_KEY = '$SECRET_VAR'" >> "$ZULIP_SECRETS"
  done
}
function zulip-setup-external-services(){
  cat <<EOF >> "$ZULIP_SETTINGS"
CACHES = {
    'default': {
        'BACKEND':  'django.core.cache.backends.memcached.PyLibMCCache',
        'LOCATION': '$MEMCACHED_HOST:$MEMCACHED_PORT',
        'TIMEOUT':  $MEMCACHED_TIMEOUT
    },
    'database': {
        'BACKEND':  'django.core.cache.backends.db.DatabaseCache',
        'LOCATION':  'third_party_api_results',
        # Basically never timeout.  Setting to 0 isn't guaranteed
        # to work, see https://code.djangoproject.com/ticket/9595
        'TIMEOUT': 2000000000,
        'OPTIONS': {
            'MAX_ENTRIES': 100000000,
            'CULL_FREQUENCY': 10,
        }
    },
}

EOF
  # Rabbitmq settings
  if [ ! -z "$RABBITMQ_USERNAME" ]; then
    cat <<EOF >> "$ZULIP_SETTINGS"
RABBITMQ_USERNAME = '$RABBITMQ_USERNAME'

EOF
  fi
  if [ ! -z "$RABBITMQ_PASSWORD" ]; then
    cat <<EOF >> "$ZULIP_SETTINGS"
RABBITMQ_PASSWORD = '$RABBITMQ_PASSWORD'

EOF
  fi
  # Redis settings
  if [ ! -z "$REDIS_RATE_LIMITING" ] && [ ! -z "$REDIS_HOST" ]; then
    if [ -z "$REDIS_PORT" ]; then
      REDIS_PORT="6379"
    fi
    case "REDIS_RATE_LIMITING" in
      [Tt][Rr][Uu][Ee])
      REDIS_RATE_LIMITING="True"
      ;;
      [Ff][Aa][Ll][Ss][Ee])
      REDIS_RATE_LIMITING="False"
      ;;
      *)
      echo "Can't parse True or Right for REDIS_RATE_LIMITING. Defaulting to True"
      REDIS_RATE_LIMITING="True"
      ;;
    esac
    cat <<EOF >> "$ZULIP_SETTINGS"
RATE_LIMITING = $REDIS_RATE_LIMITING
REDIS_HOST = '$REDIS_HOST'
REDIS_PORT = $REDIS_PORT

EOF
  fi
  # Camo settings
  if [ ! -z "$CAMO_KEY" ]; then
    cat <<EOF >> "$ZULIP_SETTINGS"
CAMO_KEY = '$CAMO_KEY'

EOF
  fi
  if [ ! -z "$CAMO_URI" ]; then
    cat <<EOF >> "$ZULIP_SETTINGS"
CAMO_URI = '$CAMO_URI'

EOF
  fi
  if [ ! -z "$ZULIP_CUSTOM_SETTINGS" ]; then
    echo -e "\n$ZULIP_CUSTOM_SETTINGS" >> "$ZULIP_SETTINGS"
  fi
}
function zulip-setup-zulip-settings(){
  # ^#?([a-zA-Z0-9_]*)[ ]*=[ ]*([\"'].*[\"']+|[\(\{]+(\n[^)]*)+.*[\)\}])$ and ^#?[ ]?([a-zA-Z0-9_]*)
  POSSIBLE_SETTINGS=($(grep -E "^#?([a-zA-Z0-9_]*)[ ]*=[ ]*([\"'].*[\"']+|[\(\{]+(\n[^)]*)+.*[\)\}])$" "$ZULIP_SETTINGS" | grep -oE "^#?[ ]?([a-zA-Z0-9_]*)") "S3_AUTH_UPLOADS_BUCKET" "S3_AVATAR_BUCKET")
  for SETTING_KEY in "${POSSIBLE_SETTINGS[@]}"; do
    KEY="ZULIP_SETTINGS_$SETTING_KEY"
    SETTING_VAR="${!KEY}"
    if [ -z "$SETTING_VAR" ]; then
      echo "No settings env var for key \"$SETTING_KEY\". Continuing."
      continue
    fi
    echo "Setting key \"$SETTING_KEY\" to value \"$SETTING_VAR\"."
    sed -i "s~#?${SETTING_KEY}[ ]*=[ ]*['\"]+.*['\"]+$~${SETTING_KEY} = '${SETTING_VAR}'~g" "$ZULIP_SETTINGS"
  done
  zulip-setup-external-services
  if [ "$ZULIP_COPY_SETTINGS" == "true" ]; then
    rm -f "$DATA_DIR/settings.py"
    cp -fT "$ZULIP_SETTINGS" "$DATA_DIR/settings.py"
    return 0
  fi
}
function zulip-create-user(){
  if [ -z "$ZULIP_USER_EMAIL" ]; then
    echo "No zulip user email given."
    return 1
  fi
    if [ -z "$ZULIP_USER_PASSWORD" ]; then
      echo "No zulip user password given."
      return 1
    fi
  if [ -z "$ZULIP_USER_FULLNAME" ]; then
    echo "No zulip user full name given. Defaulting to \"Zulip Docker\""
    ZULIP_USER_FULLNAME="Zulip Docker"
  fi
  su zulip -c " $MANAGE_PY create_user --new-email \"$ZULIP_USER_EMAIL\" --new-password \"$ZULIP_USER_PASSWORD\" --new-full-name \"$ZULIP_USER_FULLNAME\""
  su zulip -c "$MANAGE_PY knight \"$ZULIP_USER_EMAIL\" -f"
}
function rabbitmq-setup(){
  rabbitmqctl delete_user zulip || :
  rabbitmqctl delete_user guest || :
  rabbitmqctl add_user zulip "$("$ZULIP_CURRENT_DEPLOY/bin/get-django-setting" RABBITMQ_PASSWORD)" || :
  rabbitmqctl set_user_tags zulip administrator
  rabbitmqctl set_permissions -p / zulip '.*' '.*' '.*'
}

if [ ! -d "$ZULIP_DIR/uploads" ]; then
  mkdir -p "$ZULIP_DIR/uploads"
fi
if [ -d "$DATA_DIR/uploads" ]; then
  rm -rf "$ZULIP_DIR/uploads"
else
  mkdir -p "$DATA_DIR/uploads"
  mv -f "$ZULIP_DIR/uploads" "$DATA_DIR/uploads"
fi
ln -sfT "$DATA_DIR/uploads" "$ZULIP_DIR/uploads"
if [ ! -f "$DATA_DIR/.initiated" ]; then
  echo "Initiating Zulip initiation ..."
  echo "==="
  echo "Generating and setting secrets ..."
  # Generate the secrets
  /root/zulip/scripts/setup/generate_secrets.py
  zulip-add-custom-secrets
  echo "Secrets generated and set."
  echo "Setting up database settings and server ..."
  # Set database settings
  database-settings-setup
  # Init Postgres database server
  database-data-setup
  echo "Database settings and server setup done."
  echo "Setting Zulip settings ..."
  # Setup zulip settings
  zulip-setup-zulip-settings
  echo "Zulip settings setup done."
  echo "Initiating  Database ..."
  # Init database with something called data :D
  if ! database-initiation; then
    echo "Database initiation failed."
    exit 1
  fi
  touch "$DATA_DIR/.initiated"
  echo "Database initiated."
  echo "Creating zulip user account ..."
  zulip-create-user
  echo "Created zulip user account"
  echo "==="
  echo "Zulip initiation done."
fi
# Configure rabbitmq server everytime because it could be a new one ;)
rabbitmq-setup
# If there's an "update" available, then JUST DO IT!
if [ ! -f "$DATA_DIR/.zulip-$ZULIP_VERSION" ]; then
  echo "Starting zulip migration ..."
  if ! "$MANAGE_PY" migrate; then
    echo "Zulip migration error."
    exit 1
  fi
  touch "$DATA_DIR/.zulip-$ZULIP_VERSION"
  echo "Zulip migration done."
fi
echo "Starting zulip using supervisor ..."
# Start supervisord
exec supervisord