Initial commit

Setup/InstallData.php

@@ -0,0 +1,74 @@
+<?php
+namespace Shopkeeper\VendorNotes\Setup;
+
+use Magento\Eav\Setup\EavSetup;
+use Magento\Eav\Setup\EavSetupFactory;
+use Magento\Framework\Setup\InstallDataInterface;
+use Magento\Framework\Setup\ModuleContextInterface;
+use Magento\Framework\Setup\ModuleDataSetupInterface;
+use Magento\Catalog\Model\Product;
+use Magento\Eav\Model\Entity\Attribute\SetFactory as AttributeSetFactory;
+use Magento\Eav\Model\ResourceModel\Entity\Attribute\Set\CollectionFactory as AttributeSetCollectionFactory;
+
+class InstallData implements InstallDataInterface
+{
+    private $eavSetupFactory;
+    private $attributeSetCollectionFactory;
+    private $attributeSetFactory;
+
+    public function __construct(
+        EavSetupFactory $eavSetupFactory,
+        AttributeSetCollectionFactory $attributeSetCollectionFactory,
+        AttributeSetFactory $attributeSetFactory
+    ) {
+        $this->eavSetupFactory = $eavSetupFactory;
+        $this->attributeSetCollectionFactory = $attributeSetCollectionFactory;
+        $this->attributeSetFactory = $attributeSetFactory;
+    }
+
+    public function install(ModuleDataSetupInterface $setup, ModuleContextInterface $context)
+    {
+        $setup->startSetup();
+        $eavSetup = $this->eavSetupFactory->create(['setup' => $setup]);
+
+        // Create the attribute
+        $eavSetup->addAttribute(
+            Product::ENTITY,
+            'vendor',
+            [
+                'type' => 'int',
+                'label' => 'Vendor',
+                'input' => 'select',
+                'source' => \Magento\Eav\Model\Entity\Attribute\Source\Table::class,
+                'required' => false,
+                'sort_order' => 50,
+                'global' => \Magento\Eav\Model\Entity\Attribute\ScopedAttributeInterface::SCOPE_GLOBAL,
+                'group' => 'General',
+                'used_in_product_listing' => true,
+                'visible_on_front' => false,
+                'option' => [
+                    'values' => [
+                        'Thermo Fisher',
+                        'Orion',
+                        'Nalgene',
+                        'Thermo Fisher Parts'
+                    ]
+                ]
+            ]
+        );
+
+        // Assign the attribute to all existing attribute sets
+        $attributeId = $eavSetup->getAttributeId(Product::ENTITY, 'vendor');
+        $entityTypeId = $eavSetup->getEntityTypeId(Product::ENTITY);
+
+        $attributeSetCollection = $this->attributeSetCollectionFactory->create();
+        $attributeSetCollection->addFieldToFilter('entity_type_id', $entityTypeId);
+
+        foreach ($attributeSetCollection as $attributeSet) {
+            $groupId = $eavSetup->getAttributeGroupId($entityTypeId, $attributeSet->getId(), 'General');
+            $eavSetup->addAttributeToGroup($entityTypeId, $attributeSet->getId(), $groupId, $attributeId, null);
+        }
+
+        $setup->endSetup();
+    }
+}