mongo.do 0.1.0

package/dist/translator/stages/expression-translator.js

@@ -0,0 +1,467 @@
+/**
+ * Expression Translator - Translates MongoDB aggregation expressions to SQL
+ * Handles arithmetic, string, conditional, comparison, and function operators
+ * Supports multiple SQL dialects (SQLite, ClickHouse)
+ */
+import { validateFieldPath } from '../../utils/sql-safety.js';
+import { jsonExtract as dialectJsonExtract, getCastFunctions, getStringFunctions, getDateFunctions, getArrayFunctions, ifNull as dialectIfNull, } from '../dialect';
+/**
+ * Check if a value is a field reference (starts with $)
+ */
+export function isFieldReference(value) {
+    return typeof value === 'string' && value.startsWith('$') && !value.startsWith('$$');
+}
+/**
+ * Get the JSON path for a field reference
+ *
+ * SECURITY: Validates field name to prevent SQL injection attacks.
+ * @throws Error if field contains invalid characters
+ */
+export function getFieldPath(fieldRef) {
+    // Remove the leading $ and convert to JSON path
+    const field = fieldRef.substring(1);
+    // Validate the field path to prevent SQL injection
+    validateFieldPath(field);
+    const parts = field.split('.');
+    let path = '$';
+    for (const part of parts) {
+        if (/^\d+$/.test(part)) {
+            path += `[${part}]`;
+        }
+        else {
+            path += `.${part}`;
+        }
+    }
+    return path;
+}
+/**
+ * Translate an expression value (field reference, literal, or expression object)
+ */
+export function translateExpressionValue(value, params, dialect = 'sqlite') {
+    if (isFieldReference(value)) {
+        const path = getFieldPath(value);
+        return dialectJsonExtract(dialect, 'data', path);
+    }
+    if (typeof value === 'object' && value !== null && !Array.isArray(value)) {
+        return translateExpression(value, params, dialect);
+    }
+    if (typeof value === 'string') {
+        params.push(value);
+        return '?';
+    }
+    if (typeof value === 'number' || typeof value === 'boolean') {
+        return String(value);
+    }
+    if (value === null) {
+        return 'NULL';
+    }
+    params.push(JSON.stringify(value));
+    return '?';
+}
+/**
+ * Main expression translator
+ */
+export function translateExpression(expr, params, dialect = 'sqlite') {
+    const keys = Object.keys(expr);
+    if (keys.length === 0) {
+        return 'NULL';
+    }
+    const operator = keys[0];
+    // Arithmetic operators
+    if (operator === '$add') {
+        return translateArithmetic(expr.$add, params, '+', dialect);
+    }
+    if (operator === '$subtract') {
+        return translateArithmetic(expr.$subtract, params, '-', dialect);
+    }
+    if (operator === '$multiply') {
+        return translateArithmetic(expr.$multiply, params, '*', dialect);
+    }
+    if (operator === '$divide') {
+        return translateArithmetic(expr.$divide, params, '/', dialect);
+    }
+    if (operator === '$mod') {
+        return translateArithmetic(expr.$mod, params, '%', dialect);
+    }
+    // Type conversion operators
+    const castFns = getCastFunctions(dialect);
+    if (operator === '$toInt') {
+        const val = translateExpressionValue(expr.$toInt, params, dialect);
+        return castFns.toInt(val);
+    }
+    if (operator === '$toDouble') {
+        const val = translateExpressionValue(expr.$toDouble, params, dialect);
+        return castFns.toDouble(val);
+    }
+    if (operator === '$toString') {
+        const val = translateExpressionValue(expr.$toString, params, dialect);
+        return castFns.toString(val);
+    }
+    if (operator === '$toDate') {
+        const val = translateExpressionValue(expr.$toDate, params, dialect);
+        return castFns.toDate(val);
+    }
+    if (operator === '$toDecimal') {
+        const val = translateExpressionValue(expr.$toDecimal, params, dialect);
+        return castFns.toDecimal(val);
+    }
+    // String operators
+    const strFns = getStringFunctions(dialect);
+    if (operator === '$concat') {
+        return translateConcat(expr.$concat, params, dialect);
+    }
+    if (operator === '$substr') {
+        return translateSubstr(expr.$substr, params, dialect);
+    }
+    if (operator === '$toLower') {
+        const val = translateExpressionValue(expr.$toLower, params, dialect);
+        return strFns.lower(val);
+    }
+    if (operator === '$toUpper') {
+        const val = translateExpressionValue(expr.$toUpper, params, dialect);
+        return strFns.upper(val);
+    }
+    if (operator === '$indexOfBytes') {
+        const args = expr.$indexOfBytes;
+        const str = translateExpressionValue(args[0], params, dialect);
+        const search = translateExpressionValue(args[1], params, dialect);
+        return strFns.indexOf(str, search);
+    }
+    if (operator === '$strLenBytes') {
+        const val = translateExpressionValue(expr.$strLenBytes, params, dialect);
+        return strFns.strLength(val);
+    }
+    if (operator === '$replaceAll') {
+        const spec = expr.$replaceAll;
+        const input = translateExpressionValue(spec.input, params, dialect);
+        const find = translateExpressionValue(spec.find, params, dialect);
+        const replacement = translateExpressionValue(spec.replacement, params, dialect);
+        return strFns.replaceAll(input, find, replacement);
+    }
+    // Date operators
+    const dateFns = getDateFunctions(dialect);
+    if (operator === '$dateFromString') {
+        const spec = expr.$dateFromString;
+        const str = translateExpressionValue(spec.dateString, params, dialect);
+        return dateFns.fromString(str);
+    }
+    if (operator === '$dateToString') {
+        const spec = expr.$dateToString;
+        const date = translateExpressionValue(spec.date, params, dialect);
+        params.push(spec.format);
+        return dateFns.toString(date, '?');
+    }
+    if (operator === '$year') {
+        const val = translateExpressionValue(expr.$year, params, dialect);
+        return dateFns.year(val);
+    }
+    if (operator === '$month') {
+        const val = translateExpressionValue(expr.$month, params, dialect);
+        return dateFns.month(val);
+    }
+    if (operator === '$dayOfMonth') {
+        const val = translateExpressionValue(expr.$dayOfMonth, params, dialect);
+        return dateFns.day(val);
+    }
+    if (operator === '$hour') {
+        const val = translateExpressionValue(expr.$hour, params, dialect);
+        return dateFns.hour(val);
+    }
+    if (operator === '$minute') {
+        const val = translateExpressionValue(expr.$minute, params, dialect);
+        return dateFns.minute(val);
+    }
+    if (operator === '$second') {
+        const val = translateExpressionValue(expr.$second, params, dialect);
+        return dateFns.second(val);
+    }
+    if (operator === '$dateDiff') {
+        const spec = expr.$dateDiff;
+        const start = translateExpressionValue(spec.startDate, params, dialect);
+        const end = translateExpressionValue(spec.endDate, params, dialect);
+        return dateFns.dateDiff(spec.unit, start, end);
+    }
+    if (operator === '$dateAdd') {
+        const spec = expr.$dateAdd;
+        const date = translateExpressionValue(spec.startDate, params, dialect);
+        const amount = translateExpressionValue(spec.amount, params, dialect);
+        return dateFns.dateAdd(date, spec.unit, amount);
+    }
+    if (operator === '$dateTrunc') {
+        const spec = expr.$dateTrunc;
+        const date = translateExpressionValue(spec.date, params, dialect);
+        return dateFns.dateTrunc(date, spec.unit);
+    }
+    // Array operators
+    const arrFns = getArrayFunctions(dialect);
+    if (operator === '$in') {
+        const args = expr.$in;
+        const value = translateExpressionValue(args[0], params, dialect);
+        const array = translateExpressionValue(args[1], params, dialect);
+        return arrFns.in(value, array);
+    }
+    if (operator === '$concatArrays') {
+        const args = expr.$concatArrays;
+        const arrays = args.map(a => translateExpressionValue(a, params, dialect));
+        return arrFns.concat(arrays);
+    }
+    if (operator === '$filter') {
+        const spec = expr.$filter;
+        const array = translateExpressionValue(spec.input, params, dialect);
+        const cond = translateExpressionValue(spec.cond, params, dialect);
+        return arrFns.filter(array, spec.as || 'this', cond);
+    }
+    if (operator === '$map') {
+        const spec = expr.$map;
+        const array = translateExpressionValue(spec.input, params, dialect);
+        const inExpr = translateExpressionValue(spec.in, params, dialect);
+        return arrFns.map(array, spec.as || 'this', inExpr);
+    }
+    if (operator === '$reduce') {
+        const spec = expr.$reduce;
+        const array = translateExpressionValue(spec.input, params, dialect);
+        const initial = translateExpressionValue(spec.initialValue, params, dialect);
+        const inExpr = translateExpressionValue(spec.in, params, dialect);
+        return arrFns.reduce(array, initial, 'this', 'value', inExpr);
+    }
+    if (operator === '$slice') {
+        const args = expr.$slice;
+        const array = translateExpressionValue(args[0], params, dialect);
+        const count = translateExpressionValue(args[1], params, dialect);
+        return arrFns.slice(array, '0', count);
+    }
+    // Conditional operators
+    if (operator === '$cond') {
+        return translateCond(expr.$cond, params, dialect);
+    }
+    if (operator === '$ifNull') {
+        return translateIfNull(expr.$ifNull, params, dialect);
+    }
+    if (operator === '$switch') {
+        return translateSwitch(expr.$switch, params, dialect);
+    }
+    // Comparison operators (in expression context)
+    if (operator === '$eq') {
+        const args = expr.$eq;
+        const left = translateExpressionValue(args[0], params, dialect);
+        const right = translateExpressionValue(args[1], params, dialect);
+        return `(${left} = ${right})`;
+    }
+    if (operator === '$ne') {
+        const args = expr.$ne;
+        const left = translateExpressionValue(args[0], params, dialect);
+        const right = translateExpressionValue(args[1], params, dialect);
+        return `(${left} != ${right})`;
+    }
+    if (operator === '$gt') {
+        const args = expr.$gt;
+        const left = translateExpressionValue(args[0], params, dialect);
+        const right = translateExpressionValue(args[1], params, dialect);
+        return `(${left} > ${right})`;
+    }
+    if (operator === '$gte') {
+        const args = expr.$gte;
+        const left = translateExpressionValue(args[0], params, dialect);
+        const right = translateExpressionValue(args[1], params, dialect);
+        return `(${left} >= ${right})`;
+    }
+    if (operator === '$lt') {
+        const args = expr.$lt;
+        const left = translateExpressionValue(args[0], params, dialect);
+        const right = translateExpressionValue(args[1], params, dialect);
+        return `(${left} < ${right})`;
+    }
+    if (operator === '$lte') {
+        const args = expr.$lte;
+        const left = translateExpressionValue(args[0], params, dialect);
+        const right = translateExpressionValue(args[1], params, dialect);
+        return `(${left} <= ${right})`;
+    }
+    // Logical operators
+    if (operator === '$and') {
+        const conditions = expr.$and.map(c => translateExpressionValue(c, params, dialect));
+        return `(${conditions.join(' AND ')})`;
+    }
+    if (operator === '$or') {
+        const conditions = expr.$or.map(c => translateExpressionValue(c, params, dialect));
+        return `(${conditions.join(' OR ')})`;
+    }
+    if (operator === '$not') {
+        const val = translateExpressionValue(expr.$not, params, dialect);
+        return `NOT (${val})`;
+    }
+    // $expr for match conditions - extract the inner expression
+    if (operator === '$expr') {
+        return translateExpression(expr.$expr, params, dialect);
+    }
+    // $function operator - custom JavaScript function execution
+    if (operator === '$function') {
+        return translateFunction(expr.$function, params);
+    }
+    throw new Error(`Unknown expression operator: ${operator}`);
+}
+function translateArithmetic(args, params, op, dialect = 'sqlite') {
+    const parts = args.map(arg => translateExpressionValue(arg, params, dialect));
+    return `(${parts.join(` ${op} `)})`;
+}
+function translateConcat(args, params, dialect = 'sqlite') {
+    const parts = args.map(arg => translateExpressionValue(arg, params, dialect));
+    const strFns = getStringFunctions(dialect);
+    return strFns.concat(parts);
+}
+function translateSubstr(args, params, dialect = 'sqlite') {
+    const str = translateExpressionValue(args[0], params, dialect);
+    const start = translateExpressionValue(args[1], params, dialect);
+    const len = translateExpressionValue(args[2], params, dialect);
+    const strFns = getStringFunctions(dialect);
+    return strFns.substr(str, start, len);
+}
+function translateCond(cond, params, dialect = 'sqlite') {
+    let ifCond, thenVal, elseVal;
+    if (Array.isArray(cond)) {
+        [ifCond, thenVal, elseVal] = cond;
+    }
+    else {
+        ifCond = cond.if;
+        thenVal = cond.then;
+        elseVal = cond.else;
+    }
+    const condSql = translateExpressionValue(ifCond, params, dialect);
+    const thenSql = translateExpressionValue(thenVal, params, dialect);
+    const elseSql = translateExpressionValue(elseVal, params, dialect);
+    return `CASE WHEN ${condSql} THEN ${thenSql} ELSE ${elseSql} END`;
+}
+function translateIfNull(args, params, dialect = 'sqlite') {
+    const parts = args.map(arg => translateExpressionValue(arg, params, dialect));
+    return dialectIfNull(dialect, parts);
+}
+function translateSwitch(switchExpr, params, dialect = 'sqlite') {
+    const branches = switchExpr.branches;
+    const defaultVal = switchExpr.default;
+    const whenClauses = branches.map(branch => {
+        const caseSql = translateExpressionValue(branch.case, params, dialect);
+        const thenSql = translateExpressionValue(branch.then, params, dialect);
+        return `WHEN ${caseSql} THEN ${thenSql}`;
+    });
+    const elseSql = defaultVal !== undefined
+        ? translateExpressionValue(defaultVal, params, dialect)
+        : 'NULL';
+    return `CASE ${whenClauses.join(' ')} ELSE ${elseSql} END`;
+}
+/**
+ * Translate $function operator to a marker for deferred execution
+ * Returns a JSON marker string that will be processed by the aggregation executor
+ *
+ * The $function operator allows custom JavaScript functions in aggregation pipelines.
+ * Since SQLite cannot execute JavaScript, we:
+ * 1. Generate a JSON marker embedded in SQL output
+ * 2. Extract field references from args for document binding
+ * 3. Store the function expression metadata for post-processing
+ */
+function translateFunction(spec, _params) {
+    // Validate required fields
+    if (!spec.body) {
+        throw new Error('$function requires body');
+    }
+    if (!spec.args) {
+        throw new Error('$function requires args');
+    }
+    if (spec.lang !== 'js') {
+        throw new Error('$function only supports lang: "js"');
+    }
+    // Normalize body to string
+    const body = typeof spec.body === 'function'
+        ? spec.body.toString()
+        : spec.body;
+    // Process arguments - extract field paths and literal positions
+    const argPaths = [];
+    const literalArgs = {};
+    spec.args.forEach((arg, index) => {
+        if (isFieldReference(arg)) {
+            argPaths.push(getFieldPath(arg));
+        }
+        else {
+            literalArgs[index] = arg;
+        }
+    });
+    // Create marker object
+    const marker = {
+        __type: 'function',
+        body,
+        argPaths,
+        literalArgs,
+        argOrder: spec.args.map((arg, i) => isFieldReference(arg) ? { type: 'field', path: getFieldPath(arg) } : { type: 'literal', index: i })
+    };
+    // Return as a JSON string that can be detected and parsed later
+    return `'__FUNCTION__${JSON.stringify(marker).replace(/'/g, "''")}'`;
+}
+/**
+ * Function ID counter for placeholder identification
+ * Exported for testing purposes
+ */
+export let functionIdCounter = 0;
+/**
+ * Reset function ID counter (useful for testing)
+ */
+export function resetFunctionIdCounter() {
+    functionIdCounter = 0;
+}
+/**
+ * Check if a value is a $function operator
+ */
+export function isFunctionOperator(value) {
+    return (typeof value === 'object' &&
+        value !== null &&
+        '$function' in value);
+}
+/**
+ * Create a FunctionExpression from a $function spec
+ * Utility for external code that needs to work with function expressions
+ */
+export function createFunctionExpression(spec) {
+    const bodyStr = typeof spec.body === 'function'
+        ? spec.body.toString()
+        : spec.body;
+    const argPaths = [];
+    const literalArgs = new Map();
+    for (let i = 0; i < spec.args.length; i++) {
+        const arg = spec.args[i];
+        if (isFieldReference(arg)) {
+            const path = getFieldPath(arg);
+            argPaths.push(path);
+        }
+        else {
+            literalArgs.set(i, arg);
+        }
+    }
+    return {
+        __type: 'function',
+        body: bodyStr,
+        argPaths,
+        literalArgs
+    };
+}
+/**
+ * Parse a function marker from SQL output
+ * Returns the parsed function expression or null if not a function marker
+ */
+export function parseFunctionPlaceholder(sql) {
+    const match = sql.match(/'__FUNCTION__(.+?)'/);
+    if (!match || !match[1])
+        return null;
+    try {
+        // Unescape single quotes and parse JSON
+        const json = match[1].replace(/''/g, "'");
+        return JSON.parse(json);
+    }
+    catch {
+        return null;
+    }
+}
+/**
+ * Check if SQL contains function markers
+ */
+export function hasFunctionPlaceholders(sql) {
+    return /__FUNCTION__/.test(sql);
+}
+//# sourceMappingURL=expression-translator.js.map

package/dist/translator/stages/expression-translator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"expression-translator.js","sourceRoot":"","sources":["../../../src/translator/stages/expression-translator.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAA;AAC7D,OAAO,EAEL,WAAW,IAAI,kBAAkB,EACjC,gBAAgB,EAChB,kBAAkB,EAClB,gBAAgB,EAChB,iBAAiB,EACjB,MAAM,IAAI,aAAa,GACxB,MAAM,YAAY,CAAA;AAEnB;;GAEG;AACH,MAAM,UAAU,gBAAgB,CAAC,KAAc;IAC7C,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAA;AACtF,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,YAAY,CAAC,QAAgB;IAC3C,gDAAgD;IAChD,MAAM,KAAK,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;IAEnC,mDAAmD;IACnD,iBAAiB,CAAC,KAAK,CAAC,CAAA;IAExB,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAC9B,IAAI,IAAI,GAAG,GAAG,CAAA;IAEd,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,IAAI,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,IAAI,IAAI,IAAI,IAAI,GAAG,CAAA;QACrB,CAAC;aAAM,CAAC;YACN,IAAI,IAAI,IAAI,IAAI,EAAE,CAAA;QACpB,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAA;AACb,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB,CAAC,KAAc,EAAE,MAAiB,EAAE,UAAsB,QAAQ;IACxG,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;QAC5B,MAAM,IAAI,GAAG,YAAY,CAAC,KAAK,CAAC,CAAA;QAChC,OAAO,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAA;IAClD,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzE,OAAO,mBAAmB,CAAC,KAAgC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IAC/E,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;QAClB,OAAO,GAAG,CAAA;IACZ,CAAC;IAED,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;QAC5D,OAAO,MAAM,CAAC,KAAK,CAAC,CAAA;IACtB,CAAC;IAED,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACnB,OAAO,MAAM,CAAA;IACf,CAAC;IAED,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAA;IAClC,OAAO,GAAG,CAAA;AACZ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,mBAAmB,CACjC,IAA6B,EAC7B,MAAiB,EACjB,UAAsB,QAAQ;IAE9B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAE9B,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtB,OAAO,MAAM,CAAA;IACf,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAA;IAExB,uBAAuB;IACvB,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;QACxB,OAAO,mBAAmB,CAAC,IAAI,CAAC,IAAiB,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,CAAC,CAAA;IAC1E,CAAC;IACD,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QAC7B,OAAO,mBAAmB,CAAC,IAAI,CAAC,SAAsB,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,CAAC,CAAA;IAC/E,CAAC;IACD,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QAC7B,OAAO,mBAAmB,CAAC,IAAI,CAAC,SAAsB,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,CAAC,CAAA;IAC/E,CAAC;IACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,mBAAmB,CAAC,IAAI,CAAC,OAAoB,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,CAAC,CAAA;IAC7E,CAAC;IACD,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;QACxB,OAAO,mBAAmB,CAAC,IAAI,CAAC,IAAiB,EAAE,MAAM,EAAE,GAAG,EAAE,OAAO,CAAC,CAAA;IAC1E,CAAC;IAED,4BAA4B;IAC5B,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAA;IACzC,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAClE,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAC3B,CAAC;IACD,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACrE,OAAO,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAA;IAC9B,CAAC;IACD,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QAC7B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACrE,OAAO,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAA;IAC9B,CAAC;IACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACnE,OAAO,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;IAC5B,CAAC;IACD,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;QAC9B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACtE,OAAO,OAAO,CAAC,SAAS,CAAC,GAAG,CAAC,CAAA;IAC/B,CAAC;IAED,mBAAmB;IACnB,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAA;IAC1C,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,eAAe,CAAC,IAAI,CAAC,OAAoB,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IACpE,CAAC;IACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,eAAe,CAAC,IAAI,CAAC,OAAoB,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IACpE,CAAC;IACD,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;QAC5B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACpE,OAAO,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAC1B,CAAC;IACD,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;QAC5B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACpE,OAAO,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAC1B,CAAC;IACD,IAAI,QAAQ,KAAK,eAAe,EAAE,CAAC;QACjC,MAAM,IAAI,GAAG,IAAI,CAAC,aAA0B,CAAA;QAC5C,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAC9D,MAAM,MAAM,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACjE,OAAO,MAAM,CAAC,OAAO,CAAC,GAAG,EAAE,MAAM,CAAC,CAAA;IACpC,CAAC;IACD,IAAI,QAAQ,KAAK,cAAc,EAAE,CAAC;QAChC,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACxE,OAAO,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAA;IAC9B,CAAC;IACD,IAAI,QAAQ,KAAK,aAAa,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,WAAsE,CAAA;QACxF,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACnE,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACjE,MAAM,WAAW,GAAG,wBAAwB,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAC/E,OAAO,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,IAAI,EAAE,WAAW,CAAC,CAAA;IACpD,CAAC;IAED,iBAAiB;IACjB,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAA;IACzC,IAAI,QAAQ,KAAK,iBAAiB,EAAE,CAAC;QACnC,MAAM,IAAI,GAAG,IAAI,CAAC,eAA0C,CAAA;QAC5D,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACtE,OAAO,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAA;IAChC,CAAC;IACD,IAAI,QAAQ,KAAK,eAAe,EAAE,CAAC;QACjC,MAAM,IAAI,GAAG,IAAI,CAAC,aAAkD,CAAA;QACpE,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACjE,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;QACxB,OAAO,OAAO,CAAC,QAAQ,CAAC,IAAI,EAAE,GAAG,CAAC,CAAA;IACpC,CAAC;IACD,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QACzB,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACjE,OAAO,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IAC1B,CAAC;IACD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAClE,OAAO,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAA;IAC3B,CAAC;IACD,IAAI,QAAQ,KAAK,aAAa,EAAE,CAAC;QAC/B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACvE,OAAO,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;IACzB,CAAC;IACD,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QACzB,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACjE,OAAO,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IAC1B,CAAC;IACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACnE,OAAO,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;IAC5B,CAAC;IACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACnE,OAAO,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;IAC5B,CAAC;IACD,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QAC7B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAmE,CAAA;QACrF,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACvE,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACnE,OAAO,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,CAAC,CAAA;IAChD,CAAC;IACD,IAAI,QAAQ,KAAK,UAAU,EAAE,CAAC;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,QAAiE,CAAA;QACnF,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACtE,MAAM,MAAM,GAAG,wBAAwB,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACrE,OAAO,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC,CAAA;IACjD,CAAC;IACD,IAAI,QAAQ,KAAK,YAAY,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,IAAI,CAAC,UAA6C,CAAA;QAC/D,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACjE,OAAO,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAA;IAC3C,CAAC;IAED,kBAAkB;IAClB,MAAM,MAAM,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAA;IACzC,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAgB,CAAA;QAClC,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,OAAO,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,CAAA;IAChC,CAAC;IACD,IAAI,QAAQ,KAAK,eAAe,EAAE,CAAC;QACjC,MAAM,IAAI,GAAG,IAAI,CAAC,aAA0B,CAAA;QAC5C,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,wBAAwB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAA;QAC1E,OAAO,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;IAC9B,CAAC;IACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAwD,CAAA;QAC1E,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACnE,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACjE,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,IAAI,MAAM,EAAE,IAAI,CAAC,CAAA;IACtD,CAAC;IACD,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAmD,CAAA;QACrE,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACnE,MAAM,MAAM,GAAG,wBAAwB,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACjE,OAAO,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,CAAC,EAAE,IAAI,MAAM,EAAE,MAAM,CAAC,CAAA;IACrD,CAAC;IACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAiE,CAAA;QACnF,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACnE,MAAM,OAAO,GAAG,wBAAwB,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAC5E,MAAM,MAAM,GAAG,wBAAwB,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACjE,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,CAAA;IAC/D,CAAC;IACD,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,MAAmB,CAAA;QACrC,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,OAAO,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,CAAC,CAAA;IACxC,CAAC;IAED,wBAAwB;IACxB,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QACzB,OAAO,aAAa,CAAC,IAAI,CAAC,KAA4C,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IAC1F,CAAC;IACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,eAAe,CAAC,IAAI,CAAC,OAAoB,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IACpE,CAAC;IACD,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;QAC3B,OAAO,eAAe,CAAC,IAAI,CAAC,OAAkC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IAClF,CAAC;IAED,+CAA+C;IAC/C,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAgB,CAAA;QAClC,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAC/D,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,OAAO,IAAI,IAAI,MAAM,KAAK,GAAG,CAAA;IAC/B,CAAC;IACD,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAgB,CAAA;QAClC,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAC/D,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,OAAO,IAAI,IAAI,OAAO,KAAK,GAAG,CAAA;IAChC,CAAC;IACD,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAgB,CAAA;QAClC,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAC/D,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,OAAO,IAAI,IAAI,MAAM,KAAK,GAAG,CAAA;IAC/B,CAAC;IACD,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAiB,CAAA;QACnC,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAC/D,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,OAAO,IAAI,IAAI,OAAO,KAAK,GAAG,CAAA;IAChC,CAAC;IACD,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;QACvB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAgB,CAAA;QAClC,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAC/D,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,OAAO,IAAI,IAAI,MAAM,KAAK,GAAG,CAAA;IAC/B,CAAC;IACD,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;QACxB,MAAM,IAAI,GAAG,IAAI,CAAC,IAAiB,CAAA;QACnC,MAAM,IAAI,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAC/D,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,OAAO,IAAI,IAAI,OAAO,KAAK,GAAG,CAAA;IAChC,CAAC;IAED,oBAAoB;IACpB,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;QACxB,MAAM,UAAU,GAAI,IAAI,CAAC,IAAkB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAClD,wBAAwB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAC7C,CAAA;QACD,OAAO,IAAI,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAA;IACxC,CAAC;IACD,IAAI,QAAQ,KAAK,KAAK,EAAE,CAAC;QACvB,MAAM,UAAU,GAAI,IAAI,CAAC,GAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CACjD,wBAAwB,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAC7C,CAAA;QACD,OAAO,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAA;IACvC,CAAC;IACD,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;QACxB,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAChE,OAAO,QAAQ,GAAG,GAAG,CAAA;IACvB,CAAC;IAED,4DAA4D;IAC5D,IAAI,QAAQ,KAAK,OAAO,EAAE,CAAC;QACzB,OAAO,mBAAmB,CAAC,IAAI,CAAC,KAAgC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IACpF,CAAC;IAED,4DAA4D;IAC5D,IAAI,QAAQ,KAAK,WAAW,EAAE,CAAC;QAC7B,OAAO,iBAAiB,CAAC,IAAI,CAAC,SAAyB,EAAE,MAAM,CAAC,CAAA;IAClE,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,gCAAgC,QAAQ,EAAE,CAAC,CAAA;AAC7D,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAe,EAAE,MAAiB,EAAE,EAAU,EAAE,UAAsB,QAAQ;IACzG,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,wBAAwB,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAA;IAC7E,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,GAAG,CAAC,GAAG,CAAA;AACrC,CAAC;AAED,SAAS,eAAe,CAAC,IAAe,EAAE,MAAiB,EAAE,UAAsB,QAAQ;IACzF,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,wBAAwB,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAA;IAC7E,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAA;IAC1C,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;AAC7B,CAAC;AAED,SAAS,eAAe,CAAC,IAAe,EAAE,MAAiB,EAAE,UAAsB,QAAQ;IACzF,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IAC9D,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IAChE,MAAM,GAAG,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IAC9D,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAA;IAC1C,OAAO,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,EAAE,GAAG,CAAC,CAAA;AACvC,CAAC;AAED,SAAS,aAAa,CACpB,IAAyC,EACzC,MAAiB,EACjB,UAAsB,QAAQ;IAE9B,IAAI,MAAe,EAAE,OAAgB,EAAE,OAAgB,CAAA;IAEvD,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACxB,CAAC,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,GAAG,IAAI,CAAA;IACnC,CAAC;SAAM,CAAC;QACN,MAAM,GAAG,IAAI,CAAC,EAAE,CAAA;QAChB,OAAO,GAAG,IAAI,CAAC,IAAI,CAAA;QACnB,OAAO,GAAG,IAAI,CAAC,IAAI,CAAA;IACrB,CAAC;IAED,MAAM,OAAO,GAAG,wBAAwB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IACjE,MAAM,OAAO,GAAG,wBAAwB,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IAClE,MAAM,OAAO,GAAG,wBAAwB,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IAElE,OAAO,aAAa,OAAO,SAAS,OAAO,SAAS,OAAO,MAAM,CAAA;AACnE,CAAC;AAED,SAAS,eAAe,CAAC,IAAe,EAAE,MAAiB,EAAE,UAAsB,QAAQ;IACzF,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,wBAAwB,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAA;IAC7E,OAAO,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;AACtC,CAAC;AAED,SAAS,eAAe,CACtB,UAAmC,EACnC,MAAiB,EACjB,UAAsB,QAAQ;IAE9B,MAAM,QAAQ,GAAG,UAAU,CAAC,QAAmD,CAAA;IAC/E,MAAM,UAAU,GAAG,UAAU,CAAC,OAAO,CAAA;IAErC,MAAM,WAAW,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE;QACxC,MAAM,OAAO,GAAG,wBAAwB,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACtE,MAAM,OAAO,GAAG,wBAAwB,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QACtE,OAAO,QAAQ,OAAO,SAAS,OAAO,EAAE,CAAA;IAC1C,CAAC,CAAC,CAAA;IAEF,MAAM,OAAO,GAAG,UAAU,KAAK,SAAS;QACtC,CAAC,CAAC,wBAAwB,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,CAAC;QACvD,CAAC,CAAC,MAAM,CAAA;IAEV,OAAO,QAAQ,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,OAAO,MAAM,CAAA;AAC5D,CAAC;AAED;;;;;;;;;GASG;AACH,SAAS,iBAAiB,CAAC,IAAkB,EAAE,OAAkB;IAC/D,2BAA2B;IAC3B,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAA;IAC5C,CAAC;IACD,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAA;IAC5C,CAAC;IACD,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAA;IACvD,CAAC;IAED,2BAA2B;IAC3B,MAAM,IAAI,GAAG,OAAO,IAAI,CAAC,IAAI,KAAK,UAAU;QAC1C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;QACtB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAA;IAEb,gEAAgE;IAChE,MAAM,QAAQ,GAAa,EAAE,CAAA;IAC7B,MAAM,WAAW,GAA4B,EAAE,CAAA;IAE/C,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;QAC/B,IAAI,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,GAAa,CAAC,CAAC,CAAA;QAC5C,CAAC;aAAM,CAAC;YACN,WAAW,CAAC,KAAK,CAAC,GAAG,GAAG,CAAA;QAC1B,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,uBAAuB;IACvB,MAAM,MAAM,GAAG;QACb,MAAM,EAAE,UAAU;QAClB,IAAI;QACJ,QAAQ;QACR,WAAW;QACX,QAAQ,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CACjC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,CAAC,GAAa,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,CAC7G;KACF,CAAA;IAED,gEAAgE;IAChE,OAAO,gBAAgB,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAA;AACtE,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,IAAI,iBAAiB,GAAG,CAAC,CAAA;AAEhC;;GAEG;AACH,MAAM,UAAU,sBAAsB;IACpC,iBAAiB,GAAG,CAAC,CAAA;AACvB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAAC,KAAc;IAC/C,OAAO,CACL,OAAO,KAAK,KAAK,QAAQ;QACzB,KAAK,KAAK,IAAI;QACd,WAAW,IAAI,KAAK,CACrB,CAAA;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,IAAkB;IACzD,MAAM,OAAO,GAAG,OAAO,IAAI,CAAC,IAAI,KAAK,UAAU;QAC7C,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE;QACtB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAA;IAEb,MAAM,QAAQ,GAAa,EAAE,CAAA;IAC7B,MAAM,WAAW,GAAG,IAAI,GAAG,EAAmB,CAAA;IAE9C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;QAExB,IAAI,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,GAAG,YAAY,CAAC,GAAa,CAAC,CAAA;YACxC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACrB,CAAC;aAAM,CAAC;YACN,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,CAAA;QACzB,CAAC;IACH,CAAC;IAED,OAAO;QACL,MAAM,EAAE,UAAU;QAClB,IAAI,EAAE,OAAO;QACb,QAAQ;QACR,WAAW;KACZ,CAAA;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,wBAAwB,CAAC,GAAW;IAOlD,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAA;IAC9C,IAAI,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAAE,OAAO,IAAI,CAAA;IAEpC,IAAI,CAAC;QACH,wCAAwC;QACxC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAA;QACzC,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;IACzB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAA;IACb,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,uBAAuB,CAAC,GAAW;IACjD,OAAO,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;AACjC,CAAC"}

package/dist/translator/stages/facet-stage.d.ts

@@ -0,0 +1,13 @@
+/**
+ * $facet stage - Processes multiple aggregation pipelines in a single stage
+ * Translates to multiple separate queries (can be run in parallel)
+ */
+import type { StageResult, StageContext, PipelineStage } from './types';
+export interface FacetTranslator {
+    translatePipeline(stages: PipelineStage[], collection: string): {
+        sql: string;
+        params: unknown[];
+    };
+}
+export declare function translateFacetStage(facet: Record<string, PipelineStage[]>, context: StageContext, pipelineTranslator: FacetTranslator): StageResult;
+//# sourceMappingURL=facet-stage.d.ts.map

package/dist/translator/stages/facet-stage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"facet-stage.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/facet-stage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,SAAS,CAAA;AAEvE,MAAM,WAAW,eAAe;IAC9B,iBAAiB,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE,UAAU,EAAE,MAAM,GAAG;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,OAAO,EAAE,CAAA;KAAE,CAAA;CACnG;AAED,wBAAgB,mBAAmB,CACjC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,EAAE,CAAC,EACtC,OAAO,EAAE,YAAY,EACrB,kBAAkB,EAAE,eAAe,GAClC,WAAW,CAwBb"}

package/dist/translator/stages/facet-stage.js

@@ -0,0 +1,26 @@
+/**
+ * $facet stage - Processes multiple aggregation pipelines in a single stage
+ * Translates to multiple separate queries (can be run in parallel)
+ */
+export function translateFacetStage(facet, context, pipelineTranslator) {
+    const params = [];
+    const facets = {};
+    // Each facet is an independent pipeline
+    for (const [facetName, pipeline] of Object.entries(facet)) {
+        // Use the previous CTE or collection as the source for each facet
+        const source = context.previousCte || context.collection;
+        // Translate each facet pipeline independently
+        const result = pipelineTranslator.translatePipeline(pipeline, source);
+        facets[facetName] = result;
+    }
+    // The final result combines all facets
+    // In practice, this would be assembled in the application layer
+    // Here we provide the individual queries
+    return {
+        facets,
+        params,
+        selectClause: 'NULL', // Facets are handled separately
+        transformsShape: true
+    };
+}
+//# sourceMappingURL=facet-stage.js.map

package/dist/translator/stages/facet-stage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"facet-stage.js","sourceRoot":"","sources":["../../../src/translator/stages/facet-stage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAQH,MAAM,UAAU,mBAAmB,CACjC,KAAsC,EACtC,OAAqB,EACrB,kBAAmC;IAEnC,MAAM,MAAM,GAAc,EAAE,CAAA;IAC5B,MAAM,MAAM,GAAuD,EAAE,CAAA;IAErE,wCAAwC;IACxC,KAAK,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QAC1D,kEAAkE;QAClE,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,UAAU,CAAA;QAExD,8CAA8C;QAC9C,MAAM,MAAM,GAAG,kBAAkB,CAAC,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAA;QACrE,MAAM,CAAC,SAAS,CAAC,GAAG,MAAM,CAAA;IAC5B,CAAC;IAED,uCAAuC;IACvC,gEAAgE;IAChE,yCAAyC;IAEzC,OAAO;QACL,MAAM;QACN,MAAM;QACN,YAAY,EAAE,MAAM,EAAE,gCAAgC;QACtD,eAAe,EAAE,IAAI;KACtB,CAAA;AACH,CAAC"}

package/dist/translator/stages/fusion-stages.d.ts

@@ -0,0 +1,118 @@
+/**
+ * Fusion Aggregation Stages
+ *
+ * Implements $rankFusion and $scoreFusion aggregation stages for hybrid search.
+ * These stages combine results from vector search and full-text search pipelines.
+ *
+ * MongoDB Atlas $rankFusion and $scoreFusion inspired stages:
+ * - $rankFusion: Reciprocal Rank Fusion (RRF) algorithm
+ * - $scoreFusion: Weighted score combination with normalization
+ */
+import type { StageResult, StageContext, PipelineStage } from './types';
+/**
+ * Result from a fusion stage translation
+ */
+export interface FusionStageResult extends StageResult {
+    fusionType: 'rrf' | 'score';
+    vectorPipeline: PipelineStage[];
+    textPipeline: PipelineStage[];
+    limit?: number;
+    rrfK?: number;
+    weights?: {
+        vector: number;
+        text: number;
+    };
+    normalizeScores?: boolean;
+}
+/**
+ * Pipeline configuration for fusion stages
+ */
+export interface FusionPipelines {
+    vector: PipelineStage[];
+    text: PipelineStage[];
+}
+/**
+ * $rankFusion stage specification
+ */
+export interface RankFusionStageSpec {
+    input: {
+        pipelines: FusionPipelines;
+    };
+    combination?: {
+        ranker?: 'rrf';
+        k?: number;
+    };
+    limit?: number;
+}
+/**
+ * $scoreFusion stage specification
+ */
+export interface ScoreFusionStageSpec {
+    input: {
+        pipelines: FusionPipelines;
+    };
+    combination?: {
+        weights?: {
+            vector: number;
+            text: number;
+        };
+        normalizeScores?: boolean;
+    };
+    limit?: number;
+}
+/**
+ * Translate a $rankFusion aggregation stage
+ *
+ * The $rankFusion stage combines results from multiple search pipelines
+ * using Reciprocal Rank Fusion (RRF). This algorithm is particularly
+ * effective for combining results from different ranking systems
+ * (e.g., vector similarity and BM25 text search).
+ *
+ * RRF Formula: score = sum(1 / (k + rank)) for each pipeline
+ *
+ * @param spec The $rankFusion stage specification
+ * @param context Stage context including collection name
+ * @returns FusionStageResult with pipeline configurations
+ */
+export declare function translateRankFusionStage(spec: RankFusionStageSpec, _context: StageContext): FusionStageResult;
+/**
+ * Translate a $scoreFusion aggregation stage
+ *
+ * The $scoreFusion stage combines results from multiple search pipelines
+ * using weighted score combination. Scores are optionally normalized
+ * before combining.
+ *
+ * Formula: fusedScore = (vectorWeight * normalizedVectorScore) +
+ *                       (textWeight * normalizedTextScore)
+ *
+ * @param spec The $scoreFusion stage specification
+ * @param context Stage context including collection name
+ * @returns FusionStageResult with pipeline configurations
+ */
+export declare function translateScoreFusionStage(spec: ScoreFusionStageSpec, _context: StageContext): FusionStageResult;
+/**
+ * Build SQL CTE for rank fusion
+ *
+ * Creates CTEs that:
+ * 1. Execute vector search pipeline
+ * 2. Execute text search pipeline
+ * 3. Combine results using RRF
+ */
+export declare function buildRankFusionSQL(result: FusionStageResult, _collection: string): {
+    sql: string;
+    params: unknown[];
+};
+/**
+ * Build SQL CTE for score fusion
+ *
+ * Creates CTEs that:
+ * 1. Execute vector search pipeline
+ * 2. Execute text search pipeline
+ * 3. Normalize scores
+ * 4. Combine with weights
+ */
+export declare function buildScoreFusionSQL(result: FusionStageResult, _collection: string): {
+    sql: string;
+    params: unknown[];
+};
+//# sourceMappingURL=fusion-stages.d.ts.map

package/dist/translator/stages/fusion-stages.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fusion-stages.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/fusion-stages.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAMxE;;GAEG;AACH,MAAM,WAAW,iBAAkB,SAAQ,WAAW;IACpD,UAAU,EAAE,KAAK,GAAG,OAAO,CAAC;IAC5B,cAAc,EAAE,aAAa,EAAE,CAAC;IAChC,YAAY,EAAE,aAAa,EAAE,CAAC;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,IAAI,CAAC,EAAE,MAAM,CAAC;IAEd,OAAO,CAAC,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IAC3C,eAAe,CAAC,EAAE,OAAO,CAAC;CAC3B;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,MAAM,EAAE,aAAa,EAAE,CAAC;IACxB,IAAI,EAAE,aAAa,EAAE,CAAC;CACvB;AAED;;GAEG;AACH,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE;QACL,SAAS,EAAE,eAAe,CAAC;KAC5B,CAAC;IACF,WAAW,CAAC,EAAE;QACZ,MAAM,CAAC,EAAE,KAAK,CAAC;QACf,CAAC,CAAC,EAAE,MAAM,CAAC;KACZ,CAAC;IACF,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE;QACL,SAAS,EAAE,eAAe,CAAC;KAC5B,CAAC;IACF,WAAW,CAAC,EAAE;QACZ,OAAO,CAAC,EAAE;YACR,MAAM,EAAE,MAAM,CAAC;YACf,IAAI,EAAE,MAAM,CAAC;SACd,CAAC;QACF,eAAe,CAAC,EAAE,OAAO,CAAC;KAC3B,CAAC;IACF,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAgBD;;;;;;;;;;;;;GAaG;AACH,wBAAgB,wBAAwB,CACtC,IAAI,EAAE,mBAAmB,EACzB,QAAQ,EAAE,YAAY,GACrB,iBAAiB,CA2BnB;AAMD;;;;;;;;;;;;;GAaG;AACH,wBAAgB,yBAAyB,CACvC,IAAI,EAAE,oBAAoB,EAC1B,QAAQ,EAAE,YAAY,GACrB,iBAAiB,CAgCnB;AA8BD;;;;;;;GAOG;AACH,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,iBAAiB,EACzB,WAAW,EAAE,MAAM,GAClB;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,OAAO,EAAE,CAAA;CAAE,CA+BpC;AAED;;;;;;;;GAQG;AACH,wBAAgB,mBAAmB,CACjC,MAAM,EAAE,iBAAiB,EACzB,WAAW,EAAE,MAAM,GAClB;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,OAAO,EAAE,CAAA;CAAE,CA+BpC"}