Merge pull request #7 from ARCANEDEV/patch-1

Updating the middleware
ARCANEDEV · Sep 8, 2017 · e39f5ac · e39f5ac
2 parents 21b7eff + 36c2fa2
commit e39f5ac
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 7 deletions.
diff --git a/src/Middleware/BlockReferralSpam.php → src/Http/Middleware/BlockReferralSpam.php b/src/Middleware/BlockReferralSpam.php → src/Http/Middleware/BlockReferralSpam.php
@@ -1,4 +1,4 @@
-<?php namespace Arcanedev\SpamBlocker\Middleware;
+<?php namespace Arcanedev\SpamBlocker\Http\Middleware;
 
 use Arcanedev\SpamBlocker\Contracts\SpamBlocker;
 use Closure;
@@ -7,7 +7,7 @@
 /**
  * Class     BlockReferralSpam
  *
- * @package  Arcanedev\LaravelReferralSpamBlocker\Middleware
+ * @package  Arcanedev\SpamBlocker\Http\Middleware
  * @author   ARCANEDEV <[email protected]>
  */
 class BlockReferralSpam
@@ -18,7 +18,7 @@ class BlockReferralSpam
      */
 
     /** @var  \Arcanedev\SpamBlocker\Contracts\SpamBlocker */
-    private $blocker;
+    protected $blocker;
 
     /* -----------------------------------------------------------------
      |  Constructor
@@ -53,7 +53,22 @@ public function handle(Request $request, Closure $next)
         $referer = $request->headers->get('referer');
 
         return $this->blocker->isBlocked($referer)
-            ? response('Unauthorized.', 401)
+            ? $this->getBlockedResponse()
             : $next($request);
     }
+
+    /* -----------------------------------------------------------------
+     |  Other Methods
+     | -----------------------------------------------------------------
+     */
+
+    /**
+     * Get the blocked referer's response.
+     *
+     * @return \Illuminate\Contracts\Routing\ResponseFactory|\Symfony\Component\HttpFoundation\Response
+     */
+    protected function getBlockedResponse()
+    {
+        return response('Unauthorized.', 401);
+    }
 }
diff --git a/tests/Middleware/BlockReferralSpamTest.php → ...Http/Middleware/BlockReferralSpamTest.php b/tests/Middleware/BlockReferralSpamTest.php → ...Http/Middleware/BlockReferralSpamTest.php
@@ -1,11 +1,11 @@
-<?php namespace Arcanedev\SpamBlocker\Tests\Middleware;
+<?php namespace Arcanedev\SpamBlocker\Tests\Http\Middleware;
 
 use Arcanedev\SpamBlocker\Tests\TestCase;
 
 /**
  * Class     BlockReferralSpamTest
  *
- * @package  Arcanedev\SpamBlocker\Tests\Middleware
+ * @package  Arcanedev\SpamBlocker\Tests\Http\Middleware
  * @author   ARCANEDEV <[email protected]>
  */
 class BlockReferralSpamTest extends TestCase

diff --git a/tests/TestCase.php b/tests/TestCase.php
@@ -75,7 +75,7 @@ protected function getEnvironmentSetUp($app)
     private function registerRoutes($app)
     {
         $app->make(\Illuminate\Contracts\Http\Kernel::class)
-            ->pushMiddleware(\Arcanedev\SpamBlocker\Middleware\BlockReferralSpam::class);
+            ->pushMiddleware(\Arcanedev\SpamBlocker\Http\Middleware\BlockReferralSpam::class);
 
         /** @var \Illuminate\Contracts\Routing\Registrar $router */
         $router = $app['router'];