Container: explicit support for extension methods

nette · Apr 5, 2016 · 9e359e6 · 9e359e6
1 parent fa84d63
commit 9e359e6
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/Forms/Container.php b/src/Forms/Container.php
@@ -438,6 +438,27 @@ public function addContainer($name)
 	}
 
 
+	/********************* extension methods ****************d*g**/
+
+
+	public function __call($name, $args)
+	{
+		if ($callback = Nette\Utils\ObjectMixin::getExtensionMethod(__CLASS__, $name)) {
+			return Nette\Utils\Callback::invoke($callback, $this, ...$args);
+		}
+		return parent::__call($name, $args);
+	}
+
+
+	public static function extensionMethod($name, $callback = NULL)
+	{
+		if (strpos($name, '::') !== FALSE) { // back compatibility
+			list(, $name) = explode('::', $name);
+		}
+		Nette\Utils\ObjectMixin::setExtensionMethod(__CLASS__, $name, $callback);
+	}
+
+
 	/********************* interface \ArrayAccess ****************d*g**/