feat(typegen): materialized views

Author: soedirgo

src/server/routes/generators/typescript.ts

@@ -22,6 +22,8 @@ export default async (fastify: FastifyInstance) => {
     const { data: schemas, error: schemasError } = await pgMeta.schemas.list()
     const { data: tables, error: tablesError } = await pgMeta.tables.list()
     const { data: views, error: viewsError } = await pgMeta.views.list()
+    const { data: materializedViews, error: materializedViewsError } =
+      await pgMeta.materializedViews.list({ includeColumns: true })
     const { data: functions, error: functionsError } = await pgMeta.functions.list()
     const { data: types, error: typesError } = await pgMeta.types.list({
       includeArrayTypes: true,
@@ -44,6 +46,14 @@ export default async (fastify: FastifyInstance) => {
       reply.code(500)
       return { error: viewsError.message }
     }
+    if (materializedViewsError) {
+      request.log.error({
+        error: materializedViewsError,
+        request: extractRequestForLogging(request),
+      })
+      reply.code(500)
+      return { error: materializedViewsError.message }
+    }
     if (functionsError) {
       request.log.error({ error: functionsError, request: extractRequestForLogging(request) })
       reply.code(500)
@@ -63,6 +73,7 @@ export default async (fastify: FastifyInstance) => {
       ),
       tables,
       views,
+      materializedViews,
       functions: functions.filter(
         ({ return_type }) => !['trigger', 'event_trigger'].includes(return_type)
       ),

src/server/server.ts

@@ -39,6 +39,8 @@ if (EXPORT_DOCS) {
   const { data: schemas, error: schemasError } = await pgMeta.schemas.list()
   const { data: tables, error: tablesError } = await pgMeta.tables.list()
   const { data: views, error: viewsError } = await pgMeta.views.list()
+  const { data: materializedViews, error: materializedViewsError } =
+    await pgMeta.materializedViews.list({ includeColumns: true })
   const { data: functions, error: functionsError } = await pgMeta.functions.list()
   const { data: types, error: typesError } = await pgMeta.types.list({
     includeArrayTypes: true,
@@ -55,6 +57,9 @@ if (EXPORT_DOCS) {
   if (viewsError) {
     throw new Error(viewsError.message)
   }
+  if (materializedViewsError) {
+    throw new Error(materializedViewsError.message)
+  }
   if (functionsError) {
     throw new Error(functionsError.message)
   }
@@ -71,6 +76,7 @@ if (EXPORT_DOCS) {
       ),
       tables,
       views,
+      materializedViews,
       functions: functions.filter(
         ({ return_type }) => !['trigger', 'event_trigger'].includes(return_type)
       ),

src/server/templates/typescript.ts

@@ -1,6 +1,7 @@
 import prettier from 'prettier'
 import type {
   PostgresFunction,
+  PostgresMaterializedView,
   PostgresSchema,
   PostgresTable,
   PostgresType,
@@ -11,13 +12,15 @@ export const apply = ({
   schemas,
   tables,
   views,
+  materializedViews,
   functions,
   types,
   arrayTypes,
 }: {
   schemas: PostgresSchema[]
   tables: (PostgresTable & { columns: unknown[] })[]
   views: (PostgresView & { columns: unknown[] })[]
+  materializedViews: (PostgresMaterializedView & { columns: unknown[] })[]
   functions: PostgresFunction[]
   types: PostgresType[]
   arrayTypes: PostgresType[]
@@ -32,7 +35,7 @@ export interface Database {
       const schemaTables = tables
         .filter((table) => table.schema === schema.name)
         .sort(({ name: a }, { name: b }) => a.localeCompare(b))
-      const schemaViews = views
+      const schemaViews = [...views, ...materializedViews]
         .filter((view) => view.schema === schema.name)
         .sort(({ name: a }, { name: b }) => a.localeCompare(b))
       const schemaFunctions = functions
@@ -165,7 +168,7 @@ export interface Database {
                       )}
                   }
                   ${
-                    view.is_updatable
+                    'is_updatable' in view && view.is_updatable
                       ? `Insert: {
                     ${view.columns
                       .sort(({ name: a }, { name: b }) => a.localeCompare(b))
@@ -184,7 +187,7 @@ export interface Database {
                       : ''
                   }
                   ${
-                    view.is_updatable
+                    'is_updatable' in view && view.is_updatable
                       ? `Update: {
                     ${view.columns
                       .sort(({ name: a }, { name: b }) => a.localeCompare(b))