Create separate writers module

3 files changed, 17 insertions, 10 deletions

diff --git a/lib/html-generator.js b/lib/html-generator.js
index cea33e0..be1ebd2 100644
--- a/lib/html-generator.js
+++ b/lib/html-generator.js
@@ -1,11 +1,11 @@
-var htmlWriter = require("./writers/html-writer");
+var writers = require("./writers");
 
 exports.HtmlGenerator = HtmlGenerator;
 
 
 function HtmlGenerator(options) {
     var stack = [];
-    var writer = htmlWriter.writer(options);
+    var writer = writers.writer(options);
     
     function text(value) {
         if (value.length > 0) {
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);
+    }
+}