@query-farm/vgi 0.13.0 → 0.15.0

package/dist/index.js

@@ -9964,6 +9964,27 @@ function typeToString(type) {
   const s = type.toString?.();
   return s ?? String(type.typeId);
 }
+function arrowTypeString(type) {
+  if (isNull(type))
+    return "null";
+  if (isBool(type))
+    return "bool";
+  if (isUtf8(type))
+    return "string";
+  if (isBinary(type))
+    return "binary";
+  if (isInt(type)) {
+    const bw = type.bitWidth ?? 32;
+    const signed = type.isSigned ?? type.signed ?? true;
+    return `${signed ? "" : "u"}int${bw}`;
+  }
+  if (isFloat(type)) {
+    const precision = type.precision ?? 2;
+    return precision === 0 ? "halffloat" : precision === 1 ? "float" : "double";
+  }
+  const s = type.toString?.();
+  return (s ?? String(type.typeId)).toLowerCase();
+}
 function fieldMeta(field3, key) {
   return field3.metadata.get(key);
 }
@@ -10030,6 +10051,61 @@ function functionArgs(fn) {
   }
   return args;
 }
+function macroDisplayType(m) {
+  return String(m.macro_type).toUpperCase() === "SCALAR" ? "scalar" : "table";
+}
+function macroDefaultJson(value) {
+  if (typeof value === "bigint")
+    return value.toString();
+  try {
+    return JSON.stringify(value);
+  } catch {
+    return String(value);
+  }
+}
+function macroArgs(m) {
+  const defaults = new Map;
+  const defBytes = m.parameter_default_values;
+  if (defBytes && defBytes.length > 0) {
+    try {
+      const batch = deserializeBatch(defBytes);
+      for (const field3 of batch.schema.fields) {
+        try {
+          defaults.set(field3.name, batch.getChild(field3.name)?.get(0));
+        } catch {}
+      }
+    } catch {}
+  }
+  const schema3 = readSchemaSafe(m.arguments_schema);
+  const fields = schema3 ? schema3.fields : null;
+  const fieldByName = new Map;
+  if (fields)
+    for (const f of fields)
+      fieldByName.set(f.name, f);
+  const names = fields ? fields.map((f) => f.name) : m.parameters;
+  const args = [];
+  for (const name of names) {
+    const field3 = fieldByName.get(name);
+    const arg = field3 !== undefined && !isNull(field3.type) ? { name, type: arrowTypeString(field3.type) } : { name, type: "ANY" };
+    const doc = field3 !== undefined ? fieldMeta(field3, VGI_DOC_KEY) : undefined;
+    if (doc)
+      arg.desc = doc;
+    if (defaults.has(name)) {
+      arg.named = true;
+      arg.default = macroDefaultJson(defaults.get(name));
+    }
+    args.push(arg);
+  }
+  return args;
+}
+function macroToObject(m) {
+  return {
+    name: m.name,
+    type: macroDisplayType(m),
+    doc: m.comment ?? "",
+    args: macroArgs(m)
+  };
+}
 async function schemaContents(iface, aod, name, kind) {
   try {
     if (kind === "TABLE") {
@@ -10049,6 +10125,15 @@ async function schemaFunctions(iface, aod, name) {
   }
   return out;
 }
+async function schemaMacros(iface, aod, name) {
+  const out = [];
+  for (const kind of ["SCALAR_MACRO", "TABLE_MACRO"]) {
+    try {
+      out.push(...await iface.schemaContentsMacros(aod, name, kind));
+    } catch {}
+  }
+  return out;
+}
 async function buildSchemas(iface, aod) {
   let schemaInfos;
   try {
@@ -10062,6 +10147,7 @@ async function buildSchemas(iface, aod) {
     const tablesRaw = await schemaContents(iface, aod, si.name, "TABLE");
     const viewsRaw = await schemaContents(iface, aod, si.name, "VIEW");
     const funcsRaw = await schemaFunctions(iface, aod, si.name);
+    const macrosRaw = await schemaMacros(iface, aod, si.name);
     const tables = tablesRaw.map((t) => {
       const schema3 = readSchemaSafe(t.columns);
       return { name: t.name, cols: schema3 ? schema3.fields.length : 0, comment: t.comment ?? "" };
@@ -10072,7 +10158,7 @@ async function buildSchemas(iface, aod) {
       comment: v.comment ?? "",
       def: v.definition
     }));
-    const functions = funcsRaw.slice().sort((a, b) => String(a.function_type).localeCompare(String(b.function_type)) || a.name.localeCompare(b.name)).map((fn) => {
+    const functions = funcsRaw.map((fn) => {
       const obj = {
         name: fn.name,
         type: functionDisplayType(fn),
@@ -10084,7 +10170,15 @@ async function buildSchemas(iface, aod) {
         obj.returns = ret;
       return obj;
     });
-    schemas.push({ name: si.name, tables, views, functions });
+    functions.push(...macrosRaw.map((m) => macroToObject(m)));
+    functions.sort((a, b) => String(a.type).localeCompare(String(b.type)) || String(a.name).localeCompare(String(b.name)));
+    const schemaObj = { name: si.name };
+    if (si.comment)
+      schemaObj.doc = si.comment;
+    schemaObj.tables = tables;
+    schemaObj.views = views;
+    schemaObj.functions = functions;
+    schemas.push(schemaObj);
     counts.schemas += 1;
     counts.tables += tables.length;
     counts.views += views.length;
@@ -11388,4 +11482,4 @@ export {
   ArgumentValidationError
 };
 
-//# debugId=FB551171F4FDA15864756E2164756E21
+//# debugId=B7CCB77D8D0375B864756E2164756E21