app/Plugin/Api42/GraphQL/Types.php line 70

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of EC-CUBE
  5.  *
  6.  * Copyright(c) EC-CUBE CO.,LTD. All Rights Reserved.
  7.  *
  8.  * http://www.ec-cube.co.jp/
  9.  *
  10.  * For the full copyright and license information, please view the LICENSE
  11.  * file that was distributed with this source code.
  12.  */
  14. namespace Plugin\Api42\GraphQL;
  16. use Doctrine\ORM\EntityManager;
  17. use Doctrine\ORM\Mapping\ClassMetadata;
  18. use GraphQL\Type\Definition\ObjectType;
  19. use GraphQL\Type\Definition\Type;
  20. use Plugin\Api42\GraphQL\Type\Definition\DateTimeType;
  22. /**
  23.  * DoctrineのEntityからGraphQLのObjectTypeを変換するクラス.
  24.  */
  25. class Types
  26. {
  27.     /** @var EntityManager */
  28.     private $entityManager;
  30.     private $types = [];
  32.     private $allowLists = [];
  34.     /**
  35.      * Types constructor.
  36.      */
  37.     public function __construct(EntityManager $entityManager)
  38.     {
  39.         $this->entityManager = $entityManager;
  40.     }
  42.     public function addAllowList(AllowList $allowList)
  43.     {
  44.         $this->allowLists[] = $allowList;
  45.     }
  47.     /**
  48.      * Entityに対応するObjectTypeを返す.
  49.      *
  50.      * @param $className string Entityクラス名
  51.      *
  52.      * @return ObjectType
  53.      */
  54.     public function get($className)
  55.     {
  56.         if (!isset($this->types[$className])) {
  57.             $this->types[$className] = $this->createObjectType($className);
  58.         }
  60.         return $this->types[$className];
  61.     }
  63.     private function createObjectType($className)
  64.     {
  65.         return new ObjectType([
  66.             'name' => (new \ReflectionClass($className))->getShortName(),
  67.             'fields' => function () use ($className) {
  68.                 $classMetadata = $this->entityManager->getClassMetadata($className);
  69.                 $fields = array_reduce($classMetadata->fieldMappings, function ($acc, $mapping) use ($classMetadata) {
  70.                     $type = $this->convertFieldMappingToType($mapping);
  71.                     $fieldName = $mapping['fieldName'];
  73.                     $allowed = array_filter($this->allowLists, function (AllowList $al) use ($classMetadata, $fieldName) {
  74.                         return $al->isAllowed($classMetadata->name, $fieldName);
  75.                     });
  77.                     if ($allowed && $type) {
  78.                         $acc[$fieldName] = $type;
  79.                     }
  81.                     return $acc;
  82.                 }, []);
  84.                 $fields = array_reduce($classMetadata->associationMappings, function ($acc, $mapping) use ($classMetadata) {
  85.                     $fieldName = $mapping['fieldName'];
  87.                     $allowed = array_filter($this->allowLists, function (AllowList $al) use ($classMetadata, $fieldName) {
  88.                         return $al->isAllowed($classMetadata->name, $fieldName);
  89.                     });
  91.                     if ($allowed) {
  92.                         $acc[$fieldName] = [
  93.                             'type' => $this->convertAssociationMappingToType($mapping),
  94.                         ];
  95.                     }
  97.                     return $acc;
  98.                 }, $fields);
  100.                 return $fields;
  101.             },
  102.         ]);
  103.     }
  105.     private function convertFieldMappingToType($fieldMapping)
  106.     {
  107.         $type = isset($fieldMapping['id']) ? Type::id() : [
  108.             'string' => Type::string(),
  109.             'text' => Type::string(),
  110.             'integer' => Type::int(),
  111.             'decimal' => Type::float(),
  112.             'datetimetz' => DateTimeType::dateTime(),
  113.             'smallint' => Type::int(),
  114.             'boolean' => Type::boolean(),
  115.         ][$fieldMapping['type']];
  117.         if ($type) {
  118.             return $fieldMapping['nullable'] ? $type : Type::nonNull($type);
  119.         }
  121.         return null;
  122.     }
  124.     private function convertAssociationMappingToType($mapping)
  125.     {
  126.         return $this->isToManyAssociation($mapping) ? Type::listOf($this->get($mapping['targetEntity'])) : $this->get($mapping['targetEntity']);
  127.     }
  129.     private function isToManyAssociation($mapping)
  130.     {
  131.         return $mapping['type'] & ClassMetadata::TO_MANY;
  132.     }
  133. }