Fix PG typed functions

Makefile

@@ -1,4 +1,4 @@
-.PHONY: build build-endtoend test test-ci test-examples test-endtoend start psql mysqlsh proto
+.PHONY: build build-endtoend test test-ci test-examples test-endtoend start psql mysqlsh proto sqlc
 
 build:
 	go build ./...
@@ -23,17 +23,20 @@ build-endtoend:
 
 test-ci: test-examples build-endtoend vet
 
+sqlc:
+	go build -o ./bin/sqlc ./cmd/sqlc/
+
 sqlc-dev:
-	go build -o ~/bin/sqlc-dev ./cmd/sqlc/
+	go build -o ./bin/sqlc-dev ./cmd/sqlc/
 
 sqlc-pg-gen:
-	go build -o ~/bin/sqlc-pg-gen ./internal/tools/sqlc-pg-gen
+	go build -o ./bin/sqlc-pg-gen ./internal/tools/sqlc-pg-gen
 
 sqlc-gen-json:
-	go build -o ~/bin/sqlc-gen-json ./cmd/sqlc-gen-json
+	go build -o ./bin/sqlc-gen-json ./cmd/sqlc-gen-json
 
 test-json-process-plugin:
-	go build -o ~/bin/test-json-process-plugin ./scripts/test-json-process-plugin/
+	go build -o ./bin/test-json-process-plugin ./scripts/test-json-process-plugin/
 
 start:
 	docker compose up -d

bin/.gitignore

@@ -0,0 +1,2 @@
+*
+!.gitignore

internal/compiler/output_columns.go

@@ -59,6 +59,25 @@ func (c *Compiler) outputColumns(qc *QueryCatalog, node ast.Node) ([]*Column, er
 
 	targets := &ast.List{}
 	switch n := node.(type) {
+	case *ast.CallStmt:
+		fun, err := qc.catalog.ResolveFuncCall(n.FuncCall)
+		if err != nil {
+			return nil, err
+		}
+		var cols []*Column
+		for _, arg := range fun.Args {
+			if arg.Mode == ast.FuncParamOut || arg.Mode == ast.FuncParamInOut || arg.Mode == ast.FuncParamTable {
+				name := arg.Name
+				typeName := arg.Type.Name
+				if arg.Type.Names != nil && len(arg.Type.Names.Items) > 0 {
+					typeName = astutils.Join(arg.Type.Names, ".")
+				} else if arg.Type.Schema != "" {
+					typeName = arg.Type.Schema + "." + arg.Type.Name
+				}
+				cols = append(cols, &Column{Name: name, DataType: typeName, NotNull: false})
+			}
+		}
+		return cols, nil
 	case *ast.DeleteStmt:
 		targets = n.ReturningList
 	case *ast.InsertStmt:
@@ -580,8 +599,7 @@ func (c *Compiler) sourceTables(qc *QueryCatalog, node ast.Node) ([]*Table, erro
 								DataType: arg.Type.Name,
 							})
 						}
-					}
-					if fn.ReturnType != nil {
+					} else if fn.ReturnType != nil {
 						table.Columns = []*Column{
 							{
 								Name:     colName,

internal/endtoend/testdata/ddl_create_function_table/postgresql/pgx/v5/query.sql

@@ -0,0 +1,5 @@
+-- name: SelectAll :many
+SELECT * FROM public.get_test();
+
+-- name: SelectWithTime :many
+SELECT * FROM public.get_test($1::timestamp);

internal/endtoend/testdata/ddl_create_function_table/postgresql/pgx/v5/schema.sql

@@ -0,0 +1,34 @@
+-- Create mock test table
+create table if not exists public.test_data
+(
+    test_id      integer,
+    test_date    date,
+    test_time    timestamp with time zone,
+    test_string  text,
+    test_varchar character varying,
+    test_double  double precision
+);
+
+create function public.get_test(input_time timestamp without time zone DEFAULT now())
+    returns TABLE
+            (
+                test_id      integer,
+                test_date    date,
+                test_time    timestamp with time zone,
+                test_string  text,
+                test_varchar character varying,
+                test_double  double precision
+            )
+    stable
+    language sql
+as
+$$
+SELECT test_id,
+       test_date,
+       test_time,
+       test_string,
+       test_varchar,
+       test_double
+FROM public.test_data
+WHERE test_time <= input_time
+$$;

internal/endtoend/testdata/ddl_create_function_table/postgresql/pgx/v5/sqlc.json

@@ -0,0 +1,13 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "sql_package": "pgx/v5",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}