feat: add support for BEGINDATA

Author: yajra

src/ControlFileBuilder.php

@@ -6,6 +6,7 @@
 
 use Illuminate\Support\Facades\File;
 use Illuminate\Support\Str;
+use LogicException;
 use Stringable;
 
 class ControlFileBuilder implements Stringable
@@ -14,6 +15,11 @@ public function __construct(public SQLLoader $loader)
     {
     }
 
+    public function __toString(): string
+    {
+        return $this->build();
+    }
+
     public function build(): string
     {
         $template = File::get($this->getStub());
@@ -23,6 +29,7 @@ public function build(): string
             ->replace('$FILES', $this->inputFiles())
             ->replace('$METHOD', $this->method())
             ->replace('$INSERTS', $this->inserts())
+            ->replace('$BEGINDATA', $this->beginData())
             ->toString();
     }
 
@@ -38,7 +45,9 @@ protected function options(): string
 
     protected function inputFiles(): string
     {
-        return implode(PHP_EOL, $this->loader->inputFiles);
+        $inputFiles = implode(PHP_EOL, $this->loader->inputFiles);
+
+        return Str::replace("'*'", '*', $inputFiles);
     }
 
     protected function method(): string
@@ -54,8 +63,40 @@ protected function inserts(): string
         return implode(PHP_EOL, $this->loader->tables);
     }
 
-    public function __toString(): string
+    protected function beginData(): string
     {
-        return $this->build();
+        $sql = '';
+        if ($this->loader->beginData) {
+            $sql .= 'BEGINDATA'.PHP_EOL;
+            $sql .= $this->arrayToCsv($this->loader->beginData).PHP_EOL;
+        }
+
+        return $sql;
+    }
+
+    protected function arrayToCsv(
+        array $data,
+        string $delimiter = ',',
+        string $enclosure = '"',
+        string $escape_char = '\\'
+    ): string {
+        $f = fopen('php://memory', 'r+');
+
+        if (! $f) {
+            throw new LogicException('Failed to open memory stream');
+        }
+
+        foreach ($data as $item) {
+            fputcsv($f, $item, $delimiter, $enclosure, $escape_char);
+        }
+        rewind($f);
+
+        $contents = stream_get_contents($f);
+
+        if (! $contents) {
+            throw new LogicException('Failed to read memory stream');
+        }
+
+        return $contents;
     }
 }

src/SQLLoader.php

@@ -35,6 +35,8 @@ class SQLLoader
 
     protected string $logs = '';
 
+    public array $beginData = [];
+
     public function __construct(public array $options = [])
     {
     }
@@ -281,4 +283,14 @@ public function result(): ProcessResult
 
         return $this->result;
     }
+
+    public function beginData(array $data): static
+    {
+        $this->inputFiles = [];
+        $this->inFile('*');
+
+        $this->beginData = $data;
+
+        return $this;
+    }
 }

src/stubs/control.stub

@@ -3,3 +3,4 @@ LOAD DATA
 $FILES
 $METHOD
 $INSERTS
+$BEGINDATA

tests/Feature/ControlFileBuilderTest.php

@@ -95,3 +95,27 @@
         ->and($controlFile)->toContain('TIMESTAMP WITH TIME ZONE "YYYY-MM-DD HH24:MI:SS TZH:TZM"')
         ->and($controlFile)->toContain('TIMESTAMP WITH LOCAL TIME ZONE "YYYY-MM-DD HH24:MI:SS"');
 });
+
+test('it can build using begin data', function () {
+    $loader = new SQLLoader(['skip=1', 'load=2']);
+    $loader->as('users.ctl')
+        ->into(
+            table: 'users',
+            columns: ['id', 'name', 'email'],
+        )
+        ->beginData([
+            ['1', 'name-1', 'email-1'],
+            ['2', 'name-2', 'email-2'],
+            ['3', 'name,-2', 'email"-2'],
+        ]);
+
+    $ctl = new ControlFileBuilder($loader);
+    $controlFile = $ctl->build();
+
+    expect($controlFile)->toBeString()
+        ->and($controlFile)->toContain('INFILE *')
+        ->and($controlFile)->toContain("BEGINDATA\n")
+        ->and($controlFile)->toContain('1,name-1,email-1')
+        ->and($controlFile)->toContain('2,name-2,email-2')
+        ->and($controlFile)->toContain('3,"name,-2","email""-2"');
+});