From 6bd2c87511bbad513cf2eee9c7e5fd7c26e7574e Mon Sep 17 00:00:00 2001
From: Martijn <martijn@vanderlee.com>
Date: Tue, 17 Apr 2018 14:33:52 +0200
Subject: [PATCH] TypeRegistry

---
 SwaggerGen/SwaggerGen.php | 10 ++++++++++
 example/api/swagger.php   |  3 ++-
 2 files changed, 12 insertions(+), 1 deletion(-)

diff --git a/SwaggerGen/SwaggerGen.php b/SwaggerGen/SwaggerGen.php
index e02f7bf..9987eee 100644
--- a/SwaggerGen/SwaggerGen.php
+++ b/SwaggerGen/SwaggerGen.php
@@ -63,6 +63,16 @@ public function __construct($host = '', $basePath = '', $dirs = array(), $typeRe
 		$this->dirs = $dirs;
 		$this->typeRegistry = $typeRegistry;
 	}
+	
+	/**
+	 * Set a new type registry
+	 * 
+	 * @param TypeRegistry $typeRegistry
+	 */
+	public function setTypeRegistry($typeRegistry = null)
+	{
+		$this->typeRegistry = $typeRegistry;
+	}
 
 	public function define($name, $value = 1)
 	{
diff --git a/example/api/swagger.php b/example/api/swagger.php
index 65be063..9568abf 100644
--- a/example/api/swagger.php
+++ b/example/api/swagger.php
@@ -8,7 +8,8 @@
 //	\SwaggerGen\Swagger\Type\Custom\Ipv4Type::setFormats(array('ipv4'));
 	$TypeRegistry->add('\SwaggerGen\Swagger\Type\Custom\Ipv4Type');
 
-	$SwaggerGen = new \SwaggerGen\SwaggerGen($_SERVER['HTTP_HOST'], dirname($_SERVER['REQUEST_URI']), array(), $TypeRegistry);	
+	$SwaggerGen = new \SwaggerGen\SwaggerGen($_SERVER['HTTP_HOST'], dirname($_SERVER['REQUEST_URI']), array());	
+	$SwaggerGen->setTypeRegistry($TypeRegistry);
 	
 	// @todo Allow explicitly format name specification for conflict resolution.
 	// @todo Automatically scan the default types (how to register multiple type names; e.g. StringType's names)