Merge pull request #9 from ennosol/master

updateGroups method added
jenssegers · Feb 9, 2015 · 52c12d5 · 52c12d5
2 parents 4ef3366 + 8115399
commit 52c12d5
Showing 1 changed file with 55 additions and 0 deletions.
diff --git a/src/Jenssegers/Mongodb/Sentry/User.php b/src/Jenssegers/Mongodb/Sentry/User.php
@@ -527,6 +527,61 @@ public function removeGroup(GroupInterface $group)
         return true;
     }
 
+    /**
+     * Updates the user to the given group(s).
+     *
+     * @param  \Illuminate\Database\Eloquent\Collection  $groups
+     * @param  bool  $remove
+     * @return bool
+     */
+    public function updateGroups($groups, $remove = true)
+    {
+        $newGroupIds = array();
+        $removeGroupIds = array();
+        $existingGroupIds = array();
+        foreach ($groups as $group)
+        {
+            if (is_object($group))
+            {
+                $newGroupIds[] = $group->getId();
+                if ( ! $this->addGroup($group))
+                {
+                    return false;
+                }
+            }
+            else
+            {
+                $newGroupIds[] = $groups->getId();
+                if ( ! $this->addGroup($groups))
+                {
+                    return false;
+                }
+                break;
+            }
+        }
+        if ($remove)
+        {
+            foreach ($this->groups as $userGroup)
+            {
+                $existingGroupIds[] = $userGroup->getId();
+            }
+            $removeGroupIds = array_diff($existingGroupIds, $newGroupIds);
+            if ($removeGroupIds)
+            {
+                self::$groupProviderModel = self::$groupProviderModel ?: new GroupProvider;
+            }
+            foreach ($removeGroupIds as $id)
+            {
+                $group = self::$groupProviderModel->findById($id);
+                if ( ! $this->removeGroup($group))
+                {
+                    return false;
+                }
+            }
+        }
+        return true;
+    }
+
     /**
      * See if the user is in the given group.
      *