Merge pull request spatie#53 from vpratfr/find-with-uuid

Add some eloquent-ness to the trait

Author: brendt

src/Builder.php

@@ -0,0 +1,39 @@
+<?php
+
+namespace Spatie\BinaryUuid;
+
+use Illuminate\Database\Eloquent\Builder as EloquentBuilder;
+
+class Builder extends EloquentBuilder {
+  /**
+   * Find a model by its primary key.
+   *
+   * @param  mixed  $id
+   * @param  array  $columns
+   * @return \Illuminate\Database\Eloquent\Model|\Illuminate\Database\Eloquent\Collection|static[]|static|null
+   */
+  public function find($id, $columns = ['*'])
+  {
+      if (is_array($id) || $id instanceof Arrayable) {
+          return $this->findMany($id, $columns);
+      }
+
+      return $this->withUuid($id)->first($columns);
+  }
+
+  /**
+   * Find multiple models by their primary keys.
+   *
+   * @param  \Illuminate\Contracts\Support\Arrayable|array  $ids
+   * @param  array  $columns
+   * @return \Illuminate\Database\Eloquent\Collection
+   */
+  public function findMany($ids, $columns = ['*'])
+  {
+      if (empty($ids)) {
+          return $this->model->newCollection();
+      }
+
+      return $this->withUuid($ids)->get($columns);
+  }
+}

src/HasBinaryUuid.php

@@ -4,7 +4,6 @@
 
 use Ramsey\Uuid\Uuid;
 use Illuminate\Database\Eloquent\Model;
-use Illuminate\Database\Eloquent\Builder;
 
 trait HasBinaryUuid
 {
@@ -190,6 +189,11 @@ public function newQueryForRestoration($id)
         return $this->newQueryWithoutScopes()->whereKey(base64_decode($id));
     }
 
+    public function newEloquentBuilder($query)
+    {
+        return new Builder($query);
+    }
+
     public function getRouteKeyName()
     {
         $suffix = $this->getUuidSuffix();

tests/Feature/HasBinaryUuidTest.php

@@ -214,4 +214,36 @@ public function it_prevents_decoding_model_key_when_it_is_not_included_in_attrib
 
         $this->assertFalse(isset($array[$model->getKeyName()]));
     }
+
+    /** @test */
+    public function it_finds_a_model_from_its_textual_uuid_too()
+    {
+        $model = TestModel::create();
+
+        $this->assertTrue($model->is(TestModel::find($model->uuid)));
+        $this->assertTrue($model->is(TestModel::find($model->uuid_text)));
+    }
+
+    /** @test */
+    public function it_finds_many_models_from_their_textual_uuids()
+    {
+        $model1 = TestModel::create();
+        $model2 = TestModel::create();
+        $model3 = TestModel::create();
+
+        $this->assertCount(2, TestModel::find([$model1->uuid, $model3->uuid,]));
+        $this->assertCount(2, TestModel::findMany([$model1->uuid, $model3->uuid,]));
+
+        $this->assertCount(2, TestModel::find([$model1->uuid_text, $model3->uuid_text,]));
+        $this->assertCount(2, TestModel::findMany([$model1->uuid_text, $model3->uuid_text,]));
+    }
+
+    /** @test */
+    public function it_finds_or_fails_a_model_from_its_textual_uuid()
+    {
+      $model = TestModel::create();
+
+      $this->assertTrue($model->is(TestModel::findOrFail($model->uuid)));
+      $this->assertTrue($model->is(TestModel::findOrFail($model->uuid_text)));
+    }
 }