@polyglot-sql/sdk 0.3.10 → 0.4.0

package/dist/index.js

@@ -144,31 +144,34 @@ class WasmCaseBuilder {
         return this;
     }
     /**
-     * Generate SQL string (generic dialect).
+     * Generate SQL string for the given dialect.
+     * @param {string} dialect
      * @returns {string}
      */
-    to_sql() {
-        let deferred2_0;
-        let deferred2_1;
+    to_sql(dialect) {
+        let deferred3_0;
+        let deferred3_1;
         try {
             const retptr = wasm$3.__wbindgen_add_to_stack_pointer(-16);
-            wasm$3.wasmcasebuilder_to_sql(retptr, this.__wbg_ptr);
+            const ptr0 = passStringToWasm0(dialect, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+            const len0 = WASM_VECTOR_LEN;
+            wasm$3.wasmcasebuilder_to_sql(retptr, this.__wbg_ptr, ptr0, len0);
             var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
             var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
             var r2 = getDataViewMemory0().getInt32(retptr + 4 * 2, true);
             var r3 = getDataViewMemory0().getInt32(retptr + 4 * 3, true);
-            var ptr1 = r0;
-            var len1 = r1;
+            var ptr2 = r0;
+            var len2 = r1;
             if (r3) {
-                ptr1 = 0; len1 = 0;
+                ptr2 = 0; len2 = 0;
                 throw takeObject(r2);
             }
-            deferred2_0 = ptr1;
-            deferred2_1 = len1;
-            return getStringFromWasm0(ptr1, len1);
+            deferred3_0 = ptr2;
+            deferred3_1 = len2;
+            return getStringFromWasm0(ptr2, len2);
         } finally {
             wasm$3.__wbindgen_add_to_stack_pointer(16);
-            wasm$3.__wbindgen_export4(deferred2_0, deferred2_1, 1);
+            wasm$3.__wbindgen_export4(deferred3_0, deferred3_1, 1);
         }
     }
     /**
@@ -557,31 +560,34 @@ class WasmExpr {
         }
     }
     /**
-     * Generate SQL string (generic dialect).
+     * Generate SQL string for the given dialect.
+     * @param {string} dialect
      * @returns {string}
      */
-    to_sql() {
-        let deferred2_0;
-        let deferred2_1;
+    to_sql(dialect) {
+        let deferred3_0;
+        let deferred3_1;
         try {
             const retptr = wasm$3.__wbindgen_add_to_stack_pointer(-16);
-            wasm$3.wasmexpr_to_sql(retptr, this.__wbg_ptr);
+            const ptr0 = passStringToWasm0(dialect, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+            const len0 = WASM_VECTOR_LEN;
+            wasm$3.wasmexpr_to_sql(retptr, this.__wbg_ptr, ptr0, len0);
             var r0 = getDataViewMemory0().getInt32(retptr + 4 * 0, true);
             var r1 = getDataViewMemory0().getInt32(retptr + 4 * 1, true);
             var r2 = getDataViewMemory0().getInt32(retptr + 4 * 2, true);
             var r3 = getDataViewMemory0().getInt32(retptr + 4 * 3, true);
-            var ptr1 = r0;
-            var len1 = r1;
+            var ptr2 = r0;
+            var len2 = r1;
             if (r3) {
-                ptr1 = 0; len1 = 0;
+                ptr2 = 0; len2 = 0;
                 throw takeObject(r2);
             }
-            deferred2_0 = ptr1;
-            deferred2_1 = len1;
-            return getStringFromWasm0(ptr1, len1);
+            deferred3_0 = ptr2;
+            deferred3_1 = len2;
+            return getStringFromWasm0(ptr2, len2);
         } finally {
             wasm$3.__wbindgen_add_to_stack_pointer(16);
-            wasm$3.__wbindgen_export4(deferred2_0, deferred2_1, 1);
+            wasm$3.__wbindgen_export4(deferred3_0, deferred3_1, 1);
         }
     }
     /**
@@ -2382,6 +2388,87 @@ function lineage_sql_with_schema$1(sql, column, schema_json, dialect, trim_selec
     }
 }
 
+/**
+ * Build an OpenLineage columnLineage facet and dataset payload.
+ * @param {string} sql
+ * @param {string} options_json
+ * @returns {string}
+ */
+function openlineage_column_lineage$1(sql, options_json) {
+    let deferred3_0;
+    let deferred3_1;
+    try {
+        const retptr = wasm$3.__wbindgen_add_to_stack_pointer(-16);
+        const ptr0 = passStringToWasm0(sql, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passStringToWasm0(options_json, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+        const len1 = WASM_VECTOR_LEN;
+        wasm$3.openlineage_column_lineage(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);
+    }
+}
+
+/**
+ * Build an OpenLineage JobEvent payload. This does not send the event.
+ * @param {string} sql
+ * @param {string} options_json
+ * @returns {string}
+ */
+function openlineage_job_event$1(sql, options_json) {
+    let deferred3_0;
+    let deferred3_1;
+    try {
+        const retptr = wasm$3.__wbindgen_add_to_stack_pointer(-16);
+        const ptr0 = passStringToWasm0(sql, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passStringToWasm0(options_json, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+        const len1 = WASM_VECTOR_LEN;
+        wasm$3.openlineage_job_event(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);
+    }
+}
+
+/**
+ * Build an OpenLineage RunEvent payload. This does not send the event.
+ * @param {string} sql
+ * @param {string} options_json
+ * @returns {string}
+ */
+function openlineage_run_event$1(sql, options_json) {
+    let deferred3_0;
+    let deferred3_1;
+    try {
+        const retptr = wasm$3.__wbindgen_add_to_stack_pointer(-16);
+        const ptr0 = passStringToWasm0(sql, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+        const len0 = WASM_VECTOR_LEN;
+        const ptr1 = passStringToWasm0(options_json, wasm$3.__wbindgen_export, wasm$3.__wbindgen_export2);
+        const len1 = WASM_VECTOR_LEN;
+        wasm$3.openlineage_run_event(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);
+    }
+}
+
 /**
  * Parse SQL into an AST (returned as JSON).
  *
@@ -3570,6 +3657,9 @@ const get_dialects = __vite__wasmModule.get_dialects;
 const get_dialects_value = __vite__wasmModule.get_dialects_value;
 const lineage_sql = __vite__wasmModule.lineage_sql;
 const lineage_sql_with_schema = __vite__wasmModule.lineage_sql_with_schema;
+const openlineage_column_lineage = __vite__wasmModule.openlineage_column_lineage;
+const openlineage_job_event = __vite__wasmModule.openlineage_job_event;
+const openlineage_run_event = __vite__wasmModule.openlineage_run_event;
 const parse$1 = __vite__wasmModule.parse;
 const parse_value = __vite__wasmModule.parse_value;
 const plan$1 = __vite__wasmModule.plan;
@@ -3762,6 +3852,9 @@ const wasm$2 = /*#__PURE__*/Object.freeze(/*#__PURE__*/Object.defineProperty({
     lineage_sql,
     lineage_sql_with_schema,
     memory,
+    openlineage_column_lineage,
+    openlineage_job_event,
+    openlineage_run_event,
     parse: parse$1,
     parse_value,
     plan: plan$1,
@@ -3948,6 +4041,9 @@ const wasmModule = /*#__PURE__*/Object.freeze(/*#__PURE__*/Object.defineProperty
     get_dialects_value: get_dialects_value$1,
     lineage_sql: lineage_sql$1,
     lineage_sql_with_schema: lineage_sql_with_schema$1,
+    openlineage_column_lineage: openlineage_column_lineage$1,
+    openlineage_job_event: openlineage_job_event$1,
+    openlineage_run_event: openlineage_run_event$1,
     parse: parse$2,
     parse_value: parse_value$1,
     plan: plan$2,
@@ -4831,9 +4927,9 @@ class Expr {
     return new Expr(this._w.desc());
   }
   // -- Output --
-  /** Generate SQL string (generic dialect). */
-  toSql() {
-    return this._w.to_sql();
+  /** Generate SQL string. Defaults to generic dialect. */
+  toSql(dialect = "generic") {
+    return this._w.to_sql(dialect);
   }
   /** Return the expression AST as a plain JS object. */
   toJSON() {
@@ -5369,9 +5465,9 @@ class CaseBuilder {
   build() {
     return new Expr(this._w.build_expr());
   }
-  /** Generate SQL string (generic dialect). */
-  toSql() {
-    return this._w.to_sql();
+  /** Generate SQL string. Defaults to generic dialect. */
+  toSql(dialect = "generic") {
+    return this._w.to_sql(dialect);
   }
 }
 function caseWhen() {
@@ -5602,6 +5698,35 @@ function getSourceTables(column, sql, dialect = "generic") {
   return JSON.parse(resultJson);
 }
 
+function normalizeOptions(options) {
+  return {
+    dialect: "generic",
+    datasetMappings: {},
+    ...options
+  };
+}
+function openLineageColumnLineage(sql, options) {
+  const resultJson = openlineage_column_lineage$1(
+    sql,
+    JSON.stringify(normalizeOptions(options))
+  );
+  return JSON.parse(resultJson);
+}
+function openLineageJobEvent(sql, options) {
+  const resultJson = openlineage_job_event$1(
+    sql,
+    JSON.stringify(normalizeOptions(options))
+  );
+  return JSON.parse(resultJson);
+}
+function openLineageRunEvent(sql, options) {
+  const resultJson = openlineage_run_event$1(
+    sql,
+    JSON.stringify(normalizeOptions(options))
+  );
+  return JSON.parse(resultJson);
+}
+
 function diff(source, target, dialect = "generic", options = {}) {
   const { deltaOnly = false, f = 0.6, t = 0.6 } = options;
   const resultJson = diff_sql$1(source, target, dialect, deltaOnly, f, t);
@@ -5897,6 +6022,9 @@ const index = {
   lineage: lineage,
   lineageWithSchema: lineageWithSchema,
   getSourceTables: getSourceTables,
+  openLineageColumnLineage: openLineageColumnLineage,
+  openLineageJobEvent: openLineageJobEvent,
+  openLineageRunEvent: openLineageRunEvent,
   diff: diff,
   hasChanges: hasChanges,
   changesOnly: changesOnly,
@@ -5905,4 +6033,4 @@ const index = {
   Polyglot
 };
 
-export { CaseBuilder, DeleteBuilder, Dialect, Expr, InsertBuilder, MergeBuilder, Polyglot, SelectBuilder, SetOpBuilder, UpdateBuilder, ValidationSeverity, WindowDefBuilder, abs, alias, and, annotateTypes, index$1 as ast, avg, boolean, caseOf, caseWhen, cast, ceil, changesOnly, coalesce, col, concatWs, condition, count, countDistinct, currentDate, currentTime, currentTimestamp, index as default, del, deleteFrom, denseRank, diff, except, exp, extract, findAll, floor, format, formatWithOptions, func, generate, getColumns, getDialects, getInferredType, getSourceTables, getVersion, greatest, hasChanges, ifNull, init, initcap, insert, insertInto, intersect, isColumn, isFunction, isInitialized, isLiteral, isSelect, least, length, lineage, lineageWithSchema, lit, ln, lower, ltrim, max, mergeInto, min, not, nullIf, or, parse, plan, power, rank, renameColumns, replace, reverse, round, rowNumber, rtrim, select, sign, sqlExpr, sqlNull, sqrt, star, subquery, substring, sum, table, tokenize, transform, transpile, trim, union, unionAll, update, upper, validate, validateWithSchema, walk };
+export { CaseBuilder, DeleteBuilder, Dialect, Expr, InsertBuilder, MergeBuilder, Polyglot, SelectBuilder, SetOpBuilder, UpdateBuilder, ValidationSeverity, WindowDefBuilder, abs, alias, and, annotateTypes, index$1 as ast, avg, boolean, caseOf, caseWhen, cast, ceil, changesOnly, coalesce, col, concatWs, condition, count, countDistinct, currentDate, currentTime, currentTimestamp, index as default, del, deleteFrom, denseRank, diff, except, exp, extract, findAll, floor, format, formatWithOptions, func, generate, getColumns, getDialects, getInferredType, getSourceTables, getVersion, greatest, hasChanges, ifNull, init, initcap, insert, insertInto, intersect, isColumn, isFunction, isInitialized, isLiteral, isSelect, least, length, lineage, lineageWithSchema, lit, ln, lower, ltrim, max, mergeInto, min, not, nullIf, openLineageColumnLineage, openLineageJobEvent, openLineageRunEvent, or, parse, plan, power, rank, renameColumns, replace, reverse, round, rowNumber, rtrim, select, sign, sqlExpr, sqlNull, sqrt, star, subquery, substring, sum, table, tokenize, transform, transpile, trim, union, unionAll, update, upper, validate, validateWithSchema, walk };

package/package.json

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