Recursively replace .id with ._id

Author: GromNaN

src/Query/Builder.php

@@ -298,6 +298,7 @@ public function toMql(): array
         }
 
         $wheres = $this->compileWheres();
+        $wheres = $this->aliasIdForQuery($wheres);
 
         // Use MongoDB's aggregation framework when using grouping or aggregation functions.
         if ($this->groups || $this->aggregate) {
@@ -1634,13 +1635,33 @@ private function aliasIdForQuery(array $values): array
             unset($values['id']);
         }
 
+        foreach ($values as $key => $value) {
+            if (is_string($key) && str_ends_with($key, '.id')) {
+                $values[substr($key, 0, -3) . '._id'] = $value;
+                unset($values[$key]);
+            }
+        }
+
+        foreach ($values as &$value) {
+            if (is_array($value)) {
+                $value = $this->aliasIdForQuery($value);
+            }
+        }
+
         return $values;
     }
 
     private function aliasIdForResult(array $values): array
     {
         if (isset($values['_id'])) {
             $values['id'] = $values['_id'];
+            unset($values['_id']);
+        }
+
+        foreach ($values as $key => $value) {
+            if (is_array($value)) {
+                $values[$key] = $this->aliasIdForResult($value);
+            }
         }
 
         return $values;

tests/Ticket/GH2489Test.php

@@ -0,0 +1,52 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MongoDB\Laravel\Tests\Ticket;
+
+use Illuminate\Database\Eloquent\Relations\MorphOne;
+use MongoDB\Laravel\Eloquent\Model;
+use MongoDB\Laravel\Relations\MorphTo;
+use MongoDB\Laravel\Tests\Models\Location;
+use MongoDB\Laravel\Tests\TestCase;
+
+/** @see https://github.com/mongodb/laravel-mongodb/issues/2783 */
+class GH2489Test extends TestCase
+{
+    public function tearDown(): void
+    {
+        Location::truncate();
+    }
+
+    public function testQuerySubdocumentsUsingWhereInId()
+    {
+        Location::insert([
+            [
+                'name' => 'Location 1',
+                'images' => [
+                    ['_id' => 1, 'uri' => 'image1.jpg'],
+                    ['_id' => 2, 'uri' => 'image2.jpg'],
+                ],
+            ],
+            [
+                'name' => 'Location 2',
+                'images' => [
+                    ['_id' => 3, 'uri' => 'image3.jpg'],
+                    ['_id' => 4, 'uri' => 'image4.jpg'],
+                ],
+            ],
+        ]);
+
+        // With _id
+        $results = Location::whereIn('images._id', [1])->get();
+
+        $this->assertCount(1, $results);
+        $this->assertSame('Location 1', $results->first()->name);
+
+        // With id
+        $results = Location::whereIn('images.id', [1])->get();
+
+        $this->assertCount(1, $results);
+        $this->assertSame('Location 1', $results->first()->name);
+    }
+}