Fix method visibility, use enum

simonhamp · Mar 28, 2024 · b9bff60 · b9bff60
1 parent 920e58e
commit b9bff60
Showing 1 changed file with 15 additions and 13 deletions.
diff --git a/src/Traits/Payable.php b/src/Traits/Payable.php
@@ -2,14 +2,16 @@
 
 namespace SimonHamp\LaravelStripeConnect\Traits;
 
-use Stripe\Account;
-use Stripe\Balance;
-use Stripe\Transfer;
-use Stripe\StripeClient;
+
 use Illuminate\Support\Facades\App;
 use Illuminate\Support\Facades\URL;
 use Illuminate\Support\Facades\Config;
+use SimonHamp\LaravelStripeConnect\Enums\LinkType;
 use SimonHamp\LaravelStripeConnect\Interfaces\StripeConnect;
+use Stripe\Account;
+use Stripe\Balance;
+use Stripe\Transfer;
+use Stripe\StripeClient;
 
 trait Payable
 {
@@ -55,14 +57,14 @@ public function isStripeAccountActive()
     /**
      * Get the redirect URL needed to take this account through Stripe's onboarding flow
      */
-    public function getStripeAccountLink($type = 'account_onboarding'): string
+    public function getStripeAccountLink(LinkType $type = LinkType::Onboarding): string
     {
         $link = static::$stripe->accountLinks->create(
             [
                 'account' => $this->getStripeAccountId(),
                 'refresh_url' => URL::route(Config::get('stripe_connect.routes.account.refresh')),
                 'return_url' => URL::route(Config::get('stripe_connect.routes.account.return')),
-                'type' => 'account_onboarding',
+                'type' => $type,
             ]
         );
 
@@ -86,6 +88,13 @@ public function getAccountBalance(): Balance
         ]);
     }
 
+    public function setStripeAccountStatus($status)
+    {
+        $this->{$this->getStripeAccountStatusColumn()} = $status;
+
+        return $this;
+    }
+
     protected function getStripeAccountIdColumn()
     {
         return Config::get('stripe_connect.payable.account_id_column');
@@ -102,11 +111,4 @@ protected function getStripeAccountStatusColumn()
     {
         return Config::get('stripe_connect.payable.account_status_column');
     }
-
-    protected function setStripeAccountStatus($status)
-    {
-        $this->{$this->getStripeAccountStatusColumn()} = $status;
-
-        return $this;
-    }
 }