Typehint the arguments for criteria

Author: bryanvimigo

src/Criteria/OnlyTrashedCriteria.php

@@ -5,6 +5,7 @@
 use CodeOfDigital\CacheRepository\Contracts\CriteriaInterface;
 use CodeOfDigital\CacheRepository\Contracts\RepositoryInterface;
 use CodeOfDigital\CacheRepository\Exceptions\RepositoryException;
+use Illuminate\Database\Eloquent\Builder;
 use Illuminate\Database\Eloquent\Model;
 use Illuminate\Database\Eloquent\SoftDeletes;
 
@@ -13,12 +14,12 @@ class OnlyTrashedCriteria implements CriteriaInterface
     /**
      * Apply conditions into current query building
      *
-     * @param Model $model
+     * @param Model|Builder $model
      * @param RepositoryInterface $repository
      * @return mixed
      * @throws RepositoryException
      */
-    public function apply(Model $model, RepositoryInterface $repository): mixed
+    public function apply(Model|Builder $model, RepositoryInterface $repository): mixed
     {
         if (!in_array(SoftDeletes::class, class_uses($model)))
             throw new RepositoryException('Model must implement SoftDeletes Trait to use this criteria.');

src/Criteria/WithTrashedCriteria.php

@@ -5,6 +5,7 @@
 use CodeOfDigital\CacheRepository\Contracts\CriteriaInterface;
 use CodeOfDigital\CacheRepository\Contracts\RepositoryInterface;
 use CodeOfDigital\CacheRepository\Exceptions\RepositoryException;
+use Illuminate\Database\Eloquent\Builder;
 use Illuminate\Database\Eloquent\Model;
 use Illuminate\Database\Eloquent\SoftDeletes;
 
@@ -13,12 +14,12 @@ class WithTrashedCriteria implements CriteriaInterface
     /**
      * Apply conditions into current query building
      *
-     * @param Model $model
+     * @param Model|Builder $model
      * @param RepositoryInterface $repository
      * @return mixed
      * @throws RepositoryException
      */
-    public function apply(Model $model, RepositoryInterface $repository): mixed
+    public function apply(Model|Builder $model, RepositoryInterface $repository): mixed
     {
         if (!in_array(SoftDeletes::class, class_uses($model)))
             throw new RepositoryException('Model must implement SoftDeletes Trait to use this criteria.');