Moved entity export logic to own service.

This commit is contained in:
Jan Böhmer 2019-04-11 19:22:13 +02:00
parent a46bcd229b
commit b827737caa
2 changed files with 146 additions and 82 deletions

View file

@ -37,6 +37,7 @@ use App\Entity\NamedDBElement;
use App\Entity\StructuralDBElement; use App\Entity\StructuralDBElement;
use App\Form\BaseEntityAdminForm; use App\Form\BaseEntityAdminForm;
use App\Form\ExportType; use App\Form\ExportType;
use App\Services\EntityExporter;
use Doctrine\ORM\EntityManagerInterface; use Doctrine\ORM\EntityManagerInterface;
use Symfony\Bundle\FrameworkBundle\Controller\AbstractController; use Symfony\Bundle\FrameworkBundle\Controller\AbstractController;
use Symfony\Component\HttpFoundation\Request; use Symfony\Component\HttpFoundation\Request;
@ -51,6 +52,7 @@ use Symfony\Component\Serializer\SerializerInterface;
*/ */
class AttachmentTypeController extends AbstractController class AttachmentTypeController extends AbstractController
{ {
/** /**
* @Route("/{id}/edit", requirements={"id"="\d+"}, name="attachment_type_edit") * @Route("/{id}/edit", requirements={"id"="\d+"}, name="attachment_type_edit")
* @Route("/{id}/", requirements={"id"="\d+"}) * @Route("/{id}/", requirements={"id"="\d+"})
@ -113,7 +115,7 @@ class AttachmentTypeController extends AbstractController
$parent = $entity->getParent(); $parent = $entity->getParent();
//Move all sub entities to the current parent //Move all sub entities to the current parent
foreach($entity->getSubelements() as $subelement) { foreach ($entity->getSubelements() as $subelement) {
$subelement->setParent($parent); $subelement->setParent($parent);
$entityManager->persist($subelement); $entityManager->persist($subelement);
} }
@ -134,11 +136,11 @@ class AttachmentTypeController extends AbstractController
* @param EntityManagerInterface $em * @param EntityManagerInterface $em
* @return Response * @return Response
*/ */
public function exportAll(Request $request, SerializerInterface $serializer, EntityManagerInterface $em) public function exportAll(EntityManagerInterface $em, EntityExporter $exporter, Request $request)
{ {
$entities = $em->getRepository(AttachmentType::class)->findAll(); $entities = $em->getRepository(AttachmentType::class)->findAll();
return $this->exportHelper($entities, $request, $serializer); return $exporter->exportEntityFromRequest($entities,$request);
} }
/** /**
@ -146,86 +148,9 @@ class AttachmentTypeController extends AbstractController
* @param Request $request * @param Request $request
* @param AttachmentType $entity * @param AttachmentType $entity
*/ */
public function exportEntity(Request $request, AttachmentType $entity, SerializerInterface $serializer) public function exportEntity(AttachmentType $entity, EntityExporter $exporter, Request $request)
{ {
return $this->exportHelper($entity, $request, $serializer); return $exporter->exportEntityFromRequest($entity, $request);
} }
protected function exportHelper($entity, Request $request, SerializerInterface $serializer) : Response
{
$format = $request->get('format') ?? "json";
//Check if we have one of the supported formats
if (!in_array($format, ['json', 'csv', 'yaml', 'xml'])) {
throw new \InvalidArgumentException("Given format is not supported!");
}
//Check export verbosity level
$level = $request->get('level') ?? 'extended';
if (!in_array($level, ['simple', 'extended', 'full'])) {
throw new \InvalidArgumentException('Given level is not supported!');
}
//Check for include children option
$include_children = $request->get('include_children') ?? false;
//Check which groups we need to export, based on level and include_children
$groups = array($level);
if ($include_children) {
$groups[] = 'include_children';
}
//Plain text should work for all types
$content_type = "text/plain";
//Try to use better content types based on the format
switch ($format) {
case 'xml':
$content_type = "application/xml";
break;
case 'json':
$content_type = "application/json";
break;
}
$response = new Response($serializer->serialize($entity, $format,
[
'groups' => $groups,
'as_collection' => true,
'csv_delimiter' => ';', //Better for Excel
'xml_root_node_name' => 'PartDBExport'
]));
$response->headers->set('Content-Type', $content_type);
//If view option is not specified, then download the file.
if (!$request->get('view')) {
if ($entity instanceof NamedDBElement) {
$entity_name = $entity->getName();
} elseif (is_array($entity)) {
if (empty($entity)) {
throw new \InvalidArgumentException('$entity must not be empty!');
}
//Use the class name of the first element for the filename
$reflection = new \ReflectionClass($entity[0]);
$entity_name = $reflection->getShortName();
} else {
throw new \InvalidArgumentException('$entity type is not supported!');
}
$filename = "export_" . $entity_name . "_" . $level . "." . $format;
// Create the disposition of the file
$disposition = $response->headers->makeDisposition(
ResponseHeaderBag::DISPOSITION_ATTACHMENT,
$filename
);
// Set the content disposition
$response->headers->set('Content-Disposition', $disposition);
}
return $response;
}
} }

View file

@ -0,0 +1,139 @@
<?php
/**
*
* part-db version 0.1
* Copyright (C) 2005 Christoph Lechner
* http://www.cl-projects.de/
*
* part-db version 0.2+
* Copyright (C) 2009 K. Jacobs and others (see authors.php)
* http://code.google.com/p/part-db/
*
* Part-DB Version 0.4+
* Copyright (C) 2016 - 2019 Jan Böhmer
* https://github.com/jbtronics
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
*
*/
namespace App\Services;
use App\Entity\NamedDBElement;
use Symfony\Component\HttpFoundation\Request;
use Symfony\Component\HttpFoundation\Response;
use Symfony\Component\HttpFoundation\ResponseHeaderBag;
use Symfony\Component\Serializer\SerializerInterface;
/**
* Use this class to export an entity to multiple file formats.
* @package App\Services
*/
class EntityExporter
{
protected $serializer;
public function __construct(SerializerInterface $serializer)
{
$this->serializer = $serializer;
}
/**
* Exports an Entity or an array of entities to multiple file formats.
*
* @param $entity NamedDBElement|NamedDBElement[] The element/elements[] that should be exported
* @param Request $request The request that should be used for option resolving.
* @return Response The generated response containing the exported data.
* @throws \ReflectionException
*/
public function exportEntityFromRequest($entity, Request $request) : Response
{
$format = $request->get('format') ?? "json";
//Check if we have one of the supported formats
if (!in_array($format, ['json', 'csv', 'yaml', 'xml'])) {
throw new \InvalidArgumentException("Given format is not supported!");
}
//Check export verbosity level
$level = $request->get('level') ?? 'extended';
if (!in_array($level, ['simple', 'extended', 'full'])) {
throw new \InvalidArgumentException('Given level is not supported!');
}
//Check for include children option
$include_children = $request->get('include_children') ?? false;
//Check which groups we need to export, based on level and include_children
$groups = array($level);
if ($include_children) {
$groups[] = 'include_children';
}
//Plain text should work for all types
$content_type = "text/plain";
//Try to use better content types based on the format
switch ($format) {
case 'xml':
$content_type = "application/xml";
break;
case 'json':
$content_type = "application/json";
break;
}
$response = new Response($this->serializer->serialize($entity, $format,
[
'groups' => $groups,
'as_collection' => true,
'csv_delimiter' => ';', //Better for Excel
'xml_root_node_name' => 'PartDBExport'
]));
$response->headers->set('Content-Type', $content_type);
//If view option is not specified, then download the file.
if (!$request->get('view')) {
if ($entity instanceof NamedDBElement) {
$entity_name = $entity->getName();
} elseif (is_array($entity)) {
if (empty($entity)) {
throw new \InvalidArgumentException('$entity must not be empty!');
}
//Use the class name of the first element for the filename
$reflection = new \ReflectionClass($entity[0]);
$entity_name = $reflection->getShortName();
} else {
throw new \InvalidArgumentException('$entity type is not supported!');
}
$filename = "export_" . $entity_name . "_" . $level . "." . $format;
// Create the disposition of the file
$disposition = $response->headers->makeDisposition(
ResponseHeaderBag::DISPOSITION_ATTACHMENT,
$filename
);
// Set the content disposition
$response->headers->set('Content-Disposition', $disposition);
}
return $response;
}
}