NelmioApiDocBundle/Extractor/RoutingExtractor.php

<?php

/*
 * This file is part of the ApiDocBundle package.
 *
 * (c) EXSyst
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace EXSyst\Bundle\ApiDocBundle\Extractor;

use Doctrine\Common\Util\ClassUtils;
use EXSyst\Bundle\ApiDocBundle\Extractor\Routing\RouteExtractorInterface;
use gossi\swagger\Swagger;
use Symfony\Bundle\FrameworkBundle\Controller\ControllerNameParser;
use Symfony\Component\Routing\Route;
use Symfony\Component\Routing\RouterInterface;

class RoutingExtractor implements ExtractorInterface
{
    private $routeExtractors;

    /**
     * @param RouterInterface           $router
     * @param ControllerNameParser      $controllerNameParser
     * @param RouteExtractorInterface[] $extractors
     */
    public function __construct(RouterInterface $router, ControllerNameParser $controllerNameParser, array $routeExtractors)
    {
        $this->router = $router;
        $this->controllerNameParser = $controllerNameParser;
        $this->routeExtractors = $routeExtractors;
    }

    public function extractIn(Swagger $swagger)
    {
        if (0 === count($this->routeExtractors)) {
            return;
        }

        foreach ($this->getRoutes() as $route) {
            // if able to resolve the controller
            if ($method = $this->getReflectionMethod($route->getDefault('_controller'))) {
                // Extract as many informations as possible about this route
                foreach ($this->routeExtractors as $extractor) {
                    $extractor->extractIn($swagger, $route, $method);
                }
            }
        }
    }

    /**
     * Return a list of route to inspect.
     *
     * @return Route[] An array of routes
     */
    private function getRoutes()
    {
        return $this->router->getRouteCollection()->all();
    }

    /**
     * Returns the ReflectionMethod for the given controller string.
     *
     * @param string $controller
     *
     * @return \ReflectionMethod|null
     */
    private function getReflectionMethod($controller)
    {
        if (false === strpos($controller, '::') && 2 === substr_count($controller, ':')) {
            $controller = $this->controllerNameParser->parse($controller);
        }

        if (preg_match('#(.+)::([\w]+)#', $controller, $matches)) {
            $class = $matches[1];
            $method = $matches[2];
        } elseif (class_exists($controller)) {
            $class = $controller;
            $method = '__invoke';
        } else {
            if (preg_match('#(.+):([\w]+)#', $controller, $matches)) {
                $controller = $matches[1];
                $method = $matches[2];
            }

            if ($this->container->has($controller)) {
                if (class_exists(ClassUtils::class)) {
                    $class = ClassUtils::getRealClass(get_class($this->container->get($controller)));
                }

                if (!isset($method) && method_exists($class, '__invoke')) {
                    $method = '__invoke';
                }
            }
        }

        if (isset($class) && isset($method)) {
            try {
                return new \ReflectionMethod($class, $method);
            } catch (\ReflectionException $e) {
            }
        }
    }
}