@pgsql/transform 18.4.1 → 18.5.0

package/esm/facts.js

@@ -1,5 +1,5 @@
 import { walk as walkSql } from '@pgsql/traverse';
-import { transformSync, walk as walkPlpgsql } from 'plpgsql-parser';
+import { parseSql, transformSync, walk as walkPlpgsql } from 'plpgsql-parser';
 const SECURITY_TAGS = new Set([
     'CreatePolicyStmt',
     'AlterPolicyStmt',
@@ -241,6 +241,43 @@ function classifyOne(nodeTag, node) {
     collectRoles(node, facts.roles);
     return facts;
 }
+/** Read a `CreateFunctionStmt` DefElem option's scalar/list value. */
+function functionOption(node, defname) {
+    for (const opt of node.options ?? []) {
+        if (opt?.DefElem?.defname === defname)
+            return opt.DefElem.arg;
+    }
+    return undefined;
+}
+/**
+ * Collect references from a `LANGUAGE sql` function body supplied as a string
+ * literal (`AS $$ ... $$`). That body is an opaque String node the AST walker
+ * never parses, so — mirroring the schema transformer's body rewrite — parse
+ * it standalone and walk each statement with the facts visitor. The standard
+ * `BEGIN ATOMIC` / `RETURN` `sql_body` form is already part of the AST and is
+ * covered by the outer walk, so only the string form needs this.
+ */
+function collectSqlBodyReferences(node, facts) {
+    const language = functionOption(node, 'language')?.String?.sval;
+    if (typeof language !== 'string' || language.toLowerCase() !== 'sql')
+        return;
+    const asArg = functionOption(node, 'as');
+    const items = asArg?.List?.items ?? [];
+    const body = items[0]?.String?.sval;
+    if (typeof body !== 'string')
+        return;
+    try {
+        const stmts = parseSql(body)?.stmts ?? [];
+        const visitor = createFactsVisitor(facts, facts.bodyReferences);
+        for (const stmt of stmts) {
+            if (stmt?.stmt)
+                walkSql(stmt.stmt, visitor);
+        }
+    }
+    catch {
+        // A non-parseable body (C symbol name, etc.) contributes no references.
+    }
+}
 /**
  * Classify each top-level statement in a SQL script into {@link StatementFacts}.
  *
@@ -260,6 +297,9 @@ export function classifyStatements(sql) {
             if (stmtNode) {
                 walkSql(stmtNode, createFactsVisitor(facts));
             }
+            if (nodeTag === 'CreateFunctionStmt') {
+                collectSqlBodyReferences(node, facts);
+            }
             allFacts.push(facts);
         }
         for (const fn of ctx.functions ?? []) {

package/facts.js

@@ -244,6 +244,43 @@ function classifyOne(nodeTag, node) {
     collectRoles(node, facts.roles);
     return facts;
 }
+/** Read a `CreateFunctionStmt` DefElem option's scalar/list value. */
+function functionOption(node, defname) {
+    for (const opt of node.options ?? []) {
+        if (opt?.DefElem?.defname === defname)
+            return opt.DefElem.arg;
+    }
+    return undefined;
+}
+/**
+ * Collect references from a `LANGUAGE sql` function body supplied as a string
+ * literal (`AS $$ ... $$`). That body is an opaque String node the AST walker
+ * never parses, so — mirroring the schema transformer's body rewrite — parse
+ * it standalone and walk each statement with the facts visitor. The standard
+ * `BEGIN ATOMIC` / `RETURN` `sql_body` form is already part of the AST and is
+ * covered by the outer walk, so only the string form needs this.
+ */
+function collectSqlBodyReferences(node, facts) {
+    const language = functionOption(node, 'language')?.String?.sval;
+    if (typeof language !== 'string' || language.toLowerCase() !== 'sql')
+        return;
+    const asArg = functionOption(node, 'as');
+    const items = asArg?.List?.items ?? [];
+    const body = items[0]?.String?.sval;
+    if (typeof body !== 'string')
+        return;
+    try {
+        const stmts = (0, plpgsql_parser_1.parseSql)(body)?.stmts ?? [];
+        const visitor = createFactsVisitor(facts, facts.bodyReferences);
+        for (const stmt of stmts) {
+            if (stmt?.stmt)
+                (0, traverse_1.walk)(stmt.stmt, visitor);
+        }
+    }
+    catch {
+        // A non-parseable body (C symbol name, etc.) contributes no references.
+    }
+}
 /**
  * Classify each top-level statement in a SQL script into {@link StatementFacts}.
  *
@@ -263,6 +300,9 @@ function classifyStatements(sql) {
             if (stmtNode) {
                 (0, traverse_1.walk)(stmtNode, createFactsVisitor(facts));
             }
+            if (nodeTag === 'CreateFunctionStmt') {
+                collectSqlBodyReferences(node, facts);
+            }
             allFacts.push(facts);
         }
         for (const fn of ctx.functions ?? []) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pgsql/transform",
-  "version": "18.4.1",
+  "version": "18.5.0",
   "author": "Constructive <developers@constructive.io>",
   "description": "AST-based SQL transformation, qualification, classification and closure analysis for PostgreSQL",
   "main": "index.js",
@@ -49,5 +49,5 @@
     "classify",
     "plpgsql"
   ],
-  "gitHead": "df667707ab725d762cac0daab26a07d1b202d1e1"
+  "gitHead": "a2b52f04fca42931150378d5b4966b7d5a922de6"
 }