Merge pull request #48 from DivanteLtd/feature/add-endpoint-for-addre…

…ss-management

Address Management - add new controller to management customer addresses.

magento1-module/app/code/local/Divante/VueStorefrontBridge/Model/Api/Customer/Address.php

@@ -0,0 +1,165 @@
+<?php
+
+/**
+ * Class Divante_VueStorefrontBridge_Model_Api_Customer_Address
+ *
+ * @package     Divante
+ * @category    VueStorefrontBridge
+ * @author      Agata Firlejczyk <[email protected]>
+ * @copyright   Copyright (C) 2018 Divante Sp. z o.o.
+ */
+class Divante_VueStorefrontBridge_Model_Api_Customer_Address
+{
+
+    /**
+     * @param $addressData
+     *
+     * @return Mage_Customer_Model_Address
+     */
+    public function loadCustomerAddressById($addressData)
+    {
+        /* @var $address Mage_Customer_Model_Address */
+        $address = Mage::getModel('customer/address');
+
+        if (isset($addressData->id) && (int)($addressData->id)) {
+            $address->load($addressData->id);
+        }
+
+        return $address;
+    }
+
+    /**
+     * @param Mage_Customer_Model_Address  $address
+     * @param array                        $addressData
+     * @param Mage_Customer_Model_Customer $customer
+     *
+     * @throws Exception
+     */
+    public function saveAddress(
+        Mage_Customer_Model_Address $address,
+        array $addressData,
+        Mage_Customer_Model_Customer $customer
+    ) {
+        $addressData['parent_id'] = $customer->getId();
+        $addressData['customer_id'] = $customer->getId();
+        $address->addData($addressData);
+
+        if (isset($addressData['default_billing']) && ($addressData['default_billing'] === true)) {
+            $address->setIsDefaultBilling(true);
+        } else {
+            $address->setIsDefaultBilling(false);
+        }
+
+        if (isset($addressData['default_shipping']) && ($addressData['default_shipping'] === true)) {
+            $address->setIsDefaultShipping(true);
+        } else {
+            $address->setIsDefaultShipping(false);
+        }
+
+        $addressErrors = $address->validate();
+
+        if ($addressErrors !== true) {
+            $message = implode('.', $addressErrors);
+            throw new Mage_Core_Exception($message);
+        }
+
+        if (!$address->getIsDefaultBilling() && ($customer->getDefaultBilling() == $address->getId())) {
+            $customer->setDefaultBilling(null);
+        }
+
+        if (!$address->getIsDefaultShipping() && ($customer->getDefaultShipping() == $address->getId())) {
+            $customer->setDefaultShipping(null);
+        }
+
+        $address->save();
+        /** @var Mage_Customer_Model_Resource_Customer $resourceModel */
+        $resourceModel = $customer->getResource();
+
+        if ($customer->dataHasChangedFor('default_billing')) {
+            $resourceModel->saveAttribute($customer, 'default_billing');
+        }
+
+        if ($customer->dataHasChangedFor('default_shipping')) {
+            $resourceModel->saveAttribute($customer, 'default_shipping');
+        }
+    }
+
+    /**
+     * @param Mage_Customer_Model_Address  $address
+     * @param Mage_Customer_Model_Customer $customer
+     *
+     * @return mixed
+     */
+    public function prepareAddress(Mage_Customer_Model_Address $address, Mage_Customer_Model_Customer $customer)
+    {
+        $addressDTO = $address->getData();
+        $addressDTO['id'] = $addressDTO['entity_id'];
+        $region = null;
+
+        if (isset($addressDTO['region'])) {
+            $region = $addressDTO['region'];
+        }
+
+        $addressDTO['region'] = ['region' => $region];
+        $streetDTO = explode("\n", $addressDTO['street']);
+
+        if (count($streetDTO) < 2) {
+            $streetDTO[] = '';
+        }
+
+        $addressDTO['street'] = $streetDTO;
+
+        if (!$addressDTO['firstname']) {
+            $addressDTO['firstname'] = $customer->getFirstname();
+        }
+
+        if (!$addressDTO['lastname']) {
+            $addressDTO['lastname'] = $customer->getLastname();
+        }
+
+        if(!$addressDTO['city']) {
+            $addressDTO['city'] = '';
+        }
+
+        if (!$addressDTO['country_id']) {
+            $addressDTO['country_id'] = $this->getDefaultCountry();
+        }
+
+        if (!$addressDTO['postcode']) {
+            $addressDTO['postcode'] = '';
+        }
+
+        if (!$addressDTO['telephone']) {
+            $addressDTO['telephone'] = '';
+        }
+
+        if ($address->getIsDefaultBilling()) {
+            $addressDTO['default_billing'] = (bool)$address->getIsDefaultBilling();
+        } elseif ($customer->getDefaultBilling() === $address->getId()) {
+            $addressDTO['default_billing'] = true;
+        }  else {
+            $addressDTO['default_billing'] = false;
+        }
+
+        if ($address->getIsDefaultShipping()) {
+            $addressDTO['default_shipping'] = (bool)$address->getIsDefaultShipping();
+        } elseif ($customer->getDefaultShipping() === $address->getId()) {
+            $addressDTO['default_shipping'] = true;
+        } else {
+            $addressDTO['default_shipping'] = false;
+        }
+
+        return $addressDTO;
+    }
+
+    /**
+     * @return string
+     */
+    private function getDefaultCountry()
+    {
+        /** @var Mage_Core_Helper_Data $helper */
+        $helper = Mage::helper('core');
+
+        return $helper->getDefaultCountry();
+    }
+}