MAGETWO-48240: Forms on new UI component do not support Single-Store Mode

Author: arkadiych

app/code/Magento/Catalog/Ui/Component/Listing/Columns/Websites.php

@@ -41,37 +41,15 @@ public function __construct(
         $this->storeManager = $storeManager;
     }
 
-    /**
-     * {@inheritdoc}
-     */
-    public function prepareDataSource(array $dataSource)
-    {
-        $websiteNames = [];
-        foreach ($this->getData('options') as $website) {
-            $websiteNames[$website->getWebsiteId()] = $website->getName();
-        }
-        if (isset($dataSource['data']['items'])) {
-            $fieldName = $this->getData('name');
-            foreach ($dataSource['data']['items'] as & $item) {
-                $websites = [];
-                foreach ($item[$fieldName] as $websiteId) {
-                    $websites[] = $websiteNames[$websiteId];
-                }
-                $item[$fieldName] = implode(', ', $websites);
-            }
-        }
-
-        return $dataSource;
-    }
-
     /**
      * Prepare component configuration
      * @return void
      */
     public function prepare()
     {
-        if (!$this->storeManager->isSingleStoreMode()) {
-            parent::prepare();
+        parent::prepare();
+        if ($this->storeManager->isSingleStoreMode()) {
+            $this->_data['config']['componentDisabled'] = true;
         }
     }
 }

app/code/Magento/Catalog/view/adminhtml/ui_component/product_listing.xml

@@ -230,7 +230,7 @@
                 </item>
             </argument>
         </column>
-        <column name="websites">
+        <column name="websites" class="Magento\Catalog\Ui\Component\Listing\Columns\Websites">
             <argument name="data" xsi:type="array">
                 <item name="options" xsi:type="object">Magento\Store\Model\ResourceModel\Website\Collection</item>
                 <item name="config" xsi:type="array">

app/code/Magento/Cms/view/adminhtml/ui_component/cms_page_form.xml

@@ -169,7 +169,7 @@
             </argument>
         </field>
     </fieldset>
-    <fieldset name="websites">
+    <fieldset name="websites" class="Magento\Store\Ui\Component\Form\Fieldset\Websites">
         <argument name="data" xsi:type="array">
             <item name="config" xsi:type="array">
                 <item name="collapsible" xsi:type="boolean">true</item>

app/code/Magento/Customer/Ui/Component/Listing/Column/Websites.php

@@ -0,0 +1,54 @@
+<?php
+/**
+ * Copyright © 2015 Magento. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+namespace Magento\Customer\Ui\Component\Listing\Column;
+
+use Magento\Framework\View\Element\UiComponent\ContextInterface;
+use Magento\Framework\View\Element\UiComponentFactory;
+use Magento\Ui\Component\Listing\Columns\Column;
+use Magento\Store\Model\StoreManagerInterface as StoreManager;
+
+/**
+ * Class Websites
+ */
+class Websites extends Column
+{
+    /**
+     * Store manager
+     *
+     * @var StoreManager
+     */
+    protected $storeManager;
+
+    /**
+     * @param ContextInterface $context
+     * @param UiComponentFactory $uiComponentFactory
+     * @param StoreManager $storeManager
+     * @param array $components
+     * @param array $data
+     */
+    public function __construct(
+        ContextInterface $context,
+        UiComponentFactory $uiComponentFactory,
+        StoreManager $storeManager,
+        array $components = [],
+        array $data = []
+    ) {
+        $this->storeManager = $storeManager;
+        parent::__construct($context, $uiComponentFactory, $components, $data);
+    }
+
+    /**
+     * Prepare component configuration
+     * @return void
+     */
+    public function prepare()
+    {
+        parent::prepare();
+        if ($this->storeManager->isSingleStoreMode()) {
+            $this->_data['config']['componentDisabled'] = true;
+        }
+    }
+}

app/code/Magento/Customer/view/adminhtml/ui_component/customer_listing.xml

@@ -245,7 +245,7 @@
                 </item>
             </argument>
         </column>
-        <column name="website_id">
+        <column name="website_id" class="Magento\Customer\Ui\Component\Listing\Column\Websites">
             <argument name="data" xsi:type="array">
                 <item name="config" xsi:type="array">
                     <item name="filter" xsi:type="string">select</item>

app/code/Magento/Store/Ui/Component/Form/Fieldset/Websites.php

@@ -0,0 +1,56 @@
+<?php
+/**
+ * Copyright © 2015 Magento. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+namespace Magento\Store\Ui\Component\Form\Fieldset;
+
+use Magento\Framework\View\Element\UiComponentInterface;
+use Magento\Framework\View\Element\UiComponent\ContextInterface;
+use Magento\Ui\Component\Form\Fieldset;
+use Magento\Store\Model\StoreManagerInterface as StoreManager;
+
+/**
+ * Class Websites Fieldset
+ */
+class Websites extends Fieldset
+{
+    /**
+     * Store manager
+     *
+     * @var StoreManager
+     */
+    protected $storeManager;
+
+    /**
+     * Constructor
+     *
+     * @param ContextInterface $context
+     * @param StoreManager $storeManager
+     * @param UiComponentInterface[] $components
+     * @param array $data
+     */
+    public function __construct(
+        ContextInterface $context,
+        StoreManager $storeManager,
+        array $components = [],
+        array $data = []
+    ) {
+        parent::__construct($context, $components, $data);
+        $this->storeManager = $storeManager;
+
+    }
+
+    /**
+     * Prepare component configuration
+     *
+     * @return void
+     */
+    public function prepare()
+    {
+        parent::prepare();
+        if ($this->storeManager->isSingleStoreMode()) {
+            $this->_data['config']['componentDisabled'] = true;
+        }
+    }
+}

app/code/Magento/Store/Ui/Component/Listing/Column/Store.php

@@ -10,6 +10,7 @@
 use Magento\Framework\View\Element\UiComponentFactory;
 use Magento\Store\Model\System\Store as SystemStore;
 use Magento\Ui\Component\Listing\Columns\Column;
+use Magento\Store\Model\StoreManagerInterface as StoreManager;
 
 /**
  * Class Store
@@ -30,6 +31,13 @@ class Store extends Column
      */
     protected $systemStore;
 
+    /**
+     * Store manager
+     *
+     * @var StoreManager
+     */
+    protected $storeManager;
+
     /**
      * @var string
      */
@@ -40,6 +48,7 @@ class Store extends Column
      * @param UiComponentFactory $uiComponentFactory
      * @param SystemStore $systemStore
      * @param Escaper $escaper
+     * @param StoreManager $storeManager
      * @param array $components
      * @param array $data
      * @param string $storeKey
@@ -49,13 +58,15 @@ public function __construct(
         UiComponentFactory $uiComponentFactory,
         SystemStore $systemStore,
         Escaper $escaper,
+        StoreManager $storeManager,
         array $components = [],
         array $data = [],
         $storeKey = 'store_id'
     ) {
         $this->systemStore = $systemStore;
         $this->escaper = $escaper;
         $this->storeKey = $storeKey;
+        $this->storeManager = $storeManager;
         parent::__construct($context, $uiComponentFactory, $components, $data);
     }
 
@@ -111,4 +122,16 @@ protected function prepareItem(array $item)
 
         return $content;
     }
+
+    /**
+     * Prepare component configuration
+     * @return void
+     */
+    public function prepare()
+    {
+        parent::prepare();
+        if ($this->storeManager->isSingleStoreMode()) {
+            $this->_data['config']['componentDisabled'] = true;
+        }
+    }
 }