feat(typegen): add computed cols to row (#354)

* feat(typegen): add computed cols to row

* Update 00-init.sql

Co-authored-by: Bobbie Soedirgo <[email protected]>

Author: psteinroe

src/server/templates/typescript.ts

@@ -37,14 +37,26 @@ export interface Database {
                 : schemaTables.map(
                     (table) => `${JSON.stringify(table.name)}: {
                   Row: {
-                    ${table.columns.map(
-                      (column) =>
-                        `${JSON.stringify(column.name)}: ${pgTypeToTsType(
-                          column.format,
-                          types,
-                          schemas
-                        )} ${column.is_nullable ? '| null' : ''}`
-                    )}
+                    ${[
+                      ...table.columns.map(
+                        (column) =>
+                          `${JSON.stringify(column.name)}: ${pgTypeToTsType(
+                            column.format,
+                            types,
+                            schemas
+                          )} ${column.is_nullable ? '| null' : ''}`
+                      ),
+                      ...schemaFunctions
+                        .filter((fn) => fn.argument_types === table.name)
+                        .map(
+                          (fn) =>
+                            `${JSON.stringify(fn.name)}: ${pgTypeToTsType(
+                              fn.return_type,
+                              types,
+                              schemas
+                            )}`
+                        ),
+                    ]}
                   }
                   Insert: {
                     ${table.columns.map((column) => {

test/db/00-init.sql

@@ -51,3 +51,8 @@ end;
 $$ language plpgsql;
 
 CREATE VIEW todos_view AS SELECT * FROM public.todos;
+
+create function public.blurb(public.todos) returns text as
+$$
+select substring($1.details, 1, 3);
+$$ language sql stable;