@polyglot-sql/sdk 0.5.3 → 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,
@@ -10877,6 +10878,7 @@ export declare interface ProjectionFact {
     isStar: boolean;
     starTable?: string;
     transformKind: TransformKind;
+    transformFunction?: TransformFunctionFact;
     castType?: string;
     typeHint?: string;
     nullability: ProjectionNullability;
@@ -11422,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;
 }
 
 /**
@@ -12317,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
  */
@@ -13736,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,
@@ -10877,6 +10878,7 @@ export declare interface ProjectionFact {
     isStar: boolean;
     starTable?: string;
     transformKind: TransformKind;
+    transformFunction?: TransformFunctionFact;
     castType?: string;
     typeHint?: string;
     nullability: ProjectionNullability;
@@ -11422,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;
 }
 
 /**
@@ -12317,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
  */
@@ -13736,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.js

@@ -1864,6 +1864,30 @@ function ast_get_table_names$1(ast_json) {
     }
 }
 
+/**
+ * Get all table nodes from an AST (as JSON string).
+ * @param {string} ast_json
+ * @returns {string}
+ */
+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);
+    }
+}
+
 /**
  * Get all window functions from an AST (as JSON string).
  * @param {string} ast_json
@@ -2124,6 +2148,58 @@ function ast_set_limit$1(ast_json, limit) {
     }
 }
 
+/**
+ * Set the OFFSET on a SELECT or set-operation AST.
+ * @param {string} ast_json
+ * @param {number} offset
+ * @returns {string}
+ */
+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);
+    }
+}
+
+/**
+ * Set the ORDER BY clause on a SELECT or set-operation AST.
+ * @param {string} ast_json
+ * @param {string} order_by_json
+ * @returns {string}
+ */
+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);
+    }
+}
+
 /**
  * Diff two SQL statements and return edit operations.
  *
@@ -3307,10 +3383,6 @@ function __wbg_call_389efe28435a9388() { return handleError(function (arg0, arg1
     const ret = getObject(arg0).call(getObject(arg1));
     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;
@@ -3396,10 +3468,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);
@@ -3764,9 +3832,7 @@ const __vite__wasmModule = await __vite__initWasm({ "./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,
@@ -3811,6 +3877,7 @@ const ast_get_functions = __vite__wasmModule.ast_get_functions;
 const ast_get_literals = __vite__wasmModule.ast_get_literals;
 const ast_get_subqueries = __vite__wasmModule.ast_get_subqueries;
 const ast_get_table_names = __vite__wasmModule.ast_get_table_names;
+const ast_get_tables = __vite__wasmModule.ast_get_tables;
 const ast_get_window_functions = __vite__wasmModule.ast_get_window_functions;
 const ast_node_count = __vite__wasmModule.ast_node_count;
 const ast_qualify_columns = __vite__wasmModule.ast_qualify_columns;
@@ -3821,6 +3888,8 @@ const ast_rename_tables = __vite__wasmModule.ast_rename_tables;
 const ast_rename_tables_with_options = __vite__wasmModule.ast_rename_tables_with_options;
 const ast_set_distinct = __vite__wasmModule.ast_set_distinct;
 const ast_set_limit = __vite__wasmModule.ast_set_limit;
+const ast_set_offset = __vite__wasmModule.ast_set_offset;
+const ast_set_order_by = __vite__wasmModule.ast_set_order_by;
 const diff_sql = __vite__wasmModule.diff_sql;
 const format_sql = __vite__wasmModule.format_sql;
 const format_sql_value = __vite__wasmModule.format_sql_value;
@@ -4013,6 +4082,7 @@ const wasm$2 = /*#__PURE__*/Object.freeze(/*#__PURE__*/Object.defineProperty({
     ast_get_literals,
     ast_get_subqueries,
     ast_get_table_names,
+    ast_get_tables,
     ast_get_window_functions,
     ast_node_count,
     ast_qualify_columns,
@@ -4023,6 +4093,8 @@ const wasm$2 = /*#__PURE__*/Object.freeze(/*#__PURE__*/Object.defineProperty({
     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,
@@ -4211,6 +4283,7 @@ const wasmModule = /*#__PURE__*/Object.freeze(/*#__PURE__*/Object.defineProperty
     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,
@@ -4221,6 +4294,8 @@ const wasmModule = /*#__PURE__*/Object.freeze(/*#__PURE__*/Object.defineProperty
     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,
@@ -4407,6 +4482,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)) {
@@ -4576,25 +4714,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") {
@@ -4762,7 +4896,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");
@@ -4996,6 +5131,7 @@ const index$1 = /*#__PURE__*/Object.freeze(/*#__PURE__*/Object.defineProperty({
     setDistinct,
     setLimit,
     setOffset,
+    setOrderBy,
     some,
     transform,
     walk