Required method getEntityClassNames based on entityName

contributte · Feb 3, 2017 · 2134af2 · 2134af2
1 parent 12878be
commit 2134af2
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/src/Generator/Repository/RepositoryGenerator.php b/src/Generator/Repository/RepositoryGenerator.php
@@ -8,22 +8,28 @@
 use Minetro\Normgen\Resolver\IRepositoryResolver;
 use Nette\PhpGenerator\Helpers;
 use Nette\PhpGenerator\PhpNamespace;
+use Minetro\Normgen\Resolver\IEntityResolver;
 
 class RepositoryGenerator extends AbstractGenerator
 {
 
     /** @var IRepositoryResolver */
     private $resolver;
+
+    /** @var IEntityResolver */
+    private $entityResolver;
 
     /**
      * @param Config $config
      * @param IRepositoryResolver $resolver
+     * @param IEntityResolver $entityResolver
      */
-    function __construct(Config $config, IRepositoryResolver $resolver)
+    function __construct(Config $config, IRepositoryResolver $resolver, IEntityResolver $entityResolver)
     {
         parent::__construct($config);
 
         $this->resolver = $resolver;
+        $this->entityResolver = $entityResolver;
     }
 
     /**
@@ -42,6 +48,13 @@ public function generate(Database $database)
                 $class->setExtends($extends);
             }
 
+            $entityName = $this->entityResolver->resolveEntityName($table);
+            $class->addMethod("getEntityClassNames")
+				->addDocument("@return array")
+				->setVisibility('public')
+				->setStatic(true)
+				->addBody("return [$entityName::class];");
+
             // Save file
             $this->generateFile($this->resolver->resolveRepositoryFilename($table), (string)$namespace);
         }

diff --git a/src/SimpleFactory.php b/src/SimpleFactory.php
@@ -47,7 +47,7 @@ public function create()
         }
 
         $normgen->setEntityGenerator(new EntityGenerator($this->config, $resolver));
-        $normgen->setRepositoryGenerator(new RepositoryGenerator($this->config, $resolver));
+        $normgen->setRepositoryGenerator(new RepositoryGenerator($this->config, $resolver, $resolver));
         $normgen->setMapperGenerator(new MapperGenerator($this->config, $resolver));
         $normgen->setFacadeGenerator(new FacadeGenerator($this->config, $resolver));