@xano/xanoscript-language-server 11.8.3 → 11.8.5

package/parser/functions/controls/functionRunFn.js

@@ -1,6 +1,21 @@
 import { FunctionToken, RunToken } from "../../../lexer/function.js";
-import { StringLiteral } from "../../../lexer/literal.js";
-import { DotToken, Identifier } from "../../../lexer/tokens.js";
+import {
+  FloatLiteral,
+  IntegerLiteral,
+  SingleQuotedStringLiteral,
+  StringLiteral,
+} from "../../../lexer/literal.js";
+import { DotToken, FalseToken, Identifier, TrueToken } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
+
+const literalTypeMap = new Map([
+  [IntegerLiteral.name, "int"],
+  [FloatLiteral.name, "decimal"],
+  [StringLiteral.name, "text"],
+  [SingleQuotedStringLiteral.name, "text"],
+  [TrueToken.name, "bool"],
+  [FalseToken.name, "bool"],
+]);
 
 /**
  *
@@ -14,11 +29,13 @@ export function functionRunFn($) {
     $.CONSUME(DotToken); // "."
     const fnToken = $.CONSUME(RunToken); // "run"
 
-    $.OR([
-      { ALT: () => $.CONSUME(Identifier) }, // user
-      { ALT: () => $.CONSUME1(StringLiteral) }, // "user/ auth"
+    const nameToken = $.OR([
+      { ALT: () => $.CONSUME(Identifier) },
+      { ALT: () => $.CONSUME1(StringLiteral) },
     ]);
+    $.addReference("function", getVarName(nameToken), nameToken);
 
+    const captured = {};
     $.OPTION(() => {
       $.SUBRULE($.schemaParseAttributeFn, {
         ARGS: [
@@ -30,10 +47,49 @@ export function functionRunFn($) {
             "runtime_mode?": ["async-shared", "async-dedicated"],
             "mock?": { "![string]": "[expression]" },
           },
+          captured,
         ],
       });
     });
 
+    $.ACTION(() => {
+      if (captured.input) {
+        const args = {};
+        const tokens = $.input;
+        for (const [k, v] of Object.entries(captured.input)) {
+          if (k === "key" || k === "value") continue;
+          const keyOffset = v?.key?.startOffset;
+
+          // Find value token: scan from key token, skip colon, grab next token
+          let type = null;
+          let value = v?.value?.image ?? null;
+          if (keyOffset != null) {
+            for (let i = 0; i < tokens.length - 2; i++) {
+              if (tokens[i].startOffset === keyOffset) {
+                // tokens[i] = key, tokens[i+1] = colon, tokens[i+2] = value
+                const valueToken = tokens[i + 2];
+                if (valueToken) {
+                  type = literalTypeMap.get(valueToken.tokenType.name) ?? null;
+                  if (type && !value) value = valueToken.image;
+                }
+                break;
+              }
+            }
+          }
+
+          args[k] = {
+            value,
+            type,
+            startOffset: v?.key?.startOffset ?? null,
+            endOffset: v?.key?.endOffset ?? null,
+          };
+        }
+        if (Object.keys(args).length > 0) {
+          $.setReferenceArgs(args);
+        }
+      }
+    });
+
     $.SUBRULE($.asVariable, { ARGS: [fnToken] });
     $.sectionStack.pop();
   };

package/parser/functions/controls/taskCallFn.js

@@ -1,6 +1,7 @@
 import { StringLiteral } from "../../../lexer/literal.js";
 import { CallToken, TaskToken } from "../../../lexer/task.js";
 import { DotToken, Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
 
 /**
  *
@@ -17,10 +18,11 @@ export function taskCallFn($) {
     $.CONSUME(DotToken); // "."
     const fnToken = $.CONSUME(CallToken); // "call"
 
-    $.OR([
-      { ALT: () => $.CONSUME(Identifier) }, // user
-      { ALT: () => $.CONSUME1(StringLiteral) }, // "user/ auth"
+    const nameToken = $.OR([
+      { ALT: () => $.CONSUME(Identifier) },
+      { ALT: () => $.CONSUME1(StringLiteral) },
     ]);
+    $.addReference("task", getVarName(nameToken), nameToken);
 
     $.SUBRULE($.functionAttrReq, {
       ARGS: [fnToken, requiredAttrs, optionalAttrs],

package/parser/functions/db/captureFieldName.js

@@ -0,0 +1,63 @@
+import { SingleQuotedStringLiteral,StringLiteral } from "../../../lexer/literal.js";
+import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
+
+const identifierLikeTokens = new Set([
+  Identifier.name,
+  StringLiteral.name,
+  SingleQuotedStringLiteral.name,
+]);
+
+/**
+ * Extract data object keys from captured schema and attach to last reference.
+ * @param {import('../../base_parser.js').XanoBaseParser} $
+ * @param {Object} captured
+ */
+export function captureDataKeys($, captured) {
+  if (!captured.data) return;
+
+  const dataKeys = [];
+  for (const [k, v] of Object.entries(captured.data)) {
+    if (k === "key" || k === "value") continue;
+    const keyToken = v?.key;
+    if (keyToken) {
+      dataKeys.push({
+        name: k,
+        startOffset: keyToken.startOffset,
+        endOffset: keyToken.endOffset,
+      });
+    }
+  }
+  if (dataKeys.length > 0) {
+    $.setReferenceDataKeys(dataKeys);
+  }
+}
+
+/**
+ * Extract field_name value from captured schema object and attach to last reference.
+ * Only captures when the value is an identifier or string literal (not a variable expression).
+ * @param {import('../../base_parser.js').XanoBaseParser} $
+ * @param {Object} captured
+ */
+export function captureFieldName($, captured) {
+  if (!captured.field_name) return;
+
+  const keyOffset = captured.field_name.key?.startOffset;
+  if (keyOffset == null) return;
+
+  const tokens = $.input;
+  for (let i = 0; i < tokens.length - 2; i++) {
+    if (tokens[i].startOffset === keyOffset) {
+      // tokens[i] = "field_name" key, tokens[i+1] = "=", tokens[i+2] = value
+      const valueToken = tokens[i + 2];
+      if (valueToken && identifierLikeTokens.has(valueToken.tokenType.name)) {
+        $.setReferenceFieldName(
+          getVarName(valueToken),
+          valueToken.startOffset,
+          valueToken.endOffset
+        );
+      }
+      break;
+    }
+  }
+}

package/parser/functions/db/dbAddFn.js

@@ -1,6 +1,7 @@
 import { AddToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
 import { addonSchema } from "./schema.js";
 /**
  * @param {import('../base_parser.js').XanoBaseParser} $
@@ -10,13 +11,14 @@ export function dbAddFn($) {
     $.sectionStack.push("dbAddFn");
     const fnToken = $.CONSUME(AddToken); // "add"
 
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
 
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [

package/parser/functions/db/dbAddOrEditFn.js

@@ -1,6 +1,8 @@
 import { AddOrEditToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
+import { captureDataKeys,captureFieldName } from "./captureFieldName.js";
 import { addonSchema } from "./schema.js";
 
 /**
@@ -10,14 +12,16 @@ export function dbAddOrEditFn($) {
   return () => {
     $.sectionStack.push("dbAddOrEditFn");
     const fnToken = $.CONSUME(AddOrEditToken); // "edit"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
 
+    const captured = {};
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [
         fnToken,
@@ -30,9 +34,15 @@ export function dbAddOrEditFn($) {
           "addon?": [addonSchema],
           "mock?": { "![string]": "[expression]" },
         },
+        captured,
       ],
     });
 
+    $.ACTION(() => {
+      captureFieldName($, captured);
+      captureDataKeys($, captured);
+    });
+
     $.SUBRULE($.asVariable, { ARGS: [fnToken] });
     $.sectionStack.pop();
   };

package/parser/functions/db/dbBulkAddFn.js

@@ -1,6 +1,7 @@
 import { AddToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
 
 /**
  * @param {import('../base_parser.js').XanoBaseParser} $
@@ -9,13 +10,14 @@ export function dbBulkAddFn($) {
   return () => {
     $.sectionStack.push("dbBulkAddFn");
     const fnToken = $.CONSUME(AddToken); // "add"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
     const defined = {};
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [

package/parser/functions/db/dbBulkDeleteFn.js

@@ -1,6 +1,7 @@
 import { DeleteToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
 
 /**
  * @param {import('../base_parser.js').XanoBaseParser} $
@@ -9,13 +10,14 @@ export function dbBulkDeleteFn($) {
   return () => {
     $.sectionStack.push("dbBulkDeleteFn");
     const fnToken = $.CONSUME(DeleteToken); // "delete"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
 
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [

package/parser/functions/db/dbBulkPatchFn.js

@@ -1,6 +1,7 @@
 import { PatchToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
 
 /**
  * @param {import('../base_parser.js').XanoBaseParser} $
@@ -9,13 +10,14 @@ export function dbBulkPatchFn($) {
   return () => {
     $.sectionStack.push("dbBulkPatchFn");
     const fnToken = $.CONSUME(PatchToken); // "patch"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
 
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [

package/parser/functions/db/dbBulkUpdateFn.js

@@ -1,6 +1,7 @@
 import { UpdateToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
 
 /**
  * @param {import('../base_parser.js').XanoBaseParser} $
@@ -9,13 +10,14 @@ export function dbBulkUpdateFn($) {
   return () => {
     $.sectionStack.push("dbBulkUpdateFn");
     const fnToken = $.CONSUME(UpdateToken); // "update"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
 
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [

package/parser/functions/db/dbDelFn.js

@@ -1,6 +1,8 @@
 import { DelToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
+import { captureFieldName } from "./captureFieldName.js";
 
 /**
  * @param {import('../base_parser.js').XanoBaseParser} $
@@ -9,14 +11,16 @@ export function dbDelFn($) {
   return () => {
     $.sectionStack.push("dbDelFn");
     const fnToken = $.CONSUME(DelToken); // "delete"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
 
+    const captured = {};
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [
         fnToken,
@@ -26,9 +30,12 @@ export function dbDelFn($) {
           "description?": "[string]",
           "disabled?": "[boolean]",
         },
+        captured,
       ],
     });
 
+    $.ACTION(() => captureFieldName($, captured));
+
     $.sectionStack.pop();
   };
 }

package/parser/functions/db/dbEditFn.js

@@ -1,6 +1,8 @@
 import { EditToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
+import { captureDataKeys,captureFieldName } from "./captureFieldName.js";
 import { addonSchema } from "./schema.js";
 
 /**
@@ -10,14 +12,16 @@ export function dbEditFn($) {
   return () => {
     $.sectionStack.push("dbEditFn");
     const fnToken = $.CONSUME(EditToken); // "edit"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
 
+    const captured = {};
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [
         fnToken,
@@ -31,9 +35,15 @@ export function dbEditFn($) {
           "addon?": [addonSchema],
           "mock?": { "![string]": "[expression]" },
         },
+        captured,
       ],
     });
 
+    $.ACTION(() => {
+      captureFieldName($, captured);
+      captureDataKeys($, captured);
+    });
+
     $.SUBRULE($.asVariable, { ARGS: [fnToken] });
     $.sectionStack.pop();
   };

package/parser/functions/db/dbGetFn.js

@@ -1,6 +1,8 @@
 import { GetToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
+import { captureFieldName } from "./captureFieldName.js";
 import { addonSchema } from "./schema.js";
 
 /**
@@ -10,14 +12,16 @@ export function dbGetFn($) {
   return () => {
     $.sectionStack.push("dbGetFn");
     const fnToken = $.CONSUME(GetToken); // "get"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
 
+    const captured = {};
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [
         fnToken,
@@ -31,9 +35,12 @@ export function dbGetFn($) {
           "addon?": [addonSchema],
           "mock?": { "![string]": "[expression]" },
         },
+        captured,
       ],
     });
 
+    $.ACTION(() => captureFieldName($, captured));
+
     $.SUBRULE($.asVariable, { ARGS: [fnToken] });
     $.sectionStack.pop();
   };

package/parser/functions/db/dbHasFn.js

@@ -1,6 +1,8 @@
 import { HasToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
+import { captureFieldName } from "./captureFieldName.js";
 
 /**
  * @param {import('../base_parser.js').XanoBaseParser} $
@@ -9,13 +11,15 @@ export function dbHasFn($) {
   return () => {
     $.sectionStack.push("dbHasFn");
     const fnToken = $.CONSUME(HasToken); // "has"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
+    const captured = {};
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [
         fnToken,
@@ -26,8 +30,10 @@ export function dbHasFn($) {
           "disabled?": "[boolean]",
           "mock?": { "![string]": "[expression]" },
         },
+        captured,
       ],
     });
+    $.ACTION(() => captureFieldName($, captured));
     $.SUBRULE($.asVariable, { ARGS: [fnToken] });
     $.sectionStack.pop();
   };

package/parser/functions/db/dbPatchFn.js

@@ -1,6 +1,8 @@
 import { PatchToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
+import { captureFieldName } from "./captureFieldName.js";
 import { addonSchema } from "./schema.js";
 
 /**
@@ -10,14 +12,16 @@ export function dbPatchFn($) {
   return () => {
     $.sectionStack.push("dbPatchFn");
     const fnToken = $.CONSUME(PatchToken); // "patch"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
 
+    const captured = {};
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [
         fnToken,
@@ -30,9 +34,12 @@ export function dbPatchFn($) {
           "addon?": [addonSchema],
           "mock?": { "![string]": "[expression]" },
         },
+        captured,
       ],
     });
 
+    $.ACTION(() => captureFieldName($, captured));
+
     $.SUBRULE($.asVariable, { ARGS: [fnToken] });
     $.sectionStack.pop();
   };

package/parser/functions/db/dbQueryFn.js

@@ -1,6 +1,7 @@
 import { QueryToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
 import { addonSchema } from "./schema.js";
 
 /**
@@ -10,11 +11,13 @@ export function dbQueryFn($) {
   return () => {
     $.sectionStack.push("dbQueryFn");
     const fnToken = $.CONSUME(QueryToken); // "query"
-    $.OR([
-      { ALT: () => $.CONSUME(Identifier) }, // user
-      { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+    const nameToken = $.OR([
+      { ALT: () => $.CONSUME(Identifier) },
+      { ALT: () => $.CONSUME1(StringLiteral) },
     ]);
+    $.addReference("table", getVarName(nameToken), nameToken);
 
+    const captured = {};
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [
         fnToken,
@@ -51,9 +54,32 @@ export function dbQueryFn($) {
           "where?": "[query]",
           "additional_where?": "[query]",
         },
+        captured,
       ],
     });
 
+    $.ACTION(() => {
+      // Extract join table references
+      if (captured.join) {
+        for (const [k, v] of Object.entries(captured.join)) {
+          if (k === "key" || k === "value") continue;
+          const tableEntry = v?.table;
+          if (tableEntry?.value) {
+            $.addReference("table", getVarName(tableEntry.value), tableEntry.value);
+          }
+        }
+      }
+      // Extract addon name references
+      if (captured.addon?.value && Array.isArray(captured.addon.value)) {
+        for (const item of captured.addon.value) {
+          const nameEntry = item.name;
+          if (nameEntry?.value) {
+            $.addReference("function", getVarName(nameEntry.value), nameEntry.value);
+          }
+        }
+      }
+    });
+
     // db.query rule is used within an addon declaration, but
     // in this context we don't allow "as" variable assignment
     const isAddonContext = $.sectionStack.includes("addonDeclaration");

package/parser/functions/db/dbSchemaFn.js

@@ -1,6 +1,7 @@
 import { SchemaToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
 
 /**
  * @param {import('../base_parser.js').XanoBaseParser} $
@@ -12,10 +13,11 @@ export function dbSchemaFn($) {
 
     $.sectionStack.push("dbSchemaFn");
     const fnToken = $.CONSUME(SchemaToken); // "schema"
-    $.OR([
-      { ALT: () => $.CONSUME(Identifier) }, // user
-      { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+    const nameToken = $.OR([
+      { ALT: () => $.CONSUME(Identifier) },
+      { ALT: () => $.CONSUME1(StringLiteral) },
     ]);
+    $.addReference("table", getVarName(nameToken), nameToken);
     $.SUBRULE($.functionAttrReq, {
       ARGS: [fnToken, requiredAttrs, optionalAttrs],
     });

package/parser/functions/db/dbTruncateFn.js

@@ -1,6 +1,7 @@
 import { TruncateToken } from "../../../lexer/db.js";
 import { StringLiteral } from "../../../lexer/literal.js";
 import { Identifier } from "../../../lexer/tokens.js";
+import { getVarName } from "../../generic/utils.js";
 
 /**
  * @param {import('../base_parser.js').XanoBaseParser} $
@@ -9,13 +10,14 @@ export function dbTruncateFn($) {
   return () => {
     $.sectionStack.push("dbTruncateFn");
     const fnToken = $.CONSUME(TruncateToken); // "truncate"
-    $.OR({
+    const nameToken = $.OR({
       DEF: [
-        { ALT: () => $.CONSUME(Identifier) }, // user
-        { ALT: () => $.CONSUME1(StringLiteral) }, // "user table"
+        { ALT: () => $.CONSUME(Identifier) },
+        { ALT: () => $.CONSUME1(StringLiteral) },
       ],
       ERR_MSG: "Expected a table name",
     });
+    $.addReference("table", getVarName(nameToken), nameToken);
 
     $.SUBRULE($.schemaParseAttributeFn, {
       ARGS: [