add multiple driver support

Author: imaarov

config/disk-monitor.php

@@ -5,5 +5,8 @@
     /**
      * The name of the disk you want to monitor.
      */
-    'disk_name' => 'local',
+    'disk_names' => [
+        'local',
+        
+    ],
 ];

src/Commands/RecordDiskMetricCommand.php

@@ -16,16 +16,22 @@ public function handle(): int
     {
         $this->comment('Recording metrics...');
 
-        $disk_name = config('disk-monitor.disk_name');
-        $file_count = count(Storage::disk($disk_name)->allFiles());
+        collect($disk_names = config('disk-monitor.disk_names'))
+            ->each(fn(string $disk_name) => $this->recordMetrics($disk_name));
 
+        $this->comment('All done!');
+
+        return self::SUCCESS;
+    }
+
+    protected function recordMetrics(string $disk_name):void
+    {
+        $this->info("Recording metrics for disk `{$disk_name}`...");
+        $disk = Storage::disk($disk_name);
+        $file_count = count($disk->allFiles());
         DiskMonitorEntry::create([
             'disk_name' => $disk_name,
             'file_count' => $file_count,
         ]);
-
-        $this->comment('All done!');
-
-        return self::SUCCESS;
     }
 }

src/Models/DiskMonitorEntry.php

@@ -7,6 +7,9 @@
 class DiskMonitorEntry extends Model
 {
     public $guarded = [];
+    public $casts = [
+        'file_count'    => 'integer'
+    ];
 
     public static function last(): ?self
     {

tests/Feature/Commands/RecordDiskMetricsCommandTest.php

@@ -14,10 +14,11 @@ public function setUp(): void
         parent::setUp();
 
         Storage::fake('local');
+        Storage::fake('external');
     }
 
     /** @test */
-    public function it_will_record_file_count_for_a_disk()
+    public function it_will_record_file_count_for_a_single_disk()
     {
         // Storage::disk('local')->put('test.txt', 'test');
 
@@ -26,16 +27,36 @@ public function it_will_record_file_count_for_a_disk()
             ->assertExitCode(0);
 
         $this->assertCount(1, DiskMonitorEntry::get());
-
         $entry = DiskMonitorEntry::last();
+
         $this->assertEquals(0, $entry->file_count);
 
-        Storage::put('test.txt', 'text');
+        Storage::disk('local')->put('test.txt', 'text');
         $this
             ->artisan(RecordDiskMetricCommand::class)
             ->assertExitCode(0);
 
         $entry = DiskMonitorEntry::last();
         $this->assertEquals(1, $entry->file_count);
     }
+
+    /** @test */
+    public function it_will_record_file_count_for_multiple_disk()
+    {
+        config()->set('disk-monitor.disk_names', ['local', 'external']);
+        Storage::disk('external')->put('test.txt', 'test');
+
+        $this
+            ->artisan(RecordDiskMetricCommand::class)
+            ->assertExitCode(0);
+
+        $entries = DiskMonitorEntry::get();
+        $this->assertCount(2, $entries);
+
+        $this->assertEquals('local', $entries[0]->disk_name);
+        $this->assertEquals(0, $entries[0]->file_count);
+
+        $this->assertEquals('external', $entries[1]->disk_name);
+        $this->assertEquals(1, $entries[1]->file_count);
+    }
 }