From a08a7cac27aae8ac94632828e6ced0e9d65a0e7d Mon Sep 17 00:00:00 2001
From: Matthew Gordon <matthew@gordon.earth>
Date: Fri, 1 Mar 2024 22:41:02 -0400
Subject: [PATCH] Add scripts to run Postgres in docker

Run a Postgres docker image for (manual) testing
---
 .gitignore                  |  1 +
 dev                         |  8 ++++
 dev-python-requirements.txt |  7 +++
 dev.py                      | 90 +++++++++++++++++++++++++++++++++++++
 4 files changed, 106 insertions(+)
 create mode 100755 dev
 create mode 100644 dev-python-requirements.txt
 create mode 100644 dev.py

diff --git a/.gitignore b/.gitignore
index 6377adc..3d52885 100644
--- a/.gitignore
+++ b/.gitignore
@@ -20,3 +20,4 @@ Cargo.lock
 # MSVC Windows builds of rustc generate these, which store debugging information
 *.pdb
 
+/.python_venv
\ No newline at end of file
diff --git a/dev b/dev
new file mode 100755
index 0000000..aa64e4e
--- /dev/null
+++ b/dev
@@ -0,0 +1,8 @@
+#!/bin/bash
+
+ROOT_DIR=$(realpath "$(dirname "${BASH_SOURCE[0]}")")
+PYTHON_VENV="${ROOT_DIR}/.python_venv"
+if [ ! -d "${PYTHON_VENV}" ]; then
+    python3 -m venv "${PYTHON_VENV}"
+fi
+"${PYTHON_VENV}/bin/python" "${ROOT_DIR}/dev.py" $@
diff --git a/dev-python-requirements.txt b/dev-python-requirements.txt
new file mode 100644
index 0000000..0e59610
--- /dev/null
+++ b/dev-python-requirements.txt
@@ -0,0 +1,7 @@
+certifi==2024.2.2
+charset-normalizer==3.3.2
+docker==7.0.0
+idna==3.6
+packaging==23.2
+requests==2.31.0
+urllib3==2.2.1
diff --git a/dev.py b/dev.py
new file mode 100644
index 0000000..976d1aa
--- /dev/null
+++ b/dev.py
@@ -0,0 +1,90 @@
+import argparse
+import os
+import subprocess
+import shutil
+import sys
+import time
+
+root_dir = sys.path[0]
+
+
+def devupdate(args):
+    subprocess.check_call(
+        [
+            sys.executable,
+            '-m',
+            'pip',
+            'install',
+            '-r',
+            os.path.join(root_dir, 'dev-python-requirements.txt'),
+        ]
+    )
+
+
+def run(args):
+    import docker
+
+    POSTGRES_USER = 'locality'
+    POSTGRES_PASSWORD = 'wkyhjofg2837f'
+    POSTGRES_DB = 'locality'
+
+    postgres_env = {
+        'POSTGRES_USER': POSTGRES_USER,
+        'POSTGRES_PASSWORD': POSTGRES_PASSWORD,
+        'POSTGRES_DB': POSTGRES_DB,
+    }
+
+    locality_env = {
+        'LOCALITY_DATABASE_URL': 'postgres://{}:{}@localhost/{}'.format(
+            POSTGRES_USER, POSTGRES_PASSWORD, POSTGRES_DB
+        )
+    }
+
+    docker_client = docker.from_env()
+    postgres_container = docker_client.containers.run(
+        'postgres:14.11',
+        environment=postgres_env,
+        detach=True,
+        ports={'5432/tcp': ('127.0.0.1', 5432)},
+    )
+    try:
+        docker_stream = postgres_container.attach(stream=True)
+        for line_binary in docker_stream:
+            line = line_binary.decode('utf-8')
+            print(line)
+            if 'listening on IPv4 address "0.0.0.0", port 5432' in line:
+                break
+        for line_binary in docker_stream:
+            line = line_binary.decode('utf-8')
+            print(line)
+            if 'database system is ready to accept connections' in line:
+                break
+
+        cargo_bin = shutil.which('cargo')
+        locality_process = subprocess.Popen(
+            [cargo_bin, 'run'], env=locality_env, cwd=root_dir
+        )
+
+        try:
+            while locality_process.poll() is None:
+                time.sleep(0.5)
+        finally:
+            if locality_process.poll() is None:
+                locality_process.terminate()
+    finally:
+        postgres_container.stop()
+        postgres_container.remove()
+
+
+parser = argparse.ArgumentParser()
+subparsers = parser.add_subparsers(required=True)
+run_parser = subparsers.add_parser(
+    'run', help='Run a test instance of locality'
+)
+run_parser.set_defaults(func=run)
+devupdate_parser = subparsers.add_parser(
+    'devupdate', help='Install or update packages used by this script'
+)
+devupdate_parser.set_defaults(func=devupdate)
+args = parser.parse_args()
+args.func(args)