prereview/docker-entrypoint.sh at develop · PREreview/prereview · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#!/bin/sh
set -e

TRIES=30
DELAY=4
HOST="$PREREVIEW_DB_HOST"
PORT="$PREREVIEW_DB_PORT"
USER="$PREREVIEW_DB_USER"
PASS="$PREREVIEW_DB_PASS"
NAME="$PREREVIEW_DB_NAME"

# For copying from another postgres database
FROM_HOST=${IMPORT_FROM_HOST:-$PREREVIEW_DB_HOST}
FROM_PORT=${IMPORT_FROM_PORT:-$PREREVIEW_DB_PORT}
FROM_USER=${IMPORT_FROM_USER:-$PREREVIEW_DB_USER}
FROM_PASS=${IMPORT_FROM_PASS:-$PREREVIEW_DB_PASS}
FROM_NAME=${IMPORT_FROM_NAME:-production}

wait_for() {
  echo "Waiting for service => HOST: $HOST, PORT: $PORT"
  for i in $(seq $TRIES); do
    set +e
    nc -z "$HOST" "$PORT" >/dev/null 2>&1
    result=$?
    set -e

    if [ $result -eq 0 ]; then
      echo "Service is up!"
      return 0
    fi
    sleep "$DELAY"
    echo "Retrying..."
  done
  echo "Operation timed out" >&2
  exit 1
}

wait_for

clear_db() {
  echo "Force-disconnecting clients"
  env -i PGPASSWORD="$PASS" /usr/bin/psql -U "$USER" -d postgres -h "$HOST" -p "$PORT" -c "SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = '$NAME'"
  echo "Dropping $NODE_ENV database"
  env -i PGPASSWORD="$PASS" /usr/bin/psql -U "$USER" -d postgres -h "$HOST" -p "$PORT" -c "DROP DATABASE $NAME"
  echo "Creating blank $NODE_ENV database"
  env -i PGPASSWORD="$PASS" /usr/bin/psql -U "$USER" -d postgres -h "$HOST" -p "$PORT" -c "CREATE DATABASE $NAME"
}

init_db() {
  echo "Initializing database schema"
  npm run db:migrations
  npm run db:init
}

if [ $NODE_ENV == "staging" ]; then
  echo "Copying existing database $FROM_NAME to staging"
  echo "Dumping $FROM_NAME database"
  env -i PGPASSWORD="$FROM_PASS" /usr/bin/pg_dump -U "$FROM_USER" -d "$FROM_NAME" -h "$FROM_HOST" -p "$FROM_PORT" -f /tmp/import.sql

  clear_db

  echo "Importing dump to staging"
  env -i PGPASSWORD="$PASS" /usr/bin/psql -U "$USER" -d "$NAME" -h "$HOST" -p "$PORT" -f /tmp/import.sql
  echo "Done copying!"

  echo "Updating database schema"
  npm run db:migrations
elif [ $NODE_ENV == "integration" ]; then
  echo "Resetting database"
  clear_db
  init_db
  npm run db:apiKey
else
  set +e
  env -i PGPASSWORD="$PASS" /usr/bin/psql -U "$USER" -d "$NAME" -h "$HOST" -p "$PORT" -tAc "SELECT to_regclass('public.user')" | grep -q user
  result=$?
  set -e

  if [ $result -ne 0 ]; then
    init_db
  else
    echo "Updating database schema"
    npm run db:migrations
  fi
fi

echo "Running => npm run $@"

npm run "$@"