Changing logic of container logger to color the stderr with red

Author: roy-codefresh

lib/ContainerLogger.js

@@ -1,6 +1,7 @@
 'use strict';
 
 const Q              = require('q');
+const chalk          = require('chalk');
 const logger         = require('cf-logs').Logger('codefresh:containerLogger');
 const CFError        = require('cf-errors');
 const LoggerStrategy = require('./enums').LoggerStrategy;
@@ -28,18 +29,26 @@ class ContainerLogger {
                     return Q.reject(new CFError(`Strategy: ${this.loggerStrategy} is not supported`));
                 }
             })
-            .then((stream) => {
+            .then(([stdout, stderr]) => {
+                this._logMessageToFirebase('New logger started v8\n');
+                
                 logger.info(`Attached stream to container: ${this.containerId}`);
                 // Listening on the stream needs to be performed different depending if a tty is attached or not
                 // See documentation of the docker api here: https://docs.docker.com/engine/reference/api/docker_remote_api_v1.24/#/attach-to-a-container
                 if (this.tty) {
-                    this._handleTtyStream(stream);
+                    this._handleTtyStream(stdout, false);
+                    this._handleTtyStream(stderr, true);
                 } else {
-                    this._handleNonTtyStream(stream);
+                    this._handleNonTtyStream(stdout, false);
+                    this._handleNonTtyStream(stderr, true);
                 }
 
-                stream.on('end', () => {
-                    logger.info(`stream end event was fired for container: ${this.containerId}`);
+                stdout.on('end', () => {
+                    logger.info(`stdout end event was fired for container: ${this.containerId}`);
+                });
+
+                stderr.on('end', () => {
+                    logger.info(`stderr end event was fired for container: ${this.containerId}`);
                 });
             }, (err) => {
                 return Q.reject(new CFError({
@@ -50,47 +59,66 @@ class ContainerLogger {
     }
 
     _getAttachStrategyStream() {
-        return Q.ninvoke(this.containerInterface, 'attach', {
-            stream: true,
-            stdout: true,
-            stderr: true,
-            tty: true
-        });
+        return Q.all([
+            Q.ninvoke(this.containerInterface, 'attach', {
+                stream: true,
+                stdout: true,
+                stderr: false,
+                tty: true
+            }),
+            Q.ninvoke(this.containerInterface, 'attach', {
+                stream: true,
+                stdout: false,
+                stderr: true,
+                tty: true
+            })
+        ]);
     }
 
     _getLogsStrategyStream() {
-        return Q.ninvoke(this.containerInterface, 'logs', {
-            follow: 1,
-            stdout: 1,
-            stderr: 1
-        });
+        return Q.all([
+            Q.ninvoke(this.containerInterface, 'logs', {
+                follow: 1,
+                stdout: 1,
+                stderr: 0
+            }),
+            Q.ninvoke(this.containerInterface, 'logs', {
+                follow: 1,
+                stdout: 0,
+                stderr: 1
+            })
+        ]);
     }
 
-    _handleTtyStream(stream) {
+    _handleTtyStream(stream, isError) {
         stream.on('data', (chunk) => {
             const buf     = new Buffer(chunk);
             const message = buf.toString('utf8');
-            this._logMessageToFirebase(message);
+            this._logMessageToFirebase(message, isError);
         });
         logger.info(`Listening on stream 'data' event for container: ${this.containerId}`);
     }
 
-    _handleNonTtyStream(stream) {
+    _handleNonTtyStream(stream, isError) {
         stream.on('readable', () => {
             let header = stream.read(8);
             while (header !== null) {
                 const payload = stream.read(header.readUInt32BE(4));
                 if (payload === null) {
                     break;
                 }
-                this._logMessageToFirebase(new Buffer(payload).toString('utf8'));
+                this._logMessageToFirebase(new Buffer(payload).toString('utf8'), isError);
                 header = stream.read(8);
             }
         });
         logger.info(`Listening on stream 'readable' event for container: ${this.containerId}`);
     }
 
-    _logMessageToFirebase(message) {
+    _logMessageToFirebase(message, isError) {
+        if (isError) {
+            message = `\x1B[31m${message}\x1B[0m`;
+        }
+        
         this.firebaseLogger.push(message);
         this.firebaseLastUpdate.set(new Date().getTime());
     }