domokits/local/modules/TheliaSmarty/Template/Plugins/Render.php

<?php
/*************************************************************************************/
/* This file is part of the Thelia package.                                          */
/*                                                                                   */
/* Copyright (c) OpenStudio                                                          */
/* email : dev@thelia.net                                                            */
/* web : http://www.thelia.net                                                       */
/*                                                                                   */
/* For the full copyright and license information, please view the LICENSE.txt       */
/* file that was distributed with this source code.                                  */
/*************************************************************************************/

namespace TheliaSmarty\Template\Plugins;

use Symfony\Component\DependencyInjection\Container;
use Symfony\Component\HttpFoundation\ParameterBag;
use Symfony\Component\HttpFoundation\RequestStack;
use Symfony\Component\HttpFoundation\Response;
use Thelia\Core\Controller\ControllerResolver;
use TheliaSmarty\Template\AbstractSmartyPlugin;
use TheliaSmarty\Template\Exception\SmartyPluginException;
use TheliaSmarty\Template\SmartyPluginDescriptor;

/**
 * Class Render
 * @package TheliaSmarty\Template\Plugins
 * @author Benjamin Perche <bperche@openstudio.fr>
 */
class Render extends AbstractSmartyPlugin
{
    /** @var ControllerResolver */
    protected $controllerResolver;

    /** @var RequestStack */
    protected $requestStack;

    /** @var Container */
    protected $container;

    /**
     * @param ControllerResolver $controllerResolver
     * @param RequestStack       $requestStack
     * @param Container          $container
     */
    public function __construct(ControllerResolver $controllerResolver, RequestStack $requestStack, Container $container)
    {
        $this->controllerResolver = $controllerResolver;
        $this->requestStack = $requestStack;
        $this->container = $container;
    }

    /**
     * @param $params
     * @return mixed|string
     * @throws SmartyPluginException
     */
    public function processRender($params)
    {
        if (null === $params["action"]) {
            throw new SmartyPluginException(
                "You must declare the 'action' parameter in the 'render' smarty function"
            );
        }

        $request = $this->prepareRequest($params);

        $this->requestStack->push($request);

        $controller = $this->controllerResolver->getController($request);
        $controllerParameters = $this->controllerResolver->getArguments($request, $controller);

        $response = call_user_func_array($controller, $controllerParameters);

        $this->requestStack->pop();

        if ($response instanceof Response) {
            return $response->getContent();
        }

        return $response;
    }

    protected function prepareRequest(array $params)
    {
        // Get action
        $action = $this->popParameter($params, "action");

        // Then get and filter query, request and method
        $query = $this->popParameter($params, "query");
        $query = $this->filterArrayStrParam($query);
        $request = $this->popParameter($params, "request");
        $request = $this->filterArrayStrParam($request);
        $method = strtoupper($this->popParameter($params, "method", "GET"));

        // Then build the request
        $requestObject = clone $this->requestStack->getCurrentRequest();
        $requestObject->query = new ParameterBag($query);
        $requestObject->request = new ParameterBag($request);
        $requestObject->attributes = new ParameterBag(["_controller" => $action]);

        // Apply the method
        if (!empty($request) && "GET" === $method) {
            $requestObject->setMethod("POST");
        } else {
            $requestObject->setMethod($method);
        }

        // Then all the attribute parameters
        foreach ($params as $key => $attribute) {
            $requestObject->attributes->set($key, $attribute);
        }

        return $requestObject;
    }

    /**
     * @param $param
     * @return array
     *
     * If $param is an array, return it.
     * Else parser it to translate a=b&c=d&e[]=f&g[h]=i to
     * ["a"=>"b","c"=>"d","e"=>["f"],"g"=>["h"=>"i"]
     */
    protected function filterArrayStrParam($param)
    {
        if (is_array($param)) {
            return $param;
        }

        parse_str($param, $param);

        if (false === $param) {
            return [];
        }

        return $param;
    }

    /**
     * @param  array $params
     * @param $name
     * @param  null  $default
     * @return mixed
     *
     * Get a parameter then unset it
     */
    protected function popParameter(array $params, $name, $default = null)
    {
        $param = $this->getParam($params, $name, $default);

        if (array_key_exists($name, $params)) {
            unset($params[$name]);
        }

        return $param;
    }

    /**
     * @return SmartyPluginDescriptor[] an array of SmartyPluginDescriptor
     */
    public function getPluginDescriptors()
    {
        return array(
            new SmartyPluginDescriptor('function', 'render', $this, 'processRender'),
        );
    }
}