bkdotcom/PHPDebugConsole

<?php

/**
 * This file is part of PHPDebugConsole
 *
 * @package   PHPDebugConsole
 * @author    Brad Kent <bkfake-github@yahoo.com>
 * @license   http://opensource.org/licenses/MIT MIT
 * @copyright 2014-2024 Brad Kent
 * @since     3.3
 */

 namespace bdk\Debug\Collector;

 use bdk\Debug;
 use bdk\Debug\Collector\Doctrine\Driver;
 use Doctrine\Bundle\DoctrineBundle\Attribute\AsMiddleware;
 use Doctrine\DBAL\Driver as DriverInterface;
 use Doctrine\DBAL\Driver\Middleware as MiddlewareInterface;

 /**
  * Doctrine 3.x requires php 7.3+
  * Doctrine 3.2 introduces middleware for logging
  * Doctrine 3.3 adds AbstractXxxxxMiddleware classes (which we use)
  * Doctrine 3.4 deprecates SqlLogger
  */
#[AsMiddleware(priority: -10)]
class DoctrineMiddleware implements MiddlewareInterface
{
    /** @var Debug */
    protected $debug;

    /** @var string */
    protected $icon = 'fa fa-database';

    /**
     * Constructor
     *
     * @param Debug|null $debug Debug instance
     */
    public function __construct(?Debug $debug = null)
    {
        if (!$debug) {
            $debug = Debug::getChannel('Doctrine', array('channelIcon' => $this->icon));
        } elseif ($debug === $debug->rootInstance) {
            $debug = $debug->getChannel('Doctrine', array('channelIcon' => $this->icon));
        }
        $this->debug = $debug;
    }

    /**
     * {@inheritDoc}
     */
    public function wrap(DriverInterface $driver): DriverInterface
    {
        return new Driver($driver, $this->debug);
    }
}