From 189e1bf547693b2160ad5b1191660392a2677618 Mon Sep 17 00:00:00 2001
From: Ivan Shmidt <bahoc47@gmail.com>
Date: Thu, 23 Jul 2020 22:11:25 +0200
Subject: [PATCH] feat(backend): add support for PG

---
 packages/backend/package.json |  1 +
 packages/backend/src/index.ts | 34 +++++++++++++++++++++++++++++-----
 2 files changed, 30 insertions(+), 5 deletions(-)

diff --git a/packages/backend/package.json b/packages/backend/package.json
index 86d348d2a1..b040c12554 100644
--- a/packages/backend/package.json
+++ b/packages/backend/package.json
@@ -34,6 +34,7 @@
     "dockerode": "^3.2.0",
     "express": "^4.17.1",
     "knex": "^0.21.1",
+    "pg": "^8.3.0",
     "sqlite3": "^4.2.0",
     "winston": "^3.2.1"
   },
diff --git a/packages/backend/src/index.ts b/packages/backend/src/index.ts
index d060791f98..8b9edec327 100644
--- a/packages/backend/src/index.ts
+++ b/packages/backend/src/index.ts
@@ -45,11 +45,35 @@ function makeCreateEnv(loadedConfigs: AppConfig[]) {
 
   return (plugin: string): PluginEnvironment => {
     const logger = getRootLogger().child({ type: 'plugin', plugin });
-    const database = knex({
-      client: 'sqlite3',
-      connection: ':memory:',
-      useNullAsDefault: true,
-    });
+    // Supported DBs are sqlite and postgres
+    const isPg = [
+      'POSTGRES_USER',
+      'POSTGRES_HOST',
+      'POSTGRES_PASSWORD',
+    ].every(key => Object.keys(process.env).includes(key));
+
+    let knexConfig;
+
+    if (isPg) {
+      knexConfig = {
+        client: 'pg',
+        useNullAsDefault: true,
+        connection: {
+          host: process.env.PG_HOST,
+          user: process.env.PG_USER,
+          password: process.env.PG_PASSWORD,
+          database: `backstage_plugin_${plugin}`,
+        },
+      };
+    } else {
+      knexConfig = {
+        client: 'sqlite3',
+        connection: ':memory:',
+        useNullAsDefault: true,
+      };
+    }
+
+    const database = knex(knexConfig);
     database.client.pool.on('createSuccess', (_eventId: any, resource: any) => {
       resource.run('PRAGMA foreign_keys = ON', () => {});
     });