search-input-query-parser 0.1.1 → 0.1.3

package/src/search-query-to-paradedb-sql.ts

@@ -0,0 +1,346 @@
+import {
+  Expression,
+  SearchQuery,
+  parseSearchInputQuery,
+  FieldSchema,
+  WildcardPattern,
+} from "./parser";
+
+export interface SqlQueryResult {
+  text: string;
+  values: any[];
+}
+
+interface SqlState {
+  paramCounter: number;
+  values: any[];
+  searchableColumns: string[];
+  schemas: Map<string, FieldSchema>;
+  language?: string; // For tsvector configuration
+}
+
+export interface SearchQueryOptions {
+  language?: string; // PostgreSQL language configuration for tsvector
+}
+
+// Helper Functions
+const escapeRegExp = (str: string): string =>
+  str.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+
+// Helper to escape special characters for ParadeDB query syntax
+const escapeParadeDBChars = (value: string): string => {
+  const specialChars = [
+    "+",
+    "^",
+    "`",
+    ":",
+    "{",
+    "}",
+    '"',
+    "[",
+    "]",
+    "(",
+    ")",
+    "<",
+    ">",
+    "~",
+    "!",
+    "\\",
+    "*",
+  ];
+  return specialChars.reduce(
+    (escaped, char) =>
+      escaped.replace(new RegExp(escapeRegExp(char), "g"), `\\${char}`),
+    value
+  );
+};
+
+const stripQuotes = (value: string): string => {
+  if (value.startsWith('"') && value.endsWith('"')) {
+    return value.slice(1, -1);
+  }
+  return value;
+};
+
+const cleanQuotedString = (
+  value: string,
+  stripOuterQuotes: boolean = true
+): string => {
+  // First strip the outer quotes if requested
+  let cleaned = stripOuterQuotes ? stripQuotes(value) : value;
+
+  // Replace escaped quotes with regular quotes
+  cleaned = cleaned.replace(/\\"/g, '"');
+
+  // Clean up any remaining escape characters
+  cleaned = cleaned.replace(/\\\\/g, "\\");
+
+  return cleaned;
+};
+
+const prepareParadeDBString = (
+  value: string,
+  includeWildcard: boolean = false
+): string => {
+  // First clean up the string
+  const cleaned = cleanQuotedString(value);
+
+  // For ParadeDB, we need to:
+  // 1. Escape special characters (except wildcards)
+  // 2. Wrap in quotes
+  // 3. Add wildcard if needed
+  const escaped = escapeParadeDBChars(cleaned);
+  const result = `"${escaped}"`;
+  return includeWildcard ? `${result}*` : result;
+};
+
+// Create a new parameter placeholder and update state
+const nextParam = (state: SqlState): [string, SqlState] => {
+  const paramName = `$${state.paramCounter}`;
+  const newState = {
+    ...state,
+    paramCounter: state.paramCounter + 1,
+  };
+  return [paramName, newState];
+};
+
+// Add a value to the state and return updated state
+const addValue = (state: SqlState, value: any): SqlState => ({
+  ...state,
+  values: [...state.values, value],
+});
+
+/**
+ * Convert a wildcard pattern to SQL
+ */
+const wildcardPatternToSql = (
+  expr: WildcardPattern,
+  state: SqlState
+): [string, SqlState] => {
+  if (expr.prefix === "") {
+    return ["1=1", state];
+  }
+
+  const [paramName, newState] = nextParam(state);
+  const cleanedPrefix = cleanQuotedString(expr.prefix);
+
+  const queryValue = prepareParadeDBString(cleanedPrefix, true);
+  const conditions = state.searchableColumns.map(
+    (column) => `${column} @@@ ${paramName}`
+  );
+  const sql =
+    conditions.length === 1
+      ? conditions[0]
+      : `(${conditions.join(" OR ")})`;
+  return [sql, addValue(newState, queryValue)];
+};
+
+/**
+ * Convert a search term to SQL conditions based on search type
+ */
+const searchTermToSql = (
+  value: string,
+  state: SqlState
+): [string, SqlState] => {
+  const [paramName, newState] = nextParam(state);
+  const hasWildcard = value.endsWith("*");
+  const cleanedValue = cleanQuotedString(value);
+  const baseValue = hasWildcard ? cleanedValue.slice(0, -1) : cleanedValue;
+
+  const queryValue = prepareParadeDBString(baseValue, hasWildcard);
+  const conditions = state.searchableColumns.map(
+    (column) => `${column} @@@ ${paramName}`
+  );
+  const sql =
+    conditions.length === 1
+      ? conditions[0]
+      : `(${conditions.join(" OR ")})`;
+  return [sql, addValue(newState, queryValue)];
+};
+
+/**
+ * Convert a field:value pair to SQL based on search type
+ */
+const fieldValueToSql = (
+  field: string,
+  value: string,
+  state: SqlState
+): [string, SqlState] => {
+  const [paramName, newState] = nextParam(state);
+  const schema = state.schemas.get(field.toLowerCase());
+  const hasWildcard = value.endsWith("*");
+  const cleanedValue = cleanQuotedString(value);
+  const baseValue = hasWildcard ? cleanedValue.slice(0, -1) : cleanedValue;
+
+  switch (schema?.type) {
+    case "date": {
+      // Use parameter binding for dates
+      const [dateParam, dateState] = nextParam(state);
+      return [
+        `${field} @@@ '"' || ${dateParam} || '"'`,
+        addValue(dateState, baseValue),
+      ];
+    }
+    case "number":
+      return [`${field} @@@ '${baseValue}'`, newState];
+    default: {
+      const queryValue = prepareParadeDBString(baseValue, hasWildcard);
+      return [`${field} @@@ ${paramName}`, addValue(newState, queryValue)];
+    }
+  }
+};
+
+/**
+ * Convert a range expression to SQL
+ */
+const rangeToSql = (
+  field: string,
+  operator: string,
+  value: string,
+  value2: string | undefined,
+  state: SqlState
+): [string, SqlState] => {
+  const schema = state.schemas.get(field.toLowerCase());
+  const isDateField = schema?.type === "date";
+
+  if (operator === "BETWEEN" && value2) {
+    const [param1, state1] = nextParam(state);
+    const [param2, state2] = nextParam(state1);
+    let val1 = isDateField ? value : Number(value);
+    let val2 = isDateField ? value2 : Number(value2);
+    return [
+      `${field} @@@ '[' || ${param1} || ' TO ' || ${param2} || ']'`,
+      addValue(addValue(state2, val1), val2),
+    ];
+  } else {
+    const [paramName, newState] = nextParam(state);
+    const rangeOp = operator.replace(">=", ">=").replace("<=", "<=");
+    const val = isDateField ? value : Number(value);
+    return [
+      `${field} @@@ '${rangeOp}' || ${paramName}`,
+      addValue(newState, val),
+    ];
+  }
+};
+
+const inExpressionToSql = (
+  field: string,
+  values: string[],
+  state: SqlState
+): [string, SqlState] => {
+  let currentState = state;
+  const cleanedValues = values.map((v) => cleanQuotedString(v));
+
+  const paramNames: string[] = [];
+
+  for (const value of cleanedValues) {
+    const [paramName, newState] = nextParam(currentState);
+    paramNames.push(paramName);
+    currentState = addValue(newState, value);
+  }
+
+  const concatExpr = paramNames.join(" || ' ' || ");
+  return [`${field} @@@ 'IN[' || ${concatExpr} || ']'`, currentState];
+};
+
+/**
+ * Convert a binary operation (AND/OR) to SQL
+ */
+const binaryOpToSql = (
+  operator: string,
+  left: Expression,
+  right: Expression,
+  state: SqlState
+): [string, SqlState] => {
+  const [leftText, leftState] = expressionToSql(left, state);
+  const [rightText, rightState] = expressionToSql(right, leftState);
+  return [`(${leftText} ${operator} ${rightText})`, rightState];
+};
+
+/**
+ * Convert a single expression to SQL
+ */
+const expressionToSql = (
+  expr: Expression,
+  state: SqlState
+): [string, SqlState] => {
+  switch (expr.type) {
+    case "SEARCH_TERM":
+      return searchTermToSql(expr.value, state);
+
+    case "WILDCARD":
+      return wildcardPatternToSql(expr, state);
+
+    case "IN":
+      return inExpressionToSql(
+        expr.field.value,
+        expr.values.map((v) => v.value),
+        state
+      );
+
+    case "FIELD_VALUE":
+      return fieldValueToSql(expr.field.value, expr.value.value, state);
+
+    case "RANGE":
+      return rangeToSql(
+        expr.field.value,
+        expr.operator,
+        expr.value.value,
+        expr.value2?.value,
+        state
+      );
+
+    case "AND":
+      return binaryOpToSql("AND", expr.left, expr.right, state);
+
+    case "OR":
+      return binaryOpToSql("OR", expr.left, expr.right, state);
+
+    case "NOT": {
+      const [sqlText, newState] = expressionToSql(expr.expression, state);
+      return [`NOT ${sqlText}`, newState];
+    }
+  }
+};
+
+/**
+ * Convert a SearchQuery to a SQL WHERE clause with specified search type
+ */
+export const searchQueryToParadeDbSql = (
+  query: SearchQuery,
+  searchableColumns: string[],
+  schemas: FieldSchema[] = [],
+  options: SearchQueryOptions = {}
+): SqlQueryResult => {
+  const initialState: SqlState = {
+    paramCounter: 1,
+    values: [],
+    searchableColumns,
+    schemas: new Map(schemas.map((s) => [s.name.toLowerCase(), s])),
+    language: options.language,
+  };
+
+  if (!query.expression) {
+    return { text: "1=1", values: [] };
+  }
+
+  const [text, finalState] = expressionToSql(query.expression, initialState);
+  return { text, values: finalState.values };
+};
+
+/**
+ * Convert a search string directly to SQL
+ */
+export const searchStringToParadeDbSql = (
+  searchString: string,
+  searchableColumns: string[],
+  schemas: FieldSchema[] = [],
+  options: SearchQueryOptions = {}
+): SqlQueryResult => {
+  const query = parseSearchInputQuery(searchString, schemas);
+  if (query.type === "SEARCH_QUERY_ERROR") {
+    throw new Error(`Parse error: ${query.errors[0].message}`);
+  }
+
+  return searchQueryToParadeDbSql(query, searchableColumns, schemas, options);
+};