Check entity

contributte · Aug 21, 2018 · 6523e4a · 6523e4a
1 parent 885693a
commit 6523e4a
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/Annotation/Controller/ResponseMapper.php b/src/Annotation/Controller/ResponseMapper.php
@@ -2,6 +2,7 @@
 
 namespace Apitte\Core\Annotation\Controller;
 
+use Apitte\Core\Mapping\Response\IResponseEntity;
 use Doctrine\Common\Annotations\Annotation\Target;
 use Doctrine\Common\Annotations\AnnotationException;
 
@@ -20,10 +21,18 @@ final class ResponseMapper
 	 */
 	public function __construct(array $values)
 	{
-		if (!isset($values['entity'])) {
+		if (!isset($values['entity']) || empty($values['entity'])) {
 			throw new AnnotationException('Empty @ResponseMapper entity given');
 		}
 
+		if (!class_exists($values['entity'])) {
+			throw new AnnotationException(sprintf('@ResponseMapper entity "%s" does not exists', $values['entity']));
+		}
+
+		if (!isset(class_implements($values['entity'])[IResponseEntity::class])) {
+			throw new AnnotationException(sprintf('@ResponseMapper entity "%s" does not implements "%s"', $values['entity'], IResponseEntity::class));
+		}
+
 		$this->entity = $values['entity'];
 	}