whichRoles scope

``` User::WhichRoles(‘admin’)->get(); User::WhichRoles([‘admin’, ‘agent’])->get(); ``` Issue #47
artesaos · Sep 14, 2015 · 7785c68 · 7785c68
1 parent 5f6b7fa
commit 7785c68
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/Defender/Traits/Users/HasRoles.php b/src/Defender/Traits/Users/HasRoles.php
@@ -90,4 +90,21 @@ public function syncRoles(array $roles)
     {
         return $this->roles()->sync($roles);
     }
+
+    /**
+     * Take user by roles
+     *
+     * @param \Illuminate\Database\Query\Builder $query
+     * @param string|array $roles
+     *
+     * @return \Illuminate\Database\Query\Builder
+     */
+    public function scopeWhichRoles($query, $roles)
+    {
+        return $query->whereHas('roles', function($query) use($roles) {
+            $roles = (is_array($roles)) ? $roles : [$roles];
+
+            $query->whereIn('name', $roles);
+        });
+    }
 }