SinSquare/silex-doctrine-orm-provider

<?php

namespace SinSquare\Doctrine;

use Doctrine\DBAL\Logging\DebugStack;
use Psr\Log\LoggerInterface;
use Symfony\Component\Stopwatch\Stopwatch;

class DbalLogger extends DebugStack
{
    const MAX_STRING_LENGTH = 32;
    const BINARY_DATA_VALUE = '(binary value)';

    protected $logger;
    protected $stopwatch;

    public function __construct(LoggerInterface $logger = null, Stopwatch $stopwatch = null)
    {
        $this->logger = $logger;
        $this->stopwatch = $stopwatch;
    }

    public function startQuery($sql, array $params = null, array $types = null)
    {
        parent::startQuery($sql, $params, $types);

        if (null !== $this->stopwatch) {
            $this->stopwatch->start('doctrine', 'doctrine');
        }

        if (is_array($params)) {
            $params = $this->fixParams($params);
        }

        if (null !== $this->logger) {
            $this->log($sql, null === $params ? array() : $params);
        }
    }

    protected function fixParams(array $params)
    {
        foreach ($params as $index => $param) {
            $params[$index] = $this->fixParam($param);
        }

        return $params;
    }

    protected function fixParam($param)
    {
        if (!is_string($param)) {
            return $param;
        }

        // non utf-8 strings break json encoding
        if (!preg_match('#[\p{L}\p{N} ]#u', $param)) {
            return self::BINARY_DATA_VALUE;
        }

        // detect if the too long string must be shorten
        if (function_exists('mb_detect_encoding') && false !== $encoding = mb_detect_encoding($param)) {
            if (self::MAX_STRING_LENGTH < mb_strlen($param, $encoding)) {
                return mb_substr($param, 0, self::MAX_STRING_LENGTH - 6, $encoding).' [...]';
            }
        } else {
            if (self::MAX_STRING_LENGTH < strlen($param)) {
                return substr($param, 0, self::MAX_STRING_LENGTH - 6).' [...]';
            }
        }

        return $param;
    }

    public function stopQuery()
    {
        parent::stopQuery();

        if (null !== $this->stopwatch) {
            $this->stopwatch->stop('doctrine');
        }
    }

    protected function log($message, array $params)
    {
        $this->logger->debug($message, $params);
    }
}