WIP: Add functions to POST path

Author: laurenceisla

sql/components.sql

@@ -19,6 +19,7 @@ returns jsonb language sql stable as
 $$
   select oas_build_component_schemas_from_tables_and_composite_types(schemas) ||
          oas_build_component_schemas_from_functions_return_types(schemas) ||
+         oas_build_component_schemas_from_functions_arguments(schemas) ||
          oas_build_component_schemas_headers()
 $$;
 
@@ -180,6 +181,81 @@ from (
 ) x;
 $$;
 
+create or replace function oas_build_component_schemas_from_functions_arguments(schemas text[])
+returns jsonb language sql stable as
+$$
+with all_functions_with_arguments as (
+  -- Build Component Schemas for IN/INOUT function arguments
+  select *, argument_name = '' and argument_input_qty = 1 as argument_is_single_unnamed
+  from postgrest_get_all_functions(schemas)
+  where argument_input_qty > 0
+    and (argument_is_in or argument_is_inout or argument_is_variadic)
+),
+aggregated_function_arguments as (
+  select
+    function_schema,
+    function_full_name,
+    function_description,
+    argument_is_single_unnamed,
+    array_agg(argument_name order by argument_position) filter (where argument_is_required) AS required_arguments,
+    -- Build objects for functions with one or more named arguments
+    case when not argument_is_single_unnamed then
+      jsonb_object_agg(
+        argument_name,
+        case when argument_item_type_name is null and argument_is_composite then
+          oas_build_reference_to_schemas(argument_composite_full_name)
+        else
+          oas_schema_object(
+            type := postgrest_pgtype_to_oastype(argument_type_name),
+            format := argument_type_name::text,
+            items :=
+              case
+              when argument_item_type_name is null then
+                null
+              when argument_is_composite then
+                oas_build_reference_to_schemas(argument_composite_full_name)
+              else
+                oas_schema_object(
+                  type := postgrest_pgtype_to_oastype(argument_item_type_name),
+                  format := argument_item_type_name::text
+                )
+              end
+          )
+        end
+        order by argument_position
+      )
+    -- For functions with a single unnamed parameter, build the Schema according to the parameter type
+    else
+      jsonb_agg(
+        oas_schema_object(
+          type := postgrest_pgtype_to_oastype(argument_type_name),
+          format := argument_type_name::text
+        )
+      )
+    end as arguments
+  from all_functions_with_arguments
+  group by function_schema, function_full_name, function_description, argument_is_single_unnamed
+)
+select jsonb_object_agg(x.component_name, x.oas_schema)
+from (
+  select
+    'rpc.args.' || function_full_name as component_name,
+    case when not argument_is_single_unnamed then
+      oas_schema_object(
+        description := function_description,
+        properties := coalesce(arguments, '{}'),
+        type := 'object',
+        required := required_arguments
+      )
+    else
+      -- We take the first and only element, since there's just one single unnamed argument
+      arguments->0
+    end as oas_schema
+  from
+    aggregated_function_arguments
+) x;
+$$;
+
 create or replace function oas_build_component_schemas_headers()
 returns jsonb language sql stable as
 $$
@@ -304,6 +380,7 @@ returns jsonb language sql stable as
 $$
   select oas_build_component_parameters_query_params_from_tables(schemas) ||
          oas_build_component_parameters_query_params_from_function_args(schemas) ||
+         oas_build_component_parameters_query_params_from_function_ret(schemas) ||
          oas_build_component_parameters_query_params_common() ||
          oas_build_component_parameters_headers_common();
 $$;
@@ -322,14 +399,42 @@ from (
       )
     ) as param_schema
   from (
-     select table_full_name, column_name
-     from postgrest_get_all_tables_and_composite_types()
-     where table_schema = any(schemas)
-       and (is_table or is_view)
+    select table_full_name, column_name
+    from postgrest_get_all_tables_and_composite_types()
+    where (
+        table_schema = any(schemas)
+        and (is_table or is_view)
+      )
+      -- composite type columns can also be used as row filters if a function returns it
+      or exists (
+        select 1
+        from postgrest_get_all_functions(schemas)
+        where return_type_composite_relid = table_oid
+      )
   ) _
 ) x;
 $$;
 
+-- Builds "rowFilter"s for functions returning TABLE or INOUT/OUT types
+create or replace function oas_build_component_parameters_query_params_from_function_ret(schemas text[])
+returns jsonb language sql stable as
+$$
+select jsonb_object_agg(x.param_name, x.param_schema)
+from (
+  select format('rowFilter.rpc.%1$s.%2$s', function_full_name, argument_name) as param_name,
+    oas_parameter_object(
+      name := argument_name,
+      "in" := 'query',
+      schema := oas_schema_object(
+        type := 'string'
+      )
+    ) as param_schema
+  from postgrest_get_all_functions(schemas)
+  where argument_name <> ''
+    and (argument_is_inout or argument_is_out or argument_is_table)
+) x;
+$$;
+
 create or replace function oas_build_component_parameters_query_params_from_function_args(schemas text[])
 returns jsonb language sql stable as
 $$
@@ -882,7 +987,8 @@ $$;
 create or replace function oas_build_request_bodies(schemas text[])
 returns jsonb language sql stable as
 $$
-select oas_build_request_bodies_from_tables(schemas);
+select oas_build_request_bodies_from_tables(schemas) ||
+       oas_build_request_bodies_from_functions(schemas);
 $$;
 
 create or replace function oas_build_request_bodies_from_tables(schemas text[])
@@ -928,6 +1034,40 @@ from (
 ) as x;
 $$;
 
+create or replace function oas_build_request_bodies_from_functions(schemas text[])
+returns jsonb language sql stable as
+$$
+select jsonb_object_agg('rpc.' || x.function_full_name, x.oas_req_body)
+from (
+  select
+    function_full_name,
+    oas_request_body_object(
+      description := function_full_name,
+      required := argument_default_qty < argument_input_qty,
+      content := jsonb_build_object(
+        'application/json',
+        oas_media_type_object(
+          "schema" := oas_build_reference_to_schemas('rpc.args.' || function_full_name)
+        ),
+        'application/x-www-form-urlencoded',
+        oas_media_type_object(
+          "schema" := oas_build_reference_to_schemas('rpc.args.' || function_full_name)
+        ),
+        'text/csv',
+        oas_media_type_object(
+          "schema" := oas_schema_object(
+            type := 'string',
+            format := 'csv'
+          )
+        )
+      )
+    ) as oas_req_body
+  from postgrest_get_all_functions(schemas)
+  where argument_input_qty > 0
+  group by function_full_name, argument_input_qty, argument_default_qty
+) as x;
+$$;
+
 -- Security Schemes
 
 create or replace function oas_build_component_security_schemes ()

sql/paths.sql

@@ -189,13 +189,73 @@ from (
             'default',
             oas_build_reference_to_responses('defaultError', 'Error')
           )
+      ),
+      post := oas_operation_object(
+        summary := (postgrest_unfold_comment(function_description))[1],
+        description := (postgrest_unfold_comment(function_description))[2],
+        tags := array['(rpc) ' || function_name],
+        requestBody := case when argument_input_qty > 0 then oas_build_reference_to_request_bodies('rpc.' || function_full_name) end,
+        parameters :=
+          -- TODO: The row filters for functions returning TABLE, OUT, INOUT and composite types should also work for the GET path.
+          --       Right now they're not included in GET, because the argument names (in rpcParams) could clash with the name of the return type columns (in rowFilter).
+          coalesce(
+            jsonb_agg(
+              oas_build_reference_to_parameters(format('rowFilter.rpc.%1$s.%2$s', function_full_name, argument_name))
+            ) filter ( where argument_name <> '' and (argument_is_inout or argument_is_out or argument_is_table)),
+            '[]'
+          ) ||
+          return_composite_param_ref ||
+          case when return_type_is_table or return_type_is_out or return_type_composite_relid <> 0 then
+            jsonb_build_array(
+              oas_build_reference_to_parameters('select'),
+              oas_build_reference_to_parameters('order'),
+              oas_build_reference_to_parameters('limit'),
+              oas_build_reference_to_parameters('offset'),
+              oas_build_reference_to_parameters('or'),
+              oas_build_reference_to_parameters('and'),
+              oas_build_reference_to_parameters('not.or'),
+              oas_build_reference_to_parameters('not.and'),
+              oas_build_reference_to_parameters('preferPostRpc')
+            )
+          else
+            jsonb_build_array(
+              oas_build_reference_to_parameters('preferPostRpc')
+            )
+          end,
+        responses :=
+          case when return_type_is_set then
+            jsonb_build_object(
+              '200',
+              oas_build_reference_to_responses('rpc.' || function_full_name, 'OK'),
+              '206',
+              oas_build_reference_to_responses('rpc.' || function_full_name, 'Partial Content')
+            )
+          else
+            jsonb_build_object(
+              '200',
+              oas_build_reference_to_responses('rpc.' || function_full_name, 'OK')
+            )
+          end ||
+          jsonb_build_object(
+            'default',
+            oas_build_reference_to_responses('defaultError', 'Error')
+          )
       )
     ) as oas_path_item
   from (
-    select function_name, function_full_name, function_description, return_type_name, return_type_is_set, return_type_is_table, return_type_is_out, return_type_composite_relid, argument_name, argument_is_in, argument_is_inout, argument_is_variadic
-    from postgrest_get_all_functions(schemas)
+    select function_name, function_full_name, function_description, return_type_name, return_type_is_set, return_type_is_table, return_type_is_out, return_type_composite_relid, argument_name, argument_is_in, argument_is_inout, argument_is_out, argument_is_table, argument_is_variadic, argument_input_qty,
+           comp.return_composite_param_ref
+    from postgrest_get_all_functions(schemas) f
+    left join lateral (
+      select coalesce(jsonb_agg(oas_build_reference_to_parameters(format('rowFilter.%1$s.%2$s', table_full_name, column_name))),'[]') as return_composite_param_ref
+      from (
+        select c.table_full_name, c.column_name
+        from postgrest_get_all_tables_and_composite_types() c
+        where f.return_type_composite_relid = c.table_oid
+      ) _
+    ) comp on true
   ) _
-  group by function_name, function_full_name, function_description, return_type_name, return_type_is_set, return_type_is_table, return_type_is_out, return_type_composite_relid
+  group by function_name, function_full_name, function_description, return_type_name, return_type_is_set, return_type_is_table, return_type_is_out, return_type_composite_relid, argument_input_qty, return_composite_param_ref
 ) x;
 $$;
 

sql/postgrest.sql

@@ -166,6 +166,7 @@ $$;
 create or replace function postgrest_get_all_functions(schemas text[])
 returns table (
   argument_input_qty int,
+  argument_default_qty int,
   argument_name text,
   argument_reg_type oid,
   argument_type_name text,
@@ -225,6 +226,7 @@ $$
   all_functions AS (
     SELECT
       p.pronargs AS argument_input_qty,
+      p.pronargdefaults AS argument_default_qty,
       COALESCE(pa.name, '') AS argument_name,
       pa.type AS argument_reg_type,
       format_type(ta.oid, NULL::integer) AS argument_type_name,