lekoala / silverstripe-debugbar

 */

class ControllerExtension extends Extension

{

    public function onBeforeInit()

    {

        $class = get_class($this->owner);

        DebugBar::withDebugBar(function (\DebugBar\DebugBar $debugbar) use ($class) {

            // We must set the current controller when it's available and before it's pushed out of stack

            $debugbar->getCollector('silverstripe')->setController(Controller::curr());

            /** @var $timeData DebugBar\DataCollector\TimeDataCollector */

            $timeData = $debugbar['time'];

            if (!$timeData) {

                return;

            }

            if ($timeData->hasStartedMeasure('pre_request')) {

                $timeData->stopMeasure("pre_request");

            }

            $timeData->startMeasure("init", "$class init");

        });

    }

    public function onAfterInit()

    {

     * @param string $action

     * @param mixed $result (only in v4.0)

     */

    public function afterCallActionHandler($request, $action, $result)

    {

        self::clearBuffer();

        $class = get_class($this->owner);

        DebugBar::withDebugBar(function (\DebugBar\DebugBar $debugbar) use ($class, $action) {

            /* @var $timeData DebugBar\DataCollector\TimeDataCollector */

            $timeData = $debugbar['time'];

            if (!$timeData) {

                return;

            }

            if ($timeData->hasStartedMeasure("action")) {

                $timeData->stopMeasure("action");

            }

            $timeData->startMeasure(

                "after_action",

                "$class after action $action"

            );

        });

    }

    protected static function clearBuffer()

    {