Improved add attribute feature

Edujugon · Apr 23, 2018 · c3d3158 · c3d3158
1 parent f47e051
commit c3d3158
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/src/XMLMapper.php b/src/XMLMapper.php
@@ -459,27 +459,28 @@ public function wrapWith($tagName)
      *
      * @param $tagName
      * @param array $attributes
+     * @return $this
      */
     public function addAttributes(array $attributes, $tagName = null)
     {
         $obj = $this->getObj();
 
         if (is_null($tagName) || $tagName === $obj->getName()) {
             $this->addAttributesToObj($obj, $attributes);
+        } else {
+            foreach ($obj->children() as $key => $element) {
+                if ($key === $tagName) {
+                    $this->addAttributesToObj($element, $attributes);
 
-            return;
-        }
-
-        foreach ($obj->children() as $key => $element) {
-            if ($key === $tagName) {
-                $this->addAttributesToObj($element, $attributes);
-
-                return;
+                    break;
+                }
             }
         }
 
         //update xml
         $this->loadObj($obj);
+
+        return $this;
     }
 
     /**