From bad9f39fbfd9cfd49505111e148872a238244ce7 Mon Sep 17 00:00:00 2001
From: Paul Cioanca <paul.cioanca@supabase.io>
Date: Fri, 31 May 2024 11:25:14 +0300
Subject: [PATCH] chore: add PG prestart script

---
 ansible/files/postgres_prestart.sh.j2                 | 9 +++++++++
 ansible/files/postgresql_config/postgresql.service.j2 | 1 +
 ansible/tasks/internal/postgresql-prestart.yml        | 7 +++++++
 ansible/tasks/setup-supabase-internal.yml             | 3 +++
 common.vars.pkr.hcl                                   | 2 +-
 5 files changed, 21 insertions(+), 1 deletion(-)
 create mode 100644 ansible/files/postgres_prestart.sh.j2
 create mode 100644 ansible/tasks/internal/postgresql-prestart.yml

diff --git a/ansible/files/postgres_prestart.sh.j2 b/ansible/files/postgres_prestart.sh.j2
new file mode 100644
index 000000000..ae5aa1a7d
--- /dev/null
+++ b/ansible/files/postgres_prestart.sh.j2
@@ -0,0 +1,9 @@
+#!/bin/bash
+
+if [ $(cat /etc/locale.gen | grep -c en_US.UTF-8) -eq 0 ]; then
+    echo "en_US.UTF-8 UTF-8" >> /etc/locale.gen
+fi
+
+if [ $(locale -a | grep -c en_US.utf8) -eq 0 ]; then
+    locale-gen
+fi
diff --git a/ansible/files/postgresql_config/postgresql.service.j2 b/ansible/files/postgresql_config/postgresql.service.j2
index ac44e0a45..88cf62c81 100644
--- a/ansible/files/postgresql_config/postgresql.service.j2
+++ b/ansible/files/postgresql_config/postgresql.service.j2
@@ -10,6 +10,7 @@ After=database-optimizations.service
 Type=notify
 User=postgres
 ExecStart=/usr/lib/postgresql/bin/postgres -D /etc/postgresql
+ExecStartPre=+/usr/local/bin/postgres_prestart.sh
 ExecReload=/bin/kill -HUP $MAINPID
 KillMode=mixed
 KillSignal=SIGINT
diff --git a/ansible/tasks/internal/postgresql-prestart.yml b/ansible/tasks/internal/postgresql-prestart.yml
new file mode 100644
index 000000000..46671d589
--- /dev/null
+++ b/ansible/tasks/internal/postgresql-prestart.yml
@@ -0,0 +1,7 @@
+- name: postgres_prestart - create service file
+  template:
+    src: files/postgres_prestart.sh.j2
+    dest: /usr/local/bin/postgres_prestart.sh
+    mode: a+x
+    owner: root
+    group: root
diff --git a/ansible/tasks/setup-supabase-internal.yml b/ansible/tasks/setup-supabase-internal.yml
index 849c76954..0b2e717be 100644
--- a/ansible/tasks/setup-supabase-internal.yml
+++ b/ansible/tasks/setup-supabase-internal.yml
@@ -106,3 +106,6 @@
 
 - name: Install pg_egress_collect
   import_tasks: internal/pg_egress_collect.yml
+
+- name: Install PostgreSQL prestart script
+  import_tasks: internal/postgresql-prestart.yml
diff --git a/common.vars.pkr.hcl b/common.vars.pkr.hcl
index 177beb693..34c95e796 100644
--- a/common.vars.pkr.hcl
+++ b/common.vars.pkr.hcl
@@ -1 +1 @@
-postgres-version = "15.1.1.56"
+postgres-version = "15.1.1.57"