From 6f11039482bf222d4f06b038c4808ed0acb23c50 Mon Sep 17 00:00:00 2001
From: AntikCz <antik494@gmail.com>
Date: Mon, 22 Feb 2016 14:29:58 +0100
Subject: [PATCH] Registration of bar is moved to afterCompile

---
 Cronner/DI/CronnerExtension.php | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/Cronner/DI/CronnerExtension.php b/Cronner/DI/CronnerExtension.php
index 1f246fe..32396b0 100644
--- a/Cronner/DI/CronnerExtension.php
+++ b/Cronner/DI/CronnerExtension.php
@@ -6,6 +6,7 @@
 use Nette\DI\Compiler;
 use Nette\DI\CompilerExtension;
 use Nette\DI\Statement;
+use Nette\PhpGenerator\ClassType;
 use Nette\Utils\Json;
 use Nette\Utils\Validators;
 
@@ -109,10 +110,14 @@ public function beforeCompile()
 		foreach (array_keys($builder->findByTag(self::TASKS_TAG)) as $serviceName) {
 			$runner->addSetup('addTasks', array('@' . $serviceName));
 		}
+	}
+
+	public function afterCompile(ClassType $class) {
+		$builder = $this->getContainerBuilder();
+		$init = $class->getMethod('initialize');
 
 		if ($builder->hasDefinition($this->prefix('bar'))) {
-			$builder->getDefinition('tracy.bar')
-				->addSetup('addPanel', array($this->prefix('@bar')));
+			$init->addBody('$this->getByType(?)->addPanel($this->getService(?));', array('Tracy\Bar', $this->prefix('bar')));
 		}
 	}
 
@@ -126,4 +131,4 @@ public static function register(Configurator $configurator)
 		};
 	}
 
-}
+}
\ No newline at end of file