map to custom headers

lpheller · Oct 4, 2023 · e2be619 · e2be619
1 parent e93fed5
commit e2be619
Show file tree

Hide file tree

Showing 2 changed files with 43 additions and 1 deletion.
diff --git a/src/Csv.php b/src/Csv.php
@@ -22,6 +22,8 @@ class Csv
 
     protected $delimiter = ',';
 
+    protected $headers = [];
+
     public function __construct(string $filePath)
     {
         $this->filePath = $filePath;
@@ -64,9 +66,17 @@ public function delimiter(string $delimiter)
      */
     public function mapToHeaders($row = 1)
     {
-        $this->setHeaderRow($row);
         $this->mapToHeaderRow = true;
 
+        if (is_array($row)) {
+            $this->skipRows([]);
+            $this->headers = $row;
+
+            return $this;
+        }
+
+        $this->setHeaderRow($row);
+
         // Skip the header row when processing the CSV
         $this->skipRows([$this->headerRow]);
 
@@ -94,6 +104,10 @@ public function setHeaderRow($row)
      */
     public function getHeaderRow()
     {
+        if ($this->headers) {
+            return $this->headers;
+        }
+
         $handle = $this->openFile();
 
         // Skip rows until the header row

diff --git a/tests/Unit/CsvTest.php b/tests/Unit/CsvTest.php
@@ -325,3 +325,31 @@ function makeTestFile($rows = 1000)
 
     unlink($file);
 });
+
+test('It maps to custom headers', function () {
+
+    $file = makeTestFile(2);
+
+    $csv = Csv::from($file)
+        ->mapToHeaders([
+            'fii',
+            'foo',
+            'faa',
+        ])
+        ->get();
+
+    expect($csv)->toBe([
+        [
+            'fii' => 'Foo',
+            'foo' => 'Bar',
+            'faa' => 'Baz',
+        ],
+        [
+            'fii' => 'Foo0',
+            'foo' => 'Bar0',
+            'faa' => 'Baz0',
+        ],
+    ]);
+
+    unlink($file);
+});