adicionando funcao custom de export

Author: vitorbarros

src/VMBDataExport/Export/CSVExport.php

@@ -72,18 +72,11 @@ public function export()
 
     /**
      * @param array $data
+     * @param array $headers
      * @return mixed
      */
-    public function writeCustomData(array $data)
+    public function writeCustomData(array $data, array $headers)
     {
         // TODO: Implement writeCustomData() method.
     }
-
-    /**
-     * @return mixed
-     */
-    public function exportCustomData()
-    {
-        // TODO: Implement exportCustomData() method.
-    }
 }

src/VMBDataExport/Export/ExportDataServiceInterface.php

@@ -16,12 +16,8 @@ public function writeData(array $data);
 
     /**
      * @param array $data
+     * @param array $headers
      * @return mixed
      */
-    public function writeCustomData(array $data);
-
-    /**
-     * @return mixed
-     */
-    public function exportCustomData();
+    public function writeCustomData(array $data, array $headers);
 }

src/VMBDataExport/Export/PDFExport.php

@@ -28,18 +28,11 @@ public function export()
 
     /**
      * @param array $data
+     * @param array $headers
      * @return mixed
      */
-    public function writeCustomData(array $data)
+    public function writeCustomData(array $data, array $headers)
     {
         // TODO: Implement writeCustomData() method.
     }
-
-    /**
-     * @return mixed
-     */
-    public function exportCustomData()
-    {
-        // TODO: Implement exportCustomData() method.
-    }
 }

src/VMBDataExport/Export/XLSExport.php

@@ -90,18 +90,13 @@ public function export()
 
     /**
      * @param array $data
+     * @param array $headers
      * @return mixed
      */
-    public function writeCustomData(array $data)
+    public function writeCustomData(array $data, array $headers)
     {
-        // TODO: Implement writeCustomData() method.
-    }
-
-    /**
-     * @return mixed
-     */
-    public function exportCustomData()
-    {
-        // TODO: Implement exportCustomData() method.
+        $this->resultFormatedData = $data;
+        $this->headers = $headers;
+        return $this;
     }
 }

src/VMBDataExport/Module.php

@@ -1,6 +1,7 @@
 <?php
 namespace VMBDataExport;
 
+use VMBDataExport\Service\CustomExportService;
 use Zend\Mvc\ModuleRouteListener;
 use Zend\Mvc\MvcEvent;
 use VMBDataExport\Service\MainService;
@@ -32,11 +33,13 @@ public function getAutoloaderConfig()
 
     public function getServiceConfig()
     {
-
         return array(
             'factories' => array(
                 'VMBDataExport\Service\MainService' => function ($sm) {
                     return new MainService($sm->get('Doctrine\ORM\EntityManager'));
+                },
+                'VMBDataExport\Service\CustomExportService' => function ($sm) {
+                    return new CustomExportService($sm->get('Doctrine\ORM\EntityManager'));
                 }
             ),
         );

src/VMBDataExport/Service/CustomExportService.php

@@ -0,0 +1,54 @@
+<?php
+namespace VMBDataExport\Service;
+
+use Doctrine\ORM\EntityManager;
+use VMBDataExport\Export\ExportDataServiceInterface;
+
+class CustomExportService
+{
+    /**
+     * @var EntityManager
+     */
+    private $em;
+
+    /**
+     * @var ExportDataServiceInterface
+     */
+    private $class;
+
+    /**
+     * CustomExportService constructor.
+     * @param EntityManager $em
+     */
+    public function __construct(EntityManager $em)
+    {
+        $this->em = $em;
+    }
+
+    /**
+     * @param $query
+     * @param array $headerFields
+     * @param $exportClass
+     * @return bool|mixed
+     * @throws \Exception
+     */
+    public function export($query, array $headerFields, $exportClass)
+    {
+        $result = $this->em->createQuery($query)->getArrayResult();
+        
+        $type = strtoupper($exportClass);
+        $class = 'VMBDataExport\\Export\\' . $type . 'Export';
+
+        if (class_exists($class)) {
+            if (null === $this->class) {
+                $this->class = new $class($this->em);
+            }
+            if ($this->class instanceof ExportDataServiceInterface) {
+                $this->class->writeCustomData($result, $headerFields);
+                return $this->class->export();
+            }
+            throw new \Exception("Class {$class} must implements 'VMBDataExport\\Export\\ExportDataServiceInterface' ");
+        }
+        throw new \Exception("Class {$class} does not exist");
+    }
+}

src/VMBDataExport/Service/MainService.php

@@ -34,10 +34,6 @@ public function strategy($exportType, array $data)
                     $this->responsableClass = new $class($this->em);
                 }
                 if ($this->responsableClass instanceof Export\ExportDataServiceInterface) {
-                    if (isset($data['custom']) && $data['custom'] != null) {
-                        $this->responsableClass->writeCustomData($data);
-                        return $this->responsableClass->exportCustomData();
-                    }
                     $this->responsableClass->writeData($data);
                     return $this->responsableClass->export();
                 }