Allow to change Models database connection

laravel · May 9, 2020 · 555e02c · 555e02c
1 parent ca0e683
commit 555e02c
Show file tree

Hide file tree

Showing 12 changed files with 214 additions and 10 deletions.
diff --git a/config/passport.php b/config/passport.php
@@ -46,4 +46,21 @@
         'secret' => env('PASSPORT_PERSONAL_ACCESS_CLIENT_SECRET'),
     ],
 
+    /*
+    |--------------------------------------------------------------------------
+    | Passport Storage Driver
+    |--------------------------------------------------------------------------
+    |
+    | This configuration options determines the storage driver that will
+    | be used to store Passport's data. In addition, you may set any
+    | custom options as needed by the particular driver you choose.
+    |
+    */
+
+    'storage' => [
+        'database' => [
+            'connection' => env('DB_CONNECTION', 'mysql'),
+        ],
+    ],
+
 ];
diff --git a/database/migrations/2016_06_01_000001_create_oauth_auth_codes_table.php b/database/migrations/2016_06_01_000001_create_oauth_auth_codes_table.php
@@ -6,14 +6,41 @@
 
 class CreateOauthAuthCodesTable extends Migration
 {
+    /**
+     * The database schema.
+     *
+     * @var \Illuminate\Database\Schema\Builder
+     */
+    protected $schema;
+
+    /**
+     * Create a new migration instance.
+     *
+     * @return void
+     */
+    public function __construct()
+    {
+        $this->schema = Schema::connection($this->getConnection());
+    }
+
+    /**
+     * Get the migration connection name.
+     *
+     * @return string|null
+     */
+    public function getConnection()
+    {
+        return config('passport.storage.database.connection');
+    }
+
     /**
      * Run the migrations.
      *
      * @return void
      */
     public function up()
     {
-        Schema::create('oauth_auth_codes', function (Blueprint $table) {
+        $this->schema->create('oauth_auth_codes', function (Blueprint $table) {
             $table->string('id', 100)->primary();
             $table->unsignedBigInteger('user_id')->index();
             $table->unsignedBigInteger('client_id');
@@ -30,6 +57,6 @@ public function up()
      */
     public function down()
     {
-        Schema::dropIfExists('oauth_auth_codes');
+        $this->schema->dropIfExists('oauth_auth_codes');
     }
 }
diff --git a/database/migrations/2016_06_01_000002_create_oauth_access_tokens_table.php b/database/migrations/2016_06_01_000002_create_oauth_access_tokens_table.php
@@ -6,14 +6,41 @@
 
 class CreateOauthAccessTokensTable extends Migration
 {
+    /**
+     * The database schema.
+     *
+     * @var \Illuminate\Database\Schema\Builder
+     */
+    protected $schema;
+
+    /**
+     * Create a new migration instance.
+     *
+     * @return void
+     */
+    public function __construct()
+    {
+        $this->schema = Schema::connection($this->getConnection());
+    }
+
+    /**
+     * Get the migration connection name.
+     *
+     * @return string|null
+     */
+    public function getConnection()
+    {
+        return config('passport.storage.database.connection');
+    }
+
     /**
      * Run the migrations.
      *
      * @return void
      */
     public function up()
     {
-        Schema::create('oauth_access_tokens', function (Blueprint $table) {
+        $this->schema->create('oauth_access_tokens', function (Blueprint $table) {
             $table->string('id', 100)->primary();
             $table->unsignedBigInteger('user_id')->nullable()->index();
             $table->unsignedBigInteger('client_id');
@@ -32,6 +59,6 @@ public function up()
      */
     public function down()
     {
-        Schema::dropIfExists('oauth_access_tokens');
+        $this->schema->dropIfExists('oauth_access_tokens');
     }
 }
diff --git a/database/migrations/2016_06_01_000003_create_oauth_refresh_tokens_table.php b/database/migrations/2016_06_01_000003_create_oauth_refresh_tokens_table.php
@@ -6,14 +6,41 @@
 
 class CreateOauthRefreshTokensTable extends Migration
 {
+    /**
+     * The database schema.
+     *
+     * @var \Illuminate\Database\Schema\Builder
+     */
+    protected $schema;
+
+    /**
+     * Create a new migration instance.
+     *
+     * @return void
+     */
+    public function __construct()
+    {
+        $this->schema = Schema::connection($this->getConnection());
+    }
+
+    /**
+     * Get the migration connection name.
+     *
+     * @return string|null
+     */
+    public function getConnection()
+    {
+        return config('passport.storage.database.connection');
+    }
+
     /**
      * Run the migrations.
      *
      * @return void
      */
     public function up()
     {
-        Schema::create('oauth_refresh_tokens', function (Blueprint $table) {
+        $this->schema->create('oauth_refresh_tokens', function (Blueprint $table) {
             $table->string('id', 100)->primary();
             $table->string('access_token_id', 100);
             $table->boolean('revoked');
@@ -28,6 +55,6 @@ public function up()
      */
     public function down()
     {
-        Schema::dropIfExists('oauth_refresh_tokens');
+        $this->schema->dropIfExists('oauth_refresh_tokens');
     }
 }
diff --git a/database/migrations/2016_06_01_000004_create_oauth_clients_table.php b/database/migrations/2016_06_01_000004_create_oauth_clients_table.php
@@ -6,14 +6,41 @@
 
 class CreateOauthClientsTable extends Migration
 {
+    /**
+     * The database schema.
+     *
+     * @var \Illuminate\Database\Schema\Builder
+     */
+    protected $schema;
+
+    /**
+     * Create a new migration instance.
+     *
+     * @return void
+     */
+    public function __construct()
+    {
+        $this->schema = Schema::connection($this->getConnection());
+    }
+
+    /**
+     * Get the migration connection name.
+     *
+     * @return string|null
+     */
+    public function getConnection()
+    {
+        return config('passport.storage.database.connection');
+    }
+
     /**
      * Run the migrations.
      *
      * @return void
      */
     public function up()
     {
-        Schema::create('oauth_clients', function (Blueprint $table) {
+        $this->schema->create('oauth_clients', function (Blueprint $table) {
             $table->bigIncrements('id');
             $table->unsignedBigInteger('user_id')->nullable()->index();
             $table->string('name');
@@ -34,6 +61,6 @@ public function up()
      */
     public function down()
     {
-        Schema::dropIfExists('oauth_clients');
+        $this->schema->dropIfExists('oauth_clients');
     }
 }
diff --git a/database/migrations/2016_06_01_000005_create_oauth_personal_access_clients_table.php b/database/migrations/2016_06_01_000005_create_oauth_personal_access_clients_table.php
@@ -6,14 +6,41 @@
 
 class CreateOauthPersonalAccessClientsTable extends Migration
 {
+    /**
+     * The database schema.
+     *
+     * @var \Illuminate\Database\Schema\Builder
+     */
+    protected $schema;
+
+    /**
+     * Create a new migration instance.
+     *
+     * @return void
+     */
+    public function __construct()
+    {
+        $this->schema = Schema::connection($this->getConnection());
+    }
+
+    /**
+     * Get the migration connection name.
+     *
+     * @return string|null
+     */
+    public function getConnection()
+    {
+        return config('passport.storage.database.connection');
+    }
+
     /**
      * Run the migrations.
      *
      * @return void
      */
     public function up()
     {
-        Schema::create('oauth_personal_access_clients', function (Blueprint $table) {
+        $this->schema->create('oauth_personal_access_clients', function (Blueprint $table) {
             $table->bigIncrements('id');
             $table->unsignedBigInteger('client_id');
             $table->timestamps();
@@ -27,6 +54,6 @@ public function up()
      */
     public function down()
     {
-        Schema::dropIfExists('oauth_personal_access_clients');
+        $this->schema->dropIfExists('oauth_personal_access_clients');
     }
 }
diff --git a/src/AuthCode.php b/src/AuthCode.php
@@ -59,6 +59,16 @@ class AuthCode extends Model
      */
     protected $keyType = 'string';
 
+    /**
+     * Get the current connection name for the model.
+     *
+     * @return string|null
+     */
+    public function getConnectionName()
+    {
+        return config('passport.storage.database.connection') ?? $this->connection;
+    }
+
     /**
      * Get the client that owns the authentication code.
      *

diff --git a/src/Client.php b/src/Client.php
@@ -65,6 +65,16 @@ public static function boot()
         });
     }
 
+    /**
+     * Get the current connection name for the model.
+     *
+     * @return string|null
+     */
+    public function getConnectionName()
+    {
+        return config('passport.storage.database.connection') ?? $this->connection;
+    }
+
     /**
      * Get the user that the client belongs to.
      *

diff --git a/src/PersonalAccessClient.php b/src/PersonalAccessClient.php
@@ -20,6 +20,16 @@ class PersonalAccessClient extends Model
      */
     protected $guarded = [];
 
+    /**
+     * Get the current connection name for the model.
+     *
+     * @return string|null
+     */
+    public function getConnectionName()
+    {
+        return config('passport.storage.database.connection') ?? $this->connection;
+    }
+
     /**
      * Get all of the authentication codes for the client.
      *

diff --git a/src/RefreshToken.php b/src/RefreshToken.php
@@ -59,6 +59,16 @@ class RefreshToken extends Model
      */
     public $timestamps = false;
 
+    /**
+     * Get the current connection name for the model.
+     *
+     * @return string|null
+     */
+    public function getConnectionName()
+    {
+        return config('passport.storage.database.connection') ?? $this->connection;
+    }
+
     /**
      * Get the access token that the refresh token belongs to.
      *

diff --git a/src/Token.php b/src/Token.php
@@ -60,6 +60,16 @@ class Token extends Model
      */
     public $timestamps = false;
 
+    /**
+     * Get the current connection name for the model.
+     *
+     * @return string|null
+     */
+    public function getConnectionName()
+    {
+        return config('passport.storage.database.connection') ?? $this->connection;
+    }
+
     /**
      * Get the client that the token belongs to.
      *

diff --git a/tests/Feature/PassportTestCase.php b/tests/Feature/PassportTestCase.php
@@ -39,6 +39,8 @@ protected function getEnvironmentSetUp($app)
 
         $app['config']->set('database.default', 'testbench');
 
+        $app['config']->set('passport.storage.database.connection', 'testbench');
+
         $app['config']->set('database.connections.testbench', [
             'driver'   => 'sqlite',
             'database' => ':memory:',