Type checking

Author: ivanlanin

src/PhpWord/Writer/HTML/Element/Footnote.php

@@ -38,6 +38,10 @@ class Footnote extends Element
      */
     public function write()
     {
+        if (!$this->element instanceof \PhpOffice\PhpWord\Element\Footnote) {
+            return;
+        }
+
         $noteId = count($this->parentWriter->getNotes()) + 1;
         $noteMark = $this->noteType . '-' . $this->element->getRelationId();
         $this->parentWriter->addNote($noteId, $noteMark);

src/PhpWord/Writer/HTML/Element/Image.php

@@ -34,10 +34,11 @@ class Image extends Element
      */
     public function write()
     {
-        $html = '';
-        if (!$this->element instanceof ImageElement) {
-            return $html;
+        if (!$this->element instanceof \PhpOffice\PhpWord\Element\Text) {
+            return;
         }
+
+        $html = '';
         if (!$this->parentWriter->isPdf()) {
             $imageData = $this->getBase64ImageData($this->element);
             if (!is_null($imageData)) {

src/PhpWord/Writer/HTML/Element/Link.php

@@ -31,6 +31,10 @@ class Link extends Element
      */
     public function write()
     {
+        if (!$this->element instanceof \PhpOffice\PhpWord\Element\Link) {
+            return;
+        }
+
         $html = "<a href=\"{$this->element->getTarget()}\">{$this->element->getText()}</a>" . PHP_EOL;
         if (!$this->withoutP) {
             $html = '<p>' . $html . '</p>' . PHP_EOL;

src/PhpWord/Writer/HTML/Element/ListItem.php

@@ -31,6 +31,10 @@ class ListItem extends Element
      */
     public function write()
     {
+        if (!$this->element instanceof \PhpOffice\PhpWord\Element\ListItem) {
+            return;
+        }
+
         $text = htmlspecialchars($this->element->getTextObject()->getText());
         $html = '<p>' . $text . '</p>' . PHP_EOL;
 

src/PhpWord/Writer/HTML/Element/Table.php

@@ -31,6 +31,10 @@ class Table extends Element
      */
     public function write()
     {
+        if (!$this->element instanceof \PhpOffice\PhpWord\Element\Table) {
+            return;
+        }
+
         $html = '';
         $rows = $this->element->getRows();
         $rowCount = count($rows);

src/PhpWord/Writer/HTML/Element/Text.php

@@ -36,6 +36,10 @@ class Text extends Element
      */
     public function write()
     {
+        if (!$this->element instanceof \PhpOffice\PhpWord\Element\Text) {
+            return;
+        }
+
         $html = '';
         // Paragraph style
         $paragraphStyle = $this->element->getParagraphStyle();
@@ -44,6 +48,8 @@ public function write()
             $styleWriter = new ParagraphStyleWriter($paragraphStyle);
             $paragraphStyle = $styleWriter->write();
         }
+        $hasParagraphStyle = $paragraphStyle && !$this->withoutP;
+
         // Font style
         $fontStyle = $this->element->getFontStyle();
         $fontStyleIsObject = ($fontStyle instanceof Font);
@@ -52,7 +58,7 @@ public function write()
             $fontStyle = $styleWriter->write();
         }
 
-        if ($paragraphStyle && !$this->withoutP) {
+        if ($hasParagraphStyle) {
             $attribute = $pStyleIsObject ? 'style' : 'class';
             $html .= "<p {$attribute}=\"{$paragraphStyle}\">";
         }
@@ -64,7 +70,7 @@ public function write()
         if ($fontStyle) {
             $html .= '</span>';
         }
-        if ($paragraphStyle && !$this->withoutP) {
+        if ($hasParagraphStyle) {
             $html .= '</p>' . PHP_EOL;
         }
 

src/PhpWord/Writer/HTML/Element/TextRun.php

@@ -34,6 +34,10 @@ class TextRun extends Element
      */
     public function write()
     {
+        if (!$this->element instanceof \PhpOffice\PhpWord\Element\TextRun) {
+            return;
+        }
+
         $html = '';
         $elements = $this->element->getElements();
         if (count($elements) > 0) {

src/PhpWord/Writer/HTML/Element/Title.php

@@ -31,6 +31,10 @@ class Title extends Element
      */
     public function write()
     {
+        if (!$this->element instanceof \PhpOffice\PhpWord\Element\Title) {
+            return;
+        }
+
         $tag = 'h' . $this->element->getDepth();
         $text = htmlspecialchars($this->element->getText());
         $html = "<{$tag}>{$text}</{$tag}>" . PHP_EOL;

src/PhpWord/Writer/ODText/Element/Image.php

@@ -31,6 +31,10 @@ class Image extends Element
      */
     public function write()
     {
+        if (!$this->element instanceof \PhpOffice\PhpWord\Element\Image) {
+            return;
+        }
+
         $mediaIndex = $this->element->getMediaIndex();
         $target = 'Pictures/' . $this->element->getTarget();
         $style = $this->element->getStyle();

src/PhpWord/Writer/ODText/Element/Link.php

@@ -29,6 +29,10 @@ class Link extends Element
      */
     public function write()
     {
+        if (!$this->element instanceof \PhpOffice\PhpWord\Element\Link) {
+            return;
+        }
+
         if (!$this->withoutP) {
             $this->xmlWriter->startElement('text:p'); // text:p
         }