@polyglot-sql/sdk 0.5.2 → 0.5.4

package/dist/index.cjs

@@ -1699,6 +1699,24 @@ function ast_get_table_names$1(ast_json) {
     wasm$3.__wbindgen_export4(deferred2_0, deferred2_1, 1);
   }
 }
+function ast_get_tables$1(ast_json) {
+  let deferred2_0;
+  let deferred2_1;
+  try {
+    const retptr = wasm$3.__wbindgen_add_to_stack_pointer(-16);
+    const ptr0 = passStringToWasm0(ast_json, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+    const len0 = WASM_VECTOR_LEN;
+    wasm$3.ast_get_tables(retptr, ptr0, len0);
+    var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
+    var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
+    deferred2_0 = r0;
+    deferred2_1 = r1;
+    return getStringFromWasm0(r0, r1);
+  } finally {
+    wasm$3.__wbindgen_add_to_stack_pointer(16);
+    wasm$3.__wbindgen_export4(deferred2_0, deferred2_1, 1);
+  }
+}
 function ast_get_window_functions$1(ast_json) {
   let deferred2_0;
   let deferred2_1;
@@ -1891,6 +1909,44 @@ function ast_set_limit$1(ast_json, limit) {
     wasm$3.__wbindgen_export4(deferred2_0, deferred2_1, 1);
   }
 }
+function ast_set_offset$1(ast_json, offset) {
+  let deferred2_0;
+  let deferred2_1;
+  try {
+    const retptr = wasm$3.__wbindgen_add_to_stack_pointer(-16);
+    const ptr0 = passStringToWasm0(ast_json, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+    const len0 = WASM_VECTOR_LEN;
+    wasm$3.ast_set_offset(retptr, ptr0, len0, offset);
+    var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
+    var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
+    deferred2_0 = r0;
+    deferred2_1 = r1;
+    return getStringFromWasm0(r0, r1);
+  } finally {
+    wasm$3.__wbindgen_add_to_stack_pointer(16);
+    wasm$3.__wbindgen_export4(deferred2_0, deferred2_1, 1);
+  }
+}
+function ast_set_order_by$1(ast_json, order_by_json) {
+  let deferred3_0;
+  let deferred3_1;
+  try {
+    const retptr = wasm$3.__wbindgen_add_to_stack_pointer(-16);
+    const ptr0 = passStringToWasm0(ast_json, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+    const len0 = WASM_VECTOR_LEN;
+    const ptr1 = passStringToWasm0(order_by_json, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+    const len1 = WASM_VECTOR_LEN;
+    wasm$3.ast_set_order_by(retptr, ptr0, len0, ptr1, len1);
+    var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
+    var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
+    deferred3_0 = r0;
+    deferred3_1 = r1;
+    return getStringFromWasm0(r0, r1);
+  } finally {
+    wasm$3.__wbindgen_add_to_stack_pointer(16);
+    wasm$3.__wbindgen_export4(deferred3_0, deferred3_1, 1);
+  }
+}
 function diff_sql$1(source_sql, target_sql, dialect, delta_only, f, t) {
   let deferred4_0;
   let deferred4_1;
@@ -2623,10 +2679,6 @@ function __wbg_call_389efe28435a9388() {
     return addHeapObject(ret);
   }, arguments);
 }
-function __wbg_codePointAt_bf59dbf74d8db275(arg0, arg1) {
-  const ret = getObject(arg0).codePointAt(arg1 >>> 0);
-  return addHeapObject(ret);
-}
 function __wbg_done_57b39ecd9addfe81(arg0) {
   const ret = getObject(arg0).done;
   return ret;
@@ -2716,10 +2768,6 @@ function __wbg_length_35a7bace40f36eac(arg0) {
   const ret = getObject(arg0).length;
   return ret;
 }
-function __wbg_length_68dc7c5cf1b6d349(arg0) {
-  const ret = getObject(arg0).length;
-  return ret;
-}
 function __wbg_new_361308b2356cecd0() {
   const ret = new Object();
   return addHeapObject(ret);
@@ -3048,9 +3096,7 @@ const __vite__wasmImports = { "./polyglot_sql_wasm_bg.js": {
   "__wbg_call_389efe28435a9388": __wbg_call_389efe28435a9388,
   "__wbg_next_418f80d8f5303233": __wbg_next_418f80d8f5303233,
   "__wbg_next_3482f54c49e8af19": __wbg_next_3482f54c49e8af19,
-  "__wbg_codePointAt_bf59dbf74d8db275": __wbg_codePointAt_bf59dbf74d8db275,
   "__wbg_fromCodePoint_22365db7b7d6ac39": __wbg_fromCodePoint_22365db7b7d6ac39,
-  "__wbg_length_68dc7c5cf1b6d349": __wbg_length_68dc7c5cf1b6d349,
   "__wbg___wbindgen_in_47fa6863be6f2f25": __wbg___wbindgen_in_47fa6863be6f2f25,
   "__wbg___wbindgen_throw_be289d5034ed271b": __wbg___wbindgen_throw_be289d5034ed271b,
   "__wbg___wbindgen_is_null_ac34f5003991759a": __wbg___wbindgen_is_null_ac34f5003991759a,
@@ -3105,6 +3151,7 @@ async function __polyglot_init_wasm() {
   ast_get_literals = __vite__wasmModule.ast_get_literals;
   ast_get_subqueries = __vite__wasmModule.ast_get_subqueries;
   ast_get_table_names = __vite__wasmModule.ast_get_table_names;
+  ast_get_tables = __vite__wasmModule.ast_get_tables;
   ast_get_window_functions = __vite__wasmModule.ast_get_window_functions;
   ast_node_count = __vite__wasmModule.ast_node_count;
   ast_qualify_columns = __vite__wasmModule.ast_qualify_columns;
@@ -3115,6 +3162,8 @@ async function __polyglot_init_wasm() {
   ast_rename_tables_with_options = __vite__wasmModule.ast_rename_tables_with_options;
   ast_set_distinct = __vite__wasmModule.ast_set_distinct;
   ast_set_limit = __vite__wasmModule.ast_set_limit;
+  ast_set_offset = __vite__wasmModule.ast_set_offset;
+  ast_set_order_by = __vite__wasmModule.ast_set_order_by;
   diff_sql = __vite__wasmModule.diff_sql;
   format_sql = __vite__wasmModule.format_sql;
   format_sql_value = __vite__wasmModule.format_sql_value;
@@ -3304,6 +3353,7 @@ let ast_get_functions;
 let ast_get_literals;
 let ast_get_subqueries;
 let ast_get_table_names;
+let ast_get_tables;
 let ast_get_window_functions;
 let ast_node_count;
 let ast_qualify_columns;
@@ -3314,6 +3364,8 @@ let ast_rename_tables;
 let ast_rename_tables_with_options;
 let ast_set_distinct;
 let ast_set_limit;
+let ast_set_offset;
+let ast_set_order_by;
 let diff_sql;
 let format_sql;
 let format_sql_value;
@@ -3505,6 +3557,7 @@ const wasm$2 = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProper
   ast_get_literals,
   ast_get_subqueries,
   ast_get_table_names,
+  ast_get_tables,
   ast_get_window_functions,
   ast_node_count,
   ast_qualify_columns,
@@ -3515,6 +3568,8 @@ const wasm$2 = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProper
   ast_rename_tables_with_options,
   ast_set_distinct,
   ast_set_limit,
+  ast_set_offset,
+  ast_set_order_by,
   diff_sql,
   format_sql,
   format_sql_value,
@@ -3698,6 +3753,7 @@ const wasmModule = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.definePr
   ast_get_literals: ast_get_literals$1,
   ast_get_subqueries: ast_get_subqueries$1,
   ast_get_table_names: ast_get_table_names$1,
+  ast_get_tables: ast_get_tables$1,
   ast_get_window_functions: ast_get_window_functions$1,
   ast_node_count: ast_node_count$1,
   ast_qualify_columns: ast_qualify_columns$1,
@@ -3708,6 +3764,8 @@ const wasmModule = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.definePr
   ast_rename_tables_with_options: ast_rename_tables_with_options$1,
   ast_set_distinct: ast_set_distinct$1,
   ast_set_limit: ast_set_limit$1,
+  ast_set_offset: ast_set_offset$1,
+  ast_set_order_by: ast_set_order_by$1,
   diff_sql: diff_sql$1,
   format_sql: format_sql$1,
   format_sql_value: format_sql_value$1,
@@ -3891,6 +3949,69 @@ function parseAstResult(json) {
   const result = JSON.parse(json);
   return result.success ? JSON.parse(result.ast) : null;
 }
+function isQueryWithOuterClauses(node) {
+  return ["select", "union", "intersect", "except"].includes(getExprType(node));
+}
+function applyLimitExpr(node, limit) {
+  const type = getExprType(node);
+  if (!isQueryWithOuterClauses(node)) {
+    return node;
+  }
+  const data = getExprData(node);
+  if (type === "select") {
+    return makeExpr("select", {
+      ...data,
+      limit: { this: limit, percent: false, comments: [] }
+    });
+  }
+  return makeExpr(type, {
+    ...data,
+    limit
+  });
+}
+function applyOffsetExpr(node, offset) {
+  const type = getExprType(node);
+  if (!isQueryWithOuterClauses(node)) {
+    return node;
+  }
+  const data = getExprData(node);
+  if (type === "select") {
+    return makeExpr("select", {
+      ...data,
+      offset: { this: offset, rows: null }
+    });
+  }
+  return makeExpr(type, {
+    ...data,
+    offset
+  });
+}
+function applyOrderBy(node, orderBy) {
+  const type = getExprType(node);
+  if (!isQueryWithOuterClauses(node)) {
+    return node;
+  }
+  const expressions = orderBy.map((expression) => {
+    if (getExprType(expression) === "ordered") {
+      return getExprData(expression);
+    }
+    return {
+      this: expression,
+      desc: false,
+      nulls_first: null,
+      explicit_asc: false,
+      with_fill: null
+    };
+  });
+  return makeExpr(type, {
+    ...getExprData(node),
+    order_by: {
+      expressions,
+      siblings: false,
+      comments: []
+    }
+  });
+}
 function transformValue(value, config, parent) {
   if (value === null || value === void 0) return value;
   if (Array.isArray(value)) {
@@ -4060,25 +4181,21 @@ function setLimit(node, limit) {
     const result = parseAstResult(ast_set_limit$1(exprToJson$1(node), limit));
     return result ?? node;
   }
-  if (getExprType(node) !== "select") {
-    return node;
-  }
-  const selectData = getExprData(node);
-  return makeExpr("select", {
-    ...selectData,
-    limit: { this: limit }
-  });
+  return applyLimitExpr(node, limit);
 }
 function setOffset(node, offset) {
-  if (getExprType(node) !== "select") {
-    return node;
+  if (typeof offset === "number") {
+    const result = parseAstResult(ast_set_offset$1(exprToJson$1(node), offset));
+    return result ?? node;
   }
-  const selectData = getExprData(node);
-  const offsetExpr = typeof offset === "number" ? makeExpr("literal", { literal_type: "number", value: String(offset) }) : offset;
-  return makeExpr("select", {
-    ...selectData,
-    offset: { this: offsetExpr }
-  });
+  return applyOffsetExpr(node, offset);
+}
+function setOrderBy(node, orderBy) {
+  const orderByExpressions = Array.isArray(orderBy) ? orderBy : [orderBy];
+  const result = parseAstResult(
+    ast_set_order_by$1(exprToJson$1(node), JSON.stringify(orderByExpressions))
+  );
+  return result ?? applyOrderBy(node, orderByExpressions);
 }
 function removeLimitOffset(node) {
   if (getExprType(node) !== "select") {
@@ -4245,7 +4362,8 @@ function getColumns(node) {
   return findByType(node, "column");
 }
 function getTables(node) {
-  return findByType(node, "table");
+  const result = JSON.parse(ast_get_tables$1(exprToJson(node)));
+  return result.success ? JSON.parse(result.ast) : [];
 }
 function getIdentifiers(node) {
   return findByType(node, "identifier");
@@ -4478,6 +4596,7 @@ const index$1 = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.definePrope
   setDistinct,
   setLimit,
   setOffset,
+  setOrderBy,
   some,
   transform,
   walk

package/dist/index.d.cts

@@ -1065,6 +1065,7 @@ export declare namespace ast {
         setDistinct,
         setLimit,
         setOffset,
+        setOrderBy,
         transform,
         NodePath,
         NodePredicate,
@@ -2910,6 +2911,13 @@ declare type Cte = {
     comments?: Array<string>;
 };
 
+export declare interface CteFact {
+    name: string;
+    columns: string[];
+    bodySql: string;
+    outputColumns: string[];
+}
+
 /**
  * Cube
  */
@@ -10870,11 +10878,15 @@ export declare interface ProjectionFact {
     isStar: boolean;
     starTable?: string;
     transformKind: TransformKind;
+    transformFunction?: TransformFunctionFact;
     castType?: string;
     typeHint?: string;
+    nullability: ProjectionNullability;
     upstream: ColumnReferenceFact[];
 }
 
+export declare type ProjectionNullability = 'non_null' | 'nullable' | 'unknown';
+
 /**
  * Properties
  */
@@ -11006,9 +11018,11 @@ declare type Quantile = {
 export declare interface QueryAnalysis {
     shape: QueryShape;
     ctes: string[];
+    cteFacts: CteFact[];
     projections: ProjectionFact[];
     relations: RelationFact[];
     baseTables: RelationFact[];
+    starProjections: StarProjectionFact[];
     setOperations: SetOperationFact[];
 }
 
@@ -11410,9 +11424,12 @@ declare type RegrValy = {
 
 export declare interface RelationFact {
     name: string;
-    alias?: string;
+    alias?: string | null;
     kind: QueryAnalysisSourceKind;
     columns: string[];
+    catalog?: string | null;
+    schema?: string | null;
+    table?: string | null;
 }
 
 /**
@@ -12305,6 +12322,14 @@ export declare interface SetOperationFact {
 
 export declare type SetOperationType = 'union' | 'union_all' | 'intersect' | 'except';
 
+/**
+ * Set or update the ORDER BY clause.
+ *
+ * Bare expressions are normalized to ascending order entries. Existing
+ * `ordered` expression nodes preserve their direction and null ordering.
+ */
+declare function setOrderBy(node: Expression, orderBy: Expression | Expression[]): Expression;
+
 /**
  * SetProperty
  */
@@ -12654,6 +12679,12 @@ declare type Star = {
 /** Create a star (*) expression. */
 export declare function star(): Expr;
 
+export declare interface StarProjectionFact {
+    index: number;
+    table?: string;
+    expandedColumns: string[];
+}
+
 /**
  * StDistance
  */
@@ -13718,6 +13749,12 @@ declare type TransformFunc = {
     transform: Expression;
 };
 
+export declare interface TransformFunctionFact {
+    name: string;
+    literalArgs: string[];
+    columnArgs: ColumnReferenceFact[];
+}
+
 export declare type TransformKind = 'direct' | 'cast' | 'aggregation' | 'constant' | 'expression' | 'star';
 
 /**

package/dist/index.d.ts

@@ -1065,6 +1065,7 @@ export declare namespace ast {
         setDistinct,
         setLimit,
         setOffset,
+        setOrderBy,
         transform,
         NodePath,
         NodePredicate,
@@ -2910,6 +2911,13 @@ declare type Cte = {
     comments?: Array<string>;
 };
 
+export declare interface CteFact {
+    name: string;
+    columns: string[];
+    bodySql: string;
+    outputColumns: string[];
+}
+
 /**
  * Cube
  */
@@ -10870,11 +10878,15 @@ export declare interface ProjectionFact {
     isStar: boolean;
     starTable?: string;
     transformKind: TransformKind;
+    transformFunction?: TransformFunctionFact;
     castType?: string;
     typeHint?: string;
+    nullability: ProjectionNullability;
     upstream: ColumnReferenceFact[];
 }
 
+export declare type ProjectionNullability = 'non_null' | 'nullable' | 'unknown';
+
 /**
  * Properties
  */
@@ -11006,9 +11018,11 @@ declare type Quantile = {
 export declare interface QueryAnalysis {
     shape: QueryShape;
     ctes: string[];
+    cteFacts: CteFact[];
     projections: ProjectionFact[];
     relations: RelationFact[];
     baseTables: RelationFact[];
+    starProjections: StarProjectionFact[];
     setOperations: SetOperationFact[];
 }
 
@@ -11410,9 +11424,12 @@ declare type RegrValy = {
 
 export declare interface RelationFact {
     name: string;
-    alias?: string;
+    alias?: string | null;
     kind: QueryAnalysisSourceKind;
     columns: string[];
+    catalog?: string | null;
+    schema?: string | null;
+    table?: string | null;
 }
 
 /**
@@ -12305,6 +12322,14 @@ export declare interface SetOperationFact {
 
 export declare type SetOperationType = 'union' | 'union_all' | 'intersect' | 'except';
 
+/**
+ * Set or update the ORDER BY clause.
+ *
+ * Bare expressions are normalized to ascending order entries. Existing
+ * `ordered` expression nodes preserve their direction and null ordering.
+ */
+declare function setOrderBy(node: Expression, orderBy: Expression | Expression[]): Expression;
+
 /**
  * SetProperty
  */
@@ -12654,6 +12679,12 @@ declare type Star = {
 /** Create a star (*) expression. */
 export declare function star(): Expr;
 
+export declare interface StarProjectionFact {
+    index: number;
+    table?: string;
+    expandedColumns: string[];
+}
+
 /**
  * StDistance
  */
@@ -13718,6 +13749,12 @@ declare type TransformFunc = {
     transform: Expression;
 };
 
+export declare interface TransformFunctionFact {
+    name: string;
+    literalArgs: string[];
+    columnArgs: ColumnReferenceFact[];
+}
+
 export declare type TransformKind = 'direct' | 'cast' | 'aggregation' | 'constant' | 'expression' | 'star';
 
 /**