Add optional parameter headers

jwage · May 11, 2018 · c5354f5 · c5354f5
1 parent 7693a63
commit c5354f5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/lib/EasyCSV/AbstractBase.php b/lib/EasyCSV/AbstractBase.php
@@ -8,17 +8,25 @@ abstract class AbstractBase
     protected $delimiter = ',';
     protected $enclosure = '"';
 
-    public function __construct($path, $mode = 'r+', $isNeedBOM = false)
+    public function __construct($path, $mode = 'r+', $isNeedBOM = false, $headers = array())
     {
+        $isNewFile = false;
+
         if (! file_exists($path)) {
             touch($path);
+            $isNewFile = true;
         }
         $this->handle = new \SplFileObject($path, $mode);
         $this->handle->setFlags(\SplFileObject::DROP_NEW_LINE);
 
         if ($isNeedBOM) {
             $this->handle->fwrite("\xEF\xBB\xBF");
         }
+
+        if($isNewFile && isset($headers)) {
+            $headerLine = join(',', $headers) . PHP_EOL;
+            $this->handle->fwrite($headerLine, strlen($headerLine));
+        }
     }
 
     public function __destruct()

diff --git a/tests/EasyCSV/Tests/WriterTest.php b/tests/EasyCSV/Tests/WriterTest.php
@@ -21,6 +21,12 @@ public function testWriteRow()
         $this->assertEquals(18, $this->writer->writeRow('test1, test2, test3'));
     }
 
+    public function testWriteRowOnHeaders()
+    {
+        $writer = new Writer(__DIR__ . '/write.csv', 'r+', false, array('header1', 'header2', 'header3'));
+        $this->assertEquals(18, $writer->writeRow('test1, test2, test3'));
+    }
+
     public function testWriteBOMRow()
     {
         $this->assertEquals(57, $this->writerBOM->writeRow('колонка 1, колонка 2, колонка 3'));