From facdb2168a51fc3c6566598bd0c4370251378523 Mon Sep 17 00:00:00 2001
From: Sebastian <sebastian@squidex.io>
Date: Mon, 11 Sep 2023 22:22:33 +0200
Subject: [PATCH] Fix lookup.

---
 .../Contents/Operations/Extensions.cs         | 27 +++++++++++++++----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/backend/src/Squidex.Domain.Apps.Entities.MongoDb/Contents/Operations/Extensions.cs b/backend/src/Squidex.Domain.Apps.Entities.MongoDb/Contents/Operations/Extensions.cs
index da077c338..5a1cc11ca 100644
--- a/backend/src/Squidex.Domain.Apps.Entities.MongoDb/Contents/Operations/Extensions.cs
+++ b/backend/src/Squidex.Domain.Apps.Entities.MongoDb/Contents/Operations/Extensions.cs
@@ -5,6 +5,7 @@
 //  All rights reserved. Licensed under the MIT license.
 // ==========================================================================
 
+using MongoDB.Bson;
 using MongoDB.Bson.Serialization;
 using MongoDB.Bson.Serialization.Attributes;
 using MongoDB.Driver;
@@ -18,6 +19,15 @@ namespace Squidex.Domain.Apps.Entities.MongoDb.Contents.Operations;
 
 public static class Extensions
 {
+    private static readonly BsonDocument LookupLet =
+        new BsonDocument()
+            .Add("id", "$_id");
+
+    private static readonly BsonDocument LookupMatch =
+        new BsonDocument()
+            .Add("$expr", new BsonDocument()
+                .Add("$eq", new BsonArray { "$_id", "$$id" }));
+
     private static Dictionary<string, string> propertyMap;
 
     public static IReadOnlyDictionary<string, string> PropertyMap
@@ -113,8 +123,15 @@ public static class Extensions
                     .QuerySort(query)
                     .QuerySkip(query)
                     .QueryLimit(query)
-                    .Lookup<IdOnly, MongoContentEntity, IdOnly>(collection, x => x.Id, x => x.DocumentId, x => x.Joined)
-                    .SelectFields(q.Fields)
+                    .Lookup<IdOnly, MongoContentEntity, MongoContentEntity, MongoContentEntity[], IdOnly>(collection,
+                        LookupLet,
+                        PipelineDefinitionBuilder.For<MongoContentEntity>()
+                            .Match(LookupMatch)
+                            .Project(
+                                BuildProjection2<MongoContentEntity>(q.Fields)),
+                        x => x.Joined)
+                    .Project<IdOnly>(
+                        Builders<IdOnly>.Projection.Include(x => x.Joined))
                     .ToListAsync(ct);
 
             return joined.Select(x => x.Joined[0]).ToList();
@@ -147,15 +164,15 @@ public static class Extensions
 
     public static IFindFluent<T, T> SelectFields<T>(this IFindFluent<T, T> find, IEnumerable<string>? fields)
     {
-        return find.Project<T>(BuildProjection<T>(fields));
+        return find.Project<T>(BuildProjection2<T>(fields));
     }
 
     public static IAggregateFluent<T> SelectFields<T>(this IAggregateFluent<T> find, IEnumerable<string>? fields)
     {
-        return find.Project<T>(BuildProjection<T>(fields));
+        return find.Project<T>(BuildProjection2<T>(fields));
     }
 
-    private static ProjectionDefinition<T> BuildProjection<T>(IEnumerable<string>? fields)
+    private static ProjectionDefinition<T, T> BuildProjection2<T>(IEnumerable<string>? fields)
     {
         var projector = Builders<T>.Projection;
         var projections = new List<ProjectionDefinition<T>>();