WpAjaxInitializer.

Author: ProklUng

DependencyInjection/WpSymfonyRouterExtension.php

@@ -38,6 +38,7 @@ public function load(array $configs, ContainerBuilder $container) : void
         );
 
         $loader->load('services.yaml');
+        $loader->load('native.yaml');
 
         // Если не задействован FacadeBundle, то удалить фасады.
         if (!class_exists(AbstractFacade::class)) {
@@ -52,6 +53,8 @@ public function load(array $configs, ContainerBuilder $container) : void
             $container->setParameter('controller.annotations.path', []);
         }
 
+        $this->processNativeWpRoutes($container);
+
         $this->registerRouterConfiguration(
             $config,
             $container
@@ -160,4 +163,31 @@ private function registerRouterConfiguration(
                 ]);
         }
     }
+
+    /**
+     * Обработка "нативных" роутов (wp-admin/admin_ajax.php).
+     *
+     * @param ContainerBuilder $container
+     *
+     * @return void
+     *
+     * @since 13.06.2021
+     */
+    private function processNativeWpRoutes(ContainerBuilder $container): void
+    {
+        // Путь к "нативным" роутам WP.
+        if (!$container->hasParameter('yaml.native.routes.file')) {
+            $container->setParameter('yaml.native.routes.file', '/app/wp_routes.yaml');
+        }
+
+        $nativeRoutesPath = $container->getParameter('yaml.native.routes.file');
+        $root = $container->getParameter('kernel.project_dir');
+
+        // Если файл с роутами не существует, то выпилить из контейнера
+        // соответствующие сервисы.
+        if (!file_exists($root . $nativeRoutesPath)) {
+            $container->removeDefinition('wp_ajax.loader');
+            $container->removeDefinition('wp_ajax.initializer');
+        }
+    }
 }

Resources/config/native.yaml

@@ -0,0 +1,21 @@
+##########
+# Роутер
+##########
+
+services:
+  # конфигурация по умолчанию в *этом* файле
+  _defaults:
+    autowire: true
+    autoconfigure: true
+    public: true
+
+  wp_ajax.loader:
+    public: false
+    class: Symfony\Component\Routing\RouteCollection
+    factory: ['@routing.loader', 'load']
+    arguments: ['%kernel.project_dir%/%yaml.native.routes.file%']
+
+  wp_ajax.initializer:
+    class: Prokl\WpSymfonyRouterBundle\Services\NativeAjax\WpAjaxInitializer
+    arguments: ['@wp_ajax.loader', '@service_container']
+    tags: ['service.bootstrap']

Services/NativeAjax/AbstractWPAjaxController.php

@@ -0,0 +1,31 @@
+<?php
+
+namespace Prokl\WpSymfonyRouterBundle\Services\NativeAjax;
+
+use Symfony\Bundle\FrameworkBundle\Controller\AbstractController;
+use Symfony\Component\HttpFoundation\Request;
+
+/**
+ * Class AbstractWPAjaxController
+ * @package Prokl\WpSymfonyRouterBundle\Services\NativeAjax
+ */
+class AbstractWPAjaxController extends AbstractController
+{
+    /**
+     * @var Request $request
+     */
+    protected $request;
+
+    /**
+     * AbstractWPAjaxController constructor.
+     *
+     * @param Request|null $request
+     */
+    public function __construct(?Request $request = null)
+    {
+        $this->request = $request;
+        if ($request === null) {
+            $this->request = Request::createFromGlobals();
+        }
+    }
+}

Services/NativeAjax/WpAjaxInitializer.php

@@ -0,0 +1,110 @@
+<?php
+
+namespace Prokl\WpSymfonyRouterBundle\Services\NativeAjax;
+
+use ReflectionException;
+use ReflectionMethod;
+use RuntimeException;
+use Symfony\Component\DependencyInjection\ContainerInterface;
+use Symfony\Component\Routing\RouteCollection;
+
+/**
+ * Class Prokl\WpSymfonyRouterBundle\Services\NativeAjax
+ * @package Fedy\Services\Wordpress
+ *
+ * @since 12.06.2021
+ */
+class WpAjaxInitializer
+{
+    /**
+     * @var RouteCollection $routes Роуты.
+     */
+    private $routes;
+
+    /**
+     * @var ContainerInterface $container Контейнер.
+     */
+    private $container;
+
+    /**
+     * WpAjaxInitializer constructor.
+     *
+     * @param RouteCollection    $routes    Роуты.
+     * @param ContainerInterface $container Контейнер.
+     */
+    public function __construct(RouteCollection $routes, ContainerInterface $container)
+    {
+        $this->routes = $routes;
+        $this->container = $container;
+
+        $this->init();
+    }
+
+    /**
+     * Инициализация.
+     *
+     * @return void
+     */
+    private function init() : void
+    {
+        $routes = $this->routes->all();
+        foreach ($routes as $action => $route) {
+            $defaults = $route->getDefaults();
+            // Публичный роут или нет
+            $public = $defaults['_public'];
+
+            $controller = $this->parseController($defaults['_controller']);
+
+            add_action("wp_ajax_{$action}", $controller);
+            if ($public) {
+                add_action("wp_ajax_nopriv_{$action}", $controller);
+            }
+        }
+    }
+
+    /**
+     * @param array|string|object $controller
+     *
+     * @return array|false|object|string|string[]|null
+     * @throws RuntimeException Когда не получилось распарсить контроллер.
+     */
+    private function parseController($controller)
+    {
+        if (is_string($controller)) {
+            if (strpos($controller, '::') !== false) {
+                $controller = explode('::', $controller, 2);
+            } else {
+                // Invoked controller.
+                try {
+                    new ReflectionMethod($controller, '__invoke');
+                    $controller = [$controller, '__invoke'];
+                } catch (ReflectionException $e) {
+                    return [];
+                }
+            }
+        }
+
+        if (is_array($controller)) {
+            if (is_string($controller[0]) && !$this->container->has($controller[0])) {
+                throw new RuntimeException(
+                    sprintf(
+                        'Controller %s not found in container. Forgot mark him as service?',
+                        $controller[0]
+                    ),
+                );
+            }
+
+            $controller[0] = $this->container->get($controller[0]);
+
+            return $controller;
+        }
+
+        if (is_object($controller)) {
+            return [$controller];
+        }
+
+        throw new RuntimeException(
+          'Error parsing controller'
+        );
+    }
+}

Tests/Cases/SampleAjaxControllerTest.php

@@ -0,0 +1,25 @@
+<?php
+
+namespace Prokl\WpSymfonyRouterBundle\Tests\Cases;
+
+use Prokl\WordpressCi\Base\WordpressableAjaxTestCase;
+
+/**
+ * Class SampleAjaxControllerTest
+ * @package Fedy\Services\Wordpress
+ */
+class SampleAjaxControllerTest extends WordpressableAjaxTestCase
+{
+    protected function setUp() : void
+    {
+        parent::setup();
+        wp_set_current_user( 1 );
+    }
+
+    public function testAction() : void
+    {
+        $this->_handleAjax('examples_wp');
+        $result = $this->_last_response;
+        $this->assertSame('OK', $result);
+    }
+}