Fix generator to use extension manager (#2864)

hechoendrupal · Nov 1, 2016 · 3ce95b8 · 3ce95b8
1 parent f3e7a79
commit 3ce95b8
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/Generator/EntityContentGenerator.php b/src/Generator/EntityContentGenerator.php
@@ -180,32 +180,32 @@ public function generate($module, $entity_name, $entity_class, $label, $base_pat
         if ($revisionable) {
           $this->renderFile(
             'module/src/Entity/Form/entity-content-revision-delete.php.twig',
-            $this->getSite()->getFormPath($module).'/'.$entity_class.'RevisionDeleteForm.php',
+              $this->extensionManager->getModule($module)->getFormPath() .'/'.$entity_class.'RevisionDeleteForm.php',
             $parameters
           );
           $this->renderFile(
             'module/src/Entity/Form/entity-content-revision-revert-translation.php.twig',
-            $this->getSite()->getFormPath($module).'/'.$entity_class.'RevisionRevertTranslationForm.php',
+              $this->extensionManager->getModule($module)->getFormPath() .'/'.$entity_class.'RevisionRevertTranslationForm.php',
             $parameters
           );
           $this->renderFile(
             'module/src/Entity/Form/entity-content-revision-revert.php.twig',
-            $this->getSite()->getFormPath($module).'/'.$entity_class.'RevisionRevertForm.php',
+              $this->extensionManager->getModule($module)->getFormPath().'/'.$entity_class.'RevisionRevertForm.php',
             $parameters
           );
           $this->renderFile(
             'module/src/entity-storage.php.twig',
-            $this->getSite()->getSourcePath($module).'/'.$entity_class.'Storage.php',
+            $this->extensionManager->getModule($module)->getSourcePath() .'/'.$entity_class.'Storage.php',
             $parameters
           );
           $this->renderFile(
             'module/src/interface-entity-storage.php.twig',
-            $this->getSite()->getSourcePath($module).'/'.$entity_class.'StorageInterface.php',
+            $this->extensionManager->getModule($module)->getSourcePath() .'/'.$entity_class.'StorageInterface.php',
             $parameters
           );
           $this->renderFile(
             'module/src/Controller/entity-controller.php.twig',
-            $this->getSite()->getControllerPath($module).'/'.$entity_class.'Controller.php',
+            $this->extensionManager->getModule($module)->getControllerPath() .'/'.$entity_class.'Controller.php',
             $parameters
           );
         }