@polyglot-sql/sdk 0.5.2 → 0.5.4

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.2",
+  "version": "0.5.4",
   "description": "SQL dialect translator powered by WebAssembly",
   "type": "module",
   "main": "./dist/index.cjs",