add getUserId and getUser functions to websocket

Author: albertcht

src/Websocket/Authenticatable.php

@@ -3,10 +3,14 @@
 namespace SwooleTW\Http\Websocket;
 
 use InvalidArgumentException;
+use Illuminate\Support\Facades\Auth;
 use Illuminate\Contracts\Auth\Authenticatable as AuthenticatableContract;
 
 trait Authenticatable
 {
+    protected $user;
+    protected $userId;
+
     /**
      * Login using current user.
      */
@@ -53,6 +57,42 @@ public function toUserId($userIds)
         return $this;
     }
 
+    /**
+     * Get current auth user by sender's fd.
+     */
+    public function getUser()
+    {
+        if ($this->user instanceof AuthenticatableContract) {
+            return $this->user;
+        }
+
+        if (! is_null($uid = $this->getUserId()) && $user = Auth::loginUsingId($uid)) {
+            $this->user = $user;
+        }
+
+        return $this->user;
+    }
+
+    /**
+     * Get current auth user id by sender's fd.
+     */
+    public function getUserId()
+    {
+        if (! is_null($this->userId)) {
+            return $this->userId;
+        }
+
+        $rooms = $this->room->getRooms($this->getSender());
+
+        foreach ($rooms as $room) {
+            if (count($explode = explode(static::USER_PREFIX, $room)) === 2) {
+                $this->userId = $explode[1];
+            }
+        }
+
+        return $this->userId;
+    }
+
     /**
      * Check if user object implements AuthenticatableContract.
      */

src/Websocket/Websocket.php

@@ -317,6 +317,8 @@ public function reset($force = false)
 
         if ($force) {
             $this->sender = null;
+            $this->user = null;
+            $this->userId = null;
         }
 
         return $this;

tests/Websocket/WebsocketTest.php

@@ -7,6 +7,7 @@
 use InvalidArgumentException;
 use Illuminate\Pipeline\Pipeline;
 use SwooleTW\Http\Tests\TestCase;
+use Illuminate\Support\Facades\Auth;
 use SwooleTW\Http\Websocket\Websocket;
 use SwooleTW\Http\Websocket\Rooms\RoomContract;
 use Illuminate\Contracts\Auth\Authenticatable as AuthenticatableContract;
@@ -193,6 +194,36 @@ public function testToUser()
         $this->assertTrue(in_array(3, $websocket->getTo()));
     }
 
+    public function testGetUserId()
+    {
+        $room = m::mock(RoomContract::class);
+        $room->shouldReceive('getRooms')
+            ->with($sender = 1)
+            ->once()
+            ->andReturn(['uid_1']);
+
+        $websocket = $this->getWebsocket($room)->setSender($sender);
+        $this->assertEquals($sender, $websocket->getUserId());
+    }
+
+    public function testGetUser()
+    {
+        $room = m::mock(RoomContract::class);
+        $room->shouldReceive('getRooms')
+            ->with($sender = 1)
+            ->once()
+            ->andReturn(['uid_1']);
+
+        $user = m::mock(AuthenticatableContract::class);
+        Auth::shouldReceive('loginUsingId')
+            ->with($sender)
+            ->once()
+            ->andReturn($user);
+
+        $websocket = $this->getWebsocket($room)->setSender($sender);
+        $this->assertEquals($user, $websocket->getUser());
+    }
+
     public function testReset()
     {
         $websocket = $this->getWebsocket();