<?php

global $MESS;
IncludeModuleLangFile(__FILE__);

class ICMLLoader {

    public $iblocks;
    public $filename;
    public $propertiesSKU;
    public $propertiesUnitSKU;
    public $propertiesProduct;
    public $propertiesUnitProduct;
    public $application;
    public $encoding = 'utf-8';
    public $encodingDefault = 'utf-8';

    protected $fp;
    protected $mainSection = 1000000;
    protected $pageSize = 500;
    protected $offerPageSize = 50;

    protected $isLogged = false;
    protected $logFile = '/bitrix/catalog_export/i_crm_load_log.txt';
    protected $fpLog;


    protected $measurement = array (
        'mm' => 1,          // 1 mm = 1 mm
        'cm' => 10,         // 1 cm = 10 mm
        'm' => 1000,
        'mg' => 0.001,      // 0.001 g = 1 mg
        'g' => 1,
        'kg' => 1000,
    );

    protected $measurementLink = array (
        'mm' => 'mm',
        'cm' => 'mm',
        'm' => 'mm',
        'mg' => 'g',
        'g' => 'g',
        'kg' => 'g',
    );

    public function Load()
    {
            global $USER;
            if(!isset($_SESSION["SESS_AUTH"]["USER_ID"]) || !$_SESSION["SESS_AUTH"]["USER_ID"])
                $USER = new CUser;

            $this->isLogged = true;

            $defaultSite = CSite::GetList($by="def", $order="desc", Array())->Fetch();
            $this->encodingDefault = $defaultSite["CHARSET"];

            $this->PrepareSettings();

            $this->fp = $this->PrepareFile($this->filename. '.tmp');

            if ($this->isLogged) {
                $this->fpLog = $this->PrepareFile($this->logFile);
                $this->WriteLog("Start Loading");
            }

            $this->PreWriteCatalog();

            $categories = $this->GetCategories();

            $this->WriteCategories($categories);

            $this->PreWriteOffers();
            $this->BuildOffers($categories);
            $this->PostWriteOffers();

            $this->PostWriteCatalog();

            if ($this->isLogged) {
                $this->WriteLog("Loading was ended successfully (peek memory usage: " . memory_get_peak_usage() . ")");
            }

            $this->CloseFile($this->fp);
            $this->CloseFile($this->fpLog);

            unlink($_SERVER["DOCUMENT_ROOT"] . $this->filename);
            rename($_SERVER["DOCUMENT_ROOT"] . $this->filename. '.tmp', $_SERVER["DOCUMENT_ROOT"] . $this->filename);

            return true;

    }

    protected function PrepareSettings()
    {
        foreach ($this->propertiesSKU as $iblock => $arr) {
            foreach ($arr as $id => $sku) {
                $this->propertiesSKU[$iblock][$id] = strtoupper($sku);
            }
        }
        foreach ($this->propertiesProduct as $iblock => $arr) {
            foreach ($arr as $id => $prod) {
                $this->propertiesProduct[$iblock][$id] = strtoupper($prod);
            }
        }
    }

    protected function PrepareValue($text)
    {
        $newText = $this->application->ConvertCharset($text, $this->encodingDefault, $this->encoding);
        $newText = strip_tags($newText);
        $newText = str_replace("&", "&#x26;", $newText);
        return $newText;
    }

    protected function PrepareFile($filename)
    {
            $fullFilename = $_SERVER["DOCUMENT_ROOT"] . $filename;
            CheckDirPath($fullFilename);

            if ($fp = @fopen($fullFilename, "w"))
                return $fp;
            else
                return false;
    }

    protected function PreWriteCatalog()
    {
            @fwrite($this->fp, "<yml_catalog date=\"" . $this->PrepareValue(Date("Y-m-d H:i:s")) . "\">\n
                <shop>\n
                <name>" . $this->PrepareValue(COption::GetOptionString("main", "site_name", ""))."</name>\n
                <company>" . $this->PrepareValue(COption::GetOptionString("main", "site_name", ""))."</company>\n"
                    );

    }

    protected function WriteCategories($categories)
    {
            $stringCategories = "";
            @fwrite($this->fp, "<categories>\n");
            foreach ($categories as $category) {
                $stringCategories .= $this->BuildCategory($category);
            }
            @fwrite($this->fp, $stringCategories);
            @fwrite($this->fp, "</categories>\n");
    }
    protected function PreWriteOffers()
    {
            @fwrite($this->fp, "<offers>\n");
    }

        protected function PostWriteOffers()
    {
            @fwrite($this->fp, "</offers>\n");
    }

        protected function WriteOffers($offers)
    {
            @fwrite($this->fp, $offers);
    }

    protected function WriteLog($text)
    {
        if ($this->isLogged)
            @fwrite($this->fpLog, Date("Y:m:d H:i:s") . ": " . $text . "\n");
    }

    protected function PostWriteCatalog()
    {
            @fwrite($this->fp, "</shop>\n
                </yml_catalog>\n");
    }

    protected function CloseFile($fp)
    {
            @fclose($fp);
    }


    protected function GetCategories()
    {
            $categories = array();
            foreach ($this->iblocks as $id)
            {
                $filter = array("IBLOCK_ID" => $id);

                $dbRes = CIBlockSection::GetList(array("left_margin" => "asc"), $filter);
                $hasCategories = false;
                while ($arRes = $dbRes->Fetch()) {
                    $categories[$arRes['ID']] = $arRes;
                    $hasCategories = true;
                }
                if (!$hasCategories) {
                    $iblock = CIBlock::GetByID($id)->Fetch();

                    $arRes = Array();
                    $arRes['ID'] = $this->mainSection + $id;
                    $arRes['IBLOCK_SECTION_ID'] = 0;
                    $arRes['NAME'] = sprintf(GetMessage('ROOT_CATEGORY_FOR_CATALOG'), $iblock['NAME']);
                    $categories[$arRes['ID']] = $arRes;
                }
            }
            return $categories;

    }

    protected function BuildCategory($arCategory)
    {
            return "
                    <category id=\"" . $this->PrepareValue($arCategory["ID"]) . "\""
                    . ( intval($arCategory["IBLOCK_SECTION_ID"] ) > 0 ?
                            " parentId=\"" . $this->PrepareValue($arCategory["IBLOCK_SECTION_ID"]) . "\""
                            :"")
                    . ">"
                    . $this->PrepareValue($arCategory["NAME"])
                    . "</category>\n";

    }

    protected function BuildOffers(&$allCategories)
    {
            foreach ($this->iblocks as $key => $id)
            {
                    // Get Info by infoblocks
                    $iblock['IBLOCK_DB'] = CIBlock::GetByID($id)->Fetch();
                    $iblockOffer = CCatalogSKU::GetInfoByProductIBlock($id);

                    $arSelect = Array (
                        "ID",
                        "LID",
                        "IBLOCK_ID",
                        "IBLOCK_SECTION_ID",
                        "ACTIVE",
                        "ACTIVE_FROM",
                        "ACTIVE_TO",
                        "NAME",
                        "DETAIL_PICTURE",
                        "DETAIL_TEXT",
                        "DETAIL_PICTURE",
                        "LANG_DIR",
                        "DETAIL_PAGE_URL",
                        "CATALOG_GROUP_1"
                    );
                    // Set selected properties
                    foreach ($this->propertiesProduct[$id] as $key => $propProduct) {
                        if ($this->propertiesProduct[$id][$key] != "") {
                            $arSelect[] =  "PROPERTY_" . $propProduct;
                            $arSelect[] =  "PROPERTY_" . $propProduct . ".NAME";
                        }
                    }

                    $arSelectOffer = Array (
                        'ID',
                        'ACTIVE',
                        "NAME",
                        "DETAIL_TEXT",
                        "DETAIL_PAGE_URL",
                        "DETAIL_PICTURE",
                        'PROPERTY_' . $iblockOffer['SKU_PROPERTY_ID'],
                        "CATALOG_GROUP_1"
                    );
                    // Set selected properties
                    foreach ($this->propertiesSKU[$id] as $key => $propSKU) {
                        if ($this->propertiesSKU[$id][$key] != "") {
                            $arSelectOffer[] =  "PROPERTY_" . $propSKU;
                            $arSelectOffer[] =  "PROPERTY_" . $propSKU . ".NAME";
                        }
                    }

                    // Set filter
                    $filter = array(
                        "IBLOCK_ID" => $id,
                        "INCLUDE_SUBSECTIONS" => "Y",
                    );
                    $order = array("id");
                    $arNavStatParams = Array(
                        "iNumPage" => 1,
                        "nPageSize" => $this->pageSize,
                    );

                    // Cycle page to page
                    do {
                        // Get products on this page
                        $dbResProducts = CIBlockElement::GetList($order, $filter, false, $arNavStatParams, $arSelect);

                        $pictures = array();
                        $products = array();
                        while ($product = $dbResProducts->GetNext()) {

                            // Compile products to array
                            $products[$product['ID']] = $product;
                            $products[$product['ID']]['offers'] = array();

                            $detailPicture = intval($product["DETAIL_PICTURE"]);
                            $previewPicture = intval($product["PREVIEW_PICTURE"]);

                            if ($detailPicture > 0 || $previewPicture > 0)
                            {
                                $picture = $detailPicture;
                                if ($picture <= 0) {
                                        $picture = $previewPicture;
                                }

                                // Link pictureID and productID
                                $pictures[$picture] = $product['ID'];
                            }
                        }
                        unset($product);
                        unset($detailPicture, $previewPicture, $picture);

                        $pictureIDs = array_keys($pictures);

                        // Get pathes of pictures
                        $dbFiles = CFile::GetList(Array(), Array("@ID" => implode(',', $pictureIDs)));
                        while($file = $dbFiles->GetNext()) {

                            // Link picture to product
                            $products[$pictures[$file['ID']]]['PICTURE'] = ($_SERVER["HTTPS"] == 'on' ? "https://" : "http://") .
                                            $_SERVER['SERVER_NAME'] .
                                            '/upload/' . $file['SUBDIR'] .
                                            '/' . $file['FILE_NAME'] ;
                        }
                        unset($pictures);

                        if (!empty($iblockOffer['IBLOCK_ID'])) {
                            $arFilterOffer = array(
                                'IBLOCK_ID' => $iblockOffer['IBLOCK_ID'],
                                'PROPERTY_' . $iblockOffer['SKU_PROPERTY_ID'] => array_keys($products),
                            );

                            // Get all offers for products on this page
                            $dbResOffers = CIBlockElement::GetList(
                                array(),
                                $arFilterOffer,
                                false,
                                array('nTopCount' => $this->pageSize * $this->offerPageSize),
                                $arSelectOffer
                            );

                            while ($offer = $dbResOffers->GetNext()) {
                                // Link offers to products
                                $products[$offer['PROPERTY_' . $iblockOffer['SKU_PROPERTY_ID'] . '_VALUE']]['offers'][$offer['ID']] = $offer;
                            }
                            unset($offer, $dbResOffers);
                        }

                        $stringOffers = "";
                        foreach ($products as $product) {

                                // Get properties of product
                                $resPropertiesProduct = Array();
                                foreach ($this->propertiesProduct[$id] as $key => $propProduct) {
                                    $resPropertiesProduct[$key] = "";

                                    if ($propProduct != "") {

                                        if (isset ($product["PROPERTY_" . $propProduct . "_NAME"]))
                                            $resPropertiesProduct[$key] =  $product["PROPERTY_" . $propProduct . "_NAME"];
                                        elseif (isset ($product[$propProduct]))
                                            $resPropertiesProduct[$key] =  $product[$propProduct];
                                        else
                                            $resPropertiesProduct[$key] =  $product["PROPERTY_" . $propProduct . "_VALUE"];

                                        if (array_key_exists($key, $this->propertiesUnitProduct[$id])) {
                                            $resPropertiesProduct[$key] *= $this->measurement[$this->propertiesUnitProduct[$id][$key]];
                                            $resPropertiesProduct[$key . "_UNIT"] = $this->measurementLink[$this->propertiesUnitProduct[$id][$key]];
                                        }
                                    }
                                }

                                // Get categories of product
                                $categories = Array();
                                $dbResCategories = CIBlockElement::GetElementGroups($product['ID'], true);
                                while ($arResCategory = $dbResCategories->Fetch()) {
                                    $categories[$arResCategory["ID"]] = array(
                                        'ID' => $arResCategory["ID"],
                                        'NAME' => $arResCategory["NAME"],
                                        );
                                }
                                if (count($categories) == 0) {

                                    $catId = $this->mainSection + $id;
                                    $categories[$catId] = $allCategories[$catId];
                                }


                                $existOffer = false;
                                if (!empty($iblockOffer['IBLOCK_ID'])) {

                                    foreach ($product['offers'] as $offer) {


                                        $offer['PRODUCT_ID'] = $product["ID"];
                                        $offer['DETAIL_PAGE_URL'] = $product["DETAIL_PAGE_URL"];
                                        $offer['PICTURE'] = $product["PICTURE"];
                                        $offer['PRODUCT_NAME'] = $product["NAME"];
                                        $offer['PRODUCT_ACTIVE'] = $product["ACTIVE"];
                                        $offer['PRICE'] = $offer['CATALOG_PRICE_1'];
                                        $offer['QUANTITY'] = $offer["CATALOG_QUANTITY"];

                                        // Get properties of product

                                        foreach ($this->propertiesSKU[$id] as $key => $propSKU) {

                                            if ($propSKU != "") {

                                                if (isset ($offer["PROPERTY_" . $propSKU . "_NAME"]))
                                                    $offer[$key] =  $offer["PROPERTY_" . $propSKU . "_NAME"];
                                                elseif (isset ($offer[$propSKU]))
                                                    $offer[$key] = $offer[$propSKU];
                                                else
                                                    $offer[$key] =  $offer["PROPERTY_" . $propSKU . "_VALUE"];

                                                if (array_key_exists($key, $this->propertiesUnitSKU[$id])) {
                                                    $offer[$key] *= $this->measurement[$this->propertiesUnitSKU[$id][$key]];
                                                    $offer[$key . "_UNIT"] = $this->measurementLink[$this->propertiesUnitSKU[$id][$key]];
                                                }
                                            }

                                        }

                                        foreach ($resPropertiesProduct as $key => $propProduct) {
                                            if ($this->propertiesProduct[$id][$key] != "" && !isset($offer[$key]))
                                                $offer[$key] =  $propProduct;
                                        }

                                        $stringOffers .= $this->BuildOffer($offer, $categories, $iblock, $allCategories);
                                        $existOffer = true;
                                    }
                                }
                                if (!$existOffer) {

                                    $product['PRODUCT_ID'] = $product["ID"];
                                    $product['PRODUCT_NAME'] = $product["NAME"];
                                    $product['PRODUCT_ACTIVE'] = $product["ACTIVE"];
                                    $product['PRICE'] = $product['CATALOG_PRICE_1'];
                                    $product['QUANTITY'] = $product["CATALOG_QUANTITY"];

                                    foreach ($resPropertiesProduct as $key => $propProduct) {
                                        if ($this->propertiesProduct[$id][$key] != "" || $this->propertiesProduct[$id][str_replace("_UNIT", "", $key)] != "") {
                                            $product[$key] =  $propProduct;
                                        }
                                    }

                                    $stringOffers .= $this->BuildOffer($product, $categories, $iblock, $allCategories);
                                }

                        }
                        unset($products);

                        if ($this->isLogged)
                            $this->WriteLog(($this->pageSize * $arNavStatParams['iNumPage']) . " product(s) has been loaded from " . $id . " IB (memory usage: " . memory_get_usage() . ")");
                        if ($stringOffers != "") {
                            $this->WriteOffers($stringOffers);
                            $stringOffers = "";
                        }

                        $arNavStatParams['iNumPage'] = $dbResProducts->NavPageNomer + 1;
                    }
                    while ($dbResProducts->NavPageNomer < $dbResProducts->NavPageCount);
            }
    }


    protected function BuildOffer($arOffer, $categories, $iblock, &$allCategories)
    {
            $offer = "";
            $offer .= "<offer id=\"" .$this->PrepareValue($arOffer["ID"]) . "\" ".
                    "productId=\"" . $this->PrepareValue($arOffer["PRODUCT_ID"]) . "\" ".
                    "quantity=\"" . $this->PrepareValue(DoubleVal($arOffer['QUANTITY'])) . "\">\n";

            if ($arOffer['PRODUCT_ACTIVE'] == "N")
                $offer .= "<productActivity>" .  $this->PrepareValue($arOffer['PRODUCT_ACTIVE']) . "</productActivity>\n";

            $keys = array_keys($categories);
            if (strpos($arOffer['DETAIL_PAGE_URL'], "#SECTION_PATH#") !== false) {
                if (count($categories) != 0) {

                    $category = $allCategories[$keys[0]];
                    $path = $category['CODE'];

                    if(intval($category["IBLOCK_SECTION_ID"] ) != 0) {
                        while (true) {
                            $category = $allCategories[$category['IBLOCK_SECTION_ID']];
                            $path = $category['CODE'] . '/' . $path;
                            if(intval($category["IBLOCK_SECTION_ID"] ) == 0)
                                break;
                        }
                    }

                }
                $arOffer['DETAIL_PAGE_URL'] = str_replace("#SECTION_PATH#", $path, $arOffer['DETAIL_PAGE_URL']);
            }

            $offer .= "<picture>" . $this->PrepareValue($arOffer["PICTURE"]) . "</picture>\n";
            $offer .= "<url>" . ($_SERVER["HTTPS"] == 'on' ? "https://" : "http://") . $_SERVER['SERVER_NAME'] . $this->PrepareValue($arOffer['DETAIL_PAGE_URL']) . "</url>\n";

            $offer .= "<price>" . $this->PrepareValue($arOffer['PRICE']) . "</price>\n";
            foreach ($categories as $category)
                $offer .= "<categoryId>" . $category['ID'] . "</categoryId>\n";

            $offer .= "<name>" . $this->PrepareValue($arOffer["NAME"]) . "</name>\n";

            $offer .= "<xmlId>" . $this->PrepareValue($arOffer["EXTERNAL_ID"]) . "</xmlId>\n";
            $offer .= "<productName>" . $this->PrepareValue($arOffer["PRODUCT_NAME"]) . "</productName>\n";

            foreach ($this->propertiesProduct[$iblock['IBLOCK_DB']['ID']] as $key => $propProduct) {
                if ($propProduct != "" && $arOffer[$key] != null) {
                    if ($key === "manufacturer")
                        $offer .= "<vendor>" . $this->PrepareValue($arOffer[$key]) . "</vendor>\n";
                    else
                        $offer .= '<param name="' . $key . '"' . (isset($arOffer[$key . "_UNIT"]) ? ' unit="' . $arOffer[$key . "_UNIT"] . '"' : "") . ">" . $this->PrepareValue($arOffer[$key]) . "</param>\n";
                }
            }
            foreach ($this->propertiesSKU[$iblock['IBLOCK_DB']['ID']] as $key => $propProduct) {
                if ($propProduct != "" && $arOffer[$key] != null) {
                    if ($key === "manufacturer")
                        $offer .= "<vendor>" . $this->PrepareValue($arOffer[$key]) . "</vendor>\n";
                    else
                        $offer .= '<param name="' . $key . '"' . (isset($arOffer[$key . "_UNIT"]) ? ' unit="' . $arOffer[$key . "_UNIT"] . '"' : "") . ">" . $this->PrepareValue($arOffer[$key]) . "</param>\n";
                }
            }

            $offer.= "</offer>\n";
            return $offer;
    }
}