@catladder/cli/dist/cli/src/gcloud/cloudSql/copyDb.js

Panter cli tool for cloud CI/CD and DevOps

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.spawnCopyDb = void 0;
const child_process_promise_1 = require("child-process-promise");
const spawnCopyDb = async opts => {
  const script = createCopyDbScript(opts);
  return await (0, child_process_promise_1.spawn)(script, [], {
    shell: "bash",
    stdio: "inherit"
  });
};
exports.spawnCopyDb = spawnCopyDb;
const createCopyDbScript = ({
  targetPassword,
  targetPort,
  targetUsername,
  sourceUsername,
  sourcePassword,
  sourcePort,
  sourceDbName,
  targetDbName
}) => {
  const targetPSQL = command => `PGPASSWORD=${targetPassword} psql -p ${targetPort} --host=localhost --user=${targetUsername} -q ${command}`;
  // URL-encode credentials to handle special characters like @, :, /, etc.
  const encodedSourceUsername = encodeURIComponent(sourceUsername);
  const encodedSourcePassword = encodeURIComponent(sourcePassword);
  const copyDBScript = `
      set -e

      ${targetPSQL(`-c 'drop database "${targetDbName}" WITH (FORCE)' 1> /dev/null || true`)}
      ${targetPSQL(`-c 'create database "${targetDbName}"' 1> /dev/null`)}
      echo "Estimating database size..."
      DB_SIZE=$(PGPASSWORD=${encodedSourcePassword} psql --dbname=postgres://${encodedSourceUsername}:${encodedSourcePassword}@localhost:${sourcePort}/${sourceDbName} -t -A -c "SELECT pg_database_size('${sourceDbName}')")
      # pg_database_size includes indexes, TOAST, and free space — pg_dump output is roughly 40% of that
      ESTIMATED_DUMP_SIZE=$((DB_SIZE * 40 / 100))
      echo "Estimated dump size: $((ESTIMATED_DUMP_SIZE / 1024 / 1024)) MB (~40% of $((DB_SIZE / 1024 / 1024)) MB database size)"

      PROGRESS=""
      if command -v pv &> /dev/null; then
        PROGRESS="pv -s $ESTIMATED_DUMP_SIZE |"
      else
        echo "(install 'pv' for progress info)"
      fi

      echo "Dumping and restoring via pipe..."
      eval "pg_dump --dbname=postgres://${encodedSourceUsername}:${encodedSourcePassword}@localhost:${sourcePort}/${sourceDbName} --no-owner --no-privileges | $PROGRESS ${targetPSQL(`"${targetDbName}" 1> /dev/null`)}"

      echo "\n🐱 Done!"
      `;
  return copyDBScript;
};