Phraseanet/lib/Alchemy/Phrasea/Core/Provider/WebProfilerServiceProvider.php

<?php

namespace Alchemy\Phrasea\Core\Provider;

use Alchemy\Phrasea\Core\Event\Subscriber\CacheStatisticsSubscriber;
use Alchemy\Phrasea\Core\Profiler\CacheDataCollector;
use Doctrine\DBAL\Connection;
use Doctrine\DBAL\Logging\DebugStack;
use Doctrine\DBAL\Logging\LoggerChain;
use Silex\Application;
use Silex\ServiceProviderInterface;
use Sorien\DataCollector\DoctrineDataCollector;
use Sorien\Logger\DbalLogger;
use Symfony\Bundle\FrameworkBundle\DataCollector\AjaxDataCollector;
use Symfony\Bundle\WebProfilerBundle\EventListener\WebDebugToolbarListener;

class WebProfilerServiceProvider implements ServiceProviderInterface
{

    /**
     * Registers services on the given app.
     *
     * This method should only be used to configure services and parameters.
     * It should not get services.
     */
    public function register(Application $app)
    {
        // Required because the Silex provider is not up to date with Symfony Debug Toolbar
        $app['web_profiler.toolbar.listener'] = $app->share(
            $app->extend('web_profiler.toolbar.listener', function () use ($app) {
                return new WebDebugToolbarListener(
                    $app['twig'],
                    $app['web_profiler.debug_toolbar.intercept_redirects'],
                    2,
                    $app['web_profiler.debug_toolbar.position'],
                    $app['url_generator']
                );
            })
        );

        if (class_exists('Symfony\Bundle\FrameworkBundle\DataCollector\AjaxDataCollector')) {
            $app['data_collector.templates'] = $app->share($app->extend('data_collector.templates', function (array $templates) {
                $templates[] = array('ajax', '@WebProfiler/Collector/ajax.html.twig');

                return $templates;
            }));

            $app['data_collector.cache_subscriber'] = $app->share(function ($app) {
                return new CacheStatisticsSubscriber($app['cache']);
            });

            $app['data_collectors'] = $app->share($app->extend('data_collectors', function ($collectors) use ($app) {
                $collectors['ajax'] = function () {
                    return new AjaxDataCollector();
                };

                return $collectors;
            }));
        }

        $app['data_collectors.doctrine'] = $app->share(function ($app) {
            /** @var Connection $db */
            $db = $app['db'];
            return new DoctrineDataCollector($db);
        });

        $app['dbal.provider'] = $app->extend('dbal.provider', function ($provider, $app) {
            return function (array $parameters) use ($provider, $app) {
                $connection = $provider($parameters);

                if (! $connection->getConfiguration()->getSQLLogger() && $app->offsetExists('data_collectors.doctrine')) {
                    /** @var DoctrineDataCollector $collector */
                    $collector = $app['data_collectors.doctrine'];
                    $logger = new DebugStack();

                    $connection->getConfiguration()->setSQLLogger($logger);
                    $collector->addLogger($logger);
                }

                return $connection;
            };
        });

        $dataCollectors = $app['data_collectors'];
        $dataCollectors['db'] = $app->share(function ($app) {
            /** @var Connection $db */
            $db = $app['db'];
            /** @var DoctrineDataCollector $collector */
            $collector = $app['data_collectors.doctrine'];

            $loggerChain = new LoggerChain();
            $logger = new DebugStack();

            $loggerChain->addLogger($logger);
            $loggerChain->addLogger(new DbalLogger($app['logger'], $app['stopwatch']));

            $db->getConfiguration()->setSQLLogger($loggerChain);

            $collector->addLogger($logger);

            return $collector;
        });

        $dataCollectors['cache'] = $app->share(function ($app) {
            return new CacheDataCollector($app['data_collector.cache_subscriber']);
        });

        $app['data_collectors'] = $dataCollectors;

        $dataCollectorTemplates = $app['data_collector.templates'];
        $dataCollectorTemplates[] = array('db', '@DoctrineBundle/Collector/db.html.twig');
        $dataCollectorTemplates[] = array('cache', '@PhraseaProfiler/cache.html.twig');
        $app['data_collector.templates'] = $dataCollectorTemplates;

        $app['twig.loader.filesystem'] = $app->share($app->extend('twig.loader.filesystem', function ($loader, $app) {
            $loader->addPath(
                $app['root.path'] . '/vendor/sorien/silex-dbal-profiler/src/Sorien/Resources/views',
                'DoctrineBundle'
            );
            $loader->addPath($app['root.path'] . '/templates/web/debug', 'PhraseaProfiler');
            return $loader;
        }));
    }

    /**
     * Bootstraps the application.
     *
     * This method is called after all services are registered
     * and should be used for "dynamic" configuration (whenever
     * a service must be requested).
     */
    public function boot(Application $app)
    {
        $app['dispatcher']->addSubscriber($app['data_collector.cache_subscriber']);
    }
}