<?php
namespace VendorModuleSetup;

use MagentoEavSetupEavSetup;
use MagentoEavSetupEavSetupFactory;
use MagentoFrameworkSetupInstallDataInterface;
use MagentoFrameworkSetupModuleContextInterface;
use MagentoFrameworkSetupModuleDataSetupInterface;

class InstallData implements InstallDataInterface
{
    protected $eavSetupFactory;

    public function __construct(EavSetupFactory $eavSetupFactory)
    {
        $this->eavSetupFactory = $eavSetupFactory;
    }

    public function install(ModuleDataSetupInterface $setup, ModuleContextInterface $context)
    {
        $eavSetup = $this->eavSetupFactory->create(['setup' => $setup]);

        $fieldList = [
            'price',
            'special_price',
            'special_from_date',
            'special_to_date',
            'minimal_price',
            'cost',
            'tier_price',
            'weight',
        ];

        foreach ($fieldList as $field) {
            $applyTo = explode(
                ',',
                $eavSetup->getAttribute(MagentoCatalogModelProduct::ENTITY, $field, 'apply_to')
            );
            if (!in_array(VendorModuleModelProductTypeMyPT1::TYPE_CODE, $applyTo)) {
                $applyTo[] = VendorModuleModelProductTypeMyPT1::TYPE_CODE;
                $eavSetup->updateAttribute(
                    MagentoCatalogModelProduct::ENTITY,
                    $field,
                    'apply_to',
                    implode(',', $applyTo)
                );
            }

            if (!in_array(VendorModuleModelProductTypeMyPT2::TYPE_CODE, $applyTo)) {
                $applyTo[] = VendorModuleModelProductTypeMyPT2::TYPE_CODE;
                $eavSetup->updateAttribute(
                        MagentoCatalogModelProduct::ENTITY,
                        $field,
                        'apply_to',
                        implode(',', $applyTo)
                        );
            }
        }
    }
}