Extract switch team logic into HasTeams trait

Author: chefe

src/HasTeams.php

@@ -24,14 +24,30 @@ public function isCurrentTeam($team)
     public function currentTeam()
     {
         if (is_null($this->current_team_id) && $this->id) {
-            $this->forceFill([
-                'current_team_id' => $this->personalTeam()->id,
-            ])->save();
+            $this->switchTeam($this->personalTeam());
         }
 
         return $this->belongsTo(Jetstream::teamModel(), 'current_team_id');
     }
 
+    /**
+     * Switch the user's context to the given team.
+     *
+     * @return bool
+     */
+    public function switchTeam($team)
+    {
+        if (!$this->belongsToTeam($team)) {
+            return false;
+        }
+
+        $this->forceFill([
+            'current_team_id' => $team->id,
+        ])->save();
+
+        return true;
+    }
+
     /**
      * Get all of the teams the user owns or belongs to.
      *

src/Http/Controllers/CurrentTeamController.php

@@ -18,14 +18,10 @@ public function update(Request $request)
     {
         $team = Jetstream::newTeamModel()->findOrFail($request->team_id);
 
-        if (! $request->user()->belongsToTeam($team)) {
+        if (! $request->user()->switchTeam($team)) {
             abort(403);
         }
 
-        $request->user()->forceFill([
-            'current_team_id' => $team->id,
-        ])->save();
-
         return redirect(config('fortify.home'), 303);
     }
 }