@polyglot-sql/sdk 0.5.3 → 0.5.4

package/dist/index.node.js

@@ -1728,6 +1728,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;
@@ -1920,6 +1938,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;
@@ -2652,10 +2708,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;
@@ -2745,10 +2797,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);
@@ -3076,9 +3124,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,
@@ -3124,6 +3170,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;
@@ -3134,6 +3181,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;
@@ -3325,6 +3374,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,
@@ -3335,6 +3385,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,
@@ -3519,6 +3571,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,
@@ -3529,6 +3582,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,
@@ -3712,6 +3767,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)) {
@@ -3881,25 +3999,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") {
@@ -4066,7 +4180,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");
@@ -4299,6 +4414,7 @@ const index$1 = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.definePrope
   setDistinct,
   setLimit,
   setOffset,
+  setOrderBy,
   some,
   transform,
   walk

package/dist/manual.js

@@ -1824,6 +1824,30 @@ function ast_get_table_names(ast_json) {
     }
 }
 
+/**
+ * Get all table nodes from an AST (as JSON string).
+ * @param {string} ast_json
+ * @returns {string}
+ */
+function ast_get_tables(ast_json) {
+    let deferred2_0;
+    let deferred2_1;
+    try {
+        const retptr = wasm$2.__wbindgen_add_to_stack_pointer(-16);
+        const ptr0 = passStringToWasm0(ast_json, wasm$2.__wbindgen_export, wasm$2.__wbindgen_export2);
+        const len0 = WASM_VECTOR_LEN;
+        wasm$2.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$2.__wbindgen_add_to_stack_pointer(16);
+        wasm$2.__wbindgen_export4(deferred2_0, deferred2_1, 1);
+    }
+}
+
 /**
  * Get all window functions from an AST (as JSON string).
  * @param {string} ast_json
@@ -2084,6 +2108,58 @@ function ast_set_limit(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(ast_json, offset) {
+    let deferred2_0;
+    let deferred2_1;
+    try {
+        const retptr = wasm$2.__wbindgen_add_to_stack_pointer(-16);
+        const ptr0 = passStringToWasm0(ast_json, wasm$2.__wbindgen_export, wasm$2.__wbindgen_export2);
+        const len0 = WASM_VECTOR_LEN;
+        wasm$2.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$2.__wbindgen_add_to_stack_pointer(16);
+        wasm$2.__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(ast_json, order_by_json) {
+    let deferred3_0;
+    let deferred3_1;
+    try {
+        const retptr = wasm$2.__wbindgen_add_to_stack_pointer(-16);
+        const ptr0 = passStringToWasm0(ast_json, wasm$2.__wbindgen_export, wasm$2.__wbindgen_export2);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passStringToWasm0(order_by_json, wasm$2.__wbindgen_export, wasm$2.__wbindgen_export2);
+        const len1 = WASM_VECTOR_LEN;
+        wasm$2.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$2.__wbindgen_add_to_stack_pointer(16);
+        wasm$2.__wbindgen_export4(deferred3_0, deferred3_1, 1);
+    }
+}
+
 /**
  * Diff two SQL statements and return edit operations.
  *
@@ -3271,10 +3347,6 @@ function __wbg_get_imports() {
             const ret = getObject(arg0).call(getObject(arg1));
             return addHeapObject(ret);
         }, arguments); },
-        __wbg_codePointAt_bf59dbf74d8db275: function(arg0, arg1) {
-            const ret = getObject(arg0).codePointAt(arg1 >>> 0);
-            return addHeapObject(ret);
-        },
         __wbg_done_57b39ecd9addfe81: function(arg0) {
             const ret = getObject(arg0).done;
             return ret;
@@ -3360,10 +3432,6 @@ function __wbg_get_imports() {
             const ret = getObject(arg0).length;
             return ret;
         },
-        __wbg_length_68dc7c5cf1b6d349: function(arg0) {
-            const ret = getObject(arg0).length;
-            return ret;
-        },
         __wbg_new_361308b2356cecd0: function() {
             const ret = new Object();
             return addHeapObject(ret);
@@ -3827,6 +3895,7 @@ const wasmModule = /*#__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,
@@ -3837,6 +3906,8 @@ const wasmModule = /*#__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,
@@ -4026,6 +4097,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)) {
@@ -4195,25 +4329,21 @@ function setLimit(node, limit) {
     const result = parseAstResult(ast_set_limit(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(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(exprToJson$1(node), JSON.stringify(orderByExpressions))
+  );
+  return result ?? applyOrderBy(node, orderByExpressions);
 }
 function removeLimitOffset(node) {
   if (getExprType(node) !== "select") {
@@ -4381,7 +4511,8 @@ function getColumns(node) {
   return findByType(node, "column");
 }
 function getTables(node) {
-  return findByType(node, "table");
+  const result = JSON.parse(ast_get_tables(exprToJson(node)));
+  return result.success ? JSON.parse(result.ast) : [];
 }
 function getIdentifiers(node) {
   return findByType(node, "identifier");
@@ -4615,6 +4746,7 @@ const index = /*#__PURE__*/Object.freeze(/*#__PURE__*/Object.defineProperty({
     setDistinct,
     setLimit,
     setOffset,
+    setOrderBy,
     some,
     transform,
     walk

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@polyglot-sql/sdk",
-  "version": "0.5.3",
+  "version": "0.5.4",
   "description": "SQL dialect translator powered by WebAssembly",
   "type": "module",
   "main": "./dist/index.cjs",