From 5ab3970e95e44f79accad5e6fac95ff99855fdbd Mon Sep 17 00:00:00 2001
From: Christopher Jeffrey <chjjeffrey@gmail.com>
Date: Sat, 28 Jan 2012 13:09:38 -0600
Subject: [PATCH] refactor bin/marked

---
 bin/marked | 45 ++++++++++++++++++++++++++-------------------
 1 file changed, 26 insertions(+), 19 deletions(-)

diff --git a/bin/marked b/bin/marked
index dcad73e8..e6a2fe97 100755
--- a/bin/marked
+++ b/bin/marked
@@ -9,6 +9,10 @@ var fs = require('fs')
   , util = require('util')
   , marked = require('../');
 
+/**
+ * Man Page
+ */
+
 var help = function() {
   var spawn = require('child_process').spawn;
 
@@ -28,12 +32,16 @@ var help = function() {
   // });
 };
 
+/**
+ * Main
+ */
+
 var main = function(argv) {
   var files = []
+    , data = ''
     , input
     , output
     , arg
-    , data
     , tokens;
 
   var getarg = function() {
@@ -69,27 +77,26 @@ var main = function(argv) {
     }
   }
 
-  if (!input && files.length !== 2) {
+  if (!input) {
+    if (files.length <= 2) {
+      var stdin = process.stdin;
+
+      stdin.setEncoding('utf8');
+      stdin.resume();
+
+      stdin.on('data', function(text) {
+        data += text;
+      });
+
+      stdin.on('end', write);
+
+      return;
+    }
     input = files.pop();
   }
 
-  if (!input && files.length === 2) {
-    var stdin = process.openStdin()
-      , buff = [];
-
-    stdin.setEncoding('utf8');
-    stdin.on('data', function(data) {
-      buff.push(data);
-    });
-
-    stdin.on('end', function() {
-      data = buff.join('');
-      write();
-    });
-  } else {
-    data = fs.readFileSync(input, 'utf8');
-    write();
-  }
+  data = fs.readFileSync(input, 'utf8');
+  write();
 
   function write() {
     data = tokens