Utilize static properties for team and membership like User. (#119)

laravel · Sep 9, 2020 · 16840d7 · 16840d7
1 parent 872fc2c
commit 16840d7
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/Jetstream.php b/src/Jetstream.php
@@ -45,6 +45,20 @@ class Jetstream
      */
     public static $userModel = 'App\\Models\\User';
 
+    /**
+     * The team model that should be used by Jetstream.
+     *
+     * @var string
+     */
+    public static $teamModel = 'App\\Models\\Team';
+
+    /**
+     * The membership model that should be used by Jetstream.
+     *
+     * @var string
+     */
+    public static $membershipModel = 'App\\Models\\Membership';
+
     /**
      * Determine if Jetstream has registered roles.
      *
@@ -227,7 +241,7 @@ public static function useUserModel(string $model)
      */
     public static function teamModel()
     {
-        return 'App\\Models\\Team';
+        return static::$teamModel;
     }
 
     /**
@@ -249,7 +263,7 @@ public static function newTeamModel()
      */
     public static function membershipModel()
     {
-        return 'App\\Models\\Membership';
+        return static::$membershipModel;
     }
 
     /**