From 5ba9699764b5f6d17828b781d670052c41c13a53 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ivan=20=E2=80=9CCLOVIS=E2=80=9D=20Canet?=
 <ivan.canet@gmail.com>
Date: Mon, 10 Aug 2026 22:29:21 +0200
Subject: [PATCH] feat(driver-api): Missing specialization of lookup()

---
 .../jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt  | 7 +++----
 .../kotlin/CoroutineMongoAggregationPipelineImpl.kt      | 9 +++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt
index d13b4325..7c21b6bc 100644
--- a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt
+++ b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipeline.kt
@@ -22,10 +22,7 @@ package opensavvy.ktmongo.coroutines
 import opensavvy.ktmongo.api.MongoAggregationPipeline
 import opensavvy.ktmongo.dsl.LowLevelApi
 import opensavvy.ktmongo.dsl.aggregation.AccumulationOperators
-import opensavvy.ktmongo.dsl.aggregation.stages.HasUnionWithCompatibility
-import opensavvy.ktmongo.dsl.aggregation.stages.ProjectStageOperators
-import opensavvy.ktmongo.dsl.aggregation.stages.SetStageOperators
-import opensavvy.ktmongo.dsl.aggregation.stages.UnsetStageOperators
+import opensavvy.ktmongo.dsl.aggregation.stages.*
 import opensavvy.ktmongo.dsl.options.SortOptionDsl
 import opensavvy.ktmongo.dsl.path.Field
 import opensavvy.ktmongo.dsl.query.FilterQuery
@@ -70,6 +67,8 @@ interface CoroutineMongoAggregationPipeline<Document : Any> : MongoAggregationPi
 
 	override fun unionWith(other: HasUnionWithCompatibility<Document>): CoroutineMongoAggregationPipeline<Document>
 
+	override fun <ForeignDocument : Any> lookup(block: LookupStageOperators<Document, ForeignDocument>.() -> Unit): CoroutineMongoAggregationPipeline<Document>
+
 	override fun <Out : Any> group(block: AccumulationOperators<Document, Out>.() -> Unit): CoroutineMongoAggregationPipeline<Out>
 
 	override fun <Out : Any> countTo(field: Field<Out, Number>): CoroutineMongoAggregationPipeline<Out>
diff --git a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt
index 50757f44..62a0d175 100644
--- a/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt
+++ b/driver-coroutines/src/jvmMain/kotlin/CoroutineMongoAggregationPipelineImpl.kt
@@ -29,10 +29,7 @@ import opensavvy.ktmongo.dsl.aggregation.AbstractPipeline
 import opensavvy.ktmongo.dsl.aggregation.AccumulationOperators
 import opensavvy.ktmongo.dsl.aggregation.AggregationPipeline
 import opensavvy.ktmongo.dsl.aggregation.PipelineChainLink
-import opensavvy.ktmongo.dsl.aggregation.stages.HasUnionWithCompatibility
-import opensavvy.ktmongo.dsl.aggregation.stages.ProjectStageOperators
-import opensavvy.ktmongo.dsl.aggregation.stages.SetStageOperators
-import opensavvy.ktmongo.dsl.aggregation.stages.UnsetStageOperators
+import opensavvy.ktmongo.dsl.aggregation.stages.*
 import opensavvy.ktmongo.dsl.options.SortOptionDsl
 import opensavvy.ktmongo.dsl.path.Field
 import opensavvy.ktmongo.dsl.query.FilterQuery
@@ -120,6 +117,10 @@ private class CoroutineMongoAggregationPipelineImpl<Document : Any> @OptIn(LowLe
 	override fun unionWith(other: HasUnionWithCompatibility<Document>): CoroutineMongoAggregationPipelineImpl<Document> =
 		super<AggregationPipeline>.unionWith(other) as CoroutineMongoAggregationPipelineImpl<Document>
 
+	@KtMongoDsl
+	override fun <ForeignDocument : Any> lookup(block: LookupStageOperators<Document, ForeignDocument>.() -> Unit): CoroutineMongoAggregationPipelineImpl<Document> =
+		super<AggregationPipeline>.lookup(block) as CoroutineMongoAggregationPipelineImpl<Document>
+
 	@KtMongoDsl
 	override fun <Out : Any> group(block: AccumulationOperators<Document, Out>.() -> Unit): CoroutineMongoAggregationPipelineImpl<Out> =
 		super<AggregationPipeline>.group(block) as CoroutineMongoAggregationPipelineImpl<Out>
-- 
2.51.2