CsvWriter: Added option to prepend header row with column names

src/Writer/CsvWriter.php

@@ -23,20 +23,28 @@ class CsvWriter extends AbstractStreamWriter
      * @var boolean
      */
     private $utf8Encoding = false;
+    private $row = 1;
+
+    /**
+     * @var boolean
+     */
+    protected $prependHeaderRow;
 
     /**
      * @param string   $delimiter The delimiter
      * @param string   $enclosure The enclosure
      * @param resource $stream
      * @param boolean  $utf8Encoding
      */
-    public function __construct($delimiter = ';', $enclosure = '"', $stream = null, $utf8Encoding = false)
+    public function __construct($delimiter = ';', $enclosure = '"', $stream = null, $utf8Encoding = false, $prependHeaderRow = false)
     {
         parent::__construct($stream);
 
         $this->delimiter = $delimiter;
         $this->enclosure = $enclosure;
         $this->utf8Encoding = $utf8Encoding;
+
+        $this->prependHeaderRow = $prependHeaderRow;
     }
 
     /**
@@ -54,6 +62,11 @@ public function prepare()
      */
     public function writeItem(array $item)
     {
+        if ($this->prependHeaderRow && 1 == $this->row++) {
+            $headers = array_keys($item);
+            fputcsv($this->getStream(), $headers, $this->delimiter, $this->enclosure);
+        }
+
         fputcsv($this->getStream(), $item, $this->delimiter, $this->enclosure);
     }
 }