<?xml version="1.0"?>
<schema xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
        xsi:noNamespaceSchemaLocation="urn:magento:framework:Setup/Declaration/Schema/etc/schema.xsd">

    <!-- table mironsoft_loyalty_points_ledger: see chapter 3 -->
    <!-- tables mironsoft_loyalty_reward_entity and _varchar/_int/_decimal/_text/_datetime: see chapter 11 -->

    <table name="company" resource="default">
        <column xsi:type="varchar" name="loyalty_tier_override" nullable="true" length="32"
                comment="Loyalty Tier Override"/>
    </table>
</schema>

<?xml version="1.0"?>
<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
        xsi:noNamespaceSchemaLocation="urn:magento:framework:Api/etc/extension_attributes.xsd">
    <extension_attributes for="Magento\Company\Api\Data\CompanyInterface">
        <attribute code="loyalty_tier_override" type="string"/>
    </extension_attributes>
</config>

<?php

declare(strict_types=1);

namespace Mironsoft\Loyalty\Plugin\Company;

use Magento\Company\Api\CompanyRepositoryInterface;
use Magento\Company\Api\Data\CompanyExtensionFactory;
use Magento\Company\Api\Data\CompanyInterface;

/**
 * Bridges the flat loyalty_tier_override column on the company table to the
 * CompanyInterface extension attribute, in both read and write direction.
 */
class AddLoyaltyTierOverridePlugin
{
    /**
     * @param CompanyExtensionFactory $extensionFactory Creates the extension attributes container.
     */
    public function __construct(
        private readonly CompanyExtensionFactory $extensionFactory
    ) {
    }

    /**
     * Copies the raw column value into the extension attribute after loading a company.
     *
     * @param CompanyRepositoryInterface $subject Unused, required by the plugin signature.
     * @param CompanyInterface $result The freshly loaded company.
     * @return CompanyInterface
     */
    public function afterGet(CompanyRepositoryInterface $subject, CompanyInterface $result): CompanyInterface
    {
        return $this->applyExtensionAttribute($result);
    }

    /**
     * Copies the extension attribute value back onto the plain data array before saving,
     * so the resource model persists it as a normal column.
     *
     * @param CompanyRepositoryInterface $subject Unused, required by the plugin signature.
     * @param CompanyInterface $company The company about to be saved.
     * @return array<int, CompanyInterface>
     */
    public function beforeSave(CompanyRepositoryInterface $subject, CompanyInterface $company): array
    {
        $extensionAttributes = $company->getExtensionAttributes();

        if ($extensionAttributes !== null && $extensionAttributes->getLoyaltyTierOverride() !== null) {
            // @phpstan-ignore-next-line CompanyInterface has no setData() in the interface, only on the model.
            $company->setData('loyalty_tier_override', $extensionAttributes->getLoyaltyTierOverride());
        }

        return [$company];
    }

    /**
     * @param CompanyInterface $company The company to enrich with the extension attribute.
     * @return CompanyInterface
     */
    private function applyExtensionAttribute(CompanyInterface $company): CompanyInterface
    {
        $extensionAttributes = $company->getExtensionAttributes() ?? $this->extensionFactory->create();
        // @phpstan-ignore-next-line CompanyInterface has no getData() in the interface, only on the model.
        $extensionAttributes->setLoyaltyTierOverride($company->getData('loyalty_tier_override'));
        $company->setExtensionAttributes($extensionAttributes);

        return $company;
    }
}

<?xml version="1.0"?>
<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
        xsi:noNamespaceSchemaLocation="urn:magento:framework:ObjectManager/etc/config.xsd">
    <type name="Magento\Company\Api\CompanyRepositoryInterface">
        <plugin name="mironsoft_loyalty_add_tier_override"
                type="Mironsoft\Loyalty\Plugin\Company\AddLoyaltyTierOverridePlugin"/>
    </type>
</config>

<?xml version="1.0"?>
<form xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
      xsi:noNamespaceSchemaLocation="urn:magento:module:Magento_Ui:etc/ui_configuration.xsd">
    <fieldset name="company">
        <field name="loyalty_tier_override">
            <settings>
                <dataType>select</dataType>
                <formElement>select</formElement>
                <label translate="true">Loyalty Tier Override</label>
                <dataScope>loyalty_tier_override</dataScope>
                <options class="Mironsoft\Loyalty\Model\Source\LoyaltyTier"/>
            </settings>
        </field>
    </fieldset>
</form>

<?xml version="1.0"?>
<config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
        xsi:noNamespaceSchemaLocation="urn:magento:framework:Module/etc/module.xsd">
    <module name="Mironsoft_Loyalty">
        <sequence>
            <module name="Magento_Customer"/>
            <module name="Magento_Sales"/>
            <module name="Magento_Eav"/>
            <module name="Magento_Catalog"/>
            <module name="Magento_Company"/>
        </sequence>
    </module>
</config>