Fixed wrong typehint and method usage [resolves #12] (thx @xificurk)

contributte · Jul 22, 2014 · d87824f · d87824f
1 parent 0928d5b
commit d87824f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/Gopay/Extension.php b/Gopay/Extension.php
@@ -11,6 +11,7 @@
 namespace Markette\Gopay;
 
 use Nette\DI\CompilerExtension;
+use Nette\PhpGenerator;
 use Nette\Reflection\ClassType;
 
 
@@ -64,7 +65,7 @@ public function loadConfiguration()
 
 
 
-	public function afterCompile(ClassType $class)
+	public function afterCompile(PhpGenerator\ClassType $class)
 	{
 		$initialize = $class->methods['initialize'];
 		$initialize->addBody('Markette\Gopay\Service::registerAddPaymentButtons($this->getService(?));', array(

diff --git a/Gopay/Service.php b/Gopay/Service.php
@@ -547,10 +547,10 @@ private function createSignature($paymentSessionId)
 	 */
 	public static function registerAddPaymentButtons(Service $service)
 	{
-		Nette\Forms\Container::registerMethod('addPaymentButtons', function ($container, $callbacks) use ($service) {
+		Nette\Forms\Container::extensionMethod('addPaymentButtons', function ($container, $callbacks) use ($service) {
 			$service->bindPaymentButtons($container, $callbacks);
 		});
-		Nette\Forms\Container::registerMethod('addPaymentButton', function ($container, $channel) use ($service) {
+		Nette\Forms\Container::extensionMethod('addPaymentButton', function ($container, $channel) use ($service) {
 			return $service->bindPaymentButton($channel, $container);
 		});
 	}