Merge pull request #117 from graphile/fix-docker-postgres-issues

Fix docker postgres issues

Author: benjie

docker-compose.yml

@@ -14,7 +14,7 @@ services:
     user: $UID
 
     env_file:
-      - docker/docker.env
+      - docker/.env
     volumes:
       - .:/work
       # let's us run docker & docker-compose from inside container (used for yarn setup)
@@ -52,7 +52,7 @@ services:
         SETUP_MODE: dev
 
     env_file:
-      - docker/docker.env
+      - docker/.env
 
     # On Linux, this will prevent new files getting created as root, but you
     # may need to update the USER_UID and USER_GID in `Dockerfile` to match
@@ -112,6 +112,8 @@ services:
   # This runs the database that everything else connects to
   db:
     image: postgres:11
+    env_file:
+      - docker/.env
 
     # Unlike above, the Postgres image cannot start as our user account
     # otherwise we get permission denied errors. So this one has to run as
@@ -129,7 +131,10 @@ services:
     restart: unless-stopped
     networks:
       - default
-    command: postgres -c logging_collector=on -c log_destination=stderr -c log_directory=/var/lib/postgresql/data/logs -c log_rotation_age=60 -c log_truncate_on_rotation=on -c log_filename=server_log.hour.%H%M
+    command:
+      postgres -c logging_collector=on -c log_destination=stderr -c
+      log_directory=/var/lib/postgresql/data/logs -c log_rotation_age=60 -c
+      log_truncate_on_rotation=on -c log_filename=server_log.hour.%H%M
 
 networks:
   default:

docker/docker.env

@@ -1,6 +1,10 @@
 const { spawnSync: baseSpawnSync } = require("child_process");
 const { basename, dirname, resolve } = require("path");
 const platform = require("os").platform();
+const { safeRandomString } = require("../../scripts/lib/random");
+const fsp = require("fs").promises;
+
+const DOCKER_DOTENV_PATH = `${__dirname}/../.env`;
 
 if (platform !== "win32" && !process.env.UID) {
   console.error(
@@ -23,28 +27,76 @@ function spawnSync(command, args, options = {}) {
   }
 }
 
-// The `docker-compose` project name defaults to the directory name containing
-// `docker-compose.yml`, which is the root folder of our project. Let's call
-// that 'ROOT'. We're in ROOT/docker/scripts and we want to get the name of
-// ROOT:
-const projectName = basename(dirname(dirname(resolve(__dirname))));
-
-// On Windows we must run 'yarn.cmd' rather than 'yarn'
-const yarnCmd = platform === "win32" ? "yarn.cmd" : "yarn";
-
-spawnSync(yarnCmd, ["down"]);
-spawnSync(yarnCmd, ["db:up"]);
-
-// Fix permissions
-spawnSync(yarnCmd, [
-  "compose",
-  "run",
-  "server",
-  "sudo",
-  "bash",
-  "-c",
-  "chmod o+rwx /var/run/docker.sock && chown -R node /work/node_modules /work/@app/*/node_modules",
-]);
-
-// Run setup as normal
-spawnSync(yarnCmd, ["compose", "run", "server", "yarn", "setup", projectName]);
+async function main() {
+  // Check that docker/.env exists
+  try {
+    await fsp.access(DOCKER_DOTENV_PATH, fs.constants.F_OK);
+  } catch (e) {
+    // Does not exist, write it
+    const password = safeRandomString(30);
+    const data = `
+# We'd like scripts ran through Docker to pretend they're in a normal
+# interactive terminal.
+FORCE_COLOR=2
+
+# \`pg_dump\` is run from inside container, which doesn't have pg tools installed
+# so it needs a way to still run it. \`docker-compose run\` would start an
+# instance inside the current running container which doesn't work with volume
+# mappings, so we must use \`docker-compose exec\`. \`-T\` is needed because our
+# \`.gmrc\` checks for interactive TTY.
+PG_DUMP=docker-compose exec -T db pg_dump
+
+# Drops tables without asking in \`yarn setup\`. Reasoning: 1) docker-compose is
+# not tty, 2) it's a dev env anyway.
+CONFIRM_DROP=y
+
+# POSTGRES_PASSWORD is the superuser password for PostgreSQL, it's required to
+# initialize the Postgres docker volume.
+POSTGRES_PASSWORD=${password}
+
+# We're accessing Postgres via Docker, so we must use the db host and the
+# relevant password.
+DATABASE_HOST=db
+ROOT_DATABASE_URL=postgres://postgres:${password}@db/template1
+`;
+    await fsp.writeFile(DOCKER_DOTENV_PATH, data);
+  }
+
+  // The `docker-compose` project name defaults to the directory name containing
+  // `docker-compose.yml`, which is the root folder of our project. Let's call
+  // that 'ROOT'. We're in ROOT/docker/scripts and we want to get the name of
+  // ROOT:
+  const projectName = basename(dirname(dirname(resolve(__dirname))));
+
+  // On Windows we must run 'yarn.cmd' rather than 'yarn'
+  const yarnCmd = platform === "win32" ? "yarn.cmd" : "yarn";
+
+  spawnSync(yarnCmd, ["down"]);
+  spawnSync(yarnCmd, ["db:up"]);
+
+  // Fix permissions
+  spawnSync(yarnCmd, [
+    "compose",
+    "run",
+    "server",
+    "sudo",
+    "bash",
+    "-c",
+    "chmod o+rwx /var/run/docker.sock && chown -R node /work/node_modules /work/@app/*/node_modules",
+  ]);
+
+  // Run setup as normal
+  spawnSync(yarnCmd, [
+    "compose",
+    "run",
+    "server",
+    "yarn",
+    "setup",
+    projectName,
+  ]);
+}
+
+main().catch(e => {
+  console.error(e);
+  process.exit(1);
+});