feat(typegen): add CompositeTypes

Author: soedirgo

src/server/templates/typescript.ts

@@ -60,6 +60,9 @@ export interface Database {
       const schemaEnums = types
         .filter((type) => type.schema === schema.name && type.enums.length > 0)
         .sort(({ name: a }, { name: b }) => a.localeCompare(b))
+      const schemaCompositeTypes = types
+        .filter((type) => type.schema === schema.name && type.attributes.length > 0)
+        .sort(({ name: a }, { name: b }) => a.localeCompare(b))
       return `${JSON.stringify(schema.name)}: {
           Tables: {
             ${
@@ -294,6 +297,28 @@ export interface Database {
                   )
             }
           }
+          CompositeTypes: {
+            ${
+              schemaCompositeTypes.length === 0
+                ? '[_ in never]: never'
+                : schemaCompositeTypes.map(
+                    ({ name, attributes }) =>
+                      `${JSON.stringify(name)}: {
+                        ${attributes.map(({ name, type_id }) => {
+                          const type = types.find(({ id }) => id === type_id)
+                          if (type) {
+                            return `${JSON.stringify(name)}: ${pgTypeToTsType(
+                              type.name,
+                              types,
+                              schemas
+                            )}`
+                          }
+                          return 'unknown'
+                        })}
+                      }`
+                  )
+            }
+          }
         }`
     })}
 }`
@@ -305,7 +330,7 @@ export interface Database {
   return output
 }
 
-// TODO: Make this more robust. Currently doesn't handle composite types - returns them as unknown.
+// TODO: Make this more robust. Currently doesn't handle range types - returns them as unknown.
 const pgTypeToTsType = (
   pgType: string,
   types: PostgresType[],
@@ -339,12 +364,24 @@ const pgTypeToTsType = (
   } else if (pgType.startsWith('_')) {
     return `(${pgTypeToTsType(pgType.substring(1), types, schemas)})[]`
   } else {
-    const type = types.find((type) => type.name === pgType && type.enums.length > 0)
-    if (type) {
-      if (schemas.some(({ name }) => name === type.schema)) {
-        return `Database[${JSON.stringify(type.schema)}]['Enums'][${JSON.stringify(type.name)}]`
+    const enumType = types.find((type) => type.name === pgType && type.enums.length > 0)
+    if (enumType) {
+      if (schemas.some(({ name }) => name === enumType.schema)) {
+        return `Database[${JSON.stringify(enumType.schema)}]['Enums'][${JSON.stringify(
+          enumType.name
+        )}]`
+      }
+      return enumType.enums.map((variant) => JSON.stringify(variant)).join('|')
+    }
+
+    const compositeType = types.find((type) => type.name === pgType && type.attributes.length > 0)
+    if (compositeType) {
+      if (schemas.some(({ name }) => name === compositeType.schema)) {
+        return `Database[${JSON.stringify(
+          compositeType.schema
+        )}]['CompositeTypes'][${JSON.stringify(compositeType.name)}]`
       }
-      return type.enums.map((variant) => JSON.stringify(variant)).join('|')
+      return 'unknown'
     }
 
     return 'unknown'

test/lib/types.ts

@@ -5,8 +5,8 @@ test('list', async () => {
   expect(res.data?.find(({ name }) => name === 'user_status')).toMatchInlineSnapshot(
     { id: expect.any(Number) },
     `
-    Object {
-      "attributes": Array [],
+    {
+      "attributes": [],
       "comment": null,
       "enums": [
         "ACTIVE",
@@ -63,19 +63,19 @@ test('composite type attributes', async () => {
   expect(res.data?.find(({ name }) => name === 'test_composite')).toMatchInlineSnapshot(
     { id: expect.any(Number) },
     `
-    Object {
-      "attributes": Array [
-        Object {
+    {
+      "attributes": [
+        {
           "name": "id",
           "type_id": 20,
         },
-        Object {
+        {
           "name": "data",
           "type_id": 25,
         },
       ],
       "comment": null,
-      "enums": Array [],
+      "enums": [],
       "format": "test_composite",
       "id": Any<Number>,
       "name": "test_composite",