Merge pull request #16 from rixxi/bugfix-treeable

Bugfix treeable
rixxi · Nov 13, 2015 · a532672 · a532672
2 parents cdbc9ec + fd58a60
commit a532672
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/src/Rixxi/Gedmo/DI/OrmExtension.php b/src/Rixxi/Gedmo/DI/OrmExtension.php
@@ -44,18 +44,17 @@ public function getEntityMappings()
 		$config = $this->getValidatedConfig();
 
 		$annotations = array(
-			'loggable',
-			'translatable',
-			'treeable',
+			'loggable' => 'Loggable',
+			'translatable' => 'Translatable',
+			'treeable' => 'Tree',
 		);
 
 		$path = realpath(__DIR__ . '/../../../../../../gedmo/doctrine-extensions/lib/Gedmo');
 
 		$mappings = array();
-		foreach ($annotations as $annotation) {
+		foreach ($annotations as $annotation => $namespace) {
 			if ($config['all'] || $config[$annotation]) {
-				$name = ucfirst($annotation);
-				$mappings["Gedmo\\$name\\Entity"] = "$path/$name/Entity";
+				$mappings["Gedmo\\$namespace\\Entity"] = "$path/$namespace/Entity";
 			}
 		}