The start of something beautiful

vendor/doctrine/doctrine-bundle/src/Command/DoctrineCommand.php

@@ -0,0 +1,84 @@
+<?php
+
+namespace Doctrine\Bundle\DoctrineBundle\Command;
+
+use Doctrine\DBAL\Connection;
+use Doctrine\ORM\EntityManager;
+use Doctrine\ORM\Tools\EntityGenerator;
+use Doctrine\Persistence\ManagerRegistry;
+use InvalidArgumentException;
+use Symfony\Component\Console\Command\Command;
+
+/**
+ * Base class for Doctrine console commands to extend from.
+ *
+ * @internal
+ */
+abstract class DoctrineCommand extends Command
+{
+    private ManagerRegistry $doctrine;
+
+    public function __construct(ManagerRegistry $doctrine)
+    {
+        parent::__construct();
+
+        $this->doctrine = $doctrine;
+    }
+
+    /**
+     * get a doctrine entity generator
+     *
+     * @return EntityGenerator
+     *
+     * @psalm-suppress UndefinedDocblockClass ORM < 3 specific
+     */
+    protected function getEntityGenerator()
+    {
+        $entityGenerator = new EntityGenerator();
+        $entityGenerator->setGenerateAnnotations(false);
+        $entityGenerator->setGenerateStubMethods(true);
+        $entityGenerator->setRegenerateEntityIfExists(false);
+        $entityGenerator->setUpdateEntityIfExists(true);
+        $entityGenerator->setNumSpaces(4);
+        $entityGenerator->setAnnotationPrefix('ORM\\');
+
+        return $entityGenerator;
+    }
+
+    /**
+     * Get a doctrine entity manager by symfony name.
+     *
+     * @param string   $name
+     * @param int|null $shardId
+     *
+     * @return EntityManager
+     */
+    protected function getEntityManager($name, $shardId = null)
+    {
+        $manager = $this->getDoctrine()->getManager($name);
+
+        if ($shardId !== null) {
+            throw new InvalidArgumentException('Shards are not supported anymore using doctrine/dbal >= 3');
+        }
+
+        return $manager;
+    }
+
+    /**
+     * Get a doctrine dbal connection by symfony name.
+     *
+     * @param string $name
+     *
+     * @return Connection
+     */
+    protected function getDoctrineConnection($name)
+    {
+        return $this->getDoctrine()->getConnection($name);
+    }
+
+    /** @return ManagerRegistry */
+    protected function getDoctrine()
+    {
+        return $this->doctrine;
+    }
+}