add auth, change type config

Author: ctf0

README.md

@@ -20,7 +20,7 @@ return [
 
         'firebase' => [
             'driver' => 'firebase',
-            'type' => 'rtdb', // rtdb or fsdb
+            'type' => 'firestore', // database or firestore
             'databaseURL' => env('FB_DB_URL'), // the real time database url
             'creds_file' => env('FB_CREDENTIALS_FILE'), // service account json file
             'collection_name' => env('FB_COLLECTION_NAME'), // ex.notifications

src/Broadcasters/Common.php

@@ -0,0 +1,50 @@
+<?php
+
+namespace ctf0\Firebase\Broadcasters;
+
+use Illuminate\Broadcasting\Broadcasters\UsePusherChannelConventions;
+use Symfony\Component\HttpKernel\Exception\AccessDeniedHttpException;
+
+trait Common
+{
+    use UsePusherChannelConventions;
+
+    /**
+     * {@inheritdoc}
+     */
+    public function auth($request)
+    {
+        $channelName = $this->normalizeChannelName($request->channel_name);
+
+        if (
+            $this->isGuardedChannel($request->channel_name) &&
+            !$this->retrieveUser($request, $channelName)
+        ) {
+            throw new AccessDeniedHttpException();
+        }
+
+        return parent::verifyUserCanAccessChannel(
+            $request,
+            $channelName
+        );
+    }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function validAuthenticationResponse($request, $result)
+    {
+        if (is_bool($result)) {
+            return json_encode($result);
+        }
+
+        $channelName = $this->normalizeChannelName($request->channel_name);
+
+        return json_encode([
+            'channel_data' => [
+                'user_id'   => $this->retrieveUser($request, $channelName)->getAuthIdentifier(),
+                'user_info' => $result,
+            ],
+        ]);
+    }
+}

src/Broadcasters/FSDB.php

@@ -2,6 +2,7 @@
 
 namespace ctf0\Firebase\Broadcasters;
 
+use Illuminate\Support\Arr;
 use Illuminate\Support\Str;
 use Kreait\Firebase\ServiceAccount;
 use Morrislaptop\Firestore\Factory;
@@ -11,6 +12,8 @@
 
 class FSDB extends Broadcaster
 {
+    use Common;
+
     protected $db;
     protected $config;
 
@@ -28,38 +31,28 @@ public function __construct()
                         ->createFirestore();
     }
 
-    /**
-     * {@inheritdoc}
-     */
-    public function auth($request)
-    {
-    }
-
-    /**
-     * {@inheritdoc}
-     */
-    public function validAuthenticationResponse($request, $result)
-    {
-    }
-
     /**
      * {@inheritdoc}
      */
     public function broadcast(array $channels, $event, array $payload = [])
     {
-        $db = $this->db;
-
-        try {
-            $coll = $db->collection($this->config['collection_name']);
-            $doc  = $coll->document(md5(Str::uuid()));
-            $doc->set([
-                'timestamp' => round(now()->valueOf()), // return date == to js Date.now()
-                'channels'  => $this->formatChannels($channels),
-                'data'      => $payload,
-                'event'     => $event,
-            ]);
-        } catch (ApiException $e) {
-            throw new BroadcastException($e);
+        $db     = $this->db;
+        $socket = Arr::pull($payload, 'socket');
+
+        foreach ($this->formatChannels($channels) as $channel) {
+            try {
+                $coll = $db->collection($this->config['collection_name']);
+                $doc  = $coll->document(md5(Str::uuid()));
+                $doc->set([
+                    'channel'   => $channel,
+                    'data'      => $payload,
+                    'event'     => $event,
+                    'socket'    => $socket,
+                    'timestamp' => round(now()->valueOf()), // return date == to js Date.now()
+                ]);
+            } catch (ApiException $e) {
+                throw new BroadcastException($e);
+            }
         }
     }
 }

src/Broadcasters/RTDB.php

@@ -2,6 +2,7 @@
 
 namespace ctf0\Firebase\Broadcasters;
 
+use Illuminate\Support\Arr;
 use Kreait\Firebase\Factory;
 use Kreait\Firebase\ServiceAccount;
 use Kreait\Firebase\Exception\ApiException;
@@ -10,6 +11,8 @@
 
 class RTDB extends Broadcaster
 {
+    use Common;
+
     protected $db;
     protected $config;
 
@@ -28,37 +31,27 @@ public function __construct($config)
                         ->create();
     }
 
-    /**
-     * {@inheritdoc}
-     */
-    public function auth($request)
-    {
-    }
-
-    /**
-     * {@inheritdoc}
-     */
-    public function validAuthenticationResponse($request, $result)
-    {
-    }
-
     /**
      * {@inheritdoc}
      */
     public function broadcast(array $channels, $event, array $payload = [])
     {
-        $db = $this->db->getDatabase();
-
-        try {
-            $db->getReference($this->config['collection_name'])
-                ->push([
-                    'timestamp' => round(now()->valueOf()), // return date == to js Date.now()
-                    'channels'  => $this->formatChannels($channels),
-                    'data'      => $payload,
-                    'event'     => $event,
-                ]);
-        } catch (ApiException $e) {
-            throw new BroadcastException($e);
+        $db     = $this->db->getDatabase();
+        $socket = Arr::pull($payload, 'socket');
+
+        foreach ($this->formatChannels($channels) as $channel) {
+            try {
+                $db->getReference($this->config['collection_name'])
+                    ->push([
+                        'channel'   => $channel,
+                        'data'      => $payload,
+                        'event'     => $event,
+                        'socket'    => $socket,
+                        'timestamp' => round(now()->valueOf()), // return date == to js Date.now()
+                    ]);
+            } catch (ApiException $e) {
+                throw new BroadcastException($e);
+            }
         }
     }
 }

src/FireBaseBroadcastProvider.php

@@ -17,7 +17,7 @@ public function boot()
         app(BroadcastManager::class)->extend('firebase', function ($app) {
             $config = config('broadcasting.connections.firebase');
 
-            return $config['type'] == 'rtdb' ? new RTDB($config) : new FSDB($config);
+            return $config['type'] == 'database' ? new RTDB($config) : new FSDB($config);
         });
     }