replace custom script with replacestream

wearefractal · Apr 22, 2014 · e90c4e7 · e90c4e7
1 parent 275274e
commit e90c4e7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 17 deletions.
diff --git a/lib/serve/createServer.js b/lib/serve/createServer.js
@@ -4,7 +4,7 @@ var http     = require('http');
 var pf       = require('portfinder');
 var mime     = require('mime');
 var exec     = require('child_process').exec;
-var lrStream = require('../util/liveReloadStream');
+var rstream  = require('replacestream');
 
 module.exports = function (port, lrPort, open, next) {
   var server = http.createServer(function(req, res) {
@@ -26,8 +26,9 @@ module.exports = function (port, lrPort, open, next) {
 
     // if html, inject lr
     if (filePath.match(/htm(l)?$/)) {
+      console.log('html');
       fileStream
-      .pipe(lrStream(lrPort))
+      .pipe(rstream('</body>', getLivereloadTag(lrPort)+'</body>'))
       .pipe(res);
     } else {
       // stream file to response
@@ -48,6 +49,10 @@ module.exports = function (port, lrPort, open, next) {
   });
 };
 
+function getLivereloadTag(port) {
+  return "<script>document.write('<script src=\"http://' + (location.host || 'localhost').split(':')[0] + ':"+port+"/livereload.js?snipver=1\"></' + 'script>')</script>";
+}
+
 function send404(res) {
   res.writeHead(404, {'Content-type': 'text/plain'});
   res.end('Not Found');

diff --git a/lib/util/liveReloadStream.js b/lib/util/liveReloadStream.js
diff --git a/package.json b/package.json
@@ -12,7 +12,7 @@
     "portfinder": "~0.2.1",
     "minimist": "0.0.8",
     "mime": "~1.2.11",
-    "through2-map": "~1.2.1"
+    "replacestream": "~0.1.3"
   },
   "devDependencies": {
     "mocha": "*",