Merge pull request #49 from patressz/feat/custom-db-connection

feat: add ability to configure custom `database` connection

Author: MrPunyapal

src/Adapters/Laravel/Repositories/DatabaseAnalyticsRepository.php

@@ -4,7 +4,8 @@
 
 namespace Pan\Adapters\Laravel\Repositories;
 
-use Illuminate\Support\Facades\DB;
+use Illuminate\Database\Connection;
+use Illuminate\Database\DatabaseManager;
 use Pan\Contracts\AnalyticsRepository;
 use Pan\Enums\EventType;
 use Pan\PanConfiguration;
@@ -18,10 +19,10 @@
     /**
      * Creates a new analytics repository instance.
      */
-    public function __construct(private PanConfiguration $config)
-    {
-        //
-    }
+    public function __construct(
+        private DatabaseManager $databaseManager,
+        private PanConfiguration $config
+    ) {}
 
     /**
      * Returns all analytics.
@@ -31,7 +32,7 @@ public function __construct(private PanConfiguration $config)
     public function all(): array
     {
         /** @var array<int, Analytic> $all */
-        $all = DB::table('pan_analytics')->get()->map(fn (mixed $analytic): Analytic => new Analytic(
+        $all = $this->connection()->table('pan_analytics')->get()->map(fn (mixed $analytic): Analytic => new Analytic(
             id: (int) $analytic->id,
             name: $analytic->name,
             impressions: (int) $analytic->impressions,
@@ -56,22 +57,30 @@ public function increment(string $name, EventType $event): void
             return;
         }
 
-        if (DB::table('pan_analytics')->where('name', $name)->count() === 0) {
-            if (DB::table('pan_analytics')->count() < $maxAnalytics) {
-                DB::table('pan_analytics')->insert(['name' => $name, $event->column() => 1]);
+        if ($this->connection()->table('pan_analytics')->where('name', $name)->count() === 0) {
+            if ($this->connection()->table('pan_analytics')->count() < $maxAnalytics) {
+                $this->connection()->table('pan_analytics')->insert(['name' => $name, $event->column() => 1]);
             }
 
             return;
         }
 
-        DB::table('pan_analytics')->where('name', $name)->increment($event->column());
+        $this->connection()->table('pan_analytics')->where('name', $name)->increment($event->column());
     }
 
     /**
      * Flush all analytics.
      */
     public function flush(): void
     {
-        DB::table('pan_analytics')->truncate();
+        $this->connection()->table('pan_analytics')->truncate();
+    }
+
+    /**
+     * Resolve the database connection.
+     */
+    private function connection(): Connection
+    {
+        return $this->databaseManager->connection($this->config->getDatabaseConnection());
     }
 }

src/PanConfiguration.php

@@ -20,6 +20,7 @@ private function __construct(
         private int $maxAnalytics = 50,
         private array $allowedAnalytics = [],
         private string $routePrefix = 'pan',
+        private ?string $databaseConnection = null,
     ) {
         //
     }
@@ -66,6 +67,24 @@ public function setRoutePrefix(string $prefix): void
         $this->routePrefix = $prefix;
     }
 
+    /**
+     * Sets the database connection to be used.
+     *
+     * @internal
+     */
+    public function setDatabaseConnection(string $connection): void
+    {
+        $this->databaseConnection = $connection;
+    }
+
+    /**
+     * Sets the database connection to be used.
+     */
+    public static function databaseConnection(string $connection): void
+    {
+        self::instance()->setDatabaseConnection($connection);
+    }
+
     /**
      * Sets the maximum number of analytics to store.
      */
@@ -129,4 +148,14 @@ public function toArray(): array
             'route_prefix' => $this->routePrefix,
         ];
     }
+
+    /**
+     * Get the database connection to be used.
+     *
+     * @internal
+     */
+    public function getDatabaseConnection(): ?string
+    {
+        return $this->databaseConnection;
+    }
 }

tests/Unit/PanConfigurationTest.php

@@ -77,3 +77,9 @@
         'route_prefix' => 'pan',
     ]);
 });
+
+it('can set the database connection', function (): void {
+    PanConfiguration::databaseConnection('sqlite');
+
+    expect(PanConfiguration::instance()->getDatabaseConnection())->toBe('sqlite');
+});