Export from collection, array and PDOStatement added

Author: sujancse

composer.json

@@ -10,7 +10,10 @@
         }
     ],
     "minimum-stability": "dev",
-    "require": {},
+    "require": {
+        "php": "^5.5",
+        "ext-pdo": "*"
+    },
     "autoload": {
         "psr-4": {
             "Sujan\\Exporter\\": "src/"

src/Export.php

@@ -4,8 +4,10 @@
 namespace Sujan\Exporter;
 
 
+use Exception;
 use Generator;
 use Illuminate\Database\Eloquent\Builder;
+use Illuminate\Database\Eloquent\Collection;
 
 class Export
 {
@@ -27,7 +29,7 @@ class Export
     /**
      * @var string
      */
-    private $contentType;
+    private $contentType = 'application/csv';
 
     /**
      * @var string
@@ -37,43 +39,39 @@ class Export
     /**
      * @var string
      */
-    private $delimiter;
+    private $delimiter = ";";
 
     /**
      * Export constructor.
-     * @param Builder|null $model
+     * @param object | array $model
      * @param array $columns
-     * @param string $contentType
      * @param string $filename
-     * @param string $delimiter
      */
     public function __construct(
-        Builder $model = null,
+        $model,
         array $columns = [],
-        string $contentType = 'application/csv',
-        string $filename = 'export.csv',
-        string $delimiter = ";"
+        string $filename = 'export.csv'
     )
     {
         $this->setModel($model);
         $this->setColumns($columns);
         $this->setHeading($columns);
         $this->setFilename($filename);
-        $this->setContentType($contentType);
-        $this->setDelimiter($delimiter);
+        $this->setContentType();
     }
 
     /**
-     * @param $contentType
+     * Set content type
      */
-    public function setContentType($contentType)
+    public function setContentType()
     {
-        header("Content-Type: {$contentType}");
+        header("Content-Type: {$this->contentType}");
         header("Content-Disposition: attachment; filename={$this->filename};");
     }
 
     /**
      * return generated CSV
+     * @throws Exception
      */
     public function export()
     {
@@ -82,10 +80,13 @@ public function export()
         while ($generator->valid()) {
             $generator->next();
         }
+
+        die();
     }
 
     /**
      * @return Generator
+     * @throws Exception
      */
     public function write()
     {
@@ -94,10 +95,40 @@ public function write()
 
         fputcsv($file, $this->heading, $this->delimiter);
 
-        foreach ($this->model->get() as $data) {
-            $line = $this->getLine($data);
+        if (is_array($this->model)) {
+            foreach ($this->model as $data) {
+                $line = $this->getLine($data);
 
-            yield fputcsv($file, $line, $this->delimiter);
+                yield fputcsv($file, $line, $this->delimiter);
+            }
+        } else {
+            $className = !is_string($this->model) ? class_basename($this->model) : null;
+
+            switch ($className) {
+                case 'Collection':
+                    foreach ($this->model as $data) {
+                        $line = $this->getLine($data);
+
+                        yield fputcsv($file, $line, $this->delimiter);
+                    }
+                    break;
+                case 'Builder':
+                    foreach ($this->model->get() as $data) {
+                        $line = $this->getLine($data);
+
+                        yield fputcsv($file, $line, $this->delimiter);
+                    }
+                    break;
+                case 'PDOStatement':
+                    foreach ($this->model->fetchAll() as $data) {
+                        $line = $this->getLine($data);
+
+                        yield fputcsv($file, $line, $this->delimiter);
+                    }
+                    break;
+                default:
+                    throw new Exception('Type unknown');
+            }
         }
 
         fclose($file);
@@ -116,7 +147,8 @@ public function getLine($data)
                 $value = $this->getNestedData($data, $key, $k);
                 array_push($line, $value);
             } else {
-                array_push($line, $data->{$key});
+                $value = is_array($data) ? $data[$key] : $data->{$key};
+                array_push($line, $value);
             }
         }
 
@@ -132,7 +164,11 @@ public function getLine($data)
     public function getNestedData($data, $keys, $k)
     {
         foreach ($keys as $kk => $key) {
-            $data = isset($data->{$k}->{$key}) ? $data->{$k}->{$key} : '';
+            if (is_array($data)) {
+                $data = isset($data[$k][$key]) ? $data[$k][$key] : '';
+            } else {
+                $data = isset($data->{$k}->{$key}) ? $data->{$k}->{$key} : '';
+            }
 
             if (is_array($data)) {
                 $this->getNestedData($data, $key, $kk);
@@ -151,9 +187,9 @@ protected function setDelimiter(string $delimiter): void
     }
 
     /**
-     * @param Builder $model
+     * @param object | array $model
      */
-    protected function setModel(Builder $model): void
+    protected function setModel($model): void
     {
         $this->model = $model;
     }

src/Exporter.php

@@ -0,0 +1,33 @@
+<?php
+
+
+namespace Sujan\Exporter;
+
+
+use Exception;
+
+class Exporter
+{
+    protected static $exporter;
+
+    /**
+     * @param $model
+     * @param array $columns
+     * @param string $filename
+     * @throws Exception
+     */
+    public static function init($model, array $columns, string $filename): void
+    {
+        if (gettype($model) == 'string') throw new Exception('Type string for model is not allowed.');
+
+        self::$exporter = new Export($model, $columns, $filename);
+    }
+
+    /**
+     * Export the data and die
+     */
+    public static function export(): void
+    {
+        self::$exporter->export();
+    }
+}

src/contracts/ExporterContract.php

@@ -0,0 +1,12 @@
+<?php
+
+
+namespace Sujan\Exporter\contracts;
+
+
+
+interface ExporterContract
+{
+    public function init($model, array $columns, $filename): void;
+    public function export(): void;
+}