feat(relationships): include is_one_to_one

Author: soedirgo

src/lib/PostgresMetaRelationships.ts

@@ -92,6 +92,7 @@ export default class PostgresMetaRelationships {
             schema: vtkd.view_schema,
             relation: vtkd.view_name,
             columns: viewColumns,
+            is_one_to_one: r.is_one_to_one,
             referenced_schema: r.referenced_schema,
             referenced_relation: r.referenced_relation,
             referenced_columns: r.referenced_columns,
@@ -104,6 +105,7 @@ export default class PostgresMetaRelationships {
             schema: r.schema,
             relation: r.relation,
             columns: r.columns,
+            is_one_to_one: r.is_one_to_one,
             referenced_schema: tvkd.view_schema,
             referenced_relation: tvkd.view_name,
             referenced_columns: viewColumns,
@@ -119,6 +121,7 @@ export default class PostgresMetaRelationships {
                   schema: vtkd.view_schema,
                   relation: vtkd.view_name,
                   columns: viewColumns,
+                  is_one_to_one: r.is_one_to_one,
                   referenced_schema: tvkd.view_schema,
                   referenced_relation: tvkd.view_name,
                   referenced_columns: referencedViewColumns,

src/lib/sql/table_relationships.sql

@@ -1,13 +1,31 @@
 -- Adapted from
 -- https://github.com/PostgREST/postgrest/blob/f9f0f79fa914ac00c11fbf7f4c558e14821e67e2/src/PostgREST/SchemaCache.hs#L722
+WITH
+pks_uniques_cols AS (
+  SELECT
+    connamespace,
+    conrelid,
+    jsonb_agg(column_info.cols) as cols
+  FROM pg_constraint
+  JOIN lateral (
+    SELECT array_agg(cols.attname order by cols.attnum) as cols
+    FROM ( select unnest(conkey) as col) _
+    JOIN pg_attribute cols on cols.attrelid = conrelid and cols.attnum = col
+  ) column_info ON TRUE
+  WHERE
+    contype IN ('p', 'u') and
+    connamespace::regnamespace::text <> 'pg_catalog'
+  GROUP BY connamespace, conrelid
+)
 SELECT
   traint.conname AS foreign_key_name,
   ns1.nspname AS schema,
   tab.relname AS relation,
   column_info.cols AS columns,
   ns2.nspname AS referenced_schema,
   other.relname AS referenced_relation,
-  column_info.refs AS referenced_columns
+  column_info.refs AS referenced_columns,
+  (column_info.cols IN (SELECT * FROM jsonb_array_elements(pks_uqs.cols))) AS is_one_to_one
 FROM pg_constraint traint
 JOIN LATERAL (
   SELECT
@@ -21,4 +39,6 @@ JOIN pg_namespace ns1 ON ns1.oid = traint.connamespace
 JOIN pg_class tab ON tab.oid = traint.conrelid
 JOIN pg_class other ON other.oid = traint.confrelid
 JOIN pg_namespace ns2 ON ns2.oid = other.relnamespace
+LEFT JOIN pks_uniques_cols pks_uqs ON pks_uqs.connamespace = traint.connamespace AND pks_uqs.conrelid = traint.conrelid
 WHERE traint.contype = 'f'
+AND traint.conparentid = 0

src/lib/types.ts

@@ -239,6 +239,7 @@ export const postgresRelationshipSchema = Type.Object({
   schema: Type.String(),
   relation: Type.String(),
   columns: Type.Array(Type.String()),
+  is_one_to_one: Type.Boolean(),
   referenced_schema: Type.String(),
   referenced_relation: Type.String(),
   referenced_columns: Type.Array(Type.String()),

src/server/templates/typescript.ts

@@ -167,6 +167,7 @@ export interface Database {
                         (relationship) => `{
                         foreignKeyName: ${JSON.stringify(relationship.foreign_key_name)}
                         columns: ${JSON.stringify(relationship.columns)}
+                        isOneToOne: ${relationship.is_one_to_one}
                         referencedRelation: ${JSON.stringify(relationship.referenced_relation)}
                         referencedColumns: ${JSON.stringify(relationship.referenced_columns)}
                       }`
@@ -235,6 +236,7 @@ export interface Database {
                         (relationship) => `{
                         foreignKeyName: ${JSON.stringify(relationship.foreign_key_name)}
                         columns: ${JSON.stringify(relationship.columns)}
+                        isOneToOne: ${relationship.is_one_to_one}
                         referencedRelation: ${JSON.stringify(relationship.referenced_relation)}
                         referencedColumns: ${JSON.stringify(relationship.referenced_columns)}
                       }`