mongo.do 0.1.0

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

@@ -0,0 +1 @@
+{"version":3,"file":"project-stage.js","sourceRoot":"","sources":["../../../src/translator/stages/project-stage.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAA;AAC7F,OAAO,EAAE,iBAAiB,EAAE,MAAM,2BAA2B,CAAA;AAC7D,OAAO,EAAmB,WAAW,IAAI,kBAAkB,EAAE,MAAM,YAAY,CAAA;AAE/E;;GAEG;AACH,SAAS,wBAAwB,CAAC,IAAa;IAC7C,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;QAAE,OAAO,EAAE,CAAA;IAEhD,MAAM,MAAM,GAAa,EAAE,CAAA;IAC3B,MAAM,OAAO,GAAG,IAA+B,CAAA;IAE/C,0CAA0C;IAC1C,IAAI,WAAW,IAAI,OAAO,EAAE,CAAC;QAC3B,MAAM,MAAM,GAAG,OAAO,CAAC,SAAiC,CAAA;QACxD,IAAI,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YAC9C,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC9B,IAAI,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC;oBAC1B,+CAA+C;oBAC/C,MAAM,SAAS,GAAI,GAAc,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAA;oBAC5D,IAAI,SAAS,EAAE,CAAC;wBACd,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;oBACxB,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,mCAAmC;IACnC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3C,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACvC,MAAM,CAAC,IAAI,CAAC,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC,CAAA;QACjD,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAA;AACf,CAAC;AAED,MAAM,UAAU,qBAAqB,CACnC,UAAmC,EACnC,OAAqB;IAErB,MAAM,MAAM,GAAc,EAAE,CAAA;IAC5B,MAAM,OAAO,GAAe,OAAO,CAAC,OAAO,IAAI,QAAQ,CAAA;IACvD,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAA;IAEzC,sEAAsE;IACtE,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE;QAChD,IAAI,GAAG,KAAK,KAAK;YAAE,OAAO,IAAI,CAAA;QAC9B,OAAO,KAAK,KAAK,CAAC,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,IAAI,WAAW,EAAE,CAAC;QAChB,OAAO,yBAAyB,CAAC,UAAU,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;IACxE,CAAC;IAED,mCAAmC;IACnC,OAAO,yBAAyB,CAAC,UAAU,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;AACxE,CAAC;AAED,SAAS,yBAAyB,CAChC,UAAmC,EACnC,QAAsB,EACtB,MAAiB,EACjB,UAAsB,QAAQ;IAE9B,MAAM,cAAc,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC;SAC9C,MAAM,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,KAAK,KAAK,CAAC,IAAI,GAAG,KAAK,KAAK,CAAC;SACtD,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,EAAE;QACb,+CAA+C;QAC/C,iBAAiB,CAAC,GAAG,CAAC,CAAA;QACtB,OAAO,MAAM,GAAG,GAAG,CAAA;IACrB,CAAC,CAAC,CAAA;IAEJ,IAAI,YAAoB,CAAA;IACxB,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9B,IAAI,OAAO,KAAK,YAAY,EAAE,CAAC;YAC7B,qEAAqE;YACrE,YAAY,GAAG,MAAM,CAAA,CAAC,qBAAqB;QAC7C,CAAC;aAAM,CAAC;YACN,YAAY,GAAG,qBAAqB,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAA;QAC1E,CAAC;IACH,CAAC;SAAM,CAAC;QACN,YAAY,GAAG,MAAM,CAAA;IACvB,CAAC;IAED,OAAO;QACL,YAAY;QACZ,MAAM;KACP,CAAA;AACH,CAAC;AAED,SAAS,yBAAyB,CAChC,UAAmC,EACnC,QAAsB,EACtB,MAAiB,EACjB,UAAsB,QAAQ;IAE9B,MAAM,SAAS,GAAa,EAAE,CAAA;IAE9B,+DAA+D;IAC/D,oFAAoF;IACpF,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAAU,CAAA;IAC3C,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACnD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YAChD,MAAM,IAAI,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAA;YAC5C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,iBAAiB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAA;QACjD,CAAC;IACH,CAAC;IAED,6CAA6C;IAC7C,MAAM,cAAc,GAAG,IAAI,GAAG,EAAU,CAAA;IACxC,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACtD,4DAA4D;QAC5D,iBAAiB,CAAC,GAAG,CAAC,CAAA;QACtB,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC,CAAA;QACvB,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YAChB,yBAAyB;YACzB,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,GAAG,EAAE,CAAC,CAAA;YACjE,SAAS,CAAC,IAAI,CAAC,IAAI,GAAG,MAAM,SAAS,EAAE,CAAC,CAAA;QAC1C,CAAC;aAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC9D,gEAAgE;YAChE,MAAM,SAAS,GAAG,YAAY,CAAC,KAAK,CAAC,CAAA;YACrC,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,CAAA;YAChE,SAAS,CAAC,IAAI,CAAC,IAAI,GAAG,MAAM,SAAS,EAAE,CAAC,CAAA;QAC1C,CAAC;aAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;YACvD,aAAa;YACb,MAAM,OAAO,GAAG,mBAAmB,CAAC,KAAgC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAA;YACtF,SAAS,CAAC,IAAI,CAAC,IAAI,GAAG,MAAM,OAAO,EAAE,CAAC,CAAA;QACxC,CAAC;aAAM,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YACvB,gBAAgB;YAChB,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;gBAClB,SAAS,CAAC,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,CAAA;YAC/B,CAAC;iBAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;gBACnE,SAAS,CAAC,IAAI,CAAC,IAAI,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;YACtD,CAAC;QACH,CAAC;IACH,CAAC;IAED,sEAAsE;IACtE,yEAAyE;IACzE,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;QACzC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClC,oDAAoD;YACpD,iBAAiB,CAAC,QAAQ,CAAC,CAAA;YAC3B,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,QAAQ,EAAE,CAAC,CAAA;YACtE,SAAS,CAAC,IAAI,CAAC,IAAI,QAAQ,MAAM,SAAS,EAAE,CAAC,CAAA;QAC/C,CAAC;IACH,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,KAAK,YAAY;QAC3C,CAAC,CAAC,SAAS,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW;QAC1C,CAAC,CAAC,eAAe,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,CAAA;IAElD,OAAO;QACL,YAAY;QACZ,MAAM;QACN,eAAe,EAAE,IAAI;KACtB,CAAA;AACH,CAAC"}

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

@@ -0,0 +1,60 @@
+/**
+ * $search stage - MongoDB Atlas Search integration
+ *
+ * Translates MongoDB $search aggregation stage to SQLite FTS5 queries.
+ * The $search stage must be the first stage in an aggregation pipeline.
+ *
+ * Supported operators:
+ * - text: Full-text search on specified fields
+ * - phrase: Exact phrase matching
+ * - wildcard: Wildcard pattern matching
+ * - compound: Boolean combinations (must, should, mustNot, filter)
+ */
+import { type SearchOperator } from '../search-translator';
+import type { StageResult, StageContext } from './types';
+/**
+ * Extended context for search stage
+ */
+export interface SearchStageContext extends StageContext {
+    /** Include relevance score in results */
+    includeScore?: boolean;
+}
+/**
+ * Result from search stage translation
+ */
+export interface SearchStageResult extends StageResult {
+    /** FTS5 MATCH expression */
+    ftsMatch: string;
+    /** FTS table name */
+    ftsTable: string;
+    /** JOIN clause for FTS table */
+    ftsJoin: string;
+}
+/**
+ * $search stage input with optional index name
+ */
+export interface SearchStageInput extends SearchOperator {
+    /** Named search index to use */
+    index?: string;
+}
+/**
+ * Translate a $search aggregation stage to SQL
+ *
+ * @param searchSpec The $search stage specification
+ * @param context Stage context including collection name
+ * @returns SearchStageResult with FTS5 query components
+ */
+export declare function translateSearchStage(searchSpec: SearchStageInput, context: SearchStageContext): SearchStageResult;
+/**
+ * Build complete SQL for a $search stage
+ *
+ * @param searchSpec The $search specification
+ * @param collection Collection name
+ * @param includeScore Whether to include relevance score
+ * @returns Complete SQL query and parameters
+ */
+export declare function buildSearchSQL(searchSpec: SearchStageInput, collection: string, includeScore?: boolean): {
+    sql: string;
+    params: unknown[];
+};
+//# sourceMappingURL=search-stage.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"search-stage.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/search-stage.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAoB,KAAK,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC7E,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAEzD;;GAEG;AACH,MAAM,WAAW,kBAAmB,SAAQ,YAAY;IACtD,yCAAyC;IACzC,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,WAAW,iBAAkB,SAAQ,WAAW;IACpD,4BAA4B;IAC5B,QAAQ,EAAE,MAAM,CAAC;IACjB,qBAAqB;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,gCAAgC;IAChC,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;GAEG;AACH,MAAM,WAAW,gBAAiB,SAAQ,cAAc;IACtD,gCAAgC;IAChC,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;;;;;GAMG;AACH,wBAAgB,oBAAoB,CAClC,UAAU,EAAE,gBAAgB,EAC5B,OAAO,EAAE,kBAAkB,GAC1B,iBAAiB,CA8CnB;AAED;;;;;;;GAOG;AACH,wBAAgB,cAAc,CAC5B,UAAU,EAAE,gBAAgB,EAC5B,UAAU,EAAE,MAAM,EAClB,YAAY,GAAE,OAAe,GAC5B;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,OAAO,EAAE,CAAA;CAAE,CAwBpC"}

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

@@ -0,0 +1,89 @@
+/**
+ * $search stage - MongoDB Atlas Search integration
+ *
+ * Translates MongoDB $search aggregation stage to SQLite FTS5 queries.
+ * The $search stage must be the first stage in an aggregation pipeline.
+ *
+ * Supported operators:
+ * - text: Full-text search on specified fields
+ * - phrase: Exact phrase matching
+ * - wildcard: Wildcard pattern matching
+ * - compound: Boolean combinations (must, should, mustNot, filter)
+ */
+import { SearchTranslator } from '../search-translator';
+/**
+ * Translate a $search aggregation stage to SQL
+ *
+ * @param searchSpec The $search stage specification
+ * @param context Stage context including collection name
+ * @returns SearchStageResult with FTS5 query components
+ */
+export function translateSearchStage(searchSpec, context) {
+    const translator = new SearchTranslator();
+    // Extract the search operator (remove 'index' if present)
+    const { index, ...searchOperator } = searchSpec;
+    // Determine FTS table name
+    const ftsTable = `${context.collection}_fts`;
+    // Translate the search operator to FTS5 MATCH expression
+    const searchResult = translator.translateSearch(searchOperator, context.collection);
+    // Build the JOIN clause for FTS5 table
+    const ftsJoin = `JOIN ${ftsTable} ON documents.id = ${ftsTable}.rowid`;
+    // Build SELECT clause with optional score
+    let selectClause;
+    if (context.includeScore) {
+        // FTS5 bm25() returns negative values (more negative = more relevant)
+        // Negate to get positive scores where higher = more relevant
+        selectClause = `*, -bm25(${ftsTable}) AS _searchScore`;
+    }
+    // Build WHERE clause
+    const whereClause = `${ftsTable} MATCH ?`;
+    // Build ORDER BY clause for relevance ranking
+    const orderByClause = context.includeScore ? `_searchScore DESC` : undefined;
+    const result = {
+        ftsMatch: searchResult.ftsMatch,
+        ftsTable,
+        ftsJoin,
+        whereClause,
+        params: [searchResult.ftsMatch],
+    };
+    if (selectClause !== undefined) {
+        result.selectClause = selectClause;
+    }
+    if (orderByClause !== undefined) {
+        result.orderByClause = orderByClause;
+    }
+    if (context.includeScore !== undefined) {
+        result.transformsShape = context.includeScore;
+    }
+    return result;
+}
+/**
+ * Build complete SQL for a $search stage
+ *
+ * @param searchSpec The $search specification
+ * @param collection Collection name
+ * @param includeScore Whether to include relevance score
+ * @returns Complete SQL query and parameters
+ */
+export function buildSearchSQL(searchSpec, collection, includeScore = false) {
+    const context = {
+        collection,
+        cteIndex: 0,
+        existingParams: [],
+        includeScore,
+    };
+    const result = translateSearchStage(searchSpec, context);
+    const selectClause = result.selectClause || '*';
+    const sql = `
+    SELECT ${selectClause}
+    FROM documents
+    ${result.ftsJoin}
+    WHERE ${result.whereClause}
+    ${result.orderByClause ? `ORDER BY ${result.orderByClause}` : ''}
+  `.trim().replace(/\s+/g, ' ');
+    return {
+        sql,
+        params: result.params,
+    };
+}
+//# sourceMappingURL=search-stage.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"search-stage.js","sourceRoot":"","sources":["../../../src/translator/stages/search-stage.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,gBAAgB,EAAuB,MAAM,sBAAsB,CAAC;AA+B7E;;;;;;GAMG;AACH,MAAM,UAAU,oBAAoB,CAClC,UAA4B,EAC5B,OAA2B;IAE3B,MAAM,UAAU,GAAG,IAAI,gBAAgB,EAAE,CAAC;IAE1C,0DAA0D;IAC1D,MAAM,EAAE,KAAK,EAAE,GAAG,cAAc,EAAE,GAAG,UAAU,CAAC;IAEhD,2BAA2B;IAC3B,MAAM,QAAQ,GAAG,GAAG,OAAO,CAAC,UAAU,MAAM,CAAC;IAE7C,yDAAyD;IACzD,MAAM,YAAY,GAAG,UAAU,CAAC,eAAe,CAAC,cAAc,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC;IAEpF,uCAAuC;IACvC,MAAM,OAAO,GAAG,QAAQ,QAAQ,sBAAsB,QAAQ,QAAQ,CAAC;IAEvE,0CAA0C;IAC1C,IAAI,YAAgC,CAAC;IACrC,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;QACzB,sEAAsE;QACtE,6DAA6D;QAC7D,YAAY,GAAG,YAAY,QAAQ,mBAAmB,CAAC;IACzD,CAAC;IAED,qBAAqB;IACrB,MAAM,WAAW,GAAG,GAAG,QAAQ,UAAU,CAAC;IAE1C,8CAA8C;IAC9C,MAAM,aAAa,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,SAAS,CAAC;IAE7E,MAAM,MAAM,GAAsB;QAChC,QAAQ,EAAE,YAAY,CAAC,QAAQ;QAC/B,QAAQ;QACR,OAAO;QACP,WAAW;QACX,MAAM,EAAE,CAAC,YAAY,CAAC,QAAQ,CAAC;KAChC,CAAC;IACF,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;QAC/B,MAAM,CAAC,YAAY,GAAG,YAAY,CAAC;IACrC,CAAC;IACD,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;QAChC,MAAM,CAAC,aAAa,GAAG,aAAa,CAAC;IACvC,CAAC;IACD,IAAI,OAAO,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;QACvC,MAAM,CAAC,eAAe,GAAG,OAAO,CAAC,YAAY,CAAC;IAChD,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,cAAc,CAC5B,UAA4B,EAC5B,UAAkB,EAClB,eAAwB,KAAK;IAE7B,MAAM,OAAO,GAAuB;QAClC,UAAU;QACV,QAAQ,EAAE,CAAC;QACX,cAAc,EAAE,EAAE;QAClB,YAAY;KACb,CAAC;IAEF,MAAM,MAAM,GAAG,oBAAoB,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;IAEzD,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,GAAG,CAAC;IAEhD,MAAM,GAAG,GAAG;aACD,YAAY;;MAEnB,MAAM,CAAC,OAAO;YACR,MAAM,CAAC,WAAW;MACxB,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,YAAY,MAAM,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC,EAAE;GACjE,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAE9B,OAAO;QACL,GAAG;QACH,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC;AACJ,CAAC"}

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

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

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

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

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

@@ -0,0 +1,11 @@
+/**
+ * $skip stage - Skips a number of documents
+ * Translates to SQL OFFSET clause
+ */
+export function translateSkipStage(skip, _context) {
+    return {
+        offsetClause: `OFFSET ${skip}`,
+        params: []
+    };
+}
+//# sourceMappingURL=skip-stage.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"skip-stage.js","sourceRoot":"","sources":["../../../src/translator/stages/skip-stage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,MAAM,UAAU,kBAAkB,CAChC,IAAY,EACZ,QAAsB;IAEtB,OAAO;QACL,YAAY,EAAE,UAAU,IAAI,EAAE;QAC9B,MAAM,EAAE,EAAE;KACX,CAAA;AACH,CAAC"}

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

@@ -0,0 +1,7 @@
+/**
+ * $sort stage - Sorts documents by specified fields
+ * Translates to SQL ORDER BY clause
+ */
+import type { StageResult, StageContext } from './types';
+export declare function translateSortStage(sort: Record<string, 1 | -1>, _context: StageContext): StageResult;
+//# sourceMappingURL=sort-stage.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"sort-stage.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/sort-stage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,SAAS,CAAA;AAGxD,wBAAgB,kBAAkB,CAChC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAC5B,QAAQ,EAAE,YAAY,GACrB,WAAW,CAiBb"}

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

@@ -0,0 +1,21 @@
+/**
+ * $sort stage - Sorts documents by specified fields
+ * Translates to SQL ORDER BY clause
+ */
+import { getFieldPath } from './expression-translator';
+export function translateSortStage(sort, _context) {
+    const orderParts = [];
+    for (const [field, direction] of Object.entries(sort)) {
+        if (direction !== 1 && direction !== -1) {
+            throw new Error('$sort direction must be 1 or -1');
+        }
+        const path = getFieldPath('$' + field);
+        const dirStr = direction === 1 ? 'ASC' : 'DESC';
+        orderParts.push(`json_extract(data, '${path}') ${dirStr}`);
+    }
+    return {
+        orderByClause: orderParts.join(', '),
+        params: []
+    };
+}
+//# sourceMappingURL=sort-stage.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"sort-stage.js","sourceRoot":"","sources":["../../../src/translator/stages/sort-stage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAA;AAEtD,MAAM,UAAU,kBAAkB,CAChC,IAA4B,EAC5B,QAAsB;IAEtB,MAAM,UAAU,GAAa,EAAE,CAAA;IAE/B,KAAK,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;QACtD,IAAI,SAAS,KAAK,CAAC,IAAI,SAAS,KAAK,CAAC,CAAC,EAAE,CAAC;YACxC,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAA;QACpD,CAAC;QAED,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,GAAG,KAAK,CAAC,CAAA;QACtC,MAAM,MAAM,GAAG,SAAS,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAA;QAC/C,UAAU,CAAC,IAAI,CAAC,uBAAuB,IAAI,MAAM,MAAM,EAAE,CAAC,CAAA;IAC5D,CAAC;IAED,OAAO;QACL,aAAa,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;QACpC,MAAM,EAAE,EAAE;KACX,CAAA;AACH,CAAC"}

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

@@ -0,0 +1,136 @@
+/**
+ * Types for aggregation pipeline stages
+ */
+import type { SQLDialect, DialectOptions } from '../dialect';
+export interface StageResult {
+    selectClause?: string;
+    fromClause?: string;
+    whereClause?: string;
+    groupByClause?: string;
+    havingClause?: string;
+    orderByClause?: string;
+    limitClause?: string;
+    offsetClause?: string;
+    params: unknown[];
+    cteExpression?: string;
+    cteName?: string;
+    facets?: Record<string, {
+        sql: string;
+        params: unknown[];
+    }>;
+    transformsShape?: boolean;
+}
+export interface StageContext {
+    collection: string;
+    previousCte?: string;
+    cteIndex: number;
+    existingParams: unknown[];
+    /** SQL dialect to use */
+    dialect?: SQLDialect;
+    /** Dialect options */
+    dialectOptions?: DialectOptions;
+}
+export interface AggregationResult {
+    sql: string;
+    params: unknown[];
+    facets?: Record<string, {
+        sql: string;
+        params: unknown[];
+    }>;
+}
+export type PipelineStage = {
+    $match: Record<string, unknown>;
+} | {
+    $project: Record<string, unknown>;
+} | {
+    $group: GroupStage;
+} | {
+    $sort: Record<string, 1 | -1>;
+} | {
+    $limit: number;
+} | {
+    $skip: number;
+} | {
+    $count: string;
+} | {
+    $lookup: LookupStage;
+} | {
+    $unwind: string | UnwindStage;
+} | {
+    $addFields: Record<string, unknown>;
+} | {
+    $bucket: BucketStage;
+} | {
+    $facet: Record<string, PipelineStage[]>;
+} | {
+    $search: SearchStage;
+} | {
+    $vectorSearch: VectorSearchPipelineStage;
+};
+export interface SearchStage {
+    index?: string;
+    text?: {
+        query: string;
+        path?: string | string[];
+    };
+    phrase?: {
+        query: string;
+        path?: string | string[];
+    };
+    wildcard?: {
+        query: string;
+        path?: string | string[];
+    };
+    compound?: {
+        must?: SearchStage[];
+        should?: SearchStage[];
+        mustNot?: SearchStage[];
+        filter?: SearchStage[];
+    };
+}
+export interface GroupStage {
+    _id: null | string | Record<string, string>;
+    [accumulator: string]: unknown;
+}
+export interface LookupStage {
+    from: string;
+    localField?: string;
+    foreignField?: string;
+    as: string;
+    let?: Record<string, string>;
+    pipeline?: PipelineStage[];
+}
+export interface UnwindStage {
+    path: string;
+    includeArrayIndex?: string;
+    preserveNullAndEmptyArrays?: boolean;
+}
+export interface BucketStage {
+    groupBy: string;
+    boundaries: number[];
+    default?: string;
+    output?: Record<string, unknown>;
+}
+/**
+ * $vectorSearch pipeline stage for vector similarity search
+ * Compatible with MongoDB Atlas vector search
+ */
+export interface VectorSearchPipelineStage {
+    /** Name of the vector search index */
+    index: string;
+    /** Path to the vector field in documents */
+    path: string;
+    /** The query vector to search for */
+    queryVector: number[];
+    /** Number of candidates to consider (default: limit * 10) */
+    numCandidates?: number;
+    /** Maximum number of results to return */
+    limit: number;
+    /** Optional filter to apply before vector search */
+    filter?: Record<string, unknown>;
+    /** Whether to perform exact search (slower but more accurate) */
+    exact?: boolean;
+    /** Field name to store the similarity score (default: 'score') */
+    scoreField?: string;
+}
+//# sourceMappingURL=types.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,KAAK,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAE5D,MAAM,WAAW,WAAW;IAE1B,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,YAAY,CAAC,EAAE,MAAM,CAAA;IAGrB,MAAM,EAAE,OAAO,EAAE,CAAA;IAGjB,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,OAAO,CAAC,EAAE,MAAM,CAAA;IAGhB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,OAAO,EAAE,CAAA;KAAE,CAAC,CAAA;IAG3D,eAAe,CAAC,EAAE,OAAO,CAAA;CAC1B;AAED,MAAM,WAAW,YAAY;IAC3B,UAAU,EAAE,MAAM,CAAA;IAClB,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,QAAQ,EAAE,MAAM,CAAA;IAChB,cAAc,EAAE,OAAO,EAAE,CAAA;IACzB,yBAAyB;IACzB,OAAO,CAAC,EAAE,UAAU,CAAA;IACpB,sBAAsB;IACtB,cAAc,CAAC,EAAE,cAAc,CAAA;CAChC;AAED,MAAM,WAAW,iBAAiB;IAChC,GAAG,EAAE,MAAM,CAAA;IACX,MAAM,EAAE,OAAO,EAAE,CAAA;IACjB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,MAAM,EAAE,OAAO,EAAE,CAAA;KAAE,CAAC,CAAA;CAC5D;AAGD,MAAM,MAAM,aAAa,GACrB;IAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,GACnC;IAAE,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,GACrC;IAAE,MAAM,EAAE,UAAU,CAAA;CAAE,GACtB;IAAE,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAA;CAAE,GACjC;IAAE,MAAM,EAAE,MAAM,CAAA;CAAE,GAClB;IAAE,KAAK,EAAE,MAAM,CAAA;CAAE,GACjB;IAAE,MAAM,EAAE,MAAM,CAAA;CAAE,GAClB;IAAE,OAAO,EAAE,WAAW,CAAA;CAAE,GACxB;IAAE,OAAO,EAAE,MAAM,GAAG,WAAW,CAAA;CAAE,GACjC;IAAE,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,GACvC;IAAE,OAAO,EAAE,WAAW,CAAA;CAAE,GACxB;IAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,aAAa,EAAE,CAAC,CAAA;CAAE,GAC3C;IAAE,OAAO,EAAE,WAAW,CAAA;CAAE,GACxB;IAAE,aAAa,EAAE,yBAAyB,CAAA;CAAE,CAAA;AAEhD,MAAM,WAAW,WAAW;IAC1B,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,IAAI,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAA;KAAE,CAAA;IAClD,MAAM,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAA;KAAE,CAAA;IACpD,QAAQ,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAA;KAAE,CAAA;IACtD,QAAQ,CAAC,EAAE;QACT,IAAI,CAAC,EAAE,WAAW,EAAE,CAAA;QACpB,MAAM,CAAC,EAAE,WAAW,EAAE,CAAA;QACtB,OAAO,CAAC,EAAE,WAAW,EAAE,CAAA;QACvB,MAAM,CAAC,EAAE,WAAW,EAAE,CAAA;KACvB,CAAA;CACF;AAED,MAAM,WAAW,UAAU;IACzB,GAAG,EAAE,IAAI,GAAG,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC3C,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAA;CAC/B;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAA;IACZ,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,EAAE,EAAE,MAAM,CAAA;IACV,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC5B,QAAQ,CAAC,EAAE,aAAa,EAAE,CAAA;CAC3B;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAA;IACZ,iBAAiB,CAAC,EAAE,MAAM,CAAA;IAC1B,0BAA0B,CAAC,EAAE,OAAO,CAAA;CACrC;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,MAAM,CAAA;IACf,UAAU,EAAE,MAAM,EAAE,CAAA;IACpB,OAAO,CAAC,EAAE,MAAM,CAAA;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CACjC;AAED;;;GAGG;AACH,MAAM,WAAW,yBAAyB;IACxC,sCAAsC;IACtC,KAAK,EAAE,MAAM,CAAA;IACb,4CAA4C;IAC5C,IAAI,EAAE,MAAM,CAAA;IACZ,qCAAqC;IACrC,WAAW,EAAE,MAAM,EAAE,CAAA;IACrB,6DAA6D;IAC7D,aAAa,CAAC,EAAE,MAAM,CAAA;IACtB,0CAA0C;IAC1C,KAAK,EAAE,MAAM,CAAA;IACb,oDAAoD;IACpD,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAChC,iEAAiE;IACjE,KAAK,CAAC,EAAE,OAAO,CAAA;IACf,kEAAkE;IAClE,UAAU,CAAC,EAAE,MAAM,CAAA;CACpB"}

package/dist/translator/stages/types.js

@@ -0,0 +1,5 @@
+/**
+ * Types for aggregation pipeline stages
+ */
+export {};
+//# sourceMappingURL=types.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/translator/stages/types.ts"],"names":[],"mappings":"AAAA;;GAEG"}

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

@@ -0,0 +1,7 @@
+/**
+ * $unwind stage - Deconstructs an array field to output one document per element
+ * Translates to SQL using json_each for SQLite or ARRAY JOIN for ClickHouse
+ */
+import type { StageResult, StageContext, UnwindStage } from './types';
+export declare function translateUnwindStage(unwind: string | UnwindStage, context: StageContext): StageResult;
+//# sourceMappingURL=unwind-stage.d.ts.map

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

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

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

@@ -0,0 +1,61 @@
+/**
+ * $unwind stage - Deconstructs an array field to output one document per element
+ * Translates to SQL using json_each for SQLite or ARRAY JOIN for ClickHouse
+ */
+import { getFieldPath } from './expression-translator';
+import { jsonExtract as dialectJsonExtract } from '../dialect';
+export function translateUnwindStage(unwind, context) {
+    const params = [];
+    const dialect = context.dialect || 'sqlite';
+    // Normalize to UnwindStage format
+    const config = typeof unwind === 'string'
+        ? { path: unwind }
+        : unwind;
+    const { path, includeArrayIndex, preserveNullAndEmptyArrays } = config;
+    // Remove leading $ from path
+    const fieldName = path.startsWith('$') ? path.substring(1) : path;
+    const jsonPath = getFieldPath('$' + fieldName);
+    const source = context.previousCte || context.collection;
+    const cteName = `stage_${context.cteIndex}`;
+    if (dialect === 'clickhouse') {
+        // ClickHouse uses ARRAY JOIN syntax
+        const arrayExpr = dialectJsonExtract(dialect, `${source}.doc`, jsonPath);
+        const leftJoin = preserveNullAndEmptyArrays ? 'LEFT ' : '';
+        let selectParts = `*, ${fieldName}_element AS ${fieldName}`;
+        if (includeArrayIndex) {
+            selectParts = `*, ${fieldName}_element AS ${fieldName}, ${fieldName}_index AS ${includeArrayIndex}`;
+        }
+        const cteExpression = `
+      SELECT ${selectParts}
+      FROM ${source}
+      ${leftJoin}ARRAY JOIN ${arrayExpr} AS ${fieldName}_element${includeArrayIndex ? `, arrayEnumerate(${arrayExpr}) AS ${fieldName}_index` : ''}
+    `;
+        return {
+            cteExpression: cteExpression.trim(),
+            cteName,
+            params,
+            transformsShape: true
+        };
+    }
+    // SQLite uses json_each
+    // Build the join type based on preserveNullAndEmptyArrays
+    const joinType = preserveNullAndEmptyArrays ? 'LEFT JOIN' : 'JOIN';
+    // Build the SELECT clause
+    let selectParts = `json_set(${source}.data, '${jsonPath}', each.value) AS data`;
+    if (includeArrayIndex) {
+        // Include the array index in the output
+        selectParts = `json_set(json_set(${source}.data, '${jsonPath}', each.value), '$.${includeArrayIndex}', each.key) AS data`;
+    }
+    const cteExpression = `
+    SELECT ${selectParts}
+    FROM ${source}
+    ${joinType} json_each(json_extract(${source}.data, '${jsonPath}')) AS each
+  `;
+    return {
+        cteExpression: cteExpression.trim(),
+        cteName,
+        params,
+        transformsShape: true
+    };
+}
+//# sourceMappingURL=unwind-stage.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"unwind-stage.js","sourceRoot":"","sources":["../../../src/translator/stages/unwind-stage.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,EAAE,YAAY,EAAE,MAAM,yBAAyB,CAAA;AACtD,OAAO,EAAmB,WAAW,IAAI,kBAAkB,EAAE,MAAM,YAAY,CAAA;AAE/E,MAAM,UAAU,oBAAoB,CAClC,MAA4B,EAC5B,OAAqB;IAErB,MAAM,MAAM,GAAc,EAAE,CAAA;IAC5B,MAAM,OAAO,GAAe,OAAO,CAAC,OAAO,IAAI,QAAQ,CAAA;IAEvD,kCAAkC;IAClC,MAAM,MAAM,GAAgB,OAAO,MAAM,KAAK,QAAQ;QACpD,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE;QAClB,CAAC,CAAC,MAAM,CAAA;IAEV,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,0BAA0B,EAAE,GAAG,MAAM,CAAA;IAEtE,6BAA6B;IAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAA;IACjE,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,GAAG,SAAS,CAAC,CAAA;IAE9C,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,IAAI,OAAO,CAAC,UAAU,CAAA;IACxD,MAAM,OAAO,GAAG,SAAS,OAAO,CAAC,QAAQ,EAAE,CAAA;IAE3C,IAAI,OAAO,KAAK,YAAY,EAAE,CAAC;QAC7B,oCAAoC;QACpC,MAAM,SAAS,GAAG,kBAAkB,CAAC,OAAO,EAAE,GAAG,MAAM,MAAM,EAAE,QAAQ,CAAC,CAAA;QACxE,MAAM,QAAQ,GAAG,0BAA0B,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAA;QAE1D,IAAI,WAAW,GAAG,MAAM,SAAS,eAAe,SAAS,EAAE,CAAA;QAC3D,IAAI,iBAAiB,EAAE,CAAC;YACtB,WAAW,GAAG,MAAM,SAAS,eAAe,SAAS,KAAK,SAAS,aAAa,iBAAiB,EAAE,CAAA;QACrG,CAAC;QAED,MAAM,aAAa,GAAG;eACX,WAAW;aACb,MAAM;QACX,QAAQ,cAAc,SAAS,OAAO,SAAS,WAAW,iBAAiB,CAAC,CAAC,CAAC,oBAAoB,SAAS,QAAQ,SAAS,QAAQ,CAAC,CAAC,CAAC,EAAE;KAC5I,CAAA;QAED,OAAO;YACL,aAAa,EAAE,aAAa,CAAC,IAAI,EAAE;YACnC,OAAO;YACP,MAAM;YACN,eAAe,EAAE,IAAI;SACtB,CAAA;IACH,CAAC;IAED,wBAAwB;IACxB,0DAA0D;IAC1D,MAAM,QAAQ,GAAG,0BAA0B,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAA;IAElE,0BAA0B;IAC1B,IAAI,WAAW,GAAG,YAAY,MAAM,WAAW,QAAQ,wBAAwB,CAAA;IAE/E,IAAI,iBAAiB,EAAE,CAAC;QACtB,wCAAwC;QACxC,WAAW,GAAG,qBAAqB,MAAM,WAAW,QAAQ,sBAAsB,iBAAiB,sBAAsB,CAAA;IAC3H,CAAC;IAED,MAAM,aAAa,GAAG;aACX,WAAW;WACb,MAAM;MACX,QAAQ,2BAA2B,MAAM,WAAW,QAAQ;GAC/D,CAAA;IAED,OAAO;QACL,aAAa,EAAE,aAAa,CAAC,IAAI,EAAE;QACnC,OAAO;QACP,MAAM;QACN,eAAe,EAAE,IAAI;KACtB,CAAA;AACH,CAAC"}

package/dist/translator/stages/vector-search-stage.d.ts

@@ -0,0 +1,53 @@
+/**
+ * $vectorSearch stage - Performs vector similarity search
+ *
+ * Translates MongoDB Atlas $vectorSearch stage for use with Cloudflare Vectorize.
+ * This stage must be the first stage in an aggregation pipeline.
+ */
+import type { VectorSearchStage } from '../vector-translator';
+import type { StageResult, StageContext } from './types';
+/**
+ * Translated vector search result with normalized field names
+ */
+export interface TranslatedVectorSearch {
+    /** The name of the vector index to use */
+    vectorIndexName: string;
+    /** Path to the vector field in documents */
+    vectorPath: string;
+    /** Query vector for similarity search */
+    queryVector?: number[];
+    /** Text query to convert to embedding (requires AI binding) */
+    queryText?: string;
+    /** Maximum number of documents to return */
+    limit: number;
+    /** Optional filter for pre-filtering documents */
+    filter?: Record<string, unknown>;
+    /** WHERE clause generated from filter */
+    whereClause?: string;
+    /** Field to store the similarity score */
+    scoreField?: string;
+    /** Whether to perform exact (brute-force) search */
+    exact?: boolean;
+}
+/**
+ * Extended StageResult for $vectorSearch that includes vector search info
+ */
+export interface VectorSearchStageResult extends StageResult {
+    /** Flag indicating this is a vector search stage */
+    isVectorSearchStage: boolean;
+    /** The collection being searched */
+    collection: string;
+    /** Vector search stage specification (to be executed later with VectorTranslator) */
+    vectorSearchStage?: VectorSearchStage;
+    /** Translated vector search parameters */
+    vectorSearch?: TranslatedVectorSearch;
+}
+/**
+ * Translate a $vectorSearch stage
+ *
+ * @param stage - The $vectorSearch stage specification
+ * @param context - The stage context with collection and CTE info
+ * @returns Stage result with vector search parameters
+ */
+export declare function translateVectorSearchStage(stage: VectorSearchStage, context: StageContext): VectorSearchStageResult;
+//# sourceMappingURL=vector-search-stage.d.ts.map

package/dist/translator/stages/vector-search-stage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector-search-stage.d.ts","sourceRoot":"","sources":["../../../src/translator/stages/vector-search-stage.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAA;AAC7D,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,SAAS,CAAA;AAExD;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,0CAA0C;IAC1C,eAAe,EAAE,MAAM,CAAA;IACvB,4CAA4C;IAC5C,UAAU,EAAE,MAAM,CAAA;IAClB,yCAAyC;IACzC,WAAW,CAAC,EAAE,MAAM,EAAE,CAAA;IACtB,+DAA+D;IAC/D,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,4CAA4C;IAC5C,KAAK,EAAE,MAAM,CAAA;IACb,kDAAkD;IAClD,MAAM,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAChC,yCAAyC;IACzC,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,0CAA0C;IAC1C,UAAU,CAAC,EAAE,MAAM,CAAA;IACnB,oDAAoD;IACpD,KAAK,CAAC,EAAE,OAAO,CAAA;CAChB;AAED;;GAEG;AACH,MAAM,WAAW,uBAAwB,SAAQ,WAAW;IAC1D,oDAAoD;IACpD,mBAAmB,EAAE,OAAO,CAAA;IAC5B,oCAAoC;IACpC,UAAU,EAAE,MAAM,CAAA;IAClB,qFAAqF;IACrF,iBAAiB,CAAC,EAAE,iBAAiB,CAAA;IACrC,0CAA0C;IAC1C,YAAY,CAAC,EAAE,sBAAsB,CAAA;CACtC;AAmBD;;;;;;GAMG;AACH,wBAAgB,0BAA0B,CACxC,KAAK,EAAE,iBAAiB,EACxB,OAAO,EAAE,YAAY,GACpB,uBAAuB,CAgCzB"}

package/dist/translator/stages/vector-search-stage.js

@@ -0,0 +1,62 @@
+/**
+ * $vectorSearch stage - Performs vector similarity search
+ *
+ * Translates MongoDB Atlas $vectorSearch stage for use with Cloudflare Vectorize.
+ * This stage must be the first stage in an aggregation pipeline.
+ */
+/**
+ * Generate a simple WHERE clause from a filter object
+ */
+function generateWhereClause(filter) {
+    const conditions = [];
+    for (const [key, value] of Object.entries(filter)) {
+        if (typeof value === 'string') {
+            conditions.push(`${key} = '${value}'`);
+        }
+        else if (typeof value === 'number' || typeof value === 'boolean') {
+            conditions.push(`${key} = ${value}`);
+        }
+        else {
+            conditions.push(`${key} = ${JSON.stringify(value)}`);
+        }
+    }
+    return conditions.join(' AND ');
+}
+/**
+ * Translate a $vectorSearch stage
+ *
+ * @param stage - The $vectorSearch stage specification
+ * @param context - The stage context with collection and CTE info
+ * @returns Stage result with vector search parameters
+ */
+export function translateVectorSearchStage(stage, context) {
+    // Build the translated vector search object
+    const vectorSearch = {
+        vectorIndexName: stage.index,
+        vectorPath: stage.path,
+        queryVector: stage.queryVector,
+        queryText: stage.queryText,
+        limit: stage.limit,
+    };
+    // Add filter and generate WHERE clause if provided
+    if (stage.filter) {
+        vectorSearch.filter = stage.filter;
+        vectorSearch.whereClause = generateWhereClause(stage.filter);
+    }
+    // Add optional fields if present
+    if (stage.scoreField !== undefined) {
+        vectorSearch.scoreField = stage.scoreField;
+    }
+    if (stage.exact !== undefined) {
+        vectorSearch.exact = stage.exact;
+    }
+    return {
+        isVectorSearchStage: true,
+        collection: context.collection,
+        vectorSearchStage: stage,
+        vectorSearch,
+        params: [],
+        transformsShape: true
+    };
+}
+//# sourceMappingURL=vector-search-stage.js.map

package/dist/translator/stages/vector-search-stage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector-search-stage.js","sourceRoot":"","sources":["../../../src/translator/stages/vector-search-stage.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AA2CH;;GAEG;AACH,SAAS,mBAAmB,CAAC,MAA+B;IAC1D,MAAM,UAAU,GAAa,EAAE,CAAA;IAC/B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAClD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,UAAU,CAAC,IAAI,CAAC,GAAG,GAAG,OAAO,KAAK,GAAG,CAAC,CAAA;QACxC,CAAC;aAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,OAAO,KAAK,KAAK,SAAS,EAAE,CAAC;YACnE,UAAU,CAAC,IAAI,CAAC,GAAG,GAAG,MAAM,KAAK,EAAE,CAAC,CAAA;QACtC,CAAC;aAAM,CAAC;YACN,UAAU,CAAC,IAAI,CAAC,GAAG,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC,CAAA;QACtD,CAAC;IACH,CAAC;IACD,OAAO,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;AACjC,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,0BAA0B,CACxC,KAAwB,EACxB,OAAqB;IAErB,4CAA4C;IAC5C,MAAM,YAAY,GAA2B;QAC3C,eAAe,EAAE,KAAK,CAAC,KAAK;QAC5B,UAAU,EAAE,KAAK,CAAC,IAAI;QACtB,WAAW,EAAE,KAAK,CAAC,WAAW;QAC9B,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,KAAK,EAAE,KAAK,CAAC,KAAK;KACnB,CAAA;IAED,mDAAmD;IACnD,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC;QACjB,YAAY,CAAC,MAAM,GAAG,KAAK,CAAC,MAAiC,CAAA;QAC7D,YAAY,CAAC,WAAW,GAAG,mBAAmB,CAAC,KAAK,CAAC,MAAiC,CAAC,CAAA;IACzF,CAAC;IAED,iCAAiC;IACjC,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QACnC,YAAY,CAAC,UAAU,GAAG,KAAK,CAAC,UAAU,CAAA;IAC5C,CAAC;IACD,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAC9B,YAAY,CAAC,KAAK,GAAG,KAAK,CAAC,KAAK,CAAA;IAClC,CAAC;IAED,OAAO;QACL,mBAAmB,EAAE,IAAI;QACzB,UAAU,EAAE,OAAO,CAAC,UAAU;QAC9B,iBAAiB,EAAE,KAAK;QACxB,YAAY;QACZ,MAAM,EAAE,EAAE;QACV,eAAe,EAAE,IAAI;KACtB,CAAA;AACH,CAAC"}