Add rector rules for B2BKit

Author: TheMilek

config/sets/sylius/plus/b2b-suite.php

@@ -0,0 +1,44 @@
+<?php
+
+declare(strict_types=1);
+
+use Rector\Config\RectorConfig;
+use Sylius\SyliusRector\Rector\Class_\AddInterfaceToClassExtendingTypeRector;
+use Sylius\SyliusRector\Rector\Class_\AddTraitToClassExtendingTypeRector;
+
+return static function (RectorConfig $rectorConfig): void {
+    $rectorConfig->ruleWithConfiguration(AddInterfaceToClassExtendingTypeRector::class, [
+        'Sylius\Component\Core\Model\Address' => [
+            'Sylius\B2BKit\Entity\AddressInterface',
+        ],
+        'Sylius\Component\Core\Model\Customer' => [
+            'Sylius\B2BKit\Entity\CustomerInterface',
+        ],
+        'Sylius\Component\Customer\Model\CustomerGroup' => [
+            'Sylius\B2BKit\Entity\CustomerGroupInterface',
+        ],
+        'Sylius\Component\Core\Model\Order' => [
+            'Sylius\B2BKit\Entity\OrderInterface',
+        ],
+        'Sylius\Component\Core\Model\ShopUser' => [
+            'Sylius\B2BKit\Entity\ShopUserInterface',
+        ],
+    ]);
+    $rectorConfig->ruleWithConfiguration(AddTraitToClassExtendingTypeRector::class, [
+        'Sylius\Component\Core\Model\Address' => [
+            'Sylius\B2BKit\Entity\AddressAwareTrait',
+        ],
+        'Sylius\Component\Core\Model\Customer' => [
+            'Sylius\B2BKit\Entity\CustomerAwareTrait',
+        ],
+        'Sylius\Component\Customer\Model\CustomerGroup' => [
+            'Sylius\B2BKit\Entity\CustomerGroupAwareTrait',
+        ],
+        'Sylius\Component\Core\Model\Order' => [
+            'Sylius\B2BKit\Entity\OrderAwareTrait',
+        ],
+        'Sylius\Component\Core\Model\ShopUser' => [
+            'Sylius\B2BKit\Entity\ShopUserAwareTrait',
+        ],
+    ]);
+};

src/Set/SyliusPlus.php

@@ -6,6 +6,7 @@
 
 final class SyliusPlus
 {
+    public const B2B_SUITE = __DIR__ . '/../../config/sets/sylius/plus/b2b-suite.php';
     public const LOYALTY_PLUGIN = __DIR__ . '/../../config/sets/sylius/plus/loyalty-plugin.php';
     public const MULTI_SOURCE_INVENTORY_PLUGIN = __DIR__ . '/../../config/sets/sylius/plus/multi-source-inventory-plugin.php';
     public const MULTI_STORE_PLUGIN = __DIR__ . '/../../config/sets/sylius/plus/multi-store-plugin.php';

stubs/Sylius/B2BKit/Entity/AddressAwareTrait.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\B2BKit\Entity;
+
+if (trait_exists('Sylius\B2BKit\Entity\AddressAwareTrait')) {
+    return;
+}
+
+trait AddressAwareTrait
+{
+}

stubs/Sylius/B2BKit/Entity/AddressInterface.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\B2BKit\Entity;
+
+if (class_exists('Sylius\B2BKit\Entity\AddressInterface')) {
+    return;
+}
+
+interface AddressInterface
+{
+}

stubs/Sylius/B2BKit/Entity/CustomerAwareTrait.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\B2BKit\Entity;
+
+if (trait_exists('Sylius\B2BKit\Entity\CustomerAwareTrait')) {
+    return;
+}
+
+trait CustomerAwareTrait
+{
+}

stubs/Sylius/B2BKit/Entity/CustomerGroupAwareTrait.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\B2BKit\Entity;
+
+if (trait_exists('Sylius\B2BKit\Entity\CustomerGroupAwareTrait')) {
+    return;
+}
+
+trait CustomerGroupAwareTrait
+{
+}

stubs/Sylius/B2BKit/Entity/CustomerGroupInterface.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\B2BKit\Entity;
+
+if (class_exists('Sylius\B2BKit\Entity\CustomerGroupInterface')) {
+    return;
+}
+
+interface CustomerGroupInterface
+{
+}

stubs/Sylius/B2BKit/Entity/CustomerInterface.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\B2BKit\Entity;
+
+if (class_exists('Sylius\B2BKit\Entity\CustomerInterface')) {
+    return;
+}
+
+interface CustomerInterface
+{
+}

stubs/Sylius/B2BKit/Entity/OrderAwareTrait.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\B2BKit\Entity;
+
+if (trait_exists('Sylius\B2BKit\Entity\OrderAwareTrait')) {
+    return;
+}
+
+trait OrderAwareTrait
+{
+}

stubs/Sylius/B2BKit/Entity/OrderInterface.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\B2BKit\Entity;
+
+if (class_exists('Sylius\B2BKit\Entity\OrderInterface')) {
+    return;
+}
+
+interface OrderInterface
+{
+}

stubs/Sylius/B2BKit/Entity/ShopUserAwareTrait.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\B2BKit\Entity;
+
+if (trait_exists('Sylius\B2BKit\Entity\ShopUserAwareTrait')) {
+    return;
+}
+
+trait ShopUserAwareTrait
+{
+}

stubs/Sylius/B2BKit/Entity/ShopUserInterface.php

@@ -0,0 +1,13 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\B2BKit\Entity;
+
+if (class_exists('Sylius\B2BKit\Entity\ShopUserInterface')) {
+    return;
+}
+
+interface ShopUserInterface
+{
+}

stubs/Sylius/Component/Core/Model/Address.php

@@ -0,0 +1,11 @@
+<?php
+
+namespace Sylius\Component\Core\Model;
+
+if (class_exists('Sylius\Component\Core\Model\Address')) {
+    return;
+}
+
+class Address
+{
+}

stubs/Sylius/Component/Core/Model/ShopUser.php

@@ -0,0 +1,11 @@
+<?php
+
+namespace Sylius\Component\Core\Model;
+
+if (class_exists('Sylius\Component\Core\Model\ShopUser')) {
+    return;
+}
+
+class ShopUser
+{
+}

stubs/Sylius/Component/Customer/Model/CustomerGroup.php

@@ -0,0 +1,11 @@
+<?php
+
+namespace Sylius\Component\Customer\Model;
+
+if (class_exists('Sylius\Component\Customer\Model\CustomerGroup')) {
+    return;
+}
+
+class CustomerGroup
+{
+}

tests/Set/SyliusPlus/B2BSuite/B2BSuiteTest.php

@@ -0,0 +1,32 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Sylius\SyliusRector\Tests\Set\SyliusPlus\B2BKit;
+
+use Iterator;
+use Rector\Testing\PHPUnit\AbstractRectorTestCase;
+
+final class B2BSuiteTest extends AbstractRectorTestCase
+{
+    /**
+     * @dataProvider provideData()
+     */
+    public function test(string $file): void
+    {
+        $this->doTestFile($file);
+    }
+
+    /**
+     * @return Iterator<string>
+     */
+    public function provideData(): Iterator
+    {
+        return self::yieldFilesFromDirectory(__DIR__ . '/Fixture');
+    }
+
+    public function provideConfigFilePath(): string
+    {
+        return __DIR__ . '/config/configured_rule.php';
+    }
+}

tests/Set/SyliusPlus/B2BSuite/Fixture/class_extending_core_address_model.php.inc

@@ -0,0 +1,24 @@
+<?php
+
+namespace Sylius\SyliusRector\Tests\Rector\Class_\AddInterfaceToClassExtendingType\Fixture;
+
+use Sylius\Component\Core\Model\Address as BaseAddress;
+
+class Address extends BaseAddress
+{
+}
+
+?>
+-----
+<?php
+
+namespace Sylius\SyliusRector\Tests\Rector\Class_\AddInterfaceToClassExtendingType\Fixture;
+
+use Sylius\Component\Core\Model\Address as BaseAddress;
+
+class Address extends BaseAddress implements \Sylius\B2BKit\Entity\AddressInterface
+{
+    use \Sylius\B2BKit\Entity\AddressAwareTrait;
+}
+
+?>

tests/Set/SyliusPlus/B2BSuite/Fixture/class_extending_core_customer_group_model.php.inc

@@ -0,0 +1,24 @@
+<?php
+
+namespace Sylius\SyliusRector\Tests\Rector\Class_\AddInterfaceToClassExtendingType\Fixture;
+
+use Sylius\Component\Customer\Model\CustomerGroup as BaseCustomerGroup;
+
+class CustomerGroup extends BaseCustomerGroup
+{
+}
+
+?>
+-----
+<?php
+
+namespace Sylius\SyliusRector\Tests\Rector\Class_\AddInterfaceToClassExtendingType\Fixture;
+
+use Sylius\Component\Customer\Model\CustomerGroup as BaseCustomerGroup;
+
+class CustomerGroup extends BaseCustomerGroup implements \Sylius\B2BKit\Entity\CustomerGroupInterface
+{
+    use \Sylius\B2BKit\Entity\CustomerGroupAwareTrait;
+}
+
+?>

tests/Set/SyliusPlus/B2BSuite/Fixture/class_extending_core_customer_model.php.inc

@@ -0,0 +1,24 @@
+<?php
+
+namespace Sylius\SyliusRector\Tests\Rector\Class_\AddInterfaceToClassExtendingType\Fixture;
+
+use Sylius\Component\Core\Model\Customer as BaseCustomer;
+
+class Customer extends BaseCustomer
+{
+}
+
+?>
+-----
+<?php
+
+namespace Sylius\SyliusRector\Tests\Rector\Class_\AddInterfaceToClassExtendingType\Fixture;
+
+use Sylius\Component\Core\Model\Customer as BaseCustomer;
+
+class Customer extends BaseCustomer implements \Sylius\B2BKit\Entity\CustomerInterface
+{
+    use \Sylius\B2BKit\Entity\CustomerAwareTrait;
+}
+
+?>

tests/Set/SyliusPlus/B2BSuite/Fixture/class_extending_core_order_model.php.inc

@@ -0,0 +1,24 @@
+<?php
+
+namespace Sylius\SyliusRector\Tests\Rector\Class_\AddInterfaceToClassExtendingType\Fixture;
+
+use Sylius\Component\Core\Model\Order as BaseOrder;
+
+class Order extends BaseOrder
+{
+}
+
+?>
+-----
+<?php
+
+namespace Sylius\SyliusRector\Tests\Rector\Class_\AddInterfaceToClassExtendingType\Fixture;
+
+use Sylius\Component\Core\Model\Order as BaseOrder;
+
+class Order extends BaseOrder implements \Sylius\B2BKit\Entity\OrderInterface
+{
+    use \Sylius\B2BKit\Entity\OrderAwareTrait;
+}
+
+?>

tests/Set/SyliusPlus/B2BSuite/Fixture/class_extending_core_shop_user_model.php.inc

@@ -0,0 +1,24 @@
+<?php
+
+namespace Sylius\SyliusRector\Tests\Rector\Class_\AddInterfaceToClassExtendingType\Fixture;
+
+use Sylius\Component\Core\Model\ShopUser as BaseShopUser;
+
+class ShopUser extends BaseShopUser
+{
+}
+
+?>
+-----
+<?php
+
+namespace Sylius\SyliusRector\Tests\Rector\Class_\AddInterfaceToClassExtendingType\Fixture;
+
+use Sylius\Component\Core\Model\ShopUser as BaseShopUser;
+
+class ShopUser extends BaseShopUser implements \Sylius\B2BKit\Entity\ShopUserInterface
+{
+    use \Sylius\B2BKit\Entity\ShopUserAwareTrait;
+}
+
+?>

tests/Set/SyliusPlus/B2BSuite/config/configured_rule.php

@@ -0,0 +1,11 @@
+<?php
+
+declare(strict_types=1);
+
+use Rector\Config\RectorConfig;
+use Sylius\SyliusRector\Set\SyliusPlus;
+
+return static function (RectorConfig $rectorConfig): void {
+    $rectorConfig->import(__DIR__ . '/../../../../../config/config.php');
+    $rectorConfig->sets([SyliusPlus::B2B_SUITE]);
+};