@tanstack/electric-db-collection 0.1.44 → 0.2.1

package/src/sql-compiler.ts

@@ -0,0 +1,220 @@
+import { serialize } from "./pg-serializer"
+import type { SubsetParams } from "@electric-sql/client"
+import type { IR, LoadSubsetOptions } from "@tanstack/db"
+
+export type CompiledSqlRecord = Omit<SubsetParams, `params`> & {
+  params?: Array<unknown>
+}
+
+export function compileSQL<T>(options: LoadSubsetOptions): SubsetParams {
+  const { where, orderBy, limit } = options
+
+  const params: Array<T> = []
+  const compiledSQL: CompiledSqlRecord = { params }
+
+  if (where) {
+    // TODO: this only works when the where expression's PropRefs directly reference a column of the collection
+    //       doesn't work if it goes through aliases because then we need to know the entire query to be able to follow the reference until the base collection (cf. followRef function)
+    compiledSQL.where = compileBasicExpression(where, params)
+  }
+
+  if (orderBy) {
+    compiledSQL.orderBy = compileOrderBy(orderBy, params)
+  }
+
+  if (limit) {
+    compiledSQL.limit = limit
+  }
+
+  // WORKAROUND for Electric bug: Empty subset requests don't load data
+  // Add dummy "true = true" predicate when there's no where clause
+  // This is always true so doesn't filter data, just tricks Electric into loading
+  if (!where) {
+    compiledSQL.where = `true = true`
+  }
+
+  // Serialize the values in the params array into PG formatted strings
+  // and transform the array into a Record<string, string>
+  const paramsRecord = params.reduce(
+    (acc, param, index) => {
+      const serialized = serialize(param)
+      // Only include non-empty values in params
+      // Empty strings from null/undefined should be omitted
+      if (serialized !== ``) {
+        acc[`${index + 1}`] = serialized
+      }
+      return acc
+    },
+    {} as Record<string, string>
+  )
+
+  return {
+    ...compiledSQL,
+    params: paramsRecord,
+  }
+}
+
+/**
+ * Quote PostgreSQL identifiers to handle mixed case column names correctly.
+ * Electric/Postgres requires quotes for case-sensitive identifiers.
+ * @param name - The identifier to quote
+ * @returns The quoted identifier
+ */
+function quoteIdentifier(name: string): string {
+  return `"${name}"`
+}
+
+/**
+ * Compiles the expression to a SQL string and mutates the params array with the values.
+ * @param exp - The expression to compile
+ * @param params - The params array
+ * @returns The compiled SQL string
+ */
+function compileBasicExpression(
+  exp: IR.BasicExpression<unknown>,
+  params: Array<unknown>
+): string {
+  switch (exp.type) {
+    case `val`:
+      params.push(exp.value)
+      return `$${params.length}`
+    case `ref`:
+      // TODO: doesn't yet support JSON(B) values which could be accessed with nested props
+      if (exp.path.length !== 1) {
+        throw new Error(
+          `Compiler can't handle nested properties: ${exp.path.join(`.`)}`
+        )
+      }
+      return quoteIdentifier(exp.path[0]!)
+    case `func`:
+      return compileFunction(exp, params)
+    default:
+      throw new Error(`Unknown expression type`)
+  }
+}
+
+function compileOrderBy(orderBy: IR.OrderBy, params: Array<unknown>): string {
+  const compiledOrderByClauses = orderBy.map((clause: IR.OrderByClause) =>
+    compileOrderByClause(clause, params)
+  )
+  return compiledOrderByClauses.join(`,`)
+}
+
+function compileOrderByClause(
+  clause: IR.OrderByClause,
+  params: Array<unknown>
+): string {
+  // FIXME: We should handle stringSort and locale.
+  //        Correctly supporting them is tricky as it depends on Postgres' collation
+  const { expression, compareOptions } = clause
+  let sql = compileBasicExpression(expression, params)
+
+  if (compareOptions.direction === `desc`) {
+    sql = `${sql} DESC`
+  }
+
+  if (compareOptions.nulls === `first`) {
+    sql = `${sql} NULLS FIRST`
+  }
+
+  if (compareOptions.nulls === `last`) {
+    sql = `${sql} NULLS LAST`
+  }
+
+  return sql
+}
+
+function compileFunction(
+  exp: IR.Func<unknown>,
+  params: Array<unknown> = []
+): string {
+  const { name, args } = exp
+
+  const opName = getOpName(name)
+
+  const compiledArgs = args.map((arg: IR.BasicExpression) =>
+    compileBasicExpression(arg, params)
+  )
+
+  // Special case for IS NULL / IS NOT NULL - these are postfix operators
+  if (name === `isNull` || name === `isUndefined`) {
+    if (compiledArgs.length !== 1) {
+      throw new Error(`${name} expects 1 argument`)
+    }
+    return `${compiledArgs[0]} ${opName}`
+  }
+
+  // Special case for NOT - unary prefix operator
+  if (name === `not`) {
+    if (compiledArgs.length !== 1) {
+      throw new Error(`NOT expects 1 argument`)
+    }
+    // Check if the argument is IS NULL to generate IS NOT NULL
+    const arg = args[0]
+    if (arg && arg.type === `func`) {
+      const funcArg = arg
+      if (funcArg.name === `isNull` || funcArg.name === `isUndefined`) {
+        const innerArg = compileBasicExpression(funcArg.args[0]!, params)
+        return `${innerArg} IS NOT NULL`
+      }
+    }
+    return `${opName} (${compiledArgs[0]})`
+  }
+
+  if (isBinaryOp(name)) {
+    // Special handling for AND/OR which can be variadic
+    if ((name === `and` || name === `or`) && compiledArgs.length > 2) {
+      // Chain multiple arguments: (a AND b AND c) or (a OR b OR c)
+      return compiledArgs.map((arg) => `(${arg})`).join(` ${opName} `)
+    }
+
+    if (compiledArgs.length !== 2) {
+      throw new Error(`Binary operator ${name} expects 2 arguments`)
+    }
+    const [lhs, rhs] = compiledArgs
+    // Special case for = ANY operator which needs parentheses around the array parameter
+    if (name === `in`) {
+      return `${lhs} ${opName}(${rhs})`
+    }
+    return `${lhs} ${opName} ${rhs}`
+  }
+
+  return `${opName}(${compiledArgs.join(`,`)})`
+}
+
+function isBinaryOp(name: string): boolean {
+  const binaryOps = [`eq`, `gt`, `gte`, `lt`, `lte`, `and`, `or`, `in`]
+  return binaryOps.includes(name)
+}
+
+function getOpName(name: string): string {
+  const opNames = {
+    eq: `=`,
+    gt: `>`,
+    gte: `>=`,
+    lt: `<`,
+    lte: `<=`,
+    add: `+`,
+    and: `AND`,
+    or: `OR`,
+    not: `NOT`,
+    isUndefined: `IS NULL`,
+    isNull: `IS NULL`,
+    in: `= ANY`, // Use = ANY syntax for array parameters
+    like: `LIKE`,
+    ilike: `ILIKE`,
+    upper: `UPPER`,
+    lower: `LOWER`,
+    length: `LENGTH`,
+    concat: `CONCAT`,
+    coalesce: `COALESCE`,
+  }
+
+  const opName = opNames[name as keyof typeof opNames]
+
+  if (!opName) {
+    throw new Error(`Unknown operator/function: ${name}`)
+  }
+
+  return opName
+}