Build correctly for Firefox + allow to turn on logging

1 files changed, 12 insertions, 2 deletions

diff --git a/js/firefox/content-script-proxy.js b/js/firefox/content-script-proxy.js
index 58a6bc0..271bfa1 100644
--- a/js/firefox/content-script-proxy.js
+++ b/js/firefox/content-script-proxy.js
@@ -1,16 +1,26 @@
 // This file listens to messages 
 
+function log(msg) {
+	self.port.emit('log', msg);
+}
 window.addEventListener('message', function(message) {
 	if (message.data.sender !== 'page') {
 		return;
 	}
-	console.info('proxy got page message: ' + JSON.stringify(message.data));
+
+	if (message.data.logMessage) {
+		//Special handling for log messages.
+		log(message.data.logMessage);
+		return;
+	}
+
+	log('proxy got page message: ' + JSON.stringify(message.data));
 
 	//Forward the message to the background script
 	self.port.emit('message', message.data);
 })
 
 self.port.on('message', function(message) {
-	console.info('proxy got chrome message: ' + JSON.stringify(message));
+	log('proxy got chrome message: ' + JSON.stringify(message));
 	window.postMessage(message, '*');
 });
\ No newline at end of file