fix(compiler): insert-with-select statements

internal/compiler/query_catalog.go

@@ -20,7 +20,11 @@ func (comp *Compiler) buildQueryCatalog(c *catalog.Catalog, node ast.Node, embed
 	case *ast.DeleteStmt:
 		with = n.WithClause
 	case *ast.InsertStmt:
-		with = n.WithClause
+		if selectStmt, ok := n.SelectStmt.(*ast.SelectStmt); ok && selectStmt.WithClause != nil {
+			with = selectStmt.WithClause
+		} else {
+			with = n.WithClause
+		}
 	case *ast.UpdateStmt:
 		with = n.WithClause
 	case *ast.SelectStmt:

internal/endtoend/testdata/insert_cte_select/mysql/query.sql

@@ -0,0 +1,8 @@
+-- name: CreateAuthor :exec
+insert into authors (id, name, bio)
+with potential_authors as (
+  select id, name, bio
+  from dummy
+)
+select id, name, bio
+from potential_authors;

internal/endtoend/testdata/insert_cte_select/mysql/schema.sql

@@ -0,0 +1,11 @@
+create table authors (
+  id integer primary key,
+  name text not null,
+  bio text
+);
+
+create table dummy (
+  id integer primary key,
+  name text not null,
+  bio text
+);

internal/endtoend/testdata/insert_cte_select/mysql/sqlc.json

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "engine": "mysql",
+      "path": "go",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

internal/endtoend/testdata/insert_cte_select/postgresql/pgx/v4/query.sql

@@ -0,0 +1,8 @@
+-- name: CreateAuthor :exec
+insert into authors (id, name, bio)
+with potential_authors as (
+  select id, name, bio
+  from dummy
+)
+select id, name, bio
+from potential_authors;

internal/endtoend/testdata/insert_cte_select/postgresql/pgx/v4/schema.sql

@@ -0,0 +1,11 @@
+create table authors (
+  id integer primary key,
+  name text not null,
+  bio text
+);
+
+create table dummy (
+  id integer primary key,
+  name text not null,
+  bio text
+);

internal/endtoend/testdata/insert_cte_select/postgresql/pgx/v4/sqlc.json

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

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

@@ -0,0 +1,8 @@
+-- name: CreateAuthor :exec
+insert into authors (id, name, bio)
+with potential_authors as (
+  select id, name, bio
+  from dummy
+)
+select id, name, bio
+from potential_authors;

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

@@ -0,0 +1,11 @@
+create table authors (
+  id integer primary key,
+  name text not null,
+  bio text
+);
+
+create table dummy (
+  id integer primary key,
+  name text not null,
+  bio text
+);

internal/endtoend/testdata/insert_cte_select/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"
+    }
+  ]
+}