docker-zulip/entrypoint.sh

#!/bin/bash

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

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

# Some functions were originally taken from the zulip/zulip repo folder scripts
# But modified to fit the docker image :)
databaseSetup(){
  cat >> "$ZULIP_ZPROJECT_SETTINGS" <<EOF
from zerver.lib.db import TimeTrackingConnection

REMOTE_POSTGRES_HOST = '$DB_HOST'

DATABASES = {
  "default": {
    'ENGINE': 'django.db.backends.postgresql_psycopg2',
    'NAME': '$DB_NAME',
    'USER': '$DB_USER',
    'PASSWORD': '$DB_PASSWORD',
    'HOST': '$DB_HOST',
    'SCHEMA': 'zulip',
    'CONN_MAX_AGE': 600,
    'OPTIONS': {
        'connection_factory': TimeTrackingConnection,
        'sslmode': 'prefer',
    },
  },
}
EOF
  if [ -z "$PGPASSWORD" ]; then
    export PGPASSWORD="$DB_PASSWORD"
  fi
  if [ -z "$DB_PORT" ]; then
    export DB_PORT="5432"
  fi
  local timeout=60
  printf "Waiting for database server to allow connections"
  while ! /usr/bin/pg_isready -h "$DB_HOST" -p "$DB_PORT" -U "$DB_USER" -t 1 >/dev/null 2>&1
  do
    timeout=$(expr $timeout - 1)
    if [[ $timeout -eq 0 ]]; then
      echo "Could not connect to database server. Aborting..."
      exit 1
    fi
    printf "."
    sleep 1
  done
  echo """
  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" || :
  echo """
  CREATE SCHEMA zulip AUTHORIZATION zulip;
  CREATE EXTENSION tsearch_extras SCHEMA zulip;
  """ | psql -h "$DB_HOST" -p "$DB_PORT" -U "$DB_USER" "zulip" || :
}
databaseInitiation(){
  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"
}
secretsSetup(){
  local ZULIP_SECRETS="/etc/zulip/zulip-secrets.conf"
  local 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
    local KEY="ZULIP_SECRETS_$SECRET_KEY"
    local SECRET_VAR="${!KEY}"
    if [ -z "$SECRET_VAR" ]; then
      echo "No settings env var for key \"$SECRET_KEY\"."
      continue
    fi
    echo "Setting secret \"$SECRET_KEY\"."
    if [ -z "$(grep "$SECRET_KEY" "$ZULIP_SECRETS")" ]; then
      sed -ir "s~#?${SECRET_KEY}[ ]*=[ ]*['\"]+.*['\"]+$~${SECRET_KEY} = '${SECRET_VAR}'~g" "$ZULIP_SECRETS"
      continue
    fi
    echo "$SECRET_KEY = '$SECRET_VAR'" >> "$ZULIP_SECRETS"
  done
  unset SECRET_KEY
}
zulipSetup(){
  cat >> "$ZULIP_ZPROJECT_SETTINGS" <<EOF
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
  sed -i "s~pika.ConnectionParameters('localhost',~pika.ConnectionParameters(settings.RABBITMQ_HOST,~g" "$ZULIP_CURRENT_DEPLOY/zerver/lib/queue.py"
  cat >> "$ZULIP_ZPROJECT_SETTINGS" <<EOF
RABBITMQ_HOST = '$RABBITMQ_HOST'
EOF
  if [ ! -z "$RABBITMQ_USERNAME" ]; then
    cat >> "$ZULIP_ZPROJECT_SETTINGS" <<EOF
RABBITMQ_USERNAME = '$RABBITMQ_USERNAME'
EOF
  fi
  if [ ! -z "$RABBITMQ_PASSWORD" ]; then
    cat >> "$ZULIP_ZPROJECT_SETTINGS" <<EOF
RABBITMQ_PASSWORD = '$RABBITMQ_PASSWORD'
EOF
  fi
  # Redis settings
  if [ -z "$REDIS_HOST" ]; then
    export REDIS_HOST="localhost"
  fi
  if [ -z "$REDIS_PORT" ]; then
    export REDIS_PORT="6379"
  fi
  case "$REDIS_RATE_LIMITING" in
    [Tt][Rr][Uu][Ee])
    export REDIS_RATE_LIMITING="True"
    ;;
    [Ff][Aa][Ll][Ss][Ee])
    export REDIS_RATE_LIMITING="False"
    ;;
    *)
    echo "Can't parse True or Right for REDIS_RATE_LIMITING. Defaulting to True"
    export REDIS_RATE_LIMITING="True"
    ;;
  esac
  cat >> "$ZULIP_ZPROJECT_SETTINGS" <<EOF
RATE_LIMITING = $REDIS_RATE_LIMITING
REDIS_HOST = '$REDIS_HOST'
REDIS_PORT = $REDIS_PORT
EOF
  # Camo settings
  if [ ! -z "$CAMO_KEY" ]; then
    cat >> "$ZULIP_ZPROJECT_SETTINGS" <<EOF
CAMO_KEY = '$CAMO_KEY'
EOF
  fi
  if [ ! -z "$CAMO_URI" ]; then
    cat >> "$ZULIP_ZPROJECT_SETTINGS" <<EOF
CAMO_URI = '$CAMO_URI'
EOF
  fi
  if [ ! -z "$ZULIP_CUSTOM_SETTINGS" ]; then
    echo -e "\n$ZULIP_CUSTOM_SETTINGS" >> "$ZULIP_ZPROJECT_SETTINGS"
  fi
  local SET_SETTINGS=($(env | sed -nr "s/ZULIP_SETTINGS_([A-Z_]*).*/\1/p"))
  for SETTING_KEY in "${SET_SETTINGS[@]}"; do
    local KEY="ZULIP_SETTINGS_$SETTING_KEY"
    local SETTING_VAR="${!KEY}"
    if [ -z "$SETTING_VAR" ]; then
      echo "No settings env var for key \"$SETTING_KEY\"."
      continue
    fi
    echo "Setting key \"$SETTING_KEY\" to value \"$SETTING_VAR\"."
    sed -ri "s~#?${SETTING_KEY}[ ]*=[ ]*['\"]+.*['\"]+$~${SETTING_KEY} = '${SETTING_VAR}'~g" "$ZULIP_SETTINGS"
  done
  if [ "$ZULIP_COPY_SETTINGS" == "true" ]; then
    rm -f "$DATA_DIR/settings.py"
    cp -fT "$ZULIP_SETTINGS" "$DATA_DIR/settings.py"
  fi
  unset SETTING_KEY
}
zulipCreateUser(){
  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\""
    export 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"
  return 0
}
rabbitmqSetup(){
  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
  secretsSetup
  echo "Secrets generated and set."
  echo "Setting Zulip settings ..."
  # Setup zulip settings
  zulipSetup
  echo "Zulip settings setup done."
  echo "Setting up database settings and server ..."
  # setup database
  databaseSetup
  echo "Database setup done."
  echo "Initiating  Database ..."
  # Init database with something called data :D
  if ! databaseInitiation; then
    echo "Database initiation failed."
    exit 1
  fi
  echo "Database initiated."
  echo "Creating zulip user account ..."
  zulipCreateUser
  echo "Created zulip user account"
  echo "==="
  echo "Zulip initiation done."
  touch "$DATA_DIR/.initiated"
fi
# Configure rabbitmq server everytime because it could be a new one ;)
rabbitmqSetup
# 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