@uwdata/mosaic-sql 0.7.0 → 0.9.0

package/dist/mosaic-sql.js

@@ -3,13 +3,11 @@ var Ref = class {
   /**
    * Create a new Ref instance.
    * @param {string|Ref|null} table The table name.
-   * @param {string|null} column The column name.
+   * @param {string|null} [column] The column name.
    */
   constructor(table, column2) {
-    if (table)
-      this.table = String(table);
-    if (column2)
-      this.column = column2;
+    if (table) this.table = String(table);
+    if (column2) this.column = column2;
   }
   /**
    * Get the list of referenced columns. Either a single element array
@@ -48,7 +46,7 @@ function asRelation(value) {
 function relation(name) {
   return new Ref(name);
 }
-function column(table, column2) {
+function column(table, column2 = null) {
   if (arguments.length === 1) {
     column2 = table;
     table = null;
@@ -68,7 +66,7 @@ function literalToSQL(value) {
     case "boolean":
       return value ? "TRUE" : "FALSE";
     case "string":
-      return `'${value}'`;
+      return `'${value.replace(`'`, `''`)}'`;
     case "number":
       return Number.isFinite(value) ? String(value) : "NULL";
     default:
@@ -76,8 +74,7 @@ function literalToSQL(value) {
         return "NULL";
       } else if (value instanceof Date) {
         const ts = +value;
-        if (Number.isNaN(ts))
-          return "NULL";
+        if (Number.isNaN(ts)) return "NULL";
         const y2 = value.getUTCFullYear();
         const m = value.getUTCMonth();
         const d = value.getUTCDate();
@@ -98,7 +95,7 @@ function isSQLExpression(value) {
 var SQLExpression = class {
   /**
    * Create a new SQL expression instance.
-   * @param {(string|SQLExpression|Ref)[]} parts The parts of the expression.
+   * @param {(string | ParamLike | SQLExpression | import('./ref.js').Ref)[]} parts The parts of the expression.
    * @param {string[]} [columns=[]] The column dependencies
    * @param {object} [props] Additional properties for this expression.
    */
@@ -152,7 +149,7 @@ var SQLExpression = class {
   /**
    * Annotate this expression instance with additional properties.
    * @param {object[]} [props] One or more objects with properties to add.
-   * @returns {this} This SQL expression.
+   * @returns This SQL expression.
    */
   annotate(...props) {
     return Object.assign(this, ...props);
@@ -290,6 +287,16 @@ var isInfinite = functionCall("ISINF");
 
 // src/windows.js
 var WindowFunction = class _WindowFunction extends SQLExpression {
+  /**
+   * Create a new WindowFunction instance.
+   * @param {string} op The window operation indicator.
+   * @param {*} func The window function expression.
+   * @param {*} [type] The SQL data type to cast to.
+   * @param {string} [name] The window definition name.
+   * @param {*} [group] Grouping (partition by) expressions.
+   * @param {*} [order] Sorting (order by) expressions.
+   * @param {*} [frame] The window frame definition.
+   */
   constructor(op, func, type, name, group = "", order = "", frame = "") {
     let expr;
     const noWindowParams = !(group || order || frame);
@@ -304,7 +311,14 @@ var WindowFunction = class _WindowFunction extends SQLExpression {
       expr = sql`(${expr})::${type}`;
     }
     const { _expr, _deps } = expr;
-    super(_expr, _deps, { window: op, func, type, name, group, order, frame });
+    super(_expr, _deps);
+    this.window = op;
+    this.func = func;
+    this.type = type;
+    this.name = name;
+    this.group = group;
+    this.order = order;
+    this.frame = frame;
   }
   get basis() {
     return this.column;
@@ -313,10 +327,20 @@ var WindowFunction = class _WindowFunction extends SQLExpression {
     const { func } = this;
     return func.label ?? func.toString();
   }
+  /**
+   * Return an updated window function over a named window definition.
+   * @param {string} name The window definition name.
+   * @returns {WindowFunction} A new window function.
+   */
   over(name) {
     const { window: op, func, type, group, order, frame } = this;
     return new _WindowFunction(op, func, type, name, group, order, frame);
   }
+  /**
+   * Return an updated window function with the given partitioning.
+   * @param {*} expr The grouping (partition by) criteria for the window function.
+   * @returns {WindowFunction} A new window function.
+   */
   partitionby(...expr) {
     const exprs = expr.flat().filter((x2) => x2).map(asColumn);
     const group = sql(
@@ -326,6 +350,11 @@ var WindowFunction = class _WindowFunction extends SQLExpression {
     const { window: op, func, type, name, order, frame } = this;
     return new _WindowFunction(op, func, type, name, group, order, frame);
   }
+  /**
+   * Return an updated window function with the given ordering.
+   * @param {*} expr The sorting (order by) criteria for the window function.
+   * @returns {WindowFunction} A new window function.
+   */
   orderby(...expr) {
     const exprs = expr.flat().filter((x2) => x2).map(asColumn);
     const order = sql(
@@ -335,11 +364,21 @@ var WindowFunction = class _WindowFunction extends SQLExpression {
     const { window: op, func, type, name, group, frame } = this;
     return new _WindowFunction(op, func, type, name, group, order, frame);
   }
+  /**
+   * Return an updated window function with the given rows frame.
+   * @param {(number|null)[] | import('./expression.js').ParamLike} expr The row-based window frame.
+   * @returns {WindowFunction} A new window function.
+   */
   rows(expr) {
     const frame = windowFrame("ROWS", expr);
     const { window: op, func, type, name, group, order } = this;
     return new _WindowFunction(op, func, type, name, group, order, frame);
   }
+  /**
+   * Return an updated window function with the given range frame.
+   * @param {(number|null)[] | import('./expression.js').ParamLike} expr The range-based window frame.
+   * @returns {WindowFunction} A new window function.
+   */
   range(expr) {
     const frame = windowFrame("RANGE", expr);
     const { window: op, func, type, name, group, order } = this;
@@ -383,11 +422,24 @@ function agg(strings, ...exprs) {
   return sql(strings, ...exprs).annotate({ aggregate: true });
 }
 var AggregateFunction = class _AggregateFunction extends SQLExpression {
+  /**
+   * Create a new AggregateFunction instance.
+   * @param {*} op The aggregate operation.
+   * @param {*} [args] The aggregate function arguments.
+   * @param {*} [type] The SQL data type to cast to.
+   * @param {boolean} [isDistinct] Flag indicating if this is a distinct value aggregate.
+   * @param {*} [filter] Filtering expression to apply prior to aggregation.
+   */
   constructor(op, args, type, isDistinct2, filter) {
     args = (args || []).map(asColumn);
     const { strings, exprs } = aggExpr(op, args, type, isDistinct2, filter);
     const { spans, cols } = parseSQL(strings, exprs);
-    super(spans, cols, { aggregate: op, args, type, isDistinct: isDistinct2, filter });
+    super(spans, cols);
+    this.aggregate = op;
+    this.args = args;
+    this.type = type;
+    this.isDistinct = isDistinct2;
+    this.filter = filter;
   }
   get basis() {
     return this.column;
@@ -398,30 +450,63 @@ var AggregateFunction = class _AggregateFunction extends SQLExpression {
     const tail = args.length ? `(${dist}${args.map(unquoted).join(", ")})` : "";
     return `${op.toLowerCase()}${tail}`;
   }
+  /**
+   * Return a new derived aggregate function over distinct values.
+   * @returns {AggregateFunction} A new aggregate function.
+   */
   distinct() {
     const { aggregate: op, args, type, filter } = this;
     return new _AggregateFunction(op, args, type, true, filter);
   }
+  /**
+   * Return a new derived aggregate function that filters values.
+   * @param {*} filter The filter expresion.
+   * @returns {AggregateFunction} A new aggregate function.
+   */
   where(filter) {
     const { aggregate: op, args, type, isDistinct: isDistinct2 } = this;
     return new _AggregateFunction(op, args, type, isDistinct2, filter);
   }
+  /**
+   * Return a new window function over this aggregate.
+   * @returns {WindowFunction} A new aggregate function.
+   */
   window() {
     const { aggregate: op, args, type, isDistinct: isDistinct2 } = this;
     const func = new _AggregateFunction(op, args, null, isDistinct2);
     return new WindowFunction(op, func, type);
   }
+  /**
+   * Return a window function over this aggregate with the given partitioning.
+   * @param {*} expr The grouping (partition by) criteria for the window function.
+   * @returns {WindowFunction} A new window function.
+   */
   partitionby(...expr) {
     return this.window().partitionby(...expr);
   }
+  /**
+   * Return a window function over this aggregate with the given ordering.
+   * @param {*} expr The sorting (order by) criteria for the window function.
+   * @returns {WindowFunction} A new window function.
+   */
   orderby(...expr) {
     return this.window().orderby(...expr);
   }
-  rows(prev, next) {
-    return this.window().rows(prev, next);
+  /**
+   * Return a window function over this aggregate with the given row frame.
+   * @param {(number|null)[] | import('./expression.js').ParamLike} frame The row-based window frame.
+   * @returns {WindowFunction} A new window function.
+   */
+  rows(frame) {
+    return this.window().rows(frame);
   }
-  range(prev, next) {
-    return this.window().range(prev, next);
+  /**
+   * Return a window function over this aggregate with the given range frame.
+   * @param {(number|null)[] | import('./expression.js').ParamLike} frame The range-based window frame.
+   * @returns {WindowFunction} A new window function.
+   */
+  range(frame) {
+    return this.window().range(frame);
   }
 };
 function aggExpr(op, args, type, isDistinct2, filter) {
@@ -466,6 +551,7 @@ var entropy = aggf("ENTROPY");
 var varPop = aggf("VAR_POP");
 var stddevPop = aggf("STDDEV_POP");
 var corr = aggf("CORR");
+var covariance = aggf("COVAR_SAMP");
 var covarPop = aggf("COVAR_POP");
 var regrIntercept = aggf("REGR_INTERCEPT");
 var regrSlope = aggf("REGR_SLOPE");
@@ -573,12 +659,22 @@ var Query = class _Query {
       qualify: [],
       orderby: []
     };
+    this.cteFor = null;
   }
   clone() {
     const q = new _Query();
     q.query = { ...this.query };
     return q;
   }
+  /**
+   * Retrieve current WITH common table expressions (CTEs).
+   * @returns {any[]}
+   */
+  /**
+  * Add WITH common table expressions (CTEs).
+  * @param  {...any} expr Expressions to add.
+  * @returns {this}
+  */
   with(...expr) {
     const { query } = this;
     if (expr.length === 0) {
@@ -604,6 +700,15 @@ var Query = class _Query {
       return this;
     }
   }
+  /**
+   * Retrieve current SELECT expressions.
+   * @returns {any[]}
+   */
+  /**
+  * Add SELECT expressions.
+  * @param {...any} expr Expressions to add.
+  * @returns {this}
+  */
   select(...expr) {
     const { query } = this;
     if (expr.length === 0) {
@@ -624,7 +729,8 @@ var Query = class _Query {
           }
         }
       }
-      query.select = query.select.concat(list);
+      const keys = new Set(list.map((x2) => x2.as));
+      query.select = query.select.filter((x2) => !keys.has(x2.as)).concat(list.filter((x2) => x2.expr));
       return this;
     }
   }
@@ -636,6 +742,15 @@ var Query = class _Query {
     this.query.distinct = !!value;
     return this;
   }
+  /**
+   * Retrieve current from expressions.
+   * @returns {any[]}
+   */
+  /**
+  * Provide table from expressions.
+  * @param  {...any} expr
+  * @returns {this}
+  */
   from(...expr) {
     const { query } = this;
     if (expr.length === 0) {
@@ -666,6 +781,16 @@ var Query = class _Query {
     this.query.from = [];
     return this.from(...expr);
   }
+  /**
+   * Retrieve current SAMPLE settings.
+   * @returns {any[]}
+   */
+  /**
+  * Set SAMPLE settings.
+  * @param {number|object} value The percentage or number of rows to sample.
+  * @param {string} [method] The sampling method to use.
+  * @returns {this}
+  */
   sample(value, method) {
     const { query } = this;
     if (arguments.length === 0) {
@@ -679,6 +804,15 @@ var Query = class _Query {
       return this;
     }
   }
+  /**
+   * Retrieve current WHERE expressions.
+   * @returns {any[]}
+   */
+  /**
+  * Add WHERE expressions.
+  * @param  {...any} expr Expressions to add.
+  * @returns {this}
+  */
   where(...expr) {
     const { query } = this;
     if (expr.length === 0) {
@@ -694,6 +828,15 @@ var Query = class _Query {
     this.query.where = [];
     return this.where(...expr);
   }
+  /**
+   * Retrieve current GROUP BY expressions.
+   * @returns {any[]}
+   */
+  /**
+  * Add GROUP BY expressions.
+  * @param  {...any} expr Expressions to add.
+  * @returns {this}
+  */
   groupby(...expr) {
     const { query } = this;
     if (expr.length === 0) {
@@ -709,6 +852,15 @@ var Query = class _Query {
     this.query.groupby = [];
     return this.groupby(...expr);
   }
+  /**
+   * Retrieve current HAVING expressions.
+   * @returns {any[]}
+   */
+  /**
+  * Add HAVING expressions.
+  * @param  {...any} expr Expressions to add.
+  * @returns {this}
+  */
   having(...expr) {
     const { query } = this;
     if (expr.length === 0) {
@@ -720,6 +872,15 @@ var Query = class _Query {
       return this;
     }
   }
+  /**
+   * Retrieve current WINDOW definitions.
+   * @returns {any[]}
+   */
+  /**
+  * Add WINDOW definitions.
+  * @param  {...any} expr Expressions to add.
+  * @returns {this}
+  */
   window(...expr) {
     const { query } = this;
     if (expr.length === 0) {
@@ -738,6 +899,15 @@ var Query = class _Query {
       return this;
     }
   }
+  /**
+   * Retrieve current QUALIFY expressions.
+   * @returns {any[]}
+   */
+  /**
+  * Add QUALIFY expressions.
+  * @param  {...any} expr Expressions to add.
+  * @returns {this}
+  */
   qualify(...expr) {
     const { query } = this;
     if (expr.length === 0) {
@@ -749,6 +919,15 @@ var Query = class _Query {
       return this;
     }
   }
+  /**
+   * Retrieve current ORDER BY expressions.
+   * @returns {any[]}
+   */
+  /**
+  * Add ORDER BY expressions.
+  * @param  {...any} expr Expressions to add.
+  * @returns {this}
+  */
   orderby(...expr) {
     const { query } = this;
     if (expr.length === 0) {
@@ -760,6 +939,15 @@ var Query = class _Query {
       return this;
     }
   }
+  /**
+   * Retrieve current LIMIT value.
+   * @returns {number|null}
+   */
+  /**
+  * Set the query result LIMIT.
+  * @param {number} value The limit value.
+  * @returns {this}
+  */
   limit(value) {
     const { query } = this;
     if (arguments.length === 0) {
@@ -769,6 +957,15 @@ var Query = class _Query {
       return this;
     }
   }
+  /**
+   * Retrieve current OFFSET value.
+   * @returns {number|null}
+   */
+  /**
+  * Set the query result OFFSET.
+  * @param {number} value The offset value.
+  * @returns {this}
+  */
   offset(value) {
     const { query } = this;
     if (arguments.length === 0) {
@@ -827,8 +1024,7 @@ var Query = class _Query {
     }
     if (where.length) {
       const clauses = where.map(String).filter((x2) => x2).join(" AND ");
-      if (clauses)
-        sql2.push(`WHERE ${clauses}`);
+      if (clauses) sql2.push(`WHERE ${clauses}`);
     }
     if (sample) {
       const { rows, perc, method, seed } = sample;
@@ -841,8 +1037,7 @@ var Query = class _Query {
     }
     if (having.length) {
       const clauses = having.map(String).filter((x2) => x2).join(" AND ");
-      if (clauses)
-        sql2.push(`HAVING ${clauses}`);
+      if (clauses) sql2.push(`HAVING ${clauses}`);
     }
     if (window.length) {
       const windows = window.map(({ as, expr }) => `"${as}" AS (${expr})`);
@@ -850,8 +1045,7 @@ var Query = class _Query {
     }
     if (qualify.length) {
       const clauses = qualify.map(String).filter((x2) => x2).join(" AND ");
-      if (clauses)
-        sql2.push(`QUALIFY ${clauses}`);
+      if (clauses) sql2.push(`QUALIFY ${clauses}`);
     }
     if (orderby.length) {
       sql2.push(`ORDER BY ${orderby.join(", ")}`);
@@ -870,6 +1064,7 @@ var SetOperation = class _SetOperation {
     this.op = op;
     this.queries = queries.map((q) => q.clone());
     this.query = { orderby: [] };
+    this.cteFor = null;
   }
   clone() {
     const q = new _SetOperation(this.op, this.queries);
@@ -907,8 +1102,7 @@ var SetOperation = class _SetOperation {
   }
   get subqueries() {
     const { queries, cteFor } = this;
-    if (cteFor)
-      queries.forEach((q) => q.cteFor = cteFor);
+    if (cteFor) queries.forEach((q) => q.cteFor = cteFor);
     return queries;
   }
   toString() {
@@ -949,7 +1143,7 @@ function scaleLinear() {
     sqlInvert: identity
   };
 }
-function scaleLog({ base } = {}) {
+function scaleLog({ base = null } = {}) {
   if (base == null || base === Math.E) {
     return {
       apply: Math.log,
@@ -1009,6 +1203,7 @@ function scaleTime() {
   };
 }
 var scales = {
+  identity: scaleLinear,
   linear: scaleLinear,
   log: scaleLog,
   symlog: scaleSymlog,
@@ -1080,7 +1275,7 @@ function loadSpatial(tableName, fileName, options = {}) {
   const { options: opt, ...rest } = options;
   if (opt) {
     const open = Array.isArray(opt) ? opt.join(", ") : typeof opt === "string" ? opt : Object.entries(opt).map(([key, value]) => `${key}=${value}`).join(", ");
-    rest.open_options = open.toUpperCase();
+    Object.assign(rest, { open_options: open.toUpperCase() });
   }
   return load("st_read", tableName, fileName, rest);
 }
@@ -1115,6 +1310,7 @@ function toDuckDBValue(value) {
 export {
   Query,
   Ref,
+  SQLExpression,
   agg,
   all,
   and,
@@ -1135,6 +1331,7 @@ export {
   corr,
   count,
   covarPop,
+  covariance,
   create,
   cume_dist,
   dateDay,

package/dist/mosaic-sql.min.js

@@ -1 +1 @@
-var S=class{constructor(t,e){t&&(this.table=String(t)),e&&(this.column=e)}get columns(){return this.column?[this.column]:[]}toString(){let{table:t,column:e}=this;if(e){let n=e.startsWith("*")?e:`"${e}"`;return`${t?`${V(t)}.`:""}${n}`}else return t?V(t):"NULL"}};function V(r){return r.split(".").map(e=>`"${e}"`).join(".")}function k(r,t){return r instanceof S&&r.column===t}function u(r){return typeof r=="string"?K(r):r}function O(r){return typeof r=="string"?H(r):r}function H(r){return new S(r)}function K(r,t){return arguments.length===1&&(t=r,r=null),new S(r,t)}function ut(r){return new S(r,"*")}function lt(r){return typeof r=="string"?`"${r}"`:$(r)}function $(r){switch(typeof r){case"boolean":return r?"TRUE":"FALSE";case"string":return`'${r}'`;case"number":return Number.isFinite(r)?String(r):"NULL";default:if(r==null)return"NULL";if(r instanceof Date){let t=+r;if(Number.isNaN(t))return"NULL";let e=r.getUTCFullYear(),n=r.getUTCMonth(),o=r.getUTCDate();return t===Date.UTC(e,n,o)?`MAKE_DATE(${e}, ${n+1}, ${o})`:`EPOCH_MS(${t})`}else return r instanceof RegExp?`'${r.source}'`:String(r)}}var T=r=>typeof r?.addEventListener=="function";function _(r){return r instanceof N}var N=class{constructor(t,e,n){this._expr=Array.isArray(t)?t:[t],this._deps=e||[],this.annotate(n);let o=this._expr.filter(s=>T(s));o.length>0?(this._params=Array.from(new Set(o)),this._params.forEach(s=>{s.addEventListener("value",()=>pt(this,this.map?.get("value")))})):this.addEventListener=void 0}get value(){return this}get columns(){let{_params:t,_deps:e}=this;if(t){let n=new Set(t.flatMap(o=>{let s=o.value?.columns;return Array.isArray(s)?s:[]}));if(n.size){let o=new Set(e);return n.forEach(s=>o.add(s)),Array.from(o)}}return e}get column(){return this._deps.length?this._deps[0]:this.columns[0]}annotate(...t){return Object.assign(this,...t)}toString(){return this._expr.map(t=>T(t)&&!_(t)?$(t.value):t).join("")}addEventListener(t,e){let n=this.map||(this.map=new Map);(n.get(t)||(n.set(t,new Set),n.get(t))).add(e)}};function pt(r,t){if(t?.size)return Promise.allSettled(Array.from(t,e=>e(r)))}function W(r,t){let e=[r[0]],n=new Set,o=t.length;for(let s=0,i=0;s<o;){let a=t[s];T(a)?e[++i]=a:(Array.isArray(a?.columns)&&a.columns.forEach(d=>n.add(d)),e[i]+=typeof a=="string"?a:$(a));let p=r[++s];T(e[i])?e[++i]=p:e[i]+=p}return{spans:e,cols:Array.from(n)}}function c(r,...t){let{spans:e,cols:n}=W(r,t);return new N(e,n)}function ft(r){let t=u(r);return c`${t} DESC NULLS LAST`.annotate({label:t?.label,desc:!0})}var ht=r=>({value:r,toString:()=>$(r)});function D(r){r(this.op,this),this.children?.forEach(t=>t.visit(r))}function z(r,t){let e=t.filter(o=>o!=null).map(u),n=e.map((o,s)=>s?` ${r} `:"");return e.length===1?n.push(""):e.length>1&&(n[0]="(",n.push(")")),c(n,...e).annotate({op:r,children:e,visit:D})}var mt=(...r)=>z("AND",r.flat()),gt=(...r)=>z("OR",r.flat()),dt=r=>t=>c`(${r} ${u(t)})`.annotate({op:r,a:t,visit:D}),xt=dt("NOT"),J=r=>t=>c`(${u(t)} ${r})`.annotate({op:r,a:t,visit:D}),$t=J("IS NULL"),Et=J("IS NOT NULL"),w=r=>(t,e)=>c`(${u(t)} ${r} ${u(e)})`.annotate({op:r,a:t,b:e,visit:D}),yt=w("="),St=w("<>"),Nt=w("<"),wt=w(">"),At=w("<="),Tt=w(">="),Rt=w("IS DISTINCT FROM"),qt=w("IS NOT DISTINCT FROM");function Z(r,t,e,n){t=u(t);let o=r.startsWith("NOT ")?"NOT ":"";return(e?n?c`${o}(${e[0]} <= ${t} AND ${t} < ${e[1]})`:c`(${t} ${r} ${e[0]} AND ${e[1]})`:c``).annotate({op:r,visit:D,field:t,range:e})}var bt=(r,t,e)=>Z("BETWEEN",r,t,e),It=(r,t,e)=>Z("NOT BETWEEN",r,t,e);function R(r,t){return Array.from({length:r},()=>t)}function m(r,t){return(...e)=>{let n=e.map(u),o=t?`::${t}`:"";return(n.length?c([`${r}(`,...R(n.length-1,", "),`)${o}`],...n):c`${r}()${o}`).annotate({func:r,args:n})}}var Lt=m("REGEXP_MATCHES"),Ot=m("CONTAINS"),Dt=m("PREFIX"),Ct=m("SUFFIX"),Mt=m("LOWER"),_t=m("UPPER"),Pt=m("LENGTH"),Ft=m("ISNAN"),Gt=m("ISFINITE"),Ut=m("ISINF");var C=class r extends N{constructor(t,e,n,o,s="",i="",a=""){let p;if(o&&!(s||i||a))p=o?c`${e} OVER "${o}"`:c`${e} OVER ()`;else{let b=s&&i?" ":"",I=(s||i)&&a?" ":"";p=c`${e} OVER (${o?`"${o}" `:""}${s}${b}${i}${I}${a})`}n&&(p=c`(${p})::${n}`);let{_expr:y,_deps:A}=p;super(y,A,{window:t,func:e,type:n,name:o,group:s,order:i,frame:a})}get basis(){return this.column}get label(){let{func:t}=this;return t.label??t.toString()}over(t){let{window:e,func:n,type:o,group:s,order:i,frame:a}=this;return new r(e,n,o,t,s,i,a)}partitionby(...t){let e=t.flat().filter(y=>y).map(u),n=c(["PARTITION BY ",R(e.length-1,", "),""],...e),{window:o,func:s,type:i,name:a,order:p,frame:d}=this;return new r(o,s,i,a,n,p,d)}orderby(...t){let e=t.flat().filter(y=>y).map(u),n=c(["ORDER BY ",R(e.length-1,", "),""],...e),{window:o,func:s,type:i,name:a,group:p,frame:d}=this;return new r(o,s,i,a,p,n,d)}rows(t){let e=tt("ROWS",t),{window:n,func:o,type:s,name:i,group:a,order:p}=this;return new r(n,o,s,i,a,p,e)}range(t){let e=tt("RANGE",t),{window:n,func:o,type:s,name:i,group:a,order:p}=this;return new r(n,o,s,i,a,p,e)}};function tt(r,t){if(T(t)){let e=c`${t}`;return e.toString=()=>`${r} ${et(t.value)}`,e}return`${r} ${et(t)}`}function et(r){let[t,e]=r,n=t===0?"CURRENT ROW":Number.isFinite(t)?`${Math.abs(t)} PRECEDING`:"UNBOUNDED PRECEDING",o=e===0?"CURRENT ROW":Number.isFinite(e)?`${Math.abs(e)} FOLLOWING`:"UNBOUNDED FOLLOWING";return`BETWEEN ${n} AND ${o}`}function E(r,t){return(...e)=>{let n=m(r)(...e);return new C(r,n,t)}}var jt=E("ROW_NUMBER","INTEGER"),Wt=E("RANK","INTEGER"),Yt=E("DENSE_RANK","INTEGER"),Bt=E("PERCENT_RANK"),Xt=E("CUME_DIST"),vt=E("NTILE"),Qt=E("LAG"),Vt=E("LEAD"),kt=E("FIRST_VALUE"),Ht=E("LAST_VALUE"),Kt=E("NTH_VALUE");function zt(r,...t){return c(r,...t).annotate({aggregate:!0})}var Y=class r extends N{constructor(t,e,n,o,s){e=(e||[]).map(u);let{strings:i,exprs:a}=Jt(t,e,n,o,s),{spans:p,cols:d}=W(i,a);super(p,d,{aggregate:t,args:e,type:n,isDistinct:o,filter:s})}get basis(){return this.column}get label(){let{aggregate:t,args:e,isDistinct:n}=this,o=n?"DISTINCT"+(e.length?" ":""):"",s=e.length?`(${o}${e.map(Zt).join(", ")})`:"";return`${t.toLowerCase()}${s}`}distinct(){let{aggregate:t,args:e,type:n,filter:o}=this;return new r(t,e,n,!0,o)}where(t){let{aggregate:e,args:n,type:o,isDistinct:s}=this;return new r(e,n,o,s,t)}window(){let{aggregate:t,args:e,type:n,isDistinct:o}=this,s=new r(t,e,null,o);return new C(t,s,n)}partitionby(...t){return this.window().partitionby(...t)}orderby(...t){return this.window().orderby(...t)}rows(t,e){return this.window().rows(t,e)}range(t,e){return this.window().range(t,e)}};function Jt(r,t,e,n,o){let s=`)${e?`::${e}`:""}`,i=[`${r}(${n?"DISTINCT ":""}`],a=[];return t.length?(i=i.concat([...R(t.length-1,", "),`${s}${o?" FILTER (WHERE ":""}`,...o?[")"]:[]]),a=[...t,...o?[o]:[]]):i[0]+="*"+s,{exprs:a,strings:i}}function Zt(r){let t=$(r);return t&&t.startsWith('"')&&t.endsWith('"')?t.slice(1,-1):t}function l(r,t){return(...e)=>new Y(r,e,t)}var te=l("COUNT","INTEGER"),ee=l("AVG"),re=l("AVG"),ne=l("MAD"),oe=l("MAX"),se=l("MIN"),ie=l("SUM","DOUBLE"),ce=l("PRODUCT"),ae=l("MEDIAN"),ue=l("QUANTILE"),le=l("MODE"),pe=l("VARIANCE"),fe=l("STDDEV"),he=l("SKEWNESS"),me=l("KURTOSIS"),ge=l("ENTROPY"),de=l("VAR_POP"),xe=l("STDDEV_POP"),$e=l("CORR"),Ee=l("COVAR_POP"),ye=l("REGR_INTERCEPT"),Se=l("REGR_SLOPE"),Ne=l("REGR_COUNT"),we=l("REGR_R2"),Ae=l("REGR_SYY"),Te=l("REGR_SXX"),Re=l("REGR_SXY"),qe=l("REGR_AVGX"),be=l("REGR_AVGY"),Ie=l("FIRST"),Le=l("LAST"),Oe=l("ARG_MIN"),De=l("ARG_MAX"),Ce=l("STRING_AGG"),Me=l("ARRAY_AGG");function B(r,t){let e=u(r),n=c`CAST(${e} AS ${t})`;return Object.defineProperty(n,"label",{enumerable:!0,get(){return r.label}}),Object.defineProperty(n,"aggregate",{enumerable:!0,get(){return r.aggregate||!1}}),n}var _e=r=>B(r,"DOUBLE"),Pe=r=>B(r,"INTEGER");var X=r=>c`epoch_ms(${u(r)})`,Fe=r=>{let t=u(r);return c`MAKE_DATE(2012, MONTH(${t}), 1)`.annotate({label:"month"})},Ge=r=>{let t=u(r);return c`MAKE_DATE(2012, MONTH(${t}), DAY(${t}))`.annotate({label:"date"})},Ue=r=>{let t=u(r);return c`MAKE_DATE(2012, 1, DAY(${t}))`.annotate({label:"date"})};var je=m("ST_AsGeoJSON"),rt=m("ST_X"),nt=m("ST_Y"),v=m("ST_CENTROID"),We=r=>rt(v(r)),Ye=r=>nt(v(r));var F=class r{static select(...t){return new r().select(...t)}static from(...t){return new r().from(...t)}static with(...t){return new r().with(...t)}static union(...t){return new q("UNION",t.flat())}static unionAll(...t){return new q("UNION ALL",t.flat())}static intersect(...t){return new q("INTERSECT",t.flat())}static except(...t){return new q("EXCEPT",t.flat())}static describe(t){let e=t.clone(),{clone:n,toString:o}=e;return Object.assign(e,{describe:!0,clone:()=>r.describe(n.call(e)),toString:()=>`DESCRIBE ${o.call(e)}`})}constructor(){this.query={with:[],select:[],from:[],where:[],groupby:[],having:[],window:[],qualify:[],orderby:[]}}clone(){let t=new r;return t.query={...this.query},t}with(...t){let{query:e}=this;if(t.length===0)return e.with;{let n=[],o=(s,i)=>{let a=i.clone();a.cteFor=this,n.push({as:s,query:a})};return t.flat().forEach(s=>{if(s!=null)if(s.as&&s.query)o(s.as,s.query);else for(let i in s)o(i,s[i])}),e.with=e.with.concat(n),this}}select(...t){let{query:e}=this;if(t.length===0)return e.select;{let n=[];for(let o of t.flat())if(o!=null)if(typeof o=="string")n.push({as:o,expr:u(o)});else if(o instanceof S)n.push({as:o.column,expr:o});else if(Array.isArray(o))n.push({as:o[0],expr:o[1]});else for(let s in o)n.push({as:P(s),expr:u(o[s])});return e.select=e.select.concat(n),this}}$select(...t){return this.query.select=[],this.select(...t)}distinct(t=!0){return this.query.distinct=!!t,this}from(...t){let{query:e}=this;if(t.length===0)return e.from;{let n=[];return t.flat().forEach(o=>{if(o!=null)if(typeof o=="string")n.push({as:o,from:O(o)});else if(o instanceof S)n.push({as:o.table,from:o});else if(M(o)||_(o))n.push({from:o});else if(Array.isArray(o))n.push({as:P(o[0]),from:O(o[1])});else for(let s in o)n.push({as:P(s),from:O(o[s])})}),e.from=e.from.concat(n),this}}$from(...t){return this.query.from=[],this.from(...t)}sample(t,e){let{query:n}=this;if(arguments.length===0)return n.sample;{let o=t;return typeof t=="number"&&(o=t>0&&t<1?{perc:100*t,method:e}:{rows:Math.round(t),method:e}),n.sample=o,this}}where(...t){let{query:e}=this;return t.length===0?e.where:(e.where=e.where.concat(t.flat().filter(n=>n)),this)}$where(...t){return this.query.where=[],this.where(...t)}groupby(...t){let{query:e}=this;return t.length===0?e.groupby:(e.groupby=e.groupby.concat(t.flat().filter(n=>n).map(u)),this)}$groupby(...t){return this.query.groupby=[],this.groupby(...t)}having(...t){let{query:e}=this;return t.length===0?e.having:(e.having=e.having.concat(t.flat().filter(n=>n)),this)}window(...t){let{query:e}=this;if(t.length===0)return e.window;{let n=[];return t.flat().forEach(o=>{if(o!=null)for(let s in o)n.push({as:P(s),expr:o[s]})}),e.window=e.window.concat(n),this}}qualify(...t){let{query:e}=this;return t.length===0?e.qualify:(e.qualify=e.qualify.concat(t.flat().filter(n=>n)),this)}orderby(...t){let{query:e}=this;return t.length===0?e.orderby:(e.orderby=e.orderby.concat(t.flat().filter(n=>n).map(u)),this)}limit(t){let{query:e}=this;return arguments.length===0?e.limit:(e.limit=Number.isFinite(t)?t:void 0,this)}offset(t){let{query:e}=this;return arguments.length===0?e.offset:(e.offset=Number.isFinite(t)?t:void 0,this)}get subqueries(){let{query:t,cteFor:e}=this,o=(e?.query||t).with?.reduce((i,{as:a,query:p})=>(i[a]=p,i),{}),s=[];return t.from.forEach(({from:i})=>{if(M(i))s.push(i);else if(o[i.table]){let a=o[i.table];s.push(a)}}),s}toString(){let{with:t,select:e,distinct:n,from:o,sample:s,where:i,groupby:a,having:p,window:d,qualify:y,orderby:A,limit:b,offset:I}=this.query,g=[];if(t.length){let f=t.map(({as:h,query:x})=>`"${h}" AS (${x})`);g.push(`WITH ${f.join(", ")}`)}let it=e.map(({as:f,expr:h})=>k(h,f)&&!h.table?`${h}`:`${h} AS "${f}"`);if(g.push(`SELECT${n?" DISTINCT":""} ${it.join(", ")}`),o.length){let f=o.map(({as:h,from:x})=>{let L=M(x)?`(${x})`:`${x}`;return!h||h===x.table?L:`${L} AS "${h}"`});g.push(`FROM ${f.join(", ")}`)}if(i.length){let f=i.map(String).filter(h=>h).join(" AND ");f&&g.push(`WHERE ${f}`)}if(s){let{rows:f,perc:h,method:x,seed:L}=s,ct=f?`${f} ROWS`:`${h} PERCENT`,at=x?` (${x}${L!=null?`, ${L}`:""})`:"";g.push(`USING SAMPLE ${ct}${at}`)}if(a.length&&g.push(`GROUP BY ${a.join(", ")}`),p.length){let f=p.map(String).filter(h=>h).join(" AND ");f&&g.push(`HAVING ${f}`)}if(d.length){let f=d.map(({as:h,expr:x})=>`"${h}" AS (${x})`);g.push(`WINDOW ${f.join(", ")}`)}if(y.length){let f=y.map(String).filter(h=>h).join(" AND ");f&&g.push(`QUALIFY ${f}`)}return A.length&&g.push(`ORDER BY ${A.join(", ")}`),Number.isFinite(b)&&g.push(`LIMIT ${b}`),Number.isFinite(I)&&g.push(`OFFSET ${I}`),g.join(" ")}},q=class r{constructor(t,e){this.op=t,this.queries=e.map(n=>n.clone()),this.query={orderby:[]}}clone(){let t=new r(this.op,this.queries);return t.query={...this.query},t}orderby(...t){let{query:e}=this;return t.length===0?e.orderby:(e.orderby=e.orderby.concat(t.flat().filter(n=>n).map(u)),this)}limit(t){let{query:e}=this;return arguments.length===0?e.limit:(e.limit=Number.isFinite(t)?t:void 0,this)}offset(t){let{query:e}=this;return arguments.length===0?e.offset:(e.offset=Number.isFinite(t)?t:void 0,this)}get subqueries(){let{queries:t,cteFor:e}=this;return e&&t.forEach(n=>n.cteFor=e),t}toString(){let{op:t,queries:e,query:{orderby:n,limit:o,offset:s}}=this,i=[e.join(` ${t} `)];return n.length&&i.push(`ORDER BY ${n.join(", ")}`),Number.isFinite(o)&&i.push(`LIMIT ${o}`),Number.isFinite(s)&&i.push(`OFFSET ${s}`),i.join(" ")}};function M(r){return r instanceof F||r instanceof q}function Be(r){return M(r)&&r.describe}function P(r){return Xe(r)?r.slice(1,-1):r}function Xe(r){return r[0]==='"'&&r[r.length-1]==='"'}var G=r=>r;function ve(){return{apply:G,invert:G,sqlApply:u,sqlInvert:G}}function Qe({base:r}={}){if(r==null||r===Math.E)return{apply:Math.log,invert:Math.exp,sqlApply:t=>c`LN(${u(t)})`,sqlInvert:t=>c`EXP(${t})`};if(r===10)return{apply:Math.log10,invert:t=>Math.pow(10,t),sqlApply:t=>c`LOG(${u(t)})`,sqlInvert:t=>c`POW(10, ${t})`};{let t=+r;return{apply:e=>Math.log(e)/Math.log(t),invert:e=>Math.pow(t,e),sqlApply:e=>c`LN(${u(e)}) / LN(${t})`,sqlInvert:e=>c`POW(${t}, ${e})`}}}function Ve({constant:r=1}={}){let t=+r;return{apply:e=>Math.sign(e)*Math.log1p(Math.abs(e)),invert:e=>Math.sign(e)*Math.exp(Math.abs(e)-t),sqlApply:e=>(e=u(e),c`SIGN(${e}) * LN(${t} + ABS(${e}))`),sqlInvert:e=>c`SIGN(${e}) * (EXP(ABS(${e})) - ${t})`}}function ke(){return{apply:r=>Math.sign(r)*Math.sqrt(Math.abs(r)),invert:r=>Math.sign(r)*r*r,sqlApply:r=>(r=u(r),c`SIGN(${r}) * SQRT(ABS(${r}))`),sqlInvert:r=>c`SIGN(${r}) * (${r}) ** 2`}}function He({exponent:r=1}={}){let t=+r;return{apply:e=>Math.sign(e)*Math.pow(Math.abs(e),t),invert:e=>Math.sign(e)*Math.pow(Math.abs(e),1/t),sqlApply:e=>(e=u(e),c`SIGN(${e}) * POW(ABS(${e}), ${t})`),sqlInvert:e=>c`SIGN(${e}) * POW(ABS(${e}), 1/${t})`}}function ot(){return{apply:r=>+r,invert:r=>new Date(r),sqlApply:r=>r instanceof Date?+r:X(u(r)),sqlInvert:G}}var Ke={linear:ve,log:Qe,symlog:Ve,sqrt:ke,pow:He,time:ot,utc:ot};function ze(r){let t=Ke[r.type];return t?{...r,...t(r)}:null}function U(r,t,{replace:e=!1,temp:n=!0,view:o=!1}={}){return"CREATE"+(e?" OR REPLACE ":" ")+(n?"TEMP ":"")+(o?"VIEW":"TABLE")+(e?" ":" IF NOT EXISTS ")+r+" AS "+t}function Je(r){return`INSTALL ${r}; LOAD ${r}`}function st(r,{columns:t=Object.keys(r?.[0]||{})}={}){let e=[];if(Array.isArray(t)?(e=t,t=e.reduce((o,s)=>(o[s]=s,o),{})):t&&(e=Object.keys(t)),!e.length)throw new Error("Can not create table from empty column set.");let n=[];for(let o of r){let s=e.map(i=>`${$(o[i])} AS "${t[i]}"`);n.push(`(SELECT ${s.join(", ")})`)}return n.join(" UNION ALL ")}function j(r,t,e,n={},o={}){let{select:s=["*"],where:i,view:a,temp:p,replace:d,...y}=n,A=or({...o,...y}),b=`${r}('${e}'${A?", "+A:""})`,I=i?` WHERE ${i}`:"",g=`SELECT ${s.join(", ")} FROM ${b}${I}`;return U(t,g,{view:a,temp:p,replace:d})}function Ze(r,t,e){return j("read_csv",r,t,e,{auto_detect:!0,sample_size:-1})}function tr(r,t,e){return j("read_json",r,t,e,{auto_detect:!0,json_format:"auto"})}function er(r,t,e){return j("read_parquet",r,t,e)}function rr(r,t,e={}){let{options:n,...o}=e;if(n){let s=Array.isArray(n)?n.join(", "):typeof n=="string"?n:Object.entries(n).map(([i,a])=>`${i}=${a}`).join(", ");o.open_options=s.toUpperCase()}return j("st_read",r,t,o)}function nr(r,t,e={}){let{select:n=["*"],...o}=e,s=st(t),i=n.length===1&&n[0]==="*"?s:`SELECT ${n} FROM ${s}`;return U(r,i,o)}function or(r){return Object.entries(r).map(([t,e])=>`${t}=${Q(e)}`).join(", ")}function Q(r){switch(typeof r){case"boolean":return String(r);case"string":return`'${r}'`;case"undefined":case"object":return r==null?"NULL":Array.isArray(r)?"["+r.map(t=>Q(t)).join(", ")+"]":"{"+Object.entries(r).map(([t,e])=>`'${t}': ${Q(e)}`).join(", ")+"}";default:return r}}export{F as Query,S as Ref,zt as agg,ut as all,mt as and,De as argmax,Oe as argmin,Me as arrayAgg,u as asColumn,O as asRelation,ee as avg,B as cast,_e as castDouble,Pe as castInteger,v as centroid,We as centroidX,Ye as centroidY,K as column,Ot as contains,$e as corr,te as count,Ee as covarPop,U as create,Xt as cume_dist,Ue as dateDay,Fe as dateMonth,Ge as dateMonthDay,Yt as dense_rank,ft as desc,ge as entropy,X as epoch_ms,yt as eq,Ie as first,kt as first_value,je as geojson,wt as gt,Tt as gte,bt as isBetween,Be as isDescribeQuery,Rt as isDistinct,Gt as isFinite,Ut as isInfinite,Ft as isNaN,It as isNotBetween,qt as isNotDistinct,Et as isNotNull,$t as isNull,T as isParamLike,M as isQuery,_ as isSQLExpression,me as kurtosis,Qt as lag,Le as last,Ht as last_value,Vt as lead,Pt as length,ht as literal,$ as literalToSQL,Ze as loadCSV,Je as loadExtension,tr as loadJSON,nr as loadObjects,er as loadParquet,rr as loadSpatial,Mt as lower,Nt as lt,At as lte,ne as mad,oe as max,re as mean,ae as median,se as min,le as mode,St as neq,xt as not,Kt as nth_value,vt as ntile,gt as or,Bt as percent_rank,Dt as prefix,ce as product,ue as quantile,Wt as rank,Lt as regexp_matches,qe as regrAvgX,be as regrAvgY,Ne as regrCount,ye as regrIntercept,we as regrR2,Te as regrSXX,Re as regrSXY,Ae as regrSYY,Se as regrSlope,H as relation,jt as row_number,ze as scaleTransform,he as skewness,c as sql,fe as stddev,xe as stddevPop,Ce as stringAgg,Ct as suffix,ie as sum,lt as toSQL,_t as upper,de as varPop,pe as variance,rt as x,nt as y};
+var S=class{constructor(t,e){t&&(this.table=String(t)),e&&(this.column=e)}get columns(){return this.column?[this.column]:[]}toString(){let{table:t,column:e}=this;if(e){let n=e.startsWith("*")?e:`"${e}"`;return`${t?`${V(t)}.`:""}${n}`}else return t?V(t):"NULL"}};function V(r){return r.split(".").map(e=>`"${e}"`).join(".")}function k(r,t){return r instanceof S&&r.column===t}function u(r){return typeof r=="string"?K(r):r}function O(r){return typeof r=="string"?H(r):r}function H(r){return new S(r)}function K(r,t=null){return arguments.length===1&&(t=r,r=null),new S(r,t)}function lt(r){return new S(r,"*")}function pt(r){return typeof r=="string"?`"${r}"`:$(r)}function $(r){switch(typeof r){case"boolean":return r?"TRUE":"FALSE";case"string":return`'${r.replace("'","''")}'`;case"number":return Number.isFinite(r)?String(r):"NULL";default:if(r==null)return"NULL";if(r instanceof Date){let t=+r;if(Number.isNaN(t))return"NULL";let e=r.getUTCFullYear(),n=r.getUTCMonth(),o=r.getUTCDate();return t===Date.UTC(e,n,o)?`MAKE_DATE(${e}, ${n+1}, ${o})`:`EPOCH_MS(${t})`}else return r instanceof RegExp?`'${r.source}'`:String(r)}}var R=r=>typeof r?.addEventListener=="function";function _(r){return r instanceof N}var N=class{constructor(t,e,n){this._expr=Array.isArray(t)?t:[t],this._deps=e||[],this.annotate(n);let o=this._expr.filter(s=>R(s));o.length>0?(this._params=Array.from(new Set(o)),this._params.forEach(s=>{s.addEventListener("value",()=>ft(this,this.map?.get("value")))})):this.addEventListener=void 0}get value(){return this}get columns(){let{_params:t,_deps:e}=this;if(t){let n=new Set(t.flatMap(o=>{let s=o.value?.columns;return Array.isArray(s)?s:[]}));if(n.size){let o=new Set(e);return n.forEach(s=>o.add(s)),Array.from(o)}}return e}get column(){return this._deps.length?this._deps[0]:this.columns[0]}annotate(...t){return Object.assign(this,...t)}toString(){return this._expr.map(t=>R(t)&&!_(t)?$(t.value):t).join("")}addEventListener(t,e){let n=this.map||(this.map=new Map);(n.get(t)||(n.set(t,new Set),n.get(t))).add(e)}};function ft(r,t){if(t?.size)return Promise.allSettled(Array.from(t,e=>e(r)))}function W(r,t){let e=[r[0]],n=new Set,o=t.length;for(let s=0,i=0;s<o;){let a=t[s];R(a)?e[++i]=a:(Array.isArray(a?.columns)&&a.columns.forEach(d=>n.add(d)),e[i]+=typeof a=="string"?a:$(a));let p=r[++s];R(e[i])?e[++i]=p:e[i]+=p}return{spans:e,cols:Array.from(n)}}function c(r,...t){let{spans:e,cols:n}=W(r,t);return new N(e,n)}function ht(r){let t=u(r);return c`${t} DESC NULLS LAST`.annotate({label:t?.label,desc:!0})}var mt=r=>({value:r,toString:()=>$(r)});function D(r){r(this.op,this),this.children?.forEach(t=>t.visit(r))}function z(r,t){let e=t.filter(o=>o!=null).map(u),n=e.map((o,s)=>s?` ${r} `:"");return e.length===1?n.push(""):e.length>1&&(n[0]="(",n.push(")")),c(n,...e).annotate({op:r,children:e,visit:D})}var gt=(...r)=>z("AND",r.flat()),dt=(...r)=>z("OR",r.flat()),xt=r=>t=>c`(${r} ${u(t)})`.annotate({op:r,a:t,visit:D}),$t=xt("NOT"),J=r=>t=>c`(${u(t)} ${r})`.annotate({op:r,a:t,visit:D}),Et=J("IS NULL"),yt=J("IS NOT NULL"),w=r=>(t,e)=>c`(${u(t)} ${r} ${u(e)})`.annotate({op:r,a:t,b:e,visit:D}),St=w("="),Nt=w("<>"),wt=w("<"),At=w(">"),Rt=w("<="),Tt=w(">="),qt=w("IS DISTINCT FROM"),bt=w("IS NOT DISTINCT FROM");function Z(r,t,e,n){t=u(t);let o=r.startsWith("NOT ")?"NOT ":"";return(e?n?c`${o}(${e[0]} <= ${t} AND ${t} < ${e[1]})`:c`(${t} ${r} ${e[0]} AND ${e[1]})`:c``).annotate({op:r,visit:D,field:t,range:e})}var It=(r,t,e)=>Z("BETWEEN",r,t,e),Lt=(r,t,e)=>Z("NOT BETWEEN",r,t,e);function T(r,t){return Array.from({length:r},()=>t)}function m(r,t){return(...e)=>{let n=e.map(u),o=t?`::${t}`:"";return(n.length?c([`${r}(`,...T(n.length-1,", "),`)${o}`],...n):c`${r}()${o}`).annotate({func:r,args:n})}}var Ot=m("REGEXP_MATCHES"),Dt=m("CONTAINS"),Ct=m("PREFIX"),Mt=m("SUFFIX"),_t=m("LOWER"),Pt=m("UPPER"),Ft=m("LENGTH"),Gt=m("ISNAN"),jt=m("ISFINITE"),Ut=m("ISINF");var C=class r extends N{constructor(t,e,n,o,s="",i="",a=""){let p;if(o&&!(s||i||a))p=o?c`${e} OVER "${o}"`:c`${e} OVER ()`;else{let b=s&&i?" ":"",I=(s||i)&&a?" ":"";p=c`${e} OVER (${o?`"${o}" `:""}${s}${b}${i}${I}${a})`}n&&(p=c`(${p})::${n}`);let{_expr:y,_deps:A}=p;super(y,A),this.window=t,this.func=e,this.type=n,this.name=o,this.group=s,this.order=i,this.frame=a}get basis(){return this.column}get label(){let{func:t}=this;return t.label??t.toString()}over(t){let{window:e,func:n,type:o,group:s,order:i,frame:a}=this;return new r(e,n,o,t,s,i,a)}partitionby(...t){let e=t.flat().filter(y=>y).map(u),n=c(["PARTITION BY ",T(e.length-1,", "),""],...e),{window:o,func:s,type:i,name:a,order:p,frame:d}=this;return new r(o,s,i,a,n,p,d)}orderby(...t){let e=t.flat().filter(y=>y).map(u),n=c(["ORDER BY ",T(e.length-1,", "),""],...e),{window:o,func:s,type:i,name:a,group:p,frame:d}=this;return new r(o,s,i,a,p,n,d)}rows(t){let e=tt("ROWS",t),{window:n,func:o,type:s,name:i,group:a,order:p}=this;return new r(n,o,s,i,a,p,e)}range(t){let e=tt("RANGE",t),{window:n,func:o,type:s,name:i,group:a,order:p}=this;return new r(n,o,s,i,a,p,e)}};function tt(r,t){if(R(t)){let e=c`${t}`;return e.toString=()=>`${r} ${et(t.value)}`,e}return`${r} ${et(t)}`}function et(r){let[t,e]=r,n=t===0?"CURRENT ROW":Number.isFinite(t)?`${Math.abs(t)} PRECEDING`:"UNBOUNDED PRECEDING",o=e===0?"CURRENT ROW":Number.isFinite(e)?`${Math.abs(e)} FOLLOWING`:"UNBOUNDED FOLLOWING";return`BETWEEN ${n} AND ${o}`}function E(r,t){return(...e)=>{let n=m(r)(...e);return new C(r,n,t)}}var Wt=E("ROW_NUMBER","INTEGER"),Yt=E("RANK","INTEGER"),vt=E("DENSE_RANK","INTEGER"),Bt=E("PERCENT_RANK"),Qt=E("CUME_DIST"),Xt=E("NTILE"),Vt=E("LAG"),kt=E("LEAD"),Ht=E("FIRST_VALUE"),Kt=E("LAST_VALUE"),zt=E("NTH_VALUE");function Jt(r,...t){return c(r,...t).annotate({aggregate:!0})}var Y=class r extends N{constructor(t,e,n,o,s){e=(e||[]).map(u);let{strings:i,exprs:a}=Zt(t,e,n,o,s),{spans:p,cols:d}=W(i,a);super(p,d),this.aggregate=t,this.args=e,this.type=n,this.isDistinct=o,this.filter=s}get basis(){return this.column}get label(){let{aggregate:t,args:e,isDistinct:n}=this,o=n?"DISTINCT"+(e.length?" ":""):"",s=e.length?`(${o}${e.map(te).join(", ")})`:"";return`${t.toLowerCase()}${s}`}distinct(){let{aggregate:t,args:e,type:n,filter:o}=this;return new r(t,e,n,!0,o)}where(t){let{aggregate:e,args:n,type:o,isDistinct:s}=this;return new r(e,n,o,s,t)}window(){let{aggregate:t,args:e,type:n,isDistinct:o}=this,s=new r(t,e,null,o);return new C(t,s,n)}partitionby(...t){return this.window().partitionby(...t)}orderby(...t){return this.window().orderby(...t)}rows(t){return this.window().rows(t)}range(t){return this.window().range(t)}};function Zt(r,t,e,n,o){let s=`)${e?`::${e}`:""}`,i=[`${r}(${n?"DISTINCT ":""}`],a=[];return t.length?(i=i.concat([...T(t.length-1,", "),`${s}${o?" FILTER (WHERE ":""}`,...o?[")"]:[]]),a=[...t,...o?[o]:[]]):i[0]+="*"+s,{exprs:a,strings:i}}function te(r){let t=$(r);return t&&t.startsWith('"')&&t.endsWith('"')?t.slice(1,-1):t}function l(r,t){return(...e)=>new Y(r,e,t)}var ee=l("COUNT","INTEGER"),re=l("AVG"),ne=l("AVG"),oe=l("MAD"),se=l("MAX"),ie=l("MIN"),ce=l("SUM","DOUBLE"),ae=l("PRODUCT"),ue=l("MEDIAN"),le=l("QUANTILE"),pe=l("MODE"),fe=l("VARIANCE"),he=l("STDDEV"),me=l("SKEWNESS"),ge=l("KURTOSIS"),de=l("ENTROPY"),xe=l("VAR_POP"),$e=l("STDDEV_POP"),Ee=l("CORR"),ye=l("COVAR_SAMP"),Se=l("COVAR_POP"),Ne=l("REGR_INTERCEPT"),we=l("REGR_SLOPE"),Ae=l("REGR_COUNT"),Re=l("REGR_R2"),Te=l("REGR_SYY"),qe=l("REGR_SXX"),be=l("REGR_SXY"),Ie=l("REGR_AVGX"),Le=l("REGR_AVGY"),Oe=l("FIRST"),De=l("LAST"),Ce=l("ARG_MIN"),Me=l("ARG_MAX"),_e=l("STRING_AGG"),Pe=l("ARRAY_AGG");function v(r,t){let e=u(r),n=c`CAST(${e} AS ${t})`;return Object.defineProperty(n,"label",{enumerable:!0,get(){return r.label}}),Object.defineProperty(n,"aggregate",{enumerable:!0,get(){return r.aggregate||!1}}),n}var Fe=r=>v(r,"DOUBLE"),Ge=r=>v(r,"INTEGER");var B=r=>c`epoch_ms(${u(r)})`,je=r=>{let t=u(r);return c`MAKE_DATE(2012, MONTH(${t}), 1)`.annotate({label:"month"})},Ue=r=>{let t=u(r);return c`MAKE_DATE(2012, MONTH(${t}), DAY(${t}))`.annotate({label:"date"})},We=r=>{let t=u(r);return c`MAKE_DATE(2012, 1, DAY(${t}))`.annotate({label:"date"})};var Ye=m("ST_AsGeoJSON"),rt=m("ST_X"),nt=m("ST_Y"),Q=m("ST_CENTROID"),ve=r=>rt(Q(r)),Be=r=>nt(Q(r));var F=class r{static select(...t){return new r().select(...t)}static from(...t){return new r().from(...t)}static with(...t){return new r().with(...t)}static union(...t){return new q("UNION",t.flat())}static unionAll(...t){return new q("UNION ALL",t.flat())}static intersect(...t){return new q("INTERSECT",t.flat())}static except(...t){return new q("EXCEPT",t.flat())}static describe(t){let e=t.clone(),{clone:n,toString:o}=e;return Object.assign(e,{describe:!0,clone:()=>r.describe(n.call(e)),toString:()=>`DESCRIBE ${o.call(e)}`})}constructor(){this.query={with:[],select:[],from:[],where:[],groupby:[],having:[],window:[],qualify:[],orderby:[]},this.cteFor=null}clone(){let t=new r;return t.query={...this.query},t}with(...t){let{query:e}=this;if(t.length===0)return e.with;{let n=[],o=(s,i)=>{let a=i.clone();a.cteFor=this,n.push({as:s,query:a})};return t.flat().forEach(s=>{if(s!=null)if(s.as&&s.query)o(s.as,s.query);else for(let i in s)o(i,s[i])}),e.with=e.with.concat(n),this}}select(...t){let{query:e}=this;if(t.length===0)return e.select;{let n=[];for(let s of t.flat())if(s!=null)if(typeof s=="string")n.push({as:s,expr:u(s)});else if(s instanceof S)n.push({as:s.column,expr:s});else if(Array.isArray(s))n.push({as:s[0],expr:s[1]});else for(let i in s)n.push({as:P(i),expr:u(s[i])});let o=new Set(n.map(s=>s.as));return e.select=e.select.filter(s=>!o.has(s.as)).concat(n.filter(s=>s.expr)),this}}$select(...t){return this.query.select=[],this.select(...t)}distinct(t=!0){return this.query.distinct=!!t,this}from(...t){let{query:e}=this;if(t.length===0)return e.from;{let n=[];return t.flat().forEach(o=>{if(o!=null)if(typeof o=="string")n.push({as:o,from:O(o)});else if(o instanceof S)n.push({as:o.table,from:o});else if(M(o)||_(o))n.push({from:o});else if(Array.isArray(o))n.push({as:P(o[0]),from:O(o[1])});else for(let s in o)n.push({as:P(s),from:O(o[s])})}),e.from=e.from.concat(n),this}}$from(...t){return this.query.from=[],this.from(...t)}sample(t,e){let{query:n}=this;if(arguments.length===0)return n.sample;{let o=t;return typeof t=="number"&&(o=t>0&&t<1?{perc:100*t,method:e}:{rows:Math.round(t),method:e}),n.sample=o,this}}where(...t){let{query:e}=this;return t.length===0?e.where:(e.where=e.where.concat(t.flat().filter(n=>n)),this)}$where(...t){return this.query.where=[],this.where(...t)}groupby(...t){let{query:e}=this;return t.length===0?e.groupby:(e.groupby=e.groupby.concat(t.flat().filter(n=>n).map(u)),this)}$groupby(...t){return this.query.groupby=[],this.groupby(...t)}having(...t){let{query:e}=this;return t.length===0?e.having:(e.having=e.having.concat(t.flat().filter(n=>n)),this)}window(...t){let{query:e}=this;if(t.length===0)return e.window;{let n=[];return t.flat().forEach(o=>{if(o!=null)for(let s in o)n.push({as:P(s),expr:o[s]})}),e.window=e.window.concat(n),this}}qualify(...t){let{query:e}=this;return t.length===0?e.qualify:(e.qualify=e.qualify.concat(t.flat().filter(n=>n)),this)}orderby(...t){let{query:e}=this;return t.length===0?e.orderby:(e.orderby=e.orderby.concat(t.flat().filter(n=>n).map(u)),this)}limit(t){let{query:e}=this;return arguments.length===0?e.limit:(e.limit=Number.isFinite(t)?t:void 0,this)}offset(t){let{query:e}=this;return arguments.length===0?e.offset:(e.offset=Number.isFinite(t)?t:void 0,this)}get subqueries(){let{query:t,cteFor:e}=this,o=(e?.query||t).with?.reduce((i,{as:a,query:p})=>(i[a]=p,i),{}),s=[];return t.from.forEach(({from:i})=>{if(M(i))s.push(i);else if(o[i.table]){let a=o[i.table];s.push(a)}}),s}toString(){let{with:t,select:e,distinct:n,from:o,sample:s,where:i,groupby:a,having:p,window:d,qualify:y,orderby:A,limit:b,offset:I}=this.query,g=[];if(t.length){let f=t.map(({as:h,query:x})=>`"${h}" AS (${x})`);g.push(`WITH ${f.join(", ")}`)}let ct=e.map(({as:f,expr:h})=>k(h,f)&&!h.table?`${h}`:`${h} AS "${f}"`);if(g.push(`SELECT${n?" DISTINCT":""} ${ct.join(", ")}`),o.length){let f=o.map(({as:h,from:x})=>{let L=M(x)?`(${x})`:`${x}`;return!h||h===x.table?L:`${L} AS "${h}"`});g.push(`FROM ${f.join(", ")}`)}if(i.length){let f=i.map(String).filter(h=>h).join(" AND ");f&&g.push(`WHERE ${f}`)}if(s){let{rows:f,perc:h,method:x,seed:L}=s,at=f?`${f} ROWS`:`${h} PERCENT`,ut=x?` (${x}${L!=null?`, ${L}`:""})`:"";g.push(`USING SAMPLE ${at}${ut}`)}if(a.length&&g.push(`GROUP BY ${a.join(", ")}`),p.length){let f=p.map(String).filter(h=>h).join(" AND ");f&&g.push(`HAVING ${f}`)}if(d.length){let f=d.map(({as:h,expr:x})=>`"${h}" AS (${x})`);g.push(`WINDOW ${f.join(", ")}`)}if(y.length){let f=y.map(String).filter(h=>h).join(" AND ");f&&g.push(`QUALIFY ${f}`)}return A.length&&g.push(`ORDER BY ${A.join(", ")}`),Number.isFinite(b)&&g.push(`LIMIT ${b}`),Number.isFinite(I)&&g.push(`OFFSET ${I}`),g.join(" ")}},q=class r{constructor(t,e){this.op=t,this.queries=e.map(n=>n.clone()),this.query={orderby:[]},this.cteFor=null}clone(){let t=new r(this.op,this.queries);return t.query={...this.query},t}orderby(...t){let{query:e}=this;return t.length===0?e.orderby:(e.orderby=e.orderby.concat(t.flat().filter(n=>n).map(u)),this)}limit(t){let{query:e}=this;return arguments.length===0?e.limit:(e.limit=Number.isFinite(t)?t:void 0,this)}offset(t){let{query:e}=this;return arguments.length===0?e.offset:(e.offset=Number.isFinite(t)?t:void 0,this)}get subqueries(){let{queries:t,cteFor:e}=this;return e&&t.forEach(n=>n.cteFor=e),t}toString(){let{op:t,queries:e,query:{orderby:n,limit:o,offset:s}}=this,i=[e.join(` ${t} `)];return n.length&&i.push(`ORDER BY ${n.join(", ")}`),Number.isFinite(o)&&i.push(`LIMIT ${o}`),Number.isFinite(s)&&i.push(`OFFSET ${s}`),i.join(" ")}};function M(r){return r instanceof F||r instanceof q}function Qe(r){return M(r)&&r.describe}function P(r){return Xe(r)?r.slice(1,-1):r}function Xe(r){return r[0]==='"'&&r[r.length-1]==='"'}var G=r=>r;function ot(){return{apply:G,invert:G,sqlApply:u,sqlInvert:G}}function Ve({base:r=null}={}){if(r==null||r===Math.E)return{apply:Math.log,invert:Math.exp,sqlApply:t=>c`LN(${u(t)})`,sqlInvert:t=>c`EXP(${t})`};if(r===10)return{apply:Math.log10,invert:t=>Math.pow(10,t),sqlApply:t=>c`LOG(${u(t)})`,sqlInvert:t=>c`POW(10, ${t})`};{let t=+r;return{apply:e=>Math.log(e)/Math.log(t),invert:e=>Math.pow(t,e),sqlApply:e=>c`LN(${u(e)}) / LN(${t})`,sqlInvert:e=>c`POW(${t}, ${e})`}}}function ke({constant:r=1}={}){let t=+r;return{apply:e=>Math.sign(e)*Math.log1p(Math.abs(e)),invert:e=>Math.sign(e)*Math.exp(Math.abs(e)-t),sqlApply:e=>(e=u(e),c`SIGN(${e}) * LN(${t} + ABS(${e}))`),sqlInvert:e=>c`SIGN(${e}) * (EXP(ABS(${e})) - ${t})`}}function He(){return{apply:r=>Math.sign(r)*Math.sqrt(Math.abs(r)),invert:r=>Math.sign(r)*r*r,sqlApply:r=>(r=u(r),c`SIGN(${r}) * SQRT(ABS(${r}))`),sqlInvert:r=>c`SIGN(${r}) * (${r}) ** 2`}}function Ke({exponent:r=1}={}){let t=+r;return{apply:e=>Math.sign(e)*Math.pow(Math.abs(e),t),invert:e=>Math.sign(e)*Math.pow(Math.abs(e),1/t),sqlApply:e=>(e=u(e),c`SIGN(${e}) * POW(ABS(${e}), ${t})`),sqlInvert:e=>c`SIGN(${e}) * POW(ABS(${e}), 1/${t})`}}function st(){return{apply:r=>+r,invert:r=>new Date(r),sqlApply:r=>r instanceof Date?+r:B(u(r)),sqlInvert:G}}var ze={identity:ot,linear:ot,log:Ve,symlog:ke,sqrt:He,pow:Ke,time:st,utc:st};function Je(r){let t=ze[r.type];return t?{...r,...t(r)}:null}function j(r,t,{replace:e=!1,temp:n=!0,view:o=!1}={}){return"CREATE"+(e?" OR REPLACE ":" ")+(n?"TEMP ":"")+(o?"VIEW":"TABLE")+(e?" ":" IF NOT EXISTS ")+r+" AS "+t}function Ze(r){return`INSTALL ${r}; LOAD ${r}`}function it(r,{columns:t=Object.keys(r?.[0]||{})}={}){let e=[];if(Array.isArray(t)?(e=t,t=e.reduce((o,s)=>(o[s]=s,o),{})):t&&(e=Object.keys(t)),!e.length)throw new Error("Can not create table from empty column set.");let n=[];for(let o of r){let s=e.map(i=>`${$(o[i])} AS "${t[i]}"`);n.push(`(SELECT ${s.join(", ")})`)}return n.join(" UNION ALL ")}function U(r,t,e,n={},o={}){let{select:s=["*"],where:i,view:a,temp:p,replace:d,...y}=n,A=sr({...o,...y}),b=`${r}('${e}'${A?", "+A:""})`,I=i?` WHERE ${i}`:"",g=`SELECT ${s.join(", ")} FROM ${b}${I}`;return j(t,g,{view:a,temp:p,replace:d})}function tr(r,t,e){return U("read_csv",r,t,e,{auto_detect:!0,sample_size:-1})}function er(r,t,e){return U("read_json",r,t,e,{auto_detect:!0,json_format:"auto"})}function rr(r,t,e){return U("read_parquet",r,t,e)}function nr(r,t,e={}){let{options:n,...o}=e;if(n){let s=Array.isArray(n)?n.join(", "):typeof n=="string"?n:Object.entries(n).map(([i,a])=>`${i}=${a}`).join(", ");Object.assign(o,{open_options:s.toUpperCase()})}return U("st_read",r,t,o)}function or(r,t,e={}){let{select:n=["*"],...o}=e,s=it(t),i=n.length===1&&n[0]==="*"?s:`SELECT ${n} FROM ${s}`;return j(r,i,o)}function sr(r){return Object.entries(r).map(([t,e])=>`${t}=${X(e)}`).join(", ")}function X(r){switch(typeof r){case"boolean":return String(r);case"string":return`'${r}'`;case"undefined":case"object":return r==null?"NULL":Array.isArray(r)?"["+r.map(t=>X(t)).join(", ")+"]":"{"+Object.entries(r).map(([t,e])=>`'${t}': ${X(e)}`).join(", ")+"}";default:return r}}export{F as Query,S as Ref,N as SQLExpression,Jt as agg,lt as all,gt as and,Me as argmax,Ce as argmin,Pe as arrayAgg,u as asColumn,O as asRelation,re as avg,v as cast,Fe as castDouble,Ge as castInteger,Q as centroid,ve as centroidX,Be as centroidY,K as column,Dt as contains,Ee as corr,ee as count,Se as covarPop,ye as covariance,j as create,Qt as cume_dist,We as dateDay,je as dateMonth,Ue as dateMonthDay,vt as dense_rank,ht as desc,de as entropy,B as epoch_ms,St as eq,Oe as first,Ht as first_value,Ye as geojson,At as gt,Tt as gte,It as isBetween,Qe as isDescribeQuery,qt as isDistinct,jt as isFinite,Ut as isInfinite,Gt as isNaN,Lt as isNotBetween,bt as isNotDistinct,yt as isNotNull,Et as isNull,R as isParamLike,M as isQuery,_ as isSQLExpression,ge as kurtosis,Vt as lag,De as last,Kt as last_value,kt as lead,Ft as length,mt as literal,$ as literalToSQL,tr as loadCSV,Ze as loadExtension,er as loadJSON,or as loadObjects,rr as loadParquet,nr as loadSpatial,_t as lower,wt as lt,Rt as lte,oe as mad,se as max,ne as mean,ue as median,ie as min,pe as mode,Nt as neq,$t as not,zt as nth_value,Xt as ntile,dt as or,Bt as percent_rank,Ct as prefix,ae as product,le as quantile,Yt as rank,Ot as regexp_matches,Ie as regrAvgX,Le as regrAvgY,Ae as regrCount,Ne as regrIntercept,Re as regrR2,qe as regrSXX,be as regrSXY,Te as regrSYY,we as regrSlope,H as relation,Wt as row_number,Je as scaleTransform,me as skewness,c as sql,he as stddev,$e as stddevPop,_e as stringAgg,Mt as suffix,ce as sum,pt as toSQL,Pt as upper,xe as varPop,fe as variance,rt as x,nt as y};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@uwdata/mosaic-sql",
-  "version": "0.7.0",
+  "version": "0.9.0",
   "description": "SQL query construction and analysis.",
   "keywords": [
     "sql",
@@ -21,9 +21,9 @@
   "scripts": {
     "prebuild": "rimraf dist && mkdir dist",
     "build": "node ../../esbuild.js mosaic-sql",
-    "lint": "eslint src test --ext .js",
+    "lint": "eslint src test",
     "test": "mocha 'test/**/*-test.js'",
     "prepublishOnly": "npm run test && npm run lint && npm run build"
   },
-  "gitHead": "4680b922f15579b7b527f31507ed71a12230ec35"
+  "gitHead": "89bb9b0dfa747aed691eaeba35379525a6764c61"
 }

package/src/Query.js

@@ -53,6 +53,7 @@ export class Query {
       qualify: [],
       orderby: []
     };
+    this.cteFor = null;
   }
 
   clone() {
@@ -61,9 +62,18 @@ export class Query {
     return q;
   }
 
+  /**
+   * Retrieve current WITH common table expressions (CTEs).
+   * @returns {any[]}
+   *//**
+   * Add WITH common table expressions (CTEs).
+   * @param  {...any} expr Expressions to add.
+   * @returns {this}
+   */
   with(...expr) {
     const { query } = this;
     if (expr.length === 0) {
+      // @ts-ignore
       return query.with;
     } else {
       const list = [];
@@ -88,9 +98,18 @@ export class Query {
     }
   }
 
+  /**
+   * Retrieve current SELECT expressions.
+   * @returns {any[]}
+   *//**
+   * Add SELECT expressions.
+   * @param {...any} expr Expressions to add.
+   * @returns {this}
+   */
   select(...expr) {
     const { query } = this;
     if (expr.length === 0) {
+      // @ts-ignore
       return query.select;
     } else {
       const list = [];
@@ -109,7 +128,11 @@ export class Query {
           }
         }
       }
-      query.select = query.select.concat(list);
+
+      const keys = new Set(list.map(x => x.as));
+      query.select = query.select
+        .filter(x => !keys.has(x.as))
+        .concat(list.filter(x => x.expr));
       return this;
     }
   }
@@ -124,9 +147,18 @@ export class Query {
     return this;
   }
 
+  /**
+   * Retrieve current from expressions.
+   * @returns {any[]}
+   *//**
+   * Provide table from expressions.
+   * @param  {...any} expr
+   * @returns {this}
+   */
   from(...expr) {
     const { query } = this;
     if (expr.length === 0) {
+      // @ts-ignore
       return query.from;
     } else {
       const list = [];
@@ -157,9 +189,19 @@ export class Query {
     return this.from(...expr);
   }
 
+  /**
+   * Retrieve current SAMPLE settings.
+   * @returns {any[]}
+   *//**
+   * Set SAMPLE settings.
+   * @param {number|object} value The percentage or number of rows to sample.
+   * @param {string} [method] The sampling method to use.
+   * @returns {this}
+   */
   sample(value, method) {
     const { query } = this;
     if (arguments.length === 0) {
+      // @ts-ignore
       return query.sample;
     } else {
       let spec = value;
@@ -173,9 +215,18 @@ export class Query {
     }
   }
 
+  /**
+   * Retrieve current WHERE expressions.
+   * @returns {any[]}
+   *//**
+   * Add WHERE expressions.
+   * @param  {...any} expr Expressions to add.
+   * @returns {this}
+   */
   where(...expr) {
     const { query } = this;
     if (expr.length === 0) {
+      // @ts-ignore
       return query.where;
     } else {
       query.where = query.where.concat(
@@ -190,9 +241,18 @@ export class Query {
     return this.where(...expr);
   }
 
+  /**
+   * Retrieve current GROUP BY expressions.
+   * @returns {any[]}
+   *//**
+   * Add GROUP BY expressions.
+   * @param  {...any} expr Expressions to add.
+   * @returns {this}
+   */
   groupby(...expr) {
     const { query } = this;
     if (expr.length === 0) {
+      // @ts-ignore
       return query.groupby;
     } else {
       query.groupby = query.groupby.concat(
@@ -207,9 +267,18 @@ export class Query {
     return this.groupby(...expr);
   }
 
+  /**
+   * Retrieve current HAVING expressions.
+   * @returns {any[]}
+   *//**
+   * Add HAVING expressions.
+   * @param  {...any} expr Expressions to add.
+   * @returns {this}
+   */
   having(...expr) {
     const { query } = this;
     if (expr.length === 0) {
+      // @ts-ignore
       return query.having;
     } else {
       query.having = query.having.concat(
@@ -219,9 +288,18 @@ export class Query {
     }
   }
 
+  /**
+   * Retrieve current WINDOW definitions.
+   * @returns {any[]}
+   *//**
+   * Add WINDOW definitions.
+   * @param  {...any} expr Expressions to add.
+   * @returns {this}
+   */
   window(...expr) {
     const { query } = this;
     if (expr.length === 0) {
+      // @ts-ignore
       return query.window;
     } else {
       const list = [];
@@ -239,9 +317,18 @@ export class Query {
     }
   }
 
+  /**
+   * Retrieve current QUALIFY expressions.
+   * @returns {any[]}
+   *//**
+   * Add QUALIFY expressions.
+   * @param  {...any} expr Expressions to add.
+   * @returns {this}
+   */
   qualify(...expr) {
     const { query } = this;
     if (expr.length === 0) {
+      // @ts-ignore
       return query.qualify;
     } else {
       query.qualify = query.qualify.concat(
@@ -251,9 +338,18 @@ export class Query {
     }
   }
 
+  /**
+   * Retrieve current ORDER BY expressions.
+   * @returns {any[]}
+   *//**
+   * Add ORDER BY expressions.
+   * @param  {...any} expr Expressions to add.
+   * @returns {this}
+   */
   orderby(...expr) {
     const { query } = this;
     if (expr.length === 0) {
+      // @ts-ignore
       return query.orderby;
     } else {
       query.orderby = query.orderby.concat(
@@ -263,6 +359,14 @@ export class Query {
     }
   }
 
+  /**
+   * Retrieve current LIMIT value.
+   * @returns {number|null}
+   *//**
+   * Set the query result LIMIT.
+   * @param {number} value The limit value.
+   * @returns {this}
+   */
   limit(value) {
     const { query } = this;
     if (arguments.length === 0) {
@@ -273,6 +377,14 @@ export class Query {
     }
   }
 
+  /**
+   * Retrieve current OFFSET value.
+   * @returns {number|null}
+   *//**
+   * Set the query result OFFSET.
+   * @param {number} value The offset value.
+   * @returns {this}
+   */
   offset(value) {
     const { query } = this;
     if (arguments.length === 0) {
@@ -391,6 +503,7 @@ export class SetOperation {
     this.op = op;
     this.queries = queries.map(q => q.clone());
     this.query = { orderby: [] };
+    this.cteFor = null;
   }
 
   clone() {

package/src/aggregates.js

@@ -19,11 +19,24 @@ export function agg(strings, ...exprs) {
  * rather than instantiate this class.
  */
 export class AggregateFunction extends SQLExpression {
+  /**
+   * Create a new AggregateFunction instance.
+   * @param {*} op The aggregate operation.
+   * @param {*} [args] The aggregate function arguments.
+   * @param {*} [type] The SQL data type to cast to.
+   * @param {boolean} [isDistinct] Flag indicating if this is a distinct value aggregate.
+   * @param {*} [filter] Filtering expression to apply prior to aggregation.
+   */
   constructor(op, args, type, isDistinct, filter) {
     args = (args || []).map(asColumn);
     const { strings, exprs } = aggExpr(op, args, type, isDistinct, filter);
     const { spans, cols } = parseSQL(strings, exprs);
-    super(spans, cols, { aggregate: op, args, type, isDistinct, filter });
+    super(spans, cols);
+    this.aggregate = op;
+    this.args = args;
+    this.type = type;
+    this.isDistinct = isDistinct;
+    this.filter = filter;
   }
 
   get basis() {
@@ -37,36 +50,69 @@ export class AggregateFunction extends SQLExpression {
     return `${op.toLowerCase()}${tail}`;
   }
 
+  /**
+   * Return a new derived aggregate function over distinct values.
+   * @returns {AggregateFunction} A new aggregate function.
+   */
   distinct() {
     const { aggregate: op, args, type, filter } = this;
     return new AggregateFunction(op, args, type, true, filter);
   }
 
+  /**
+   * Return a new derived aggregate function that filters values.
+   * @param {*} filter The filter expresion.
+   * @returns {AggregateFunction} A new aggregate function.
+   */
   where(filter) {
     const { aggregate: op, args, type, isDistinct } = this;
     return new AggregateFunction(op, args, type, isDistinct, filter);
   }
 
+  /**
+   * Return a new window function over this aggregate.
+   * @returns {WindowFunction} A new aggregate function.
+   */
   window() {
     const { aggregate: op, args, type, isDistinct } = this;
     const func = new AggregateFunction(op, args, null, isDistinct);
     return new WindowFunction(op, func, type);
   }
 
+  /**
+   * Return a window function over this aggregate with the given partitioning.
+   * @param {*} expr The grouping (partition by) criteria for the window function.
+   * @returns {WindowFunction} A new window function.
+   */
   partitionby(...expr) {
     return this.window().partitionby(...expr);
   }
 
+  /**
+   * Return a window function over this aggregate with the given ordering.
+   * @param {*} expr The sorting (order by) criteria for the window function.
+   * @returns {WindowFunction} A new window function.
+   */
   orderby(...expr) {
     return this.window().orderby(...expr);
   }
 
-  rows(prev, next) {
-    return this.window().rows(prev, next);
+  /**
+   * Return a window function over this aggregate with the given row frame.
+   * @param {(number|null)[] | import('./expression.js').ParamLike} frame The row-based window frame.
+   * @returns {WindowFunction} A new window function.
+   */
+  rows(frame) {
+    return this.window().rows(frame);
   }
 
-  range(prev, next) {
-    return this.window().range(prev, next);
+  /**
+   * Return a window function over this aggregate with the given range frame.
+   * @param {(number|null)[] | import('./expression.js').ParamLike} frame The range-based window frame.
+   * @returns {WindowFunction} A new window function.
+   */
+  range(frame) {
+    return this.window().range(frame);
   }
 }
 
@@ -117,6 +163,7 @@ export const varPop = aggf('VAR_POP');
 export const stddevPop = aggf('STDDEV_POP');
 
 export const corr = aggf('CORR');
+export const covariance = aggf('COVAR_SAMP');
 export const covarPop = aggf('COVAR_POP');
 export const regrIntercept = aggf('REGR_INTERCEPT');
 export const regrSlope = aggf('REGR_SLOPE');

package/src/desc.js

@@ -4,8 +4,8 @@ import { asColumn } from './ref.js';
 /**
  * Annotate an expression to indicate descending sort order.
  * Null values are ordered last.
- * @param {SQLExpression|string} expr A SQL expression or column name string.
- * @returns {SQLExpression} An expression with descending order.
+ * @param {import('./expression.js').SQLExpression|string} expr A SQL expression or column name string.
+ * @returns {import('./expression.js').SQLExpression} An expression with descending order.
  */
 export function desc(expr) {
   const e = asColumn(expr);

package/src/expression.js

@@ -1,16 +1,25 @@
 import { literalToSQL } from './to-sql.js';
 
+/**
+ * @typedef {{
+ *   value: any;
+ *   addEventListener(type: string, callback: Function): any;
+ *   column?: string,
+ *   columns?: string[]
+ * }} ParamLike
+ */
+
 /**
  * Test if a value is parameter-like. Parameters have addEventListener methods.
  * @param {*} value The value to test.
- * @returns True if the value is param-like, false otherwise.
+ * @returns {value is ParamLike} True if the value is param-like, false otherwise.
  */
 export const isParamLike = value => typeof value?.addEventListener === 'function';
 
 /**
  * Test if a value is a SQL expression instance.
  * @param {*} value The value to test.
- * @returns {boolean} True if value is a SQL expression, false otherwise.
+ * @returns {value is SQLExpression} True if value is a SQL expression, false otherwise.
  */
 export function isSQLExpression(value) {
   return value instanceof SQLExpression;
@@ -24,7 +33,7 @@ export class SQLExpression {
 
   /**
    * Create a new SQL expression instance.
-   * @param {(string|SQLExpression|Ref)[]} parts The parts of the expression.
+   * @param {(string | ParamLike | SQLExpression | import('./ref.js').Ref)[]} parts The parts of the expression.
    * @param {string[]} [columns=[]] The column dependencies
    * @param {object} [props] Additional properties for this expression.
    */
@@ -35,6 +44,8 @@ export class SQLExpression {
 
     const params = this._expr.filter(part => isParamLike(part));
     if (params.length > 0) {
+      /** @type {ParamLike[]} */
+      // @ts-ignore
       this._params = Array.from(new Set(params));
       this._params.forEach(param => {
         param.addEventListener('value', () => update(this, this.map?.get('value')));
@@ -87,7 +98,7 @@ export class SQLExpression {
   /**
    * Annotate this expression instance with additional properties.
    * @param {object[]} [props] One or more objects with properties to add.
-   * @returns {this} This SQL expression.
+   * @returns This SQL expression.
    */
   annotate(...props) {
     return Object.assign(this, ...props);

package/src/index.js

@@ -10,7 +10,8 @@ export {
 export {
   isSQLExpression,
   isParamLike,
-  sql
+  sql,
+  SQLExpression
 } from './expression.js';
 
 export {
@@ -47,6 +48,7 @@ export {
   avg,
   corr,
   count,
+  covariance,
   covarPop,
   entropy,
   first,

package/src/load/load.js

@@ -38,7 +38,7 @@ export function loadSpatial(tableName, fileName, options = {}) {
       : Object.entries(opt)
           .map(([key, value]) => `${key}=${value}`)
           .join(', ');
-    rest.open_options = open.toUpperCase();
+    Object.assign(rest, { open_options: open.toUpperCase() });
   }
   // TODO: handle box_2d for spatial_filter_box option
   // TODO: handle wkb_blob for spatial_filter option

package/src/load/sql-from.js

@@ -1,5 +1,12 @@
 import { literalToSQL } from '../to-sql.js';
 
+/**
+ * Create a SQL query that embeds the given data for loading.
+ * @param {*} data The dataset
+ * @param {object} [options] Loading options
+ * @param {string[]|object} [options.columns] The columns to include
+ * @returns {string} SQL query string to load data
+ */
 export function sqlFrom(data, {
   columns = Object.keys(data?.[0] || {})
 } = {}) {

package/src/ref.js

@@ -5,7 +5,7 @@ export class Ref {
   /**
    * Create a new Ref instance.
    * @param {string|Ref|null} table The table name.
-   * @param {string|null} column The column name.
+   * @param {string|null} [column] The column name.
    */
   constructor(table, column) {
     if (table) this.table = String(table);
@@ -87,11 +87,11 @@ export function relation(name) {
 
 /**
  * Create a column reference.
- * @param {string} [table] The table name (optional).
- * @param {string} column The column name.
+ * @param {string} table The table name (optional).
+ * @param {string} [column] The column name.
  * @returns {Ref} The generated column reference.
  */
-export function column(table, column) {
+export function column(table, column = null) {
   if (arguments.length === 1) {
     column = table;
     table = null;

package/src/scales.js

@@ -13,7 +13,7 @@ function scaleLinear() {
   };
 }
 
-function scaleLog({ base } = {}) {
+function scaleLog({ base = null } = {}) {
   if (base == null || base === Math.E) {
     return {
       apply: Math.log,
@@ -78,6 +78,7 @@ function scaleTime() {
 }
 
 const scales = {
+  identity: scaleLinear,
   linear: scaleLinear,
   log: scaleLog,
   symlog: scaleSymlog,

package/src/to-sql.js

@@ -27,7 +27,7 @@ export function literalToSQL(value) {
     case 'boolean':
       return value ? 'TRUE' : 'FALSE';
     case 'string':
-      return `'${value}'`;
+      return `'${value.replace(`'`, `''`)}'`;
     case 'number':
       return Number.isFinite(value) ? String(value) : 'NULL';
     default:

package/src/windows.js

@@ -9,6 +9,16 @@ import { repeat } from './repeat.js';
  * rather than instantiate this class.
  */
 export class WindowFunction extends SQLExpression {
+  /**
+   * Create a new WindowFunction instance.
+   * @param {string} op The window operation indicator.
+   * @param {*} func The window function expression.
+   * @param {*} [type] The SQL data type to cast to.
+   * @param {string} [name] The window definition name.
+   * @param {*} [group] Grouping (partition by) expressions.
+   * @param {*} [order] Sorting (order by) expressions.
+   * @param {*} [frame] The window frame definition.
+   */
   constructor(op, func, type, name, group = '', order = '', frame = '') {
     // build and parse expression
     let expr;
@@ -24,7 +34,14 @@ export class WindowFunction extends SQLExpression {
       expr = sql`(${expr})::${type}`;
     }
     const { _expr, _deps } = expr;
-    super(_expr, _deps, { window: op, func, type, name, group, order, frame });
+    super(_expr, _deps);
+    this.window = op;
+    this.func = func;
+    this.type = type;
+    this.name = name;
+    this.group = group;
+    this.order = order;
+    this.frame = frame;
   }
 
   get basis() {
@@ -36,11 +53,21 @@ export class WindowFunction extends SQLExpression {
     return func.label ?? func.toString();
   }
 
+  /**
+   * Return an updated window function over a named window definition.
+   * @param {string} name The window definition name.
+   * @returns {WindowFunction} A new window function.
+   */
   over(name) {
     const { window: op, func, type, group, order, frame } = this;
     return new WindowFunction(op, func, type, name, group, order, frame);
   }
 
+  /**
+   * Return an updated window function with the given partitioning.
+   * @param {*} expr The grouping (partition by) criteria for the window function.
+   * @returns {WindowFunction} A new window function.
+   */
   partitionby(...expr) {
     const exprs = expr.flat().filter(x => x).map(asColumn);
     const group = sql(
@@ -51,6 +78,11 @@ export class WindowFunction extends SQLExpression {
     return new WindowFunction(op, func, type, name, group, order, frame);
   }
 
+  /**
+   * Return an updated window function with the given ordering.
+   * @param {*} expr The sorting (order by) criteria for the window function.
+   * @returns {WindowFunction} A new window function.
+   */
   orderby(...expr) {
     const exprs = expr.flat().filter(x => x).map(asColumn);
     const order = sql(
@@ -61,12 +93,22 @@ export class WindowFunction extends SQLExpression {
     return new WindowFunction(op, func, type, name, group, order, frame);
   }
 
+  /**
+   * Return an updated window function with the given rows frame.
+   * @param {(number|null)[] | import('./expression.js').ParamLike} expr The row-based window frame.
+   * @returns {WindowFunction} A new window function.
+   */
   rows(expr) {
     const frame = windowFrame('ROWS', expr);
     const { window: op, func, type, name, group, order } = this;
     return new WindowFunction(op, func, type, name, group, order, frame);
   }
 
+  /**
+   * Return an updated window function with the given range frame.
+   * @param {(number|null)[] | import('./expression.js').ParamLike} expr The range-based window frame.
+   * @returns {WindowFunction} A new window function.
+   */
   range(expr) {
     const frame = windowFrame('RANGE', expr);
     const { window: op, func, type, name, group, order } = this;