From fa2496c229e123e67048ca1143522c6402034251 Mon Sep 17 00:00:00 2001
From: Anders Kaseorg <anders@zulip.com>
Date: Mon, 15 Jun 2020 16:36:02 -0700
Subject: [PATCH] terminate-psql-sessions: Rely on the caller to set PGHOST,
 PGUSER.

Signed-off-by: Anders Kaseorg <anders@zulip.com>
---
 scripts/setup/postgres-init-db        |  3 ++-
 scripts/setup/restore-backup          |  4 ++--
 scripts/setup/terminate-psql-sessions | 22 +---------------------
 tools/rebuild-dev-database            |  3 ++-
 tools/rebuild-test-database           |  3 ++-
 tools/setup/generate-fixtures         |  3 ++-
 tools/setup/postgres-init-dev-db      |  3 ++-
 7 files changed, 13 insertions(+), 28 deletions(-)

diff --git a/scripts/setup/postgres-init-db b/scripts/setup/postgres-init-db
index fcee219eb9..016349a50a 100755
--- a/scripts/setup/postgres-init-db
+++ b/scripts/setup/postgres-init-db
@@ -33,7 +33,8 @@ if [ -e "/var/run/supervisor.sock" ]; then
 fi
 
 # Drop any open connections to any old database.
-"$(dirname "$0")/terminate-psql-sessions" postgres zulip zulip_base
+su -s /usr/bin/env - -- "$POSTGRES_USER" \
+   "$(dirname "$0")/terminate-psql-sessions" zulip zulip_base
 
 (
     cd /  # Make sure the current working directory is readable by postgres
diff --git a/scripts/setup/restore-backup b/scripts/setup/restore-backup
index 432cf4a61c..15446778ed 100755
--- a/scripts/setup/restore-backup
+++ b/scripts/setup/restore-backup
@@ -84,17 +84,17 @@ def restore_backup(tarball_file: IO[bytes]) -> None:
         db_dir = os.path.join(tmp, "zulip-backup", "database")
         os.setresuid(0, 0, 0)
         run(["chown", "-R", POSTGRES_USER, "--", tmp])
+        as_postgres = ["su", "-s", "/usr/bin/env", "-", "--", POSTGRES_USER]
         run(
             [
+                *as_postgres,
                 os.path.join(
                     settings.DEPLOY_ROOT, "scripts", "setup", "terminate-psql-sessions",
                 ),
                 "zulip",
-                "zulip",
                 "zulip_base",
             ],
         )
-        as_postgres = ["su", "-s", "/usr/bin/env", "-", "--", POSTGRES_USER]
         run(as_postgres + ["dropdb", "--if-exists", "--", db_name])
         run(as_postgres + ["createdb", "-O", "zulip", "-T", "template0", "--", db_name])
 
diff --git a/scripts/setup/terminate-psql-sessions b/scripts/setup/terminate-psql-sessions
index c6340d270c..f952eae14d 100755
--- a/scripts/setup/terminate-psql-sessions
+++ b/scripts/setup/terminate-psql-sessions
@@ -1,32 +1,12 @@
 #!/usr/bin/env bash
 set -e
 
-POSTGRES_USER="postgres"
-if [ "$(uname)" = "OpenBSD" ]; then
-  POSTGRES_USER="_postgresql"
-fi
-
 cd /
 
-username=$1
-
-shift
 tables="$(printf "'%s'," "${@//\'/\'\'}")"
 tables="${tables%,}"
 
-case "$(id -un)" in
-root)
-    psql=(su -s /usr/bin/env - -- "$POSTGRES_USER" psql postgres "$POSTGRES_USER")
-    ;;
-"$POSTGRES_USER")
-    psql=(psql postgres "$POSTGRES_USER")
-    ;;
-*)
-    psql=(psql -h localhost postgres "$username")
-    ;;
-esac
-
-"${psql[@]}" -v ON_ERROR_STOP=1 <<EOF
+psql postgres -v ON_ERROR_STOP=1 <<EOF
 SELECT pg_terminate_backend(s.pid)
     FROM pg_stat_activity s, pg_roles r
     WHERE
diff --git a/tools/rebuild-dev-database b/tools/rebuild-dev-database
index cedca211e3..68baf634fa 100755
--- a/tools/rebuild-dev-database
+++ b/tools/rebuild-dev-database
@@ -2,7 +2,8 @@
 set -e
 set -x
 
-"$(dirname "$0")/../scripts/setup/terminate-psql-sessions" zulip zulip zulip_base
+PGHOST=localhost PGUSER=zulip \
+    "$(dirname "$0")/../scripts/setup/terminate-psql-sessions" zulip zulip_base
 
 psql -v ON_ERROR_STOP=1 -e -h localhost postgres zulip <<EOF
 DROP DATABASE IF EXISTS zulip;
diff --git a/tools/rebuild-test-database b/tools/rebuild-test-database
index a6e9214d6b..990552d8d4 100755
--- a/tools/rebuild-test-database
+++ b/tools/rebuild-test-database
@@ -22,7 +22,8 @@ EOF
 
 mkdir -p zerver/tests/fixtures
 
-"$(dirname "$0")/../scripts/setup/terminate-psql-sessions" zulip zulip_test zulip_test_base zulip_test_template
+PGHOST=localhost PGUSER=zulip \
+    "$(dirname "$0")/../scripts/setup/terminate-psql-sessions" zulip_test zulip_test_base zulip_test_template
 
 create_zulip_test
 
diff --git a/tools/setup/generate-fixtures b/tools/setup/generate-fixtures
index b77aec85ac..11a02894cc 100755
--- a/tools/setup/generate-fixtures
+++ b/tools/setup/generate-fixtures
@@ -3,7 +3,8 @@ set -e
 
 run()
 {
-    "$(dirname "$0")/../../scripts/setup/terminate-psql-sessions" zulip zulip_test zulip_test_base zulip_test_template
+    PGHOST=localhost PGUSER=zulip \
+        "$(dirname "$0")/../../scripts/setup/terminate-psql-sessions" zulip_test zulip_test_base zulip_test_template
     psql -v ON_ERROR_STOP=1 -h localhost postgres zulip_test << EOF
 DROP DATABASE IF EXISTS zulip_test;
 CREATE DATABASE zulip_test TEMPLATE zulip_test_template;
diff --git a/tools/setup/postgres-init-dev-db b/tools/setup/postgres-init-dev-db
index c0cc6b1c35..7e3cb90330 100755
--- a/tools/setup/postgres-init-dev-db
+++ b/tools/setup/postgres-init-dev-db
@@ -93,7 +93,8 @@ else
 fi
 chmod go-rw ~/.pgpass
 
-"$(dirname "$0")/../../scripts/setup/terminate-psql-sessions" "$USERNAME" "$DBNAME" "$DBNAME_BASE"
+PGHOST=localhost PGUSER="$USERNAME" \
+    "$(dirname "$0")/../../scripts/setup/terminate-psql-sessions" "$DBNAME" "$DBNAME_BASE"
 
 psql -v ON_ERROR_STOP=1 -e -h localhost postgres "$USERNAME" <<EOF
 DROP DATABASE IF EXISTS $DBNAME;