Bugfix: wrong case for $beantype

lutsen · Jun 4, 2017 · 44c90ac · 44c90ac
1 parent 0bbf5ba
commit 44c90ac
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/routes/functions.php b/routes/functions.php
@@ -9,11 +9,16 @@
  *
  * @var string $beantype The type of bean.
  *
- * @return string The name of the controller.
+ * @return object The controller.
  */
 function setupBeanModel($beantype) {
+	$beantype = ucfirst( strtolower( $beantype ) );
+	if ( !file_exists( ROOT_PATH. '/models/lagan/'.$beantype.'.php' ) ) {
+		throw new \Exception('The '.$beantype.' model does not exist.');
+	}
+
 	// Return model
-	$model_name = '\Lagan\Model\\' . ucfirst($beantype);
+	$model_name = '\Lagan\Model\\' . $beantype;
 	return new $model_name();
 }