1: <?php
2:
3: 4: 5: 6:
7:
8: namespace Tracy\Bridges\Nette;
9:
10: use Nette;
11:
12:
13: 14: 15:
16: class TracyExtension extends Nette\DI\CompilerExtension
17: {
18: public $defaults = [
19: 'email' => NULL,
20: 'fromEmail' => NULL,
21: 'logSeverity' => NULL,
22: 'editor' => NULL,
23: 'browser' => NULL,
24: 'errorTemplate' => NULL,
25: 'strictMode' => NULL,
26: 'showBar' => NULL,
27: 'maxLen' => NULL,
28: 'maxDepth' => NULL,
29: 'showLocation' => NULL,
30: 'scream' => NULL,
31: 'bar' => [],
32: 'blueScreen' => [],
33: ];
34:
35:
36: private $debugMode;
37:
38:
39: private $cliMode;
40:
41:
42: public function __construct($debugMode = FALSE, $cliMode = FALSE)
43: {
44: $this->debugMode = $debugMode;
45: $this->cliMode = $cliMode;
46: }
47:
48:
49: public function loadConfiguration()
50: {
51: $this->validateConfig($this->defaults);
52: $builder = $this->getContainerBuilder();
53:
54: $builder->addDefinition($this->prefix('logger'))
55: ->setClass('Tracy\ILogger')
56: ->setFactory('Tracy\Debugger::getLogger');
57:
58: $builder->addDefinition($this->prefix('blueScreen'))
59: ->setFactory('Tracy\Debugger::getBlueScreen');
60:
61: $builder->addDefinition($this->prefix('bar'))
62: ->setFactory('Tracy\Debugger::getBar');
63: }
64:
65:
66: public function afterCompile(Nette\PhpGenerator\ClassType $class)
67: {
68: $initialize = $class->getMethod('initialize');
69: $builder = $this->getContainerBuilder();
70: $class = method_exists('Nette\DI\Helpers', 'filterArguments') ? 'Nette\DI\Helpers' : 'Nette\DI\Compiler';
71:
72: $options = $this->config;
73: unset($options['bar'], $options['blueScreen']);
74: if (isset($options['logSeverity'])) {
75: $res = 0;
76: foreach ((array) $options['logSeverity'] as $level) {
77: $res |= is_int($level) ? $level : constant($level);
78: }
79: $options['logSeverity'] = $res;
80: }
81: foreach ($options as $key => $value) {
82: if ($value !== NULL) {
83: $key = ($key === 'fromEmail' ? 'getLogger()->' : '$') . $key;
84: $initialize->addBody($builder->formatPhp(
85: 'Tracy\Debugger::' . $key . ' = ?;',
86: $class::filterArguments([$value])
87: ));
88: }
89: }
90:
91: $logger = $builder->getDefinition($this->prefix('logger'));
92: if ($logger->getFactory()->getEntity() !== 'Tracy\Debugger::getLogger') {
93: $initialize->addBody($builder->formatPhp('Tracy\Debugger::setLogger(?);', [$logger]));
94: }
95:
96: if ($this->debugMode) {
97: foreach ((array) $this->config['bar'] as $item) {
98: $initialize->addBody($builder->formatPhp(
99: '$this->getService(?)->addPanel(?);',
100: $class::filterArguments([
101: $this->prefix('bar'),
102: is_string($item) ? new Nette\DI\Statement($item) : $item,
103: ])
104: ));
105: }
106:
107: if (!$this->cliMode) {
108: $initialize->addBody('if ($tmp = $this->getByType("Nette\Http\Session", FALSE)) { $tmp->start(); Tracy\Debugger::dispatch(); };');
109: }
110: }
111:
112: foreach ((array) $this->config['blueScreen'] as $item) {
113: $initialize->addBody($builder->formatPhp(
114: '$this->getService(?)->addPanel(?);',
115: $class::filterArguments([$this->prefix('blueScreen'), $item])
116: ));
117: }
118: }
119:
120: }
121: