From e7388891e16c0b9571007ac160c3adda50ead59d Mon Sep 17 00:00:00 2001 From: wing328 Date: Mon, 18 Apr 2016 16:51:47 +0800 Subject: [PATCH] add exception handling for template dir that does not exist --- .../io/swagger/codegen/config/CodegenConfigurator.java | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/modules/swagger-codegen/src/main/java/io/swagger/codegen/config/CodegenConfigurator.java b/modules/swagger-codegen/src/main/java/io/swagger/codegen/config/CodegenConfigurator.java index 98c57ab341a..641b161857c 100644 --- a/modules/swagger-codegen/src/main/java/io/swagger/codegen/config/CodegenConfigurator.java +++ b/modules/swagger-codegen/src/main/java/io/swagger/codegen/config/CodegenConfigurator.java @@ -19,6 +19,7 @@ import org.slf4j.LoggerFactory; import java.io.File; import java.io.IOException; +import java.nio.file.Files; import java.nio.file.Paths; import java.util.HashMap; import java.util.HashSet; @@ -148,7 +149,14 @@ public class CodegenConfigurator { } public CodegenConfigurator setTemplateDir(String templateDir) { - this.templateDir = new File(templateDir).getAbsolutePath(); + File f = new File(templateDir); + + // check to see if the folder exists + if (!(f != null && f.exists() && f.isDirectory())) { + throw new IllegalArgumentException("Template directory " + templateDir + " does not exist."); + } + + this.templateDir = f.getAbsolutePath(); return this; }