Merge 0f38b9b into 429487b

eiriksm · Dec 18, 2020 · c9a43fd · c9a43fd
2 parents 429487b + 0f38b9b
commit c9a43fd
Showing 1 changed file with 46 additions and 11 deletions.
diff --git a/src/AdType.php b/src/AdType.php
@@ -45,6 +45,12 @@ abstract class AdType extends XmlBase implements AdTypeInterface
 
     protected $objectHeadLocationCountryCode;
 
+    protected $objectHeadLocationAddress2;
+
+    protected $objectHeadLocationAddress3;
+
+    protected $objectHeadLocationAddress4;
+
     protected $objectHeadHeading;
 
     protected $objectHeadSegment;
@@ -65,6 +71,12 @@ abstract class AdType extends XmlBase implements AdTypeInterface
 
     protected $countryCode;
 
+    protected $address2;
+
+    protected $address3;
+
+    protected $address4;
+
     protected $heading;
 
     protected $adBody;
@@ -130,6 +142,24 @@ public function setZipCode($zipCode)
         $this->objectHeadLocationZipcode->nodeValue = $zipCode;
     }
 
+    public function setAddress2($address)
+    {
+        $this->address2 = $address;
+        $this->objectHeadLocationAddress2->nodeValue = $address;
+    }
+
+    public function setAddress3($address)
+    {
+        $this->address3 = $address;
+        $this->objectHeadLocationAddress3->nodeValue = $address;
+    }
+
+    public function setAddress4($address)
+    {
+        $this->address4 = $address;
+        $this->objectHeadLocationAddress4->nodeValue = $address;
+    }
+
   /**
    * @param mixed $countryCode
    */
@@ -187,27 +217,32 @@ public function createObjectHead()
     {
         $this->createObjectHeadTrait();
         $mappings = [
-        'ORDERNO' => 'objectHeadOrderNo',
-        'USER_REFERENCE' => 'objectHeadUserReference',
-        'PROVIDER_REFERENCE' => 'objectHeadProviderReference',
-        'OVERWRITE_MMO' => 'objectHeadOverWriteMmo',
-        'VERSIONNO' => 'objectHeadVersionNo',
-        'FROMDATE' => 'objectHeadFromDate',
-        'TODATE' => 'objectHeadToDate',
-        'OBJECT_LOCATION' => 'objectHeadLocation',
-        'HEADING' => 'objectHeadHeading',
+            'ORDERNO' => 'objectHeadOrderNo',
+            'USER_REFERENCE' => 'objectHeadUserReference',
+            'PROVIDER_REFERENCE' => 'objectHeadProviderReference',
+            'OVERWRITE_MMO' => 'objectHeadOverWriteMmo',
+            'VERSIONNO' => 'objectHeadVersionNo',
+            'FROMDATE' => 'objectHeadFromDate',
+            'TODATE' => 'objectHeadToDate',
+            'OBJECT_LOCATION' => 'objectHeadLocation',
+            'HEADING' => 'objectHeadHeading',
         ];
         foreach ($mappings as $field => $property) {
             $this->{$property} = $this->dom->createElement($field);
             $this->objectHeadBody->appendChild($this->{$property});
         }
-      // Overwrite mmo is empty, but has a property.
+        // Overwrite mmo is empty, but has a property.
         $this->objectHeadOverWriteMmo->setAttribute('MODUS', 'all');
-      // Also, create the children of the location.
+        // Also, create the children of the location.
         $this->objectHeadLocationZipcode = $this->dom->createElement('ZIPCODE');
         $this->objectHeadLocation->appendChild($this->objectHeadLocationZipcode);
         $this->objectHeadLocationCountryCode = $this->dom->createElement('COUNTRYCODE');
         $this->objectHeadLocation->appendChild($this->objectHeadLocationCountryCode);
+        foreach ([2, 3, 4] as $address_item) {
+            $key = 'objectHeadLocationAddress' . $address_item;
+            $this->{$key} = $this->dom->createElement('ADDRESS' . $address_item);
+            $this->objectHeadLocation->appendChild($this->{$key});
+        }
         $this->objectBody->appendChild($this->objectHeadBody);
     }
 }