From d0ca0432809d276b246e9276b4a75fecbc7f876b Mon Sep 17 00:00:00 2001
From: Vladimir Pouzanov <farcaller@gmail.com>
Date: Fri, 16 Apr 2021 16:04:45 +0000
Subject: [PATCH] Add support for iBazel

---
 README.md      |  1 +
 index.js       | 20 +++++++++++++++++++-
 live-server.js |  7 ++++++-
 3 files changed, 26 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index cd7760a..5c5bfaf 100644
--- a/README.md
+++ b/README.md
@@ -61,6 +61,7 @@ Command line parameters:
 * `--wait=MILLISECONDS` - (default 100ms) wait for all changes, before reloading
 * `--htpasswd=PATH` - Enables http-auth expecting htpasswd file located at PATH
 * `--cors` - Enables CORS for any origin (reflects request origin, requests with credentials are supported)
+* `--ibazel` - Enables the [iBazel notification protocol](https://github.com/bazelbuild/bazel-watcher)
 * `--https=PATH` - PATH to a HTTPS configuration module
 * `--https-module=MODULE_NAME` - Custom HTTPS module (e.g. `spdy`)
 * `--proxy=ROUTE:URL` - proxy all requests for ROUTE to URL
diff --git a/index.js b/index.js
index 77a2a7e..76d2005 100644
--- a/index.js
+++ b/index.js
@@ -11,7 +11,8 @@ var fs = require('fs'),
 	open = require('opn'),
 	es = require("event-stream"),
 	os = require('os'),
-	chokidar = require('chokidar');
+	chokidar = require('chokidar'),
+	readline = require('readline');
 require('colors');
 
 var INJECTED_CODE = fs.readFileSync(path.join(__dirname, "injected.html"), "utf8");
@@ -369,6 +370,23 @@ LiveServer.start = function(options) {
 				ws.send(cssChange ? 'refreshcss' : 'reload');
 		});
 	}
+
+	if (options.ibazelListener) {
+		var rl = readline.createInterface({
+			input: process.stdin,
+			terminal: false,
+		});
+
+		rl.on('line', function (line) {
+			if (line === 'IBAZEL_BUILD_COMPLETED SUCCESS') {
+				clients.forEach(function (ws) {
+					if (ws)
+						ws.send(cssChange ? 'refreshcss' : 'reload');
+				});
+			}
+		})
+	}
+
 	LiveServer.watcher
 		.on("change", handleChange)
 		.on("add", handleChange)
diff --git a/live-server.js b/live-server.js
index 9eac8f1..f8f5218 100755
--- a/live-server.js
+++ b/live-server.js
@@ -12,6 +12,7 @@ var opts = {
 	proxy: [],
 	middleware: [],
 	logLevel: 2,
+	ibazelListener: false,
 };
 
 var homeDir = process.env[(process.platform === 'win32') ? 'USERPROFILE' : 'HOME'];
@@ -146,8 +147,12 @@ for (var i = process.argv.length - 1; i >= 2; --i) {
 		opts.middleware.push(arg.substring(13));
 		process.argv.splice(i, 1);
 	}
+	else if (arg === "--ibazel") {
+		opts.ibazelListener = true;
+		process.argv.splice(i, 1);
+	}
 	else if (arg === "--help" || arg === "-h") {
-		console.log('Usage: live-server [-v|--version] [-h|--help] [-q|--quiet] [--port=PORT] [--host=HOST] [--open=PATH] [--no-browser] [--browser=BROWSER] [--ignore=PATH] [--ignorePattern=RGXP] [--no-css-inject] [--entry-file=PATH] [--spa] [--mount=ROUTE:PATH] [--wait=MILLISECONDS] [--htpasswd=PATH] [--cors] [--https=PATH] [--https-module=MODULE_NAME] [--proxy=PATH] [PATH]');
+		console.log('Usage: live-server [-v|--version] [-h|--help] [-q|--quiet] [--port=PORT] [--host=HOST] [--open=PATH] [--no-browser] [--browser=BROWSER] [--ignore=PATH] [--ignorePattern=RGXP] [--no-css-inject] [--entry-file=PATH] [--spa] [--mount=ROUTE:PATH] [--wait=MILLISECONDS] [--htpasswd=PATH] [--cors] [--ibazel] [--https=PATH] [--https-module=MODULE_NAME] [--proxy=PATH] [PATH]');
 		process.exit();
 	}
 	else if (arg === "--test") {