setdefaultvaluestepper.php

<?php
 
namespace Bitrix\Catalog\Product\SystemField\ProductMapping;
 
use Bitrix\Catalog\Product\SystemField\ProductMapping;
use Bitrix\Catalog\ProductTable;
use Bitrix\Main\Application;
use Bitrix\Main\DB\Result;
use Bitrix\Main\DB\SqlExpression;
use Bitrix\Main\Localization\Loc;
use Bitrix\Main\Update\Stepper;
use Bitrix\Main\UserField\Internal\UserFieldHelper;
use Throwable;
 
Loc::loadMessages(__FILE__);
 
class SetDefaultValueStepper extends Stepper
{
    protected static $moduleId = "catalog";
 
    public static function getTitle()
    {
        return Loc::getMessage('CATALOG_PRODUCT_SYSTEMFIELD_PRODUCTMAPPING_SET_DEFAULT_VALUE_STEPPER_TITLE');
    }
 
    public function execute(array &$option)
    {
        $userFieldManager = UserFieldHelper::getInstance()->getManager();
 
        if (!ProductMapping::isAllowed())
        {
            return self::FINISH_EXECUTION;
        }
        $fieldDescription = ProductMapping::getUserFieldBaseParam();
 
        // first run
        if (!isset($option['is_started']))
        {
            return $this->firstRun($option);
        }
 
        if (empty($option['count']))
        {
            return self::FINISH_EXECUTION;
        }
 
        // processing
        $defaultValues = $this->getDefaultValues();
        if (!$defaultValues)
        {
            return self::FINISH_EXECUTION;
        }
 
        $lastId = (int)($option['last_id'] ?? 0);
        $products = $this->getProductsWithEmptyValue($lastId);
 
        $stepCount = 0;
 
        $db = Application::getConnection();
        try
        {
            $db->startTransaction();
 
            $entityId = ProductTable::getUfId();
            $fieldId = $fieldDescription['FIELD_NAME'];
            foreach ($products as $row)
            {
                $productId = (int)$row['id'];
                $userFieldManager->Update(
                    $entityId,
                    $productId,
                    [
                        $fieldId => $defaultValues,
                    ]
                );
                $lastId = $productId;
                $stepCount++;
            }
 
            $db->commitTransaction();
        }
        catch (Throwable $e)
        {
            $db->rollbackTransaction();
 
            throw $e;
        }
 
        if ($stepCount === 0)
        {
            return self::FINISH_EXECUTION;
        }
 
        $option['steps'] += $stepCount;
        $option['last_id'] = $lastId;
 
        return self::CONTINUE_EXECUTION;
    }
 
    private function firstRun(array & $option): bool
    {
        $connection = Application::getConnection();
        $helper = $connection->getSqlHelper();
 
        $existUfTable = $connection->query("SHOW TABLES LIKE 'b_uts_product'")->getSelectedRowsCount() > 0;
        if (!$existUfTable)
        {
            return self::FINISH_EXECUTION;
        }
 
        $existUfColumn = $connection->query(
            "SHOW COLUMNS FROM " . $helper->quote('b_uts_product') . " LIKE 'UF_PRODUCT_MAPPING'"
            )->getSelectedRowsCount() > 0;
        if (!$existUfColumn)
        {
            return self::FINISH_EXECUTION;
        }
 
        if (!$this->isNotEmptyProducts())
        {
            return self::FINISH_EXECUTION;
        }
 
        $count = $this->getProductsToBeProcessedTotalCount();
        if ($count === 0)
        {
            return self::FINISH_EXECUTION;
        }
 
        $option['count'] = $count;
        $option['title'] = self::getTitle();
        $option['is_started'] = true;
        $option['last_id'] = null;
 
        return self::CONTINUE_EXECUTION;
    }
 
    private function getProductsQuery(bool $isCountSelect = false, int $lastId = 0): string
    {
        $connection = Application::getConnection();
        $helper = $connection->getSqlHelper();
 
        $typeIds = join(',', [
            ProductTable::TYPE_PRODUCT,
            ProductTable::TYPE_SET,
            ProductTable::TYPE_SKU,
        ]);
 
        $select = 'b_catalog_product.id as ' . $helper->quote('id');
        if ($isCountSelect)
        {
            $select = 'COUNT(b_catalog_product.id)';
        }
 
        $where = '';
        if ($lastId)
        {
            $where = new SqlExpression("AND b_catalog_product.id > ?i", $lastId);
        }
 
        return trim("
        SELECT {$select}
        FROM b_catalog_product
        LEFT JOIN b_uts_product ON b_catalog_product.id = b_uts_product.value_id
        WHERE b_catalog_product.TYPE in ({$typeIds})
        AND b_uts_product.UF_PRODUCT_MAPPING IS NULL {$where}
        " . ($isCountSelect ? "" : "ORDER BY b_catalog_product.id ASC") . "
        ");
    }
 
    private function getProductsWithEmptyValue(int $lastId): Result
    {
        $limit = 10;
        $sql = $this->getProductsQuery(false, $lastId);
 
        return Application::getConnection()->query($sql, $limit);
    }
 
    private function getProductsToBeProcessedTotalCount(): int
    {
        $sql = $this->getProductsQuery(true);
 
        return (int)Application::getConnection()->queryScalar($sql);
    }
 
    private function isNotEmptyProducts(): bool
    {
        return ProductTable::getRow([
            'select' => [
                'ID',
            ]
        ]) !== null;
    }
 
    private function getDefaultValues(): ?array
    {
        $config = ProductMapping::getConfig();
        $tableName = $config['HIGHLOADBLOCK']['TABLE_NAME'] ?? null;
        if (!$tableName)
        {
            return null;
        }
 
        $values = [];
        $values[] = Application::getConnection()->queryScalar(
            new SqlExpression("SELECT ID FROM ?# WHERE UF_XML_ID = 'LANDING'", $tableName)
        );
 
        return $values;
    }
 
    public static function clearAbandonedSteppersAgent(): string
    {
        $option = \Bitrix\Main\Config\Option::get(
            'main.stepper.catalog',
            __CLASS__,
            null,
            ''
        );
        $needRemove = false;
        if ($option !== null)
        {
            if ($option === '')
            {
                $needRemove = true;
            }
            if (!CheckSerializedData($option))
            {
                $needRemove = true;
            }
            else
            {
                $option = unserialize($option, ['allowed_classes' => false]);
                if (empty($option) || !is_array($option))
                {
                    $needRemove = true;
                }
                else
                {
                    if (empty($option['count']))
                    {
                        $needRemove = true;
                    }
                }
            }
        }
 
        if ($needRemove)
        {
            \Bitrix\Main\Config\Option::delete(
                'main.stepper.catalog',
                [
                    'name' => __CLASS__
                ]
            );
        }
 
        return '';
    }
}