fix: include materializedView types

Author: avallete

src/lib/sql/types.sql.ts

@@ -47,7 +47,7 @@ from
         t.typrelid = 0
         or (
           select
-            c.relkind ${props.includeTableTypes ? `in ('c', 'r', 'v')` : `= 'c'`}
+            c.relkind ${props.includeTableTypes ? `in ('c', 'r', 'v', 'm')` : `= 'c'`}
           from
             pg_class c
           where

test/db/00-init.sql

@@ -203,6 +203,13 @@ AS $$
   SELECT * FROM public.users_audit WHERE user_id = user_row.id;
 $$;
 
+CREATE OR REPLACE FUNCTION public.get_todos_by_matview(todos_matview)
+RETURNS SETOF todos ROWS 1
+LANGUAGE SQL STABLE
+AS $$
+  SELECT * FROM public.todos LIMIT 1;
+$$;
+
 CREATE OR REPLACE FUNCTION public.get_todos_setof_rows(user_row users)
 RETURNS SETOF todos
 LANGUAGE SQL STABLE

test/lib/functions.ts

@@ -118,7 +118,7 @@ test('list set-returning function with multiples definitions', async () => {
         "definition": "
       SELECT * FROM public.todos WHERE "user-id" = user_row.id;
     ",
-        "id": 16507,
+        "id": 16508,
         "identity_argument_types": "user_row users",
         "is_set_returning_function": true,
         "language": "sql",
@@ -153,7 +153,7 @@ test('list set-returning function with multiples definitions', async () => {
         "definition": "
       SELECT * FROM public.todos WHERE "user-id" = todo_row."user-id";
     ",
-        "id": 16508,
+        "id": 16509,
         "identity_argument_types": "todo_row todos",
         "is_set_returning_function": true,
         "language": "sql",

test/server/typegen.ts

@@ -306,6 +306,11 @@ test('typegen: typescript', async () => {
               details: string | null
               id: number | null
               "user-id": number | null
+              get_todos_by_matview: {
+                details: string | null
+                id: number
+                "user-id": number
+              } | null
             }
             Relationships: [
               {
@@ -628,6 +633,20 @@ test('typegen: typescript', async () => {
                   user_status: Database["public"]["Enums"]["user_status"] | null
                 }
               }
+          get_todos_by_matview: {
+            Args: { "": unknown }
+            Returns: {
+              details: string | null
+              id: number
+              "user-id": number
+            }
+            SetofOptions: {
+              from: "todos_matview"
+              to: "todos"
+              isOneToOne: true
+              isSetofReturn: true
+            }
+          }
           get_todos_from_user:
             | {
                 Args: {
@@ -1347,6 +1366,11 @@ test('typegen w/ one-to-one relationships', async () => {
               details: string | null
               id: number | null
               "user-id": number | null
+              get_todos_by_matview: {
+                details: string | null
+                id: number
+                "user-id": number
+              } | null
             }
             Relationships: [
               {
@@ -1681,6 +1705,20 @@ test('typegen w/ one-to-one relationships', async () => {
                   user_status: Database["public"]["Enums"]["user_status"] | null
                 }
               }
+          get_todos_by_matview: {
+            Args: { "": unknown }
+            Returns: {
+              details: string | null
+              id: number
+              "user-id": number
+            }
+            SetofOptions: {
+              from: "todos_matview"
+              to: "todos"
+              isOneToOne: true
+              isSetofReturn: true
+            }
+          }
           get_todos_from_user:
             | {
                 Args: {
@@ -2400,6 +2438,11 @@ test('typegen: typescript w/ one-to-one relationships', async () => {
               details: string | null
               id: number | null
               "user-id": number | null
+              get_todos_by_matview: {
+                details: string | null
+                id: number
+                "user-id": number
+              } | null
             }
             Relationships: [
               {
@@ -2734,6 +2777,20 @@ test('typegen: typescript w/ one-to-one relationships', async () => {
                   user_status: Database["public"]["Enums"]["user_status"] | null
                 }
               }
+          get_todos_by_matview: {
+            Args: { "": unknown }
+            Returns: {
+              details: string | null
+              id: number
+              "user-id": number
+            }
+            SetofOptions: {
+              from: "todos_matview"
+              to: "todos"
+              isOneToOne: true
+              isSetofReturn: true
+            }
+          }
           get_todos_from_user:
             | {
                 Args: {
@@ -3458,6 +3515,11 @@ test('typegen: typescript w/ postgrestVersion', async () => {
               details: string | null
               id: number | null
               "user-id": number | null
+              get_todos_by_matview: {
+                details: string | null
+                id: number
+                "user-id": number
+              } | null
             }
             Relationships: [
               {
@@ -3792,6 +3854,20 @@ test('typegen: typescript w/ postgrestVersion', async () => {
                   user_status: Database["public"]["Enums"]["user_status"] | null
                 }
               }
+          get_todos_by_matview: {
+            Args: { "": unknown }
+            Returns: {
+              details: string | null
+              id: number
+              "user-id": number
+            }
+            SetofOptions: {
+              from: "todos_matview"
+              to: "todos"
+              isOneToOne: true
+              isSetofReturn: true
+            }
+          }
           get_todos_from_user:
             | {
                 Args: {