Color runtime container's stderr in RED

Color runtime container's stderr in RED

Author: roy-codefresh

codefresh.yml

@@ -2,18 +2,18 @@ version: '1.0'
 
 steps:
   unit-test-step:
-    image: codefresh/buildpacks:all-5
+    image: node:6.9.2
     environment:
         - COVERALLS_TOKEN=${{COVERALLS_TOKEN}}
     working-directory: ${{initial-clone}}
     commands:
         - npm install
-        - gulp lint
-        - gulp no.onlys
-        - gulp unit_test
-        - gulp coveralls
-        - gulp integ_test
-        - gulp clean
+        - npm run gulp lint
+        - npm run gulp no.onlys
+        - npm run gulp unit_test
+        - npm run gulp coveralls
+        - npm run gulp integ_test
+        - npm run gulp clean
 
   build-step:
     type: build

lib/ContainerLogger.js

@@ -28,18 +28,24 @@ class ContainerLogger {
                     return Q.reject(new CFError(`Strategy: ${this.loggerStrategy} is not supported`));
                 }
             })
-            .then((stream) => {
+            .then(([stdout, stderr]) => {
                 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 +56,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());
     }

package.json

@@ -31,6 +31,7 @@
     "sinon-chai": "^2.8.0"
   },
   "scripts": {
+    "gulp": "gulp",
     "start": "node lib/index.js"
   }
 }