Create separate writers module

2 files changed, 15 insertions, 8 deletions

diff --git a/lib/writers/html-writer.js b/lib/writers/html-writer.js
index 33441cb..f1761c9 100644
--- a/lib/writers/html-writer.js
+++ b/lib/writers/html-writer.js
@@ -5,15 +5,9 @@ exports.writer = writer;
 
 function writer(options) {
     options = options || {};
-    if(options.prettyPrint) {
+    if (options.prettyPrint) {
         return prettyWriter();
-    }
-    else if(options.markdown) {
-        var markdown = require("./markdown-writer");
-
-        return markdown.writer();
-    }
-    else {
+    } else {
         return simpleWriter();
     }
 }
diff --git a/lib/writers/index.js b/lib/writers/index.js
index 8b13789..b34d28f 100644
--- a/lib/writers/index.js
+++ b/lib/writers/index.js
@@ -1 +1,14 @@
+var htmlWriter = require("./html-writer");
+var markdownWriter = require("./markdown-writer");
 
+exports.writer = writer;
+
+
+function writer(options) {
+    options = options || {};
+    if (options.markdown) {
+        return markdownWriter.writer();
+    } else {
+        return htmlWriter.writer(options);
+    }
+}