ClassType: added getMethod() and getProperty()

Author: dg

src/PhpGenerator/ClassType.php

@@ -446,6 +446,18 @@ public function getProperties()
 	}
 
 
+	/**
+	 * @return Property
+	 */
+	public function getProperty($name)
+	{
+		if (!isset($this->properties[$name])) {
+			throw new Nette\InvalidArgumentException("Property '$name' not found.");
+		}
+		return $this->properties[$name];
+	}
+
+
 	/**
 	 * @param  string  without $
 	 * @param  mixed
@@ -483,6 +495,18 @@ public function getMethods()
 	}
 
 
+	/**
+	 * @return Method
+	 */
+	public function getMethod($name)
+	{
+		if (!isset($this->methods[$name])) {
+			throw new Nette\InvalidArgumentException("Method '$name' not found.");
+		}
+		return $this->methods[$name];
+	}
+
+
 	/**
 	 * @param  string
 	 * @return Method

tests/PhpGenerator/ClassType.phpt

@@ -34,15 +34,19 @@ $class->addProperty('handle')
 $class->addProperty('order')
 	->setValue(new PhpLiteral('RecursiveIteratorIterator::SELF_FIRST'));
 
-$class->addProperty('sections', array('first' => TRUE))
+$p = $class->addProperty('sections', array('first' => TRUE))
 	->setStatic(TRUE);
 
-$class->addMethod('getHandle')
+Assert::same($p, $class->getProperty('sections'));
+
+$m = $class->addMethod('getHandle')
 	->addDocument('Returns file handle.')
 	->addDocument('@return resource')
 	->setFinal(TRUE)
 	->setBody('return $this->?;', array('handle'));
 
+Assert::same($m, $class->getMethod('getHandle'));
+
 $class->addMethod('getSections')
 	->setStatic(TRUE)
 	->setVisibility('protected')