From 760dc88ef3930d8a9f72da67760ab93c431821d7 Mon Sep 17 00:00:00 2001
From: Martin McKeaveney <martin@shogunsystems.co.uk>
Date: Mon, 14 Jun 2021 20:05:18 +0100
Subject: [PATCH] SQL for fetching primary keys, needs merged into column SQL

---
 .../app/[application]/data/_layout.svelte     |  6 ++---
 .../server/src/integrations/plus/postgres.js  | 23 ++++++++++++++++---
 2 files changed, 23 insertions(+), 6 deletions(-)

diff --git a/packages/builder/src/pages/builder/app/[application]/data/_layout.svelte b/packages/builder/src/pages/builder/app/[application]/data/_layout.svelte
index 6910d3e925..ef360efd3f 100644
--- a/packages/builder/src/pages/builder/app/[application]/data/_layout.svelte
+++ b/packages/builder/src/pages/builder/app/[application]/data/_layout.svelte
@@ -8,7 +8,7 @@
 
   const tabs = [
     {
-      title: "Internal",
+      title: "Databases",
       key: "table",
     },
     {
@@ -17,7 +17,7 @@
     },
   ]
 
-  let selected = $isActive("./datasource") ? "External" : "Internal"
+  let selected = $isActive("./datasource") ? "External" : "Databases"
 
   function selectFirstTableOrSource({ detail }) {
     const { key } = tabs.find(t => t.title === detail)
@@ -35,7 +35,7 @@
 <div class="root">
   <div class="nav">
     <Tabs {selected} on:select={selectFirstTableOrSource}>
-      <Tab title="Internal">
+      <Tab title="Databases">
         <div class="tab-content-padding">
           <TableNavigator />
           <Modal bind:this={modal}>
diff --git a/packages/server/src/integrations/plus/postgres.js b/packages/server/src/integrations/plus/postgres.js
index d1df611c74..3ac49dc925 100644
--- a/packages/server/src/integrations/plus/postgres.js
+++ b/packages/server/src/integrations/plus/postgres.js
@@ -59,6 +59,16 @@ class PostgresPlus extends Sql {
   COLUMNS_SQL =
     "select * from information_schema.columns where table_schema = 'public'"
 
+  PRIMARY_KEYS_SQL = `
+    select tc.table_schema, tc.table_name, tc.column_name, kc.column_name as primary_key 
+    from information_schema.table_constraints tc
+    join 
+      information_schema.key_column_usage kc on kc.table_name = tc.table_name 
+      and kc.table_schema = tc.table_schema 
+      and kc.constraint_name = tc.constraint_name
+    where tc.constraint_type = 'PRIMARY KEY';
+  `
+
   constructor(config) {
     super("pg")
     this.config = config
@@ -70,17 +80,24 @@ class PostgresPlus extends Sql {
   }
 
   async init() {
-    const response = await this.client.query(this.COLUMNS_SQL)
+    const primaryKeysResponse = await this.client.query(this.PRIMARY_KEYS_SQL)
+    const primaryKeys = {}
 
+    for (let table of primaryKeysResponse.rows) {
+      primaryKeys[table.primary_key] = table.column_name
+    }
+
+    const columnsResponse = await this.client.query(this.COLUMNS_SQL)
     const tables = {}
-    for (let column of response.rows) {
+
+    for (let column of columnsResponse.rows) {
       const tableName = column.table_name
       const columnName = column.column_name
 
       // table key doesn't exist yet
       if (!tables[tableName]) {
         tables[tableName] = {
-          _id: "something",
+          _id: primaryKeys[tableName],
           name: tableName,
           schema: {},
         }