Copy Files From External Storage To Local Storage for Import

src/Snapshot.php

@@ -3,13 +3,15 @@
 namespace Spatie\DbSnapshots;
 
 use Carbon\Carbon;
+use Illuminate\Contracts\Filesystem\Factory;
 use Illuminate\Filesystem\FilesystemAdapter as Disk;
 use Illuminate\Support\Facades\DB;
 use Illuminate\Support\LazyCollection;
 use Spatie\DbSnapshots\Events\DeletedSnapshot;
 use Spatie\DbSnapshots\Events\DeletingSnapshot;
 use Spatie\DbSnapshots\Events\LoadedSnapshot;
 use Spatie\DbSnapshots\Events\LoadingSnapshot;
+use Spatie\TemporaryDirectory\TemporaryDirectory;
 
 class Snapshot
 {
@@ -25,6 +27,8 @@ class Snapshot
 
     public const STREAM_BUFFER_SIZE = 16384;
 
+    protected Factory $filesystemFactory;
+
     public function __construct(Disk $disk, string $fileName)
     {
         $this->disk = $disk;
@@ -39,6 +43,8 @@ public function __construct(Disk $disk, string $fileName)
         }
 
         $this->name = pathinfo($fileName, PATHINFO_FILENAME);
+
+        $this->filesystemFactory = app(Factory::class);
     }
 
     public function useStream(): self
@@ -88,49 +94,73 @@ protected function shouldIgnoreLine(string $line): bool
         return empty($line) || $this->isASqlComment($line);
     }
 
-    protected function loadStream(?string $connectionName = null): void
+    protected function loadStream(string $connectionName = null)
     {
-        LazyCollection::make(function () {
-            $stream = $this->compressionExtension === 'gz'
-                ? gzopen($this->disk->path($this->fileName), 'r')
-                : $this->disk->readStream($this->fileName);
-
-            $statement = '';
-            while (! feof($stream)) {
-                $chunk = $this->compressionExtension === 'gz'
+        $directory = (new TemporaryDirectory(config('db-snapshots.temporary_directory_path')))->create();
+
+        config([
+            'filesystems.disks.' . self::class => [
+                'driver' => 'local',
+                'root' => $directory->path(),
+                'throw' => false,
+            ]
+        ]);
+
+        $localDisk = $this->filesystemFactory->disk(self::class);
+
+        try {
+            LazyCollection::make(function () use ($localDisk) {
+                $localDisk->writeStream($this->fileName, $this->disk->readStream($this->fileName));
+
+                $stream = $this->compressionExtension === 'gz'
+                    ? gzopen($localDisk->path($this->fileName), 'r')
+                    : $localDisk->readStream($this->fileName);
+
+                $statement = '';
+                while (!feof($stream)) {
+                    $chunk = $this->compressionExtension === 'gz'
                         ? gzread($stream, self::STREAM_BUFFER_SIZE)
                         : fread($stream, self::STREAM_BUFFER_SIZE);
 
-                $lines = explode("\n", $chunk);
-                foreach ($lines as $idx => $line) {
-                    if ($this->shouldIgnoreLine($line)) {
-                        continue;
+                    $lines = explode("\n", $chunk);
+                    foreach ($lines as $idx => $line) {
+                        if ($this->shouldIgnoreLine($line)) {
+                            continue;
+                        }
+
+                        $statement .= $line;
+
+                        // Carry-over the last line to the next chunk since it
+                        // is possible that this chunk finished mid-line right on
+                        // a semi-colon.
+                        if (count($lines) == $idx + 1) {
+                            break;
+                        }
+
+                        if (str_ends_with(trim($statement), ';')) {
+                            yield $statement;
+                            $statement = '';
+                        }
                     }
+                }
 
-                    $statement .= $line;
-
-                    // Carry-over the last line to the next chunk since it
-                    // is possible that this chunk finished mid-line right on
-                    // a semi-colon.
-                    if (count($lines) == $idx + 1) {
-                        break;
-                    }
+                if (str_ends_with(trim($statement), ';')) {
+                    yield $statement;
+                }
 
-                    if (str_ends_with(trim($statement), ';')) {
-                        yield $statement;
-                        $statement = '';
-                    }
+                if ($this->compressionExtension === 'gz') {
+                    gzclose($stream);
+                } else {
+                    fclose($stream);
                 }
-            }
-
-            if (str_ends_with(trim($statement), ';')) {
-                yield $statement;
-            }
-        })->each(function (string $statement) use ($connectionName) {
-            DB::connection($connectionName)->unprepared($statement);
-        });
+            })->each(function (string $statement) use ($connectionName) {
+                DB::connection($connectionName)->unprepared($statement);
+            });
+        } finally {
+            $directory->delete();
+        }
     }
-
+  
     public function delete(): void
     {
         event(new DeletingSnapshot($this));