fix: WithSchemaPlugin is adding schema to function arguments and causes db errors.

src/plugin/with-schema/with-schema-transformer.ts

@@ -1,4 +1,6 @@
+import { AggregateFunctionNode } from '../../operation-node/aggregate-function-node.js'
 import { AliasNode } from '../../operation-node/alias-node.js'
+import { FunctionNode } from '../../operation-node/function-node.js'
 import { IdentifierNode } from '../../operation-node/identifier-node.js'
 import { OperationNodeTransformer } from '../../operation-node/operation-node-transformer.js'
 import { OperationNode } from '../../operation-node/operation-node.js'
@@ -34,6 +36,11 @@ const ROOT_OPERATION_NODES: Record<RootOperationNode['kind'], true> = freeze({
   MergeQueryNode: true,
 })
 
+const SCHEMALESS_FUNCTIONS: Record<string, true> = {
+  json_agg: true,
+  to_json: true,
+}
+
 export class WithSchemaTransformer extends OperationNodeTransformer {
   readonly #schema: string
   readonly #schemableIds = new Set<string>()
@@ -105,6 +112,45 @@ export class WithSchemaTransformer extends OperationNodeTransformer {
     }
   }
 
+  protected override transformAggregateFunction(
+    node: AggregateFunctionNode
+  ): AggregateFunctionNode {
+    const transformed = super.transformAggregateFunction(node)
+
+    return SCHEMALESS_FUNCTIONS[node.func]
+      ? {
+          ...transformed,
+          aggregated: this.#revertTableTransformations(
+            node.aggregated,
+            transformed.aggregated
+          ),
+        }
+      : transformed
+  }
+
+  protected override transformFunction(node: FunctionNode): FunctionNode {
+    const transformed = super.transformFunction(node)
+
+    return SCHEMALESS_FUNCTIONS[node.func]
+      ? {
+          ...transformed,
+          arguments: this.#revertTableTransformations(
+            node.arguments,
+            transformed.arguments
+          ),
+        }
+      : transformed
+  }
+
+  #revertTableTransformations(
+    before: readonly OperationNode[],
+    after: readonly OperationNode[]
+  ): OperationNode[] {
+    return after.map((transformed, index) =>
+      TableNode.is(before[index]) ? before[index] : transformed
+    )
+  }
+
   #isRootOperationNode(node: OperationNode): node is RootOperationNode {
     return node.kind in ROOT_OPERATION_NODES
   }

test/node/src/with-schema.test.ts

@@ -218,6 +218,52 @@ for (const dialect of DIALECTS.filter(
 
         await query.execute()
       })
+
+      if (dialect === 'postgres') {
+        it('should not add schema for json_agg parameters', async () => {
+          const query = ctx.db
+            .withSchema('mammals')
+            .selectFrom('pet')
+            .select((eb) => [
+              eb.fn.jsonAgg('pet').as('one'),
+              eb.fn.jsonAgg(eb.table('pet')).as('two'),
+            ])
+
+          testSql(query, dialect, {
+            postgres: {
+              sql: 'select json_agg("pet") as "one", json_agg("pet") as "two" from "mammals"."pet"',
+              parameters: [],
+            },
+            mysql: NOT_SUPPORTED,
+            mssql: NOT_SUPPORTED,
+            sqlite: NOT_SUPPORTED,
+          })
+
+          await query.execute()
+        })
+
+        it('should not add schema for to_json parameters', async () => {
+          const query = ctx.db
+            .withSchema('mammals')
+            .selectFrom('pet')
+            .select((eb) => [
+              eb.fn.toJson('pet').as('one'),
+              eb.fn.toJson(eb.table('pet')).as('two'),
+            ])
+
+          testSql(query, dialect, {
+            postgres: {
+              sql: 'select to_json("pet") as "one", to_json("pet") as "two" from "mammals"."pet"',
+              parameters: [],
+            },
+            mysql: NOT_SUPPORTED,
+            mssql: NOT_SUPPORTED,
+            sqlite: NOT_SUPPORTED,
+          })
+
+          await query.execute()
+        })
+      }
     })
 
     describe('insert into', () => {