Introduce API interfaces for Assigning Stocks to Sales channels #151

app/code/Magento/InventoryApi/Api/Data/StockExtension.php

@@ -0,0 +1,33 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\InventoryApi\Api\Data;
+
+/**
+ * TODO: temporal fix of extension classes generation during installation
+ * Extension class for @see \Magento\InventoryApi\Api\Data\StockInterface
+ */
+class StockExtension extends \Magento\Framework\Api\AbstractSimpleObject implements StockExtensionInterface
+{
+    /**
+     * @return \Magento\InventorySalesApi\Api\Data\SalesChannelInterface[]|null
+     */
+    public function getSalesChannels()
+    {
+        return $this->_get('sales_channels');
+    }
+
+    /**
+     * @param \Magento\InventorySalesApi\Api\Data\SalesChannelInterface[] $salesChannels
+     * @return $this
+     */
+    public function setSalesChannels($salesChannels)
+    {
+        $this->setData('sales_channels', $salesChannels);
+        return $this;
+    }
+}

app/code/Magento/InventoryApi/Api/Data/StockExtensionInterface.php

@@ -0,0 +1,26 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\InventoryApi\Api\Data;
+
+/**
+ * TODO: temporal fix of extension classes generation during installation
+ * ExtensionInterface class for @see \Magento\InventoryApi\Api\Data\StockInterface
+ */
+interface StockExtensionInterface extends \Magento\Framework\Api\ExtensionAttributesInterface
+{
+    /**
+     * @return \Magento\InventorySalesApi\Api\Data\SalesChannelInterface[]|null
+     */
+    public function getSalesChannels();
+
+    /**
+     * @param \Magento\InventorySalesApi\Api\Data\SalesChannelInterface[] $salesChannels
+     * @return $this
+     */
+    public function setSalesChannels($salesChannels);
+}

app/code/Magento/InventoryApi/Test/Api/SourceRepository/CreateTest.php

@@ -72,7 +72,7 @@ protected function tearDown()
     {
         /** @var ResourceConnection $connection */
         $connection = Bootstrap::getObjectManager()->get(ResourceConnection::class);
-        $connection->getConnection()->delete('inventory_source', [
+        $connection->getConnection()->delete($connection->getTableName('inventory_source'), [
             SourceInterface::NAME . ' IN (?)' => ['source-name-1'],
         ]);
         parent::tearDown();

app/code/Magento/InventoryApi/Test/Api/SourceRepository/RegionProcessingTest.php

@@ -82,7 +82,7 @@ protected function tearDown()
     {
         /** @var ResourceConnection $connection */
         $connection = Bootstrap::getObjectManager()->get(ResourceConnection::class);
-        $connection->getConnection()->delete('inventory_source', [
+        $connection->getConnection()->delete($connection->getTableName('inventory_source'), [
             SourceInterface::NAME . ' IN (?)' => ['source-name-1'],
         ]);
         parent::tearDown();

app/code/Magento/InventoryCatalog/etc/module.xml

@@ -6,5 +6,9 @@
  */
 -->
 <config xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="urn:magento:framework:Module/etc/module.xsd">
-    <module name="Magento_InventoryCatalog" setup_version="1.0.0" />
+    <module name="Magento_InventoryCatalog" setup_version="2.0.0">
+        <sequence>
+            <module name="InventorySales"/>
+        </sequence>
+    </module>
 </config>

app/code/Magento/InventorySales/Model/GetAssignedSalesChannelsForStock.php

@@ -0,0 +1,58 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\InventorySales\Model;
+
+use Magento\InventorySales\Model\ResourceModel\GetAssignedSalesChannelsDataForStock;
+use Magento\InventorySalesApi\Api\Data\SalesChannelInterfaceFactory;
+use Magento\InventorySalesApi\Api\Data\SalesChannelInterface;
+
+/**
+ * @inheritdoc
+ */
+class GetAssignedSalesChannelsForStock implements GetAssignedSalesChannelsForStockInterface
+{
+    /**
+     * @var GetAssignedSalesChannelsDataForStock
+     */
+    private $getAssignedSalesChannelsDataForStock;
+
+    /**
+     * @var SalesChannelInterfaceFactory
+     */
+    private $salesChannelFactory;
+
+    /**
+     * @param GetAssignedSalesChannelsDataForStock $getAssignedSalesChannelsDataForStock
+     * @param SalesChannelInterfaceFactory $salesChannelFactory
+     */
+    public function __construct(
+        GetAssignedSalesChannelsDataForStock $getAssignedSalesChannelsDataForStock,
+        SalesChannelInterfaceFactory $salesChannelFactory
+    ) {
+        $this->getAssignedSalesChannelsDataForStock = $getAssignedSalesChannelsDataForStock;
+        $this->salesChannelFactory = $salesChannelFactory;
+    }
+
+    /**
+     * @inheritdoc
+     */
+    public function execute(int $stockId) : array
+    {
+        $salesChannelsData = $this->getAssignedSalesChannelsDataForStock->execute($stockId);
+
+        $salesChannels = [];
+        foreach ($salesChannelsData as $salesChannelData) {
+            /** @var SalesChannelInterface $salesChannel */
+            $salesChannel = $this->salesChannelFactory->create();
+            $salesChannel->setType($salesChannelData[SalesChannelInterface::TYPE]);
+            $salesChannel->setCode($salesChannelData[SalesChannelInterface::CODE]);
+            $salesChannels[] = $salesChannel;
+        }
+        return $salesChannels;
+    }
+}

app/code/Magento/InventorySales/Model/GetAssignedSalesChannelsForStockInterface.php

@@ -0,0 +1,27 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\InventorySales\Model;
+
+use Magento\InventorySalesApi\Api\Data\SalesChannelInterface;
+
+/**
+ * Get assigned Sales Channels for Stock (Service Provider Interface - SPI)
+ * Provide own implementation of this interface if you would like to replace channels management strategy
+ *
+ * @api
+ */
+interface GetAssignedSalesChannelsForStockInterface
+{
+    /**
+     * Get linked sales channels for Stock
+     *
+     * @param int $stockId
+     * @return SalesChannelInterface[]
+     */
+    public function execute(int $stockId): array;
+}

app/code/Magento/InventorySales/Model/OptionSource/WebsiteSource.php

@@ -42,7 +42,7 @@ public function toOptionArray(): array
             }
             $websites[] = [
                 'value' => $website->getCode(),
-                'label' => $website->getName()
+                'label' => $website->getName(),
             ];
         }
         return $websites;

app/code/Magento/InventorySales/Model/ReplaceSalesChannelsForStockInterface.php

@@ -0,0 +1,26 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\InventorySales\Model;
+
+/**
+ * Implementation of links replacement between Stock and Sales Channels (Service Provider Interface - SPI)
+ * Provide own implementation of this interface if you would like to replace channels management strategy
+ *
+ * @api
+ */
+interface ReplaceSalesChannelsForStockInterface
+{
+    /**
+     * Replace Sales Channels for Stock
+     *
+     * @param array $salesChannels
+     * @param int $stockId
+     * @return void
+     */
+    public function execute(array $salesChannels, int $stockId);
+}

app/code/Magento/InventorySales/Model/ResourceModel/GetAssignedSalesChannelsDataForStock.php

@@ -0,0 +1,49 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\InventorySales\Model\ResourceModel;
+
+use Magento\Framework\App\ResourceConnection;
+use Magento\InventorySales\Setup\Operation\CreateSalesChannelTable;
+
+/**
+ * Provides linked sales channels by given stock id
+ */
+class GetAssignedSalesChannelsDataForStock
+{
+    /**
+     * @var ResourceConnection
+     */
+    private $resourceConnection;
+
+    /**
+     * @param ResourceConnection $resourceConnection
+     */
+    public function __construct(
+        ResourceConnection $resourceConnection
+    ) {
+        $this->resourceConnection = $resourceConnection;
+    }
+
+    /**
+     * Given a stock id, return array of sales channels assigned to it
+     *
+     * @param int $stockId
+     * @return array
+     */
+    public function execute(int $stockId): array
+    {
+        $connection = $this->resourceConnection->getConnection();
+        $tableName = $this->resourceConnection->getTableName(CreateSalesChannelTable::TABLE_NAME_SALES_CHANNEL);
+
+        $select = $connection->select()
+            ->from($tableName)
+            ->where(CreateSalesChannelTable::STOCK_ID . ' = ?', $stockId);
+
+        return $connection->fetchAll($select);
+    }
+}

app/code/Magento/InventorySales/Model/ResourceModel/ReplaceSalesChannelsDataForStock.php

@@ -0,0 +1,63 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\InventorySales\Model\ResourceModel;
+
+use Magento\Framework\App\ResourceConnection;
+use Magento\InventorySales\Model\ReplaceSalesChannelsForStockInterface;
+use Magento\InventorySales\Setup\Operation\CreateSalesChannelTable;
+use Magento\InventorySalesApi\Api\Data\SalesChannelInterface;
+
+/**
+ * Implementation of links replacement between Stock and Sales Channels for specific db layer
+ *
+ * There is no additional business logic on SPI (Service Provider Interface) level so could use resource model as
+ * SPI implementation directly
+ */
+class ReplaceSalesChannelsDataForStock implements ReplaceSalesChannelsForStockInterface
+{
+    /**
+     * @var ResourceConnection
+     */
+    private $resourceConnection;
+
+    /**
+     * @param ResourceConnection $resourceConnection
+     */
+    public function __construct(
+        ResourceConnection $resourceConnection
+    ) {
+        $this->resourceConnection = $resourceConnection;
+    }
+
+    /**
+     * Replace Sales Channels for Stock
+     *
+     * @param SalesChannelInterface[] $salesChannels
+     * @param int $stockId
+     * @return void
+     */
+    public function execute(array $salesChannels, int $stockId)
+    {
+        $connection = $this->resourceConnection->getConnection();
+        $tableName = $this->resourceConnection->getTableName(CreateSalesChannelTable::TABLE_NAME_SALES_CHANNEL);
+
+        $connection->delete($tableName, [CreateSalesChannelTable::STOCK_ID . ' = ?' => $stockId]);
+
+        if (count($salesChannels)) {
+            $salesChannelsToInsert = [];
+            foreach ($salesChannels as $salesChannel) {
+                $salesChannelsToInsert[] = [
+                    SalesChannelInterface::TYPE => $salesChannel->getType(),
+                    SalesChannelInterface::CODE => $salesChannel->getCode(),
+                    CreateSalesChannelTable::STOCK_ID => $stockId,
+                ];
+            }
+            $connection->insertMultiple($tableName, $salesChannelsToInsert);
+        }
+    }
+}

app/code/Magento/InventorySales/Model/ResourceModel/StockIdResolver.php

@@ -0,0 +1,56 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\InventorySales\Model\ResourceModel;
+
+use Magento\Framework\App\ResourceConnection;
+use Magento\Framework\Exception\NoSuchEntityException;
+use Magento\InventorySales\Setup\Operation\CreateSalesChannelTable;
+use Magento\InventorySalesApi\Api\Data\SalesChannelInterface;
+
+/**
+ * This resource model is responsible for retrieving Stock items by sales channel type and code
+ * Used by Service Contracts that are agnostic to the Data Access Layer
+ */
+class StockIdResolver
+{
+    /**
+     * @var ResourceConnection
+     */
+    private $resourceConnection;
+
+    /**
+     * @param ResourceConnection $resourceConnection
+     */
+    public function __construct(
+        ResourceConnection $resourceConnection
+    ) {
+        $this->resourceConnection = $resourceConnection;
+    }
+
+    /**
+     * Returns the linked stock id by given a sales channel type and code
+     *
+     * @param string $type
+     * @param string $code
+     * @throws NoSuchEntityException
+     * @return int|null
+     */
+    public function resolve(string $type, string $code)
+    {
+        $connection = $this->resourceConnection->getConnection();
+        $tableName = $this->resourceConnection->getTableName(CreateSalesChannelTable::TABLE_NAME_SALES_CHANNEL);
+
+        $select = $connection->select()
+            ->from($tableName, CreateSalesChannelTable::STOCK_ID)
+            ->where(SalesChannelInterface::TYPE . ' = ?', $type)
+            ->where(SalesChannelInterface::CODE . ' = ?', $code);
+
+        $stockId = $connection->fetchOne($select);
+        return false === $stockId ? null : (int)$stockId;
+    }
+}