Adjusting project

Author: fernandomoraes

.npmignore

@@ -0,0 +1,3 @@
+node_modules
+.git/
+.gitignore

package.json

@@ -1,17 +1,18 @@
 {
-  "name": "tiny-livereload",
+  "name": "@moraes/tiny-livereload-server",
   "version": "1.0.0",
   "description": "tiny web live reload",
   "main": "src/index.js",
   "repository": "git@github.com:fernandomoraes/tiny-livereload.git",
   "author": "Fernando Moraes <fernandosdemoraes@gmail.com>",
   "license": "MIT",
   "private": false,
+  "bin": "src/index.js",
   "dependencies": {
     "chokidar": "^3.4.2",
     "event-stream": "^4.0.1",
     "open": "^7.1.0",
     "send": "^0.17.1",
-    "ws": "^7.3.1"
+    "ws": "^7.3.1"  
   }
 }

src/index.js

@@ -1,3 +1,5 @@
+#! /usr/bin/env node
+
 const path = require('path');
 const es = require('event-stream');
 
@@ -58,7 +60,7 @@ function startServer() {
         send(req, filePath)
             .on('stream', (stream) => {
                 if (injectLiveReload) {
-                    injectLiveReloadFragment(stream);
+                    injectLiveReloadFragment(stream, res);
                 }
             })
             .pipe(res);
@@ -68,12 +70,12 @@ function startServer() {
         .addListener('upgrade', (req, socket, head) =>
             ws.handleUpgrade(req, socket, head, (connection) => liveReloadWsConnections.push(connection))
         )
-        .addListener('listening', () => console.log('server running...'))
+        .addListener('listening', () => console.log(`server running on http://localhost:${SERVER_PORT} ...`))
         .addListener('listening', () => open(`http://localhost:${SERVER_PORT}`))
         .listen(SERVER_PORT);
 }
 
-function injectLiveReloadFragment(stream) {
+function injectLiveReloadFragment(stream, res) {
     const newContentLength = res.getHeader('content-length') + LIVERELOAD_FRAGMENT.length;
     res.setHeader('content-length', newContentLength);