Fix ListItem camel-case classname

shadz3rg · Aug 7, 2017 · 5f29c5c · 5f29c5c
1 parent 5717779
commit 5f29c5c
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/src/PHPStamp/Document/WordDocument.php b/src/PHPStamp/Document/WordDocument.php
@@ -70,12 +70,16 @@ public function cleanup(\DOMDocument $template)
      */
     public function getExpression($id, Tag $tag)
     {
-        $className = 'PHPStamp\\Document\\WordDocument\\Extension\\' . ucfirst($id);
-
-        if (class_exists($className) === false) {
-            throw new InvalidArgumentException('Class by id ' . $id . ' not found.');
-        }
-
+        $available = array(
+			'cell' => 'PHPStamp\\Document\\WordDocument\\Extension\\Cell',
+			'listitem' => 'PHPStamp\\Document\\WordDocument\\Extension\\ListItem',
+		);
+
+		if (isset($available[$id]) === false) {
+			throw new InvalidArgumentException('Class by id "' . $id . '" not found.');
+		}
+
+        $className = $available[$id];
         return new $className($tag);
     }
-}
+}