make it more flexible.

imanghafoori1 · Feb 15, 2020 · 7c83e28 · 7c83e28
1 parent 462d041
commit 7c83e28
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/src/TwoFactorAuthServiceProvider.php b/src/TwoFactorAuthServiceProvider.php
@@ -11,9 +11,7 @@
 use Imanghafoori\TokenizedLogin\Facades\TokenStoreFacade;
 use Imanghafoori\TokenizedLogin\Facades\UserProviderFacade;
 use Imanghafoori\TokenizedLogin\Http\ResponderFacade;
-use Imanghafoori\TokenizedLogin\Http\Responses\Responses;
 use Imanghafoori\TokenizedLogin\TokenGenerators\FakeTokenGenerator;
-use Imanghafoori\TokenizedLogin\TokenGenerators\TokenGenerator;
 use Imanghafoori\TokenizedLogin\TokenStore\FakeTokenStore;
 use Imanghafoori\TokenizedLogin\TokenStore\TokenStore;
 
@@ -32,10 +30,10 @@ public function register()
             $tokenSender = FakeTokenSender::class;
         } else {
             $tokenSender = config('tokenized_login.token_sender');
-            $tokenGenerator = TokenGenerator::class;
+            $tokenGenerator = config('tokenized_login.token_generator');
             $tokenStore = TokenStore::class;
         }
-        ResponderFacade::shouldProxyTo(Responses::class);
+        ResponderFacade::shouldProxyTo(config('tokenized_login.responses'));
         TokenGeneratorFacade::shouldProxyTo($tokenGenerator);
         TokenStoreFacade::shouldProxyTo($tokenStore);
         TokenSenderFacade::shouldProxyTo($tokenSender);

diff --git a/src/config/tokenized_login.php b/src/config/tokenized_login.php
@@ -29,4 +29,15 @@
      * You may define your own token sender class.
      */
     'token_sender' => \Imanghafoori\TokenizedLogin\TokenSender::class,
+
+    /**
+     * You can change the way you generate the token by define you own class.
+     */
+    'token_generator' => \Imanghafoori\TokenizedLogin\TokenGenerators\TokenGenerator::class,
+
+    /**
+     * You can extend Responses class and override
+     * it's methods, to define your own responses
+     */
+    'responses' => \Imanghafoori\TokenizedLogin\Http\Responses\Responses::class
 ];