Improve Doctrine integration Fix #146, #152

src/Doctrine/DoctrineHelper.php

@@ -0,0 +1,148 @@
+<?php
+
+/*
+ * This file is part of the Symfony MakerBundle package.
+ *
+ * (c) Fabien Potencier <fabien@symfony.com>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\Bundle\MakerBundle\Doctrine;
+
+use Doctrine\Common\Persistence\Mapping\Driver\MappingDriver;
+use Doctrine\Common\Persistence\Mapping\Driver\MappingDriverChain;
+use Doctrine\ORM\EntityManagerInterface;
+use Doctrine\Common\Persistence\Mapping\ClassMetadata;
+use Doctrine\ORM\Tools\DisconnectedClassMetadataFactory;
+use Symfony\Bridge\Doctrine\ManagerRegistry;
+use Symfony\Bundle\MakerBundle\Util\ClassNameDetails;
+
+/**
+ * @author Fabien Potencier <fabien@symfony.com>
+ * @author Ryan Weaver <ryan@knpuniversity.com>
+ * @author Sadicov Vladimir <sadikoff@gmail.com>
+ *
+ * @internal
+ */
+final class DoctrineHelper
+{
+    /**
+     * @var ManagerRegistry
+     */
+    private $registry;
+
+    public function __construct(ManagerRegistry $registry = null)
+    {
+        $this->registry = $registry;
+    }
+
+    public function getRegistry(): ManagerRegistry
+    {
+        // this should never happen: we will have checked for the
+        // DoctrineBundle dependency before calling this
+        if (null === $this->registry) {
+            throw new \Exception('Somehow the doctrine service is missing. Is DoctrineBundle installed?');
+        }
+
+        return $this->registry;
+    }
+
+    private function isDoctrineInstalled(): bool
+    {
+        return null !== $this->registry;
+    }
+
+    public function getMappingDriverForClass(string $className): ?MappingDriver
+    {
+        /** @var EntityManagerInterface $em */
+        $em = $this->getRegistry()->getManagerForClass($className);
+
+        if (null === $em) {
+            throw new \InvalidArgumentException(sprintf('Cannot find the entity manager for class "%s"', $className));
+        }
+
+        $metadataDriver = $em->getConfiguration()->getMetadataDriverImpl();
+
+        if (!$metadataDriver instanceof MappingDriverChain) {
+            return $metadataDriver;
+        }
+
+        foreach ($metadataDriver->getDrivers() as $namespace => $driver) {
+            if (0 === strpos($className, $namespace)) {
+                return $driver;
+            }
+        }
+
+        return $metadataDriver->getDefaultDriver();
+    }
+
+    public function getEntitiesForAutocomplete(): array
+    {
+        $entities = [];
+
+        if ($this->isDoctrineInstalled()) {
+            $allMetadata = $this->getMetadata();
+
+            /* @var ClassMetadata $metadata */
+            foreach (array_keys($allMetadata) as $classname) {
+                $entityClassDetails = new ClassNameDetails($classname, 'App\\Entity');
+                $entities[] = $entityClassDetails->getRelativeName();
+            }
+        }
+
+        return $entities;
+    }
+
+    /**
+     * @param string|null $classOrNamespace
+     * @param bool        $disconnected
+     *
+     * @return array|ClassMetadata
+     */
+    public function getMetadata(string $classOrNamespace = null, bool $disconnected = false)
+    {
+        $metadata = [];
+
+        /** @var EntityManagerInterface $em */
+        foreach ($this->getRegistry()->getManagers() as $em) {
+            if ($disconnected) {
+                $cmf = new DisconnectedClassMetadataFactory();
+                $cmf->setEntityManager($em);
+            } else {
+                $cmf = $em->getMetadataFactory();
+            }
+
+            foreach ($cmf->getAllMetadata() as $m) {
+                if (null === $classOrNamespace) {
+                    $metadata[$m->getName()] = $m;
+                } else {
+                    if ($m->getName() == $classOrNamespace) {
+                        return $m;
+                    }
+
+                    if (0 === strpos($m->getName(), $classOrNamespace)) {
+                        $metadata[$m->getName()] = $m;
+                    }
+                }
+            }
+        }
+
+        return $metadata;
+    }
+
+    /**
+     * @param string $entityClassName
+     */
+    public function createDoctrineDetails(string $entityClassName): ?EntityDetails
+    {
+        $metadata = $this->getMetadata($entityClassName);
+
+        if ($metadata instanceof ClassMetadata) {
+            return new EntityDetails($metadata);
+        }
+
+        return null;
+    }
+}