mongo.do 0.1.0 → 0.1.1

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

@@ -1,201 +0,0 @@
-/**
- * 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
- */
-// ============================================================
-// Default Constants
-// ============================================================
-/** Default RRF k constant */
-const DEFAULT_RRF_K = 60;
-/** Default fusion weights */
-const DEFAULT_WEIGHTS = { vector: 0.5, text: 0.5 };
-// ============================================================
-// $rankFusion Stage Translation
-// ============================================================
-/**
- * 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 function translateRankFusionStage(spec, _context) {
-    // Validate input
-    if (!spec.input?.pipelines) {
-        throw new Error('$rankFusion requires input.pipelines');
-    }
-    const { pipelines } = spec.input;
-    if (!pipelines.vector || !pipelines.text) {
-        throw new Error('$rankFusion requires both vector and text pipelines');
-    }
-    // Extract RRF k constant
-    const rrfK = spec.combination?.k ?? DEFAULT_RRF_K;
-    const result = {
-        fusionType: 'rrf',
-        vectorPipeline: pipelines.vector,
-        textPipeline: pipelines.text,
-        rrfK,
-        params: [],
-        transformsShape: true,
-    };
-    if (spec.limit !== undefined) {
-        result.limit = spec.limit;
-    }
-    return result;
-}
-// ============================================================
-// $scoreFusion Stage Translation
-// ============================================================
-/**
- * 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 function translateScoreFusionStage(spec, _context) {
-    // Validate input
-    if (!spec.input?.pipelines) {
-        throw new Error('$scoreFusion requires input.pipelines');
-    }
-    const { pipelines } = spec.input;
-    if (!pipelines.vector || !pipelines.text) {
-        throw new Error('$scoreFusion requires both vector and text pipelines');
-    }
-    // Extract and normalize weights
-    let weights = spec.combination?.weights ?? { ...DEFAULT_WEIGHTS };
-    weights = normalizeWeights(weights);
-    // Extract normalization option
-    const normalizeScores = spec.combination?.normalizeScores ?? true;
-    const result = {
-        fusionType: 'score',
-        vectorPipeline: pipelines.vector,
-        textPipeline: pipelines.text,
-        weights,
-        normalizeScores,
-        params: [],
-        transformsShape: true,
-    };
-    if (spec.limit !== undefined) {
-        result.limit = spec.limit;
-    }
-    return result;
-}
-// ============================================================
-// Utility Functions
-// ============================================================
-/**
- * Normalize weights so they sum to 1.0
- */
-function normalizeWeights(weights) {
-    const sum = weights.vector + weights.text;
-    if (sum === 0) {
-        return { vector: 0.5, text: 0.5 };
-    }
-    if (Math.abs(sum - 1) < 0.001) {
-        // Already normalized (within tolerance)
-        return weights;
-    }
-    return {
-        vector: weights.vector / sum,
-        text: weights.text / sum,
-    };
-}
-/**
- * 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 function buildRankFusionSQL(result, _collection) {
-    const k = result.rrfK ?? DEFAULT_RRF_K;
-    // This is a template - actual implementation will require
-    // executing both pipelines and combining results in JavaScript
-    // since SQLite doesn't support row_number() in all contexts
-    const sql = `
-    WITH vector_ranked AS (
-      -- Vector search results will be inserted here
-      SELECT docId, row_number() OVER (ORDER BY score DESC) as rank
-      FROM vector_results
-    ),
-    text_ranked AS (
-      -- Text search results will be inserted here
-      SELECT docId, row_number() OVER (ORDER BY score DESC) as rank
-      FROM text_results
-    ),
-    rrf_scores AS (
-      SELECT
-        COALESCE(v.docId, t.docId) as docId,
-        COALESCE(1.0 / (${k} + v.rank), 0) + COALESCE(1.0 / (${k} + t.rank), 0) as fusedScore
-      FROM vector_ranked v
-      FULL OUTER JOIN text_ranked t ON v.docId = t.docId
-    )
-    SELECT docId, fusedScore
-    FROM rrf_scores
-    ORDER BY fusedScore DESC
-    ${result.limit ? `LIMIT ${result.limit}` : ''}
-  `.trim();
-    return { sql, params: [] };
-}
-/**
- * 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 function buildScoreFusionSQL(result, _collection) {
-    const { vector: vectorWeight, text: textWeight } = result.weights ?? DEFAULT_WEIGHTS;
-    // This is a template - actual implementation will require
-    // executing both pipelines and combining results in JavaScript
-    const sql = `
-    WITH vector_scores AS (
-      -- Vector search results with normalized scores
-      SELECT docId, score as vectorScore
-      FROM vector_results
-    ),
-    text_scores AS (
-      -- Text search results with normalized scores
-      SELECT docId, score as textScore
-      FROM text_results
-    ),
-    fused_scores AS (
-      SELECT
-        COALESCE(v.docId, t.docId) as docId,
-        (${vectorWeight} * COALESCE(v.vectorScore, 0)) +
-        (${textWeight} * COALESCE(t.textScore, 0)) as fusedScore
-      FROM vector_scores v
-      FULL OUTER JOIN text_scores t ON v.docId = t.docId
-    )
-    SELECT docId, fusedScore
-    FROM fused_scores
-    ORDER BY fusedScore DESC
-    ${result.limit ? `LIMIT ${result.limit}` : ''}
-  `.trim();
-    return { sql, params: [] };
-}
-//# sourceMappingURL=fusion-stages.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"fusion-stages.js","sourceRoot":"","sources":["../../../src/translator/stages/fusion-stages.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AA8DH,+DAA+D;AAC/D,oBAAoB;AACpB,+DAA+D;AAE/D,6BAA6B;AAC7B,MAAM,aAAa,GAAG,EAAE,CAAC;AAEzB,6BAA6B;AAC7B,MAAM,eAAe,GAAG,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;AAEnD,+DAA+D;AAC/D,gCAAgC;AAChC,+DAA+D;AAE/D;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,wBAAwB,CACtC,IAAyB,EACzB,QAAsB;IAEtB,iBAAiB;IACjB,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;IAC1D,CAAC;IAED,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;IAEjC,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QACzC,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;IACzE,CAAC;IAED,yBAAyB;IACzB,MAAM,IAAI,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,IAAI,aAAa,CAAC;IAElD,MAAM,MAAM,GAAsB;QAChC,UAAU,EAAE,KAAK;QACjB,cAAc,EAAE,SAAS,CAAC,MAAM;QAChC,YAAY,EAAE,SAAS,CAAC,IAAI;QAC5B,IAAI;QACJ,MAAM,EAAE,EAAE;QACV,eAAe,EAAE,IAAI;KACtB,CAAC;IACF,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAC7B,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;IAC5B,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,+DAA+D;AAC/D,iCAAiC;AACjC,+DAA+D;AAE/D;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,yBAAyB,CACvC,IAA0B,EAC1B,QAAsB;IAEtB,iBAAiB;IACjB,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,EAAE,CAAC;QAC3B,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;IAC3D,CAAC;IAED,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC;IAEjC,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QACzC,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;IAC1E,CAAC;IAED,gCAAgC;IAChC,IAAI,OAAO,GAAG,IAAI,CAAC,WAAW,EAAE,OAAO,IAAI,EAAE,GAAG,eAAe,EAAE,CAAC;IAClE,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAEpC,+BAA+B;IAC/B,MAAM,eAAe,GAAG,IAAI,CAAC,WAAW,EAAE,eAAe,IAAI,IAAI,CAAC;IAElE,MAAM,MAAM,GAAsB;QAChC,UAAU,EAAE,OAAO;QACnB,cAAc,EAAE,SAAS,CAAC,MAAM;QAChC,YAAY,EAAE,SAAS,CAAC,IAAI;QAC5B,OAAO;QACP,eAAe;QACf,MAAM,EAAE,EAAE;QACV,eAAe,EAAE,IAAI;KACtB,CAAC;IACF,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAC7B,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC;IAC5B,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,+DAA+D;AAC/D,oBAAoB;AACpB,+DAA+D;AAE/D;;GAEG;AACH,SAAS,gBAAgB,CAAC,OAAyC;IAIjE,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAE1C,IAAI,GAAG,KAAK,CAAC,EAAE,CAAC;QACd,OAAO,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;IACpC,CAAC;IAED,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,KAAK,EAAE,CAAC;QAC9B,wCAAwC;QACxC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,OAAO;QACL,MAAM,EAAE,OAAO,CAAC,MAAM,GAAG,GAAG;QAC5B,IAAI,EAAE,OAAO,CAAC,IAAI,GAAG,GAAG;KACzB,CAAC;AACJ,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,kBAAkB,CAChC,MAAyB,EACzB,WAAmB;IAEnB,MAAM,CAAC,GAAG,MAAM,CAAC,IAAI,IAAI,aAAa,CAAC;IAEvC,0DAA0D;IAC1D,+DAA+D;IAC/D,4DAA4D;IAC5D,MAAM,GAAG,GAAG;;;;;;;;;;;;;;0BAcY,CAAC,oCAAoC,CAAC;;;;;;;MAO1D,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;GAC9C,CAAC,IAAI,EAAE,CAAC;IAET,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;AAC7B,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,mBAAmB,CACjC,MAAyB,EACzB,WAAmB;IAEnB,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,MAAM,CAAC,OAAO,IAAI,eAAe,CAAC;IAErF,0DAA0D;IAC1D,+DAA+D;IAC/D,MAAM,GAAG,GAAG;;;;;;;;;;;;;;WAcH,YAAY;WACZ,UAAU;;;;;;;MAOf,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;GAC9C,CAAC,IAAI,EAAE,CAAC;IAET,OAAO,EAAE,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;AAC7B,CAAC"}

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

@@ -1,8 +0,0 @@
-/**
- * $group stage - Groups documents by a specified expression
- * Translates to SQL GROUP BY with aggregate functions
- * Supports multiple SQL dialects (SQLite, ClickHouse)
- */
-import type { StageResult, StageContext, GroupStage } from './types';
-export declare function translateGroupStage(group: GroupStage, context: StageContext): StageResult;
-//# sourceMappingURL=group-stage.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"group-stage.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/group-stage.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,SAAS,CAAA;AAKpE,wBAAgB,mBAAmB,CACjC,KAAK,EAAE,UAAU,EACjB,OAAO,EAAE,YAAY,GACpB,WAAW,CAkEb"}

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

@@ -1,123 +0,0 @@
-/**
- * $group stage - Groups documents by a specified expression
- * Translates to SQL GROUP BY with aggregate functions
- * Supports multiple SQL dialects (SQLite, ClickHouse)
- */
-import { isFieldReference, getFieldPath, translateExpressionValue } from './expression-translator';
-import { validateFieldPath } from '../../utils/sql-safety.js';
-import { getAggregationFunctions, jsonExtract as dialectJsonExtract } from '../dialect';
-export function translateGroupStage(group, context) {
-    const params = [];
-    const dialect = context.dialect || 'sqlite';
-    // Validate _id is present
-    if (!('_id' in group)) {
-        throw new Error('$group requires _id field');
-    }
-    const { _id, ...accumulators } = group;
-    // Build GROUP BY clause
-    const groupByFields = [];
-    const selectParts = [];
-    if (_id === null) {
-        // Total aggregation - no GROUP BY
-        selectParts.push("'_id', NULL");
-    }
-    else if (typeof _id === 'string' && isFieldReference(_id)) {
-        // Simple field grouping
-        const path = getFieldPath(_id);
-        const fieldExpr = dialectJsonExtract(dialect, 'data', path);
-        groupByFields.push(fieldExpr);
-        selectParts.push(`'_id', ${fieldExpr}`);
-    }
-    else if (typeof _id === 'object' && _id !== null) {
-        // Compound _id
-        const idParts = [];
-        for (const [key, value] of Object.entries(_id)) {
-            // Validate key to prevent SQL injection
-            validateFieldPath(key);
-            if (typeof value === 'string' && isFieldReference(value)) {
-                // getFieldPath validates the field reference
-                const path = getFieldPath(value);
-                const fieldExpr = dialectJsonExtract(dialect, 'data', path);
-                groupByFields.push(fieldExpr);
-                idParts.push(`'${key}', ${fieldExpr}`);
-            }
-        }
-        if (dialect === 'clickhouse') {
-            selectParts.push(`'_id', tuple(${idParts.join(', ')})`);
-        }
-        else {
-            selectParts.push(`'_id', json_object(${idParts.join(', ')})`);
-        }
-    }
-    // Build accumulators
-    for (const [field, accumulator] of Object.entries(accumulators)) {
-        // Validate field name to prevent SQL injection
-        validateFieldPath(field);
-        const accSql = translateAccumulator(accumulator, params, dialect);
-        selectParts.push(`'${field}', ${accSql}`);
-    }
-    const selectClause = dialect === 'clickhouse'
-        ? `tuple(${selectParts.join(', ')}) AS data`
-        : `json_object(${selectParts.join(', ')}) AS data`;
-    const result = {
-        selectClause,
-        params,
-        transformsShape: true
-    };
-    if (groupByFields.length > 0) {
-        result.groupByClause = groupByFields.join(', ');
-    }
-    return result;
-}
-function translateAccumulator(accumulator, params, dialect = 'sqlite') {
-    const operator = Object.keys(accumulator)[0];
-    const value = accumulator[operator];
-    const aggFns = getAggregationFunctions(dialect);
-    switch (operator) {
-        case '$sum': {
-            if (typeof value === 'number') {
-                // $sum: 1 counts documents
-                if (value === 1) {
-                    return aggFns.count();
-                }
-                return aggFns.sum(String(value));
-            }
-            const expr = translateExpressionValue(value, params, dialect);
-            return aggFns.sum(expr);
-        }
-        case '$avg': {
-            const expr = translateExpressionValue(value, params, dialect);
-            return aggFns.avg(expr);
-        }
-        case '$min': {
-            const expr = translateExpressionValue(value, params, dialect);
-            return aggFns.min(expr);
-        }
-        case '$max': {
-            const expr = translateExpressionValue(value, params, dialect);
-            return aggFns.max(expr);
-        }
-        case '$count': {
-            return aggFns.count();
-        }
-        case '$first': {
-            const expr = translateExpressionValue(value, params, dialect);
-            return aggFns.first(expr);
-        }
-        case '$last': {
-            const expr = translateExpressionValue(value, params, dialect);
-            return aggFns.last(expr);
-        }
-        case '$push': {
-            const expr = translateExpressionValue(value, params, dialect);
-            return aggFns.push(expr);
-        }
-        case '$addToSet': {
-            const expr = translateExpressionValue(value, params, dialect);
-            return aggFns.addToSet(expr);
-        }
-        default:
-            throw new Error(`Unknown accumulator operator: ${operator}`);
-    }
-}
-//# sourceMappingURL=group-stage.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"group-stage.js","sourceRoot":"","sources":["../../../src/translator/stages/group-stage.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,gBAAgB,EAAE,YAAY,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAA;AAClG,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAA;AAC7D,OAAO,EAAmB,uBAAuB,EAAE,WAAW,IAAI,kBAAkB,EAAE,MAAM,YAAY,CAAA;AAExG,MAAM,UAAU,mBAAmB,CACjC,KAAiB,EACjB,OAAqB;IAErB,MAAM,MAAM,GAAc,EAAE,CAAA;IAC5B,MAAM,OAAO,GAAe,OAAO,CAAC,OAAO,IAAI,QAAQ,CAAA;IAEvD,0BAA0B;IAC1B,IAAI,CAAC,CAAC,KAAK,IAAI,KAAK,CAAC,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAA;IAC9C,CAAC;IAED,MAAM,EAAE,GAAG,EAAE,GAAG,YAAY,EAAE,GAAG,KAAK,CAAA;IAEtC,wBAAwB;IACxB,MAAM,aAAa,GAAa,EAAE,CAAA;IAClC,MAAM,WAAW,GAAa,EAAE,CAAA;IAEhC,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;QACjB,kCAAkC;QAClC,WAAW,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;IACjC,CAAC;SAAM,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC;QAC5D,wBAAwB;QACxB,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,CAAA;QAC9B,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAA;QAC3D,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAC7B,WAAW,CAAC,IAAI,CAAC,UAAU,SAAS,EAAE,CAAC,CAAA;IACzC,CAAC;SAAM,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;QACnD,eAAe;QACf,MAAM,OAAO,GAAa,EAAE,CAAA;QAC5B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YAC/C,wCAAwC;YACxC,iBAAiB,CAAC,GAAG,CAAC,CAAA;YACtB,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzD,6CAA6C;gBAC7C,MAAM,IAAI,GAAG,YAAY,CAAC,KAAK,CAAC,CAAA;gBAChC,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,CAAA;gBAC3D,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;gBAC7B,OAAO,CAAC,IAAI,CAAC,IAAI,GAAG,MAAM,SAAS,EAAE,CAAC,CAAA;YACxC,CAAC;QACH,CAAC;QACD,IAAI,OAAO,KAAK,YAAY,EAAE,CAAC;YAC7B,WAAW,CAAC,IAAI,CAAC,gBAAgB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QACzD,CAAC;aAAM,CAAC;YACN,WAAW,CAAC,IAAI,CAAC,sBAAsB,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;QAC/D,CAAC;IACH,CAAC;IAED,qBAAqB;IACrB,KAAK,MAAM,CAAC,KAAK,EAAE,WAAW,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,YAAY,CAAC,EAAE,CAAC;QAChE,+CAA+C;QAC/C,iBAAiB,CAAC,KAAK,CAAC,CAAA;QACxB,MAAM,MAAM,GAAG,oBAAoB,CAAC,WAAsC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;QAC5F,WAAW,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,MAAM,EAAE,CAAC,CAAA;IAC3C,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,KAAK,YAAY;QAC3C,CAAC,CAAC,SAAS,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW;QAC5C,CAAC,CAAC,eAAe,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAA;IAEpD,MAAM,MAAM,GAAgB;QAC1B,YAAY;QACZ,MAAM;QACN,eAAe,EAAE,IAAI;KACtB,CAAA;IACD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,MAAM,CAAC,aAAa,GAAG,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACjD,CAAC;IACD,OAAO,MAAM,CAAA;AACf,CAAC;AAED,SAAS,oBAAoB,CAC3B,WAAoC,EACpC,MAAiB,EACjB,UAAsB,QAAQ;IAE9B,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAE,CAAA;IAC7C,MAAM,KAAK,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAA;IACnC,MAAM,MAAM,GAAG,uBAAuB,CAAC,OAAO,CAAC,CAAA;IAE/C,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,2BAA2B;gBAC3B,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;oBAChB,OAAO,MAAM,CAAC,KAAK,EAAE,CAAA;gBACvB,CAAC;gBACD,OAAO,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAA;YAClC,CAAC;YACD,MAAM,IAAI,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;YAC7D,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QACzB,CAAC;QAED,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,MAAM,IAAI,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;YAC7D,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QACzB,CAAC;QAED,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,MAAM,IAAI,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;YAC7D,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QACzB,CAAC;QAED,KAAK,MAAM,CAAC,CAAC,CAAC;YACZ,MAAM,IAAI,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;YAC7D,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;QACzB,CAAC;QAED,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,OAAO,MAAM,CAAC,KAAK,EAAE,CAAA;QACvB,CAAC;QAED,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,MAAM,IAAI,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;YAC7D,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAC3B,CAAC;QAED,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,MAAM,IAAI,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;YAC7D,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC1B,CAAC;QAED,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,MAAM,IAAI,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;YAC7D,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC1B,CAAC;QAED,KAAK,WAAW,CAAC,CAAC,CAAC;YACjB,MAAM,IAAI,GAAG,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;YAC7D,OAAO,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAA;QAC9B,CAAC;QAED;YACE,MAAM,IAAI,KAAK,CAAC,iCAAiC,QAAQ,EAAE,CAAC,CAAA;IAChE,CAAC;AACH,CAAC"}

package/dist/translator/stages/index.d.ts

@@ -1,24 +0,0 @@
-/**
- * Stage translators index
- */
-export * from './types';
-export * from './expression-translator';
-export * from './match-stage';
-export * from './project-stage';
-export * from './group-stage';
-export * from './sort-stage';
-export * from './limit-stage';
-export * from './skip-stage';
-export * from './count-stage';
-export * from './lookup-stage';
-export * from './unwind-stage';
-export * from './add-fields-stage';
-export * from './bucket-stage';
-export * from './facet-stage';
-export * from './search-stage';
-export * from './optimizer';
-export * from './parallel-facet';
-export * from './join-optimizer';
-export * from './fusion-stages';
-export * from './vector-search-stage';
-//# sourceMappingURL=index.d.ts.map

package/dist/translator/stages/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,cAAc,SAAS,CAAA;AACvB,cAAc,yBAAyB,CAAA;AACvC,cAAc,eAAe,CAAA;AAC7B,cAAc,iBAAiB,CAAA;AAC/B,cAAc,eAAe,CAAA;AAC7B,cAAc,cAAc,CAAA;AAC5B,cAAc,eAAe,CAAA;AAC7B,cAAc,cAAc,CAAA;AAC5B,cAAc,eAAe,CAAA;AAC7B,cAAc,gBAAgB,CAAA;AAC9B,cAAc,gBAAgB,CAAA;AAC9B,cAAc,oBAAoB,CAAA;AAClC,cAAc,gBAAgB,CAAA;AAC9B,cAAc,eAAe,CAAA;AAC7B,cAAc,gBAAgB,CAAA;AAC9B,cAAc,aAAa,CAAA;AAC3B,cAAc,kBAAkB,CAAA;AAChC,cAAc,kBAAkB,CAAA;AAChC,cAAc,iBAAiB,CAAA;AAC/B,cAAc,uBAAuB,CAAA"}

package/dist/translator/stages/index.js

@@ -1,24 +0,0 @@
-/**
- * Stage translators index
- */
-export * from './types';
-export * from './expression-translator';
-export * from './match-stage';
-export * from './project-stage';
-export * from './group-stage';
-export * from './sort-stage';
-export * from './limit-stage';
-export * from './skip-stage';
-export * from './count-stage';
-export * from './lookup-stage';
-export * from './unwind-stage';
-export * from './add-fields-stage';
-export * from './bucket-stage';
-export * from './facet-stage';
-export * from './search-stage';
-export * from './optimizer';
-export * from './parallel-facet';
-export * from './join-optimizer';
-export * from './fusion-stages';
-export * from './vector-search-stage';
-//# sourceMappingURL=index.js.map

package/dist/translator/stages/index.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/translator/stages/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,cAAc,SAAS,CAAA;AACvB,cAAc,yBAAyB,CAAA;AACvC,cAAc,eAAe,CAAA;AAC7B,cAAc,iBAAiB,CAAA;AAC/B,cAAc,eAAe,CAAA;AAC7B,cAAc,cAAc,CAAA;AAC5B,cAAc,eAAe,CAAA;AAC7B,cAAc,cAAc,CAAA;AAC5B,cAAc,eAAe,CAAA;AAC7B,cAAc,gBAAgB,CAAA;AAC9B,cAAc,gBAAgB,CAAA;AAC9B,cAAc,oBAAoB,CAAA;AAClC,cAAc,gBAAgB,CAAA;AAC9B,cAAc,eAAe,CAAA;AAC7B,cAAc,gBAAgB,CAAA;AAC9B,cAAc,aAAa,CAAA;AAC3B,cAAc,kBAAkB,CAAA;AAChC,cAAc,kBAAkB,CAAA;AAChC,cAAc,iBAAiB,CAAA;AAC/B,cAAc,uBAAuB,CAAA"}

package/dist/translator/stages/join-optimizer.d.ts

@@ -1,37 +0,0 @@
-/**
- * Join Optimizer - Optimizes $lookup stages for better performance
- *
- * Strategies:
- * 1. Index hints for join conditions
- * 2. Predicate pushdown into subqueries
- * 3. Query rewriting for specific patterns
- */
-import type { LookupStage, PipelineStage } from './types';
-export interface JoinOptimization {
-    /** Suggested index for the lookup */
-    suggestedIndex?: {
-        collection: string;
-        fields: string[];
-    };
-    /** Optimized lookup configuration */
-    optimizedLookup: LookupStage;
-    /** Pre-filter that can be applied to foreign collection */
-    foreignPreFilter?: Record<string, unknown>;
-}
-/**
- * Analyze a $lookup stage and suggest optimizations
- */
-export declare function optimizeLookup(lookup: LookupStage, followingStages: PipelineStage[]): JoinOptimization;
-/**
- * Check if a lookup can use an index-based strategy
- */
-export declare function canUseIndexLookup(lookup: LookupStage): boolean;
-/**
- * Estimate lookup cost based on heuristics
- */
-export declare function estimateLookupCost(lookup: LookupStage, estimatedDocumentCount?: number): number;
-/**
- * Suggest a rewritten lookup for specific patterns
- */
-export declare function suggestLookupRewrite(_lookup: LookupStage): LookupStage | null;
-//# sourceMappingURL=join-optimizer.d.ts.map

package/dist/translator/stages/join-optimizer.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"join-optimizer.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/join-optimizer.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,SAAS,CAAA;AAEzD,MAAM,WAAW,gBAAgB;IAC/B,qCAAqC;IACrC,cAAc,CAAC,EAAE;QACf,UAAU,EAAE,MAAM,CAAA;QAClB,MAAM,EAAE,MAAM,EAAE,CAAA;KACjB,CAAA;IACD,qCAAqC;IACrC,eAAe,EAAE,WAAW,CAAA;IAC5B,2DAA2D;IAC3D,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAC3C;AAED;;GAEG;AACH,wBAAgB,cAAc,CAC5B,MAAM,EAAE,WAAW,EACnB,eAAe,EAAE,aAAa,EAAE,GAC/B,gBAAgB,CAuDlB;AAED;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAG9D;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,WAAW,EACnB,sBAAsB,GAAE,MAAa,GACpC,MAAM,CAeR;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,OAAO,EAAE,WAAW,GACnB,WAAW,GAAG,IAAI,CAQpB"}

package/dist/translator/stages/join-optimizer.js

@@ -1,93 +0,0 @@
-/**
- * Join Optimizer - Optimizes $lookup stages for better performance
- *
- * Strategies:
- * 1. Index hints for join conditions
- * 2. Predicate pushdown into subqueries
- * 3. Query rewriting for specific patterns
- */
-/**
- * Analyze a $lookup stage and suggest optimizations
- */
-export function optimizeLookup(lookup, followingStages) {
-    const result = {
-        optimizedLookup: { ...lookup }
-    };
-    // Suggest index for the join condition
-    if (lookup.foreignField) {
-        result.suggestedIndex = {
-            collection: lookup.from,
-            fields: [lookup.foreignField]
-        };
-    }
-    // Look for $match stages that filter on the lookup result
-    // These can potentially be pushed into the lookup subquery
-    for (const stage of followingStages) {
-        if ('$match' in stage) {
-            const match = stage.$match;
-            const lookupFieldPrefix = `${lookup.as}.`;
-            // Extract conditions that reference the lookup field
-            const pushableConditions = {};
-            for (const [field, condition] of Object.entries(match)) {
-                if (field.startsWith(lookupFieldPrefix)) {
-                    const foreignField = field.substring(lookupFieldPrefix.length);
-                    pushableConditions[foreignField] = condition;
-                }
-            }
-            if (Object.keys(pushableConditions).length > 0) {
-                result.foreignPreFilter = pushableConditions;
-                // If we have a pipeline lookup, add the filter to the pipeline
-                if (lookup.pipeline) {
-                    result.optimizedLookup = {
-                        ...lookup,
-                        pipeline: [
-                            { $match: pushableConditions },
-                            ...lookup.pipeline
-                        ]
-                    };
-                }
-            }
-            // Only check the first $match after the lookup
-            break;
-        }
-        // Stop if we hit a stage that transforms the data
-        if ('$group' in stage || '$project' in stage || '$unwind' in stage) {
-            break;
-        }
-    }
-    return result;
-}
-/**
- * Check if a lookup can use an index-based strategy
- */
-export function canUseIndexLookup(lookup) {
-    // Simple lookups on _id fields are most efficient
-    return lookup.foreignField === '_id';
-}
-/**
- * Estimate lookup cost based on heuristics
- */
-export function estimateLookupCost(lookup, estimatedDocumentCount = 1000) {
-    let cost = estimatedDocumentCount;
-    // Nested lookups are more expensive
-    if (lookup.pipeline) {
-        // Each pipeline stage adds overhead
-        cost *= (lookup.pipeline.length + 1);
-    }
-    // Lookups on indexed fields are cheaper
-    if (canUseIndexLookup(lookup)) {
-        cost *= 0.1;
-    }
-    return cost;
-}
-/**
- * Suggest a rewritten lookup for specific patterns
- */
-export function suggestLookupRewrite(_lookup) {
-    // Pattern: Simple equality join can be converted to correlated subquery
-    // This is already what we do, so no change needed
-    // Pattern: Self-join can sometimes be replaced with window functions
-    // SQLite doesn't support this well, so skip
-    return null;
-}
-//# sourceMappingURL=join-optimizer.js.map

package/dist/translator/stages/join-optimizer.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"join-optimizer.js","sourceRoot":"","sources":["../../../src/translator/stages/join-optimizer.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAgBH;;GAEG;AACH,MAAM,UAAU,cAAc,CAC5B,MAAmB,EACnB,eAAgC;IAEhC,MAAM,MAAM,GAAqB;QAC/B,eAAe,EAAE,EAAE,GAAG,MAAM,EAAE;KAC/B,CAAA;IAED,uCAAuC;IACvC,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QACxB,MAAM,CAAC,cAAc,GAAG;YACtB,UAAU,EAAE,MAAM,CAAC,IAAI;YACvB,MAAM,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC;SAC9B,CAAA;IACH,CAAC;IAED,0DAA0D;IAC1D,2DAA2D;IAC3D,KAAK,MAAM,KAAK,IAAI,eAAe,EAAE,CAAC;QACpC,IAAI,QAAQ,IAAI,KAAK,EAAE,CAAC;YACtB,MAAM,KAAK,GAAG,KAAK,CAAC,MAAiC,CAAA;YACrD,MAAM,iBAAiB,GAAG,GAAG,MAAM,CAAC,EAAE,GAAG,CAAA;YAEzC,qDAAqD;YACrD,MAAM,kBAAkB,GAA4B,EAAE,CAAA;YACtD,KAAK,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACvD,IAAI,KAAK,CAAC,UAAU,CAAC,iBAAiB,CAAC,EAAE,CAAC;oBACxC,MAAM,YAAY,GAAG,KAAK,CAAC,SAAS,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAA;oBAC9D,kBAAkB,CAAC,YAAY,CAAC,GAAG,SAAS,CAAA;gBAC9C,CAAC;YACH,CAAC;YAED,IAAI,MAAM,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/C,MAAM,CAAC,gBAAgB,GAAG,kBAAkB,CAAA;gBAE5C,+DAA+D;gBAC/D,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;oBACpB,MAAM,CAAC,eAAe,GAAG;wBACvB,GAAG,MAAM;wBACT,QAAQ,EAAE;4BACR,EAAE,MAAM,EAAE,kBAAkB,EAAmB;4BAC/C,GAAG,MAAM,CAAC,QAAQ;yBACnB;qBACF,CAAA;gBACH,CAAC;YACH,CAAC;YAED,+CAA+C;YAC/C,MAAK;QACP,CAAC;QAED,kDAAkD;QAClD,IAAI,QAAQ,IAAI,KAAK,IAAI,UAAU,IAAI,KAAK,IAAI,SAAS,IAAI,KAAK,EAAE,CAAC;YACnE,MAAK;QACP,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAA;AACf,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,MAAmB;IACnD,kDAAkD;IAClD,OAAO,MAAM,CAAC,YAAY,KAAK,KAAK,CAAA;AACtC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,kBAAkB,CAChC,MAAmB,EACnB,yBAAiC,IAAI;IAErC,IAAI,IAAI,GAAG,sBAAsB,CAAA;IAEjC,oCAAoC;IACpC,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;QACpB,oCAAoC;QACpC,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;IACtC,CAAC;IAED,wCAAwC;IACxC,IAAI,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC;QAC9B,IAAI,IAAI,GAAG,CAAA;IACb,CAAC;IAED,OAAO,IAAI,CAAA;AACb,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAClC,OAAoB;IAEpB,wEAAwE;IACxE,kDAAkD;IAElD,qEAAqE;IACrE,4CAA4C;IAE5C,OAAO,IAAI,CAAA;AACb,CAAC"}

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

@@ -1,7 +0,0 @@
-/**
- * $limit stage - Limits the number of documents
- * Translates to SQL LIMIT clause
- */
-import type { StageResult, StageContext } from './types';
-export declare function translateLimitStage(limit: number, _context: StageContext): StageResult;
-//# sourceMappingURL=limit-stage.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"limit-stage.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/limit-stage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,SAAS,CAAA;AAExD,wBAAgB,mBAAmB,CACjC,KAAK,EAAE,MAAM,EACb,QAAQ,EAAE,YAAY,GACrB,WAAW,CAKb"}

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

@@ -1,11 +0,0 @@
-/**
- * $limit stage - Limits the number of documents
- * Translates to SQL LIMIT clause
- */
-export function translateLimitStage(limit, _context) {
-    return {
-        limitClause: `LIMIT ${limit}`,
-        params: []
-    };
-}
-//# sourceMappingURL=limit-stage.js.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"limit-stage.js","sourceRoot":"","sources":["../../../src/translator/stages/limit-stage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,MAAM,UAAU,mBAAmB,CACjC,KAAa,EACb,QAAsB;IAEtB,OAAO;QACL,WAAW,EAAE,SAAS,KAAK,EAAE;QAC7B,MAAM,EAAE,EAAE;KACX,CAAA;AACH,CAAC"}

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

@@ -1,7 +0,0 @@
-/**
- * $lookup stage - Performs a left outer join with another collection
- * Translates to SQL LEFT JOIN or subquery
- */
-import type { StageResult, StageContext, LookupStage } from './types';
-export declare function translateLookupStage(lookup: LookupStage, context: StageContext): StageResult;
-//# sourceMappingURL=lookup-stage.d.ts.map

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

@@ -1 +0,0 @@
-{"version":3,"file":"lookup-stage.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/lookup-stage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,EAAE,MAAM,SAAS,CAAA;AAGrE,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,WAAW,EACnB,OAAO,EAAE,YAAY,GACpB,WAAW,CAab"}