Merge pull request #107 from codefresh-io/moshe_levi

Handle firebase disconnect

Author: danielm-codefresh

lib/logic/logs.helper.js

@@ -7,6 +7,23 @@ const Firebase = require('firebase');
 const rp = require('request-promise');
 const { CfReceiverService } = require('@codefresh-io/cf-receiver');
 
+const { Http } = require('../../helpers/http');
+
+const http = Http();
+
+const getStatus = async (firebaseAuth, progressJobId) => {
+    const { url, accessToken: token } = firebaseAuth;
+    const userOptions = {
+        url: `${url}/${progressJobId}/status.json?auth=${token}`,
+        method: 'GET',
+        options: {
+            timeout: 10000,
+        },
+    };
+    const status = await http(userOptions);
+    return status;
+};
+
 const _connectToFirebase = async (firebaseAuth) => new Promise((resolve, reject) => {
     const jobIdRef = new Firebase(firebaseAuth.url);
     jobIdRef.authWithCustomToken(firebaseAuth.accessToken, (err) => {
@@ -72,7 +89,8 @@ function _isBuildFinished(status) {
     return ['success', 'error', 'terminated'].includes(status);
 }
 
-const _printFollowFirebaseLogs = async (firebaseAuth, progressJobId, removeTimestamps) => new Promise((resolve, reject) => {
+// eslint-disable-next-line max-len
+const _printFollowFirebaseLogs = async (firebaseAuth, progressJobId, removeTimestamps, workflowId, sdk) => new Promise((resolve, reject) => {
     const jobIdRef = new Firebase(`${_getFinalFirebaseUrl(firebaseAuth)}/${progressJobId}`);
 
     const errorCallback = (err) => {
@@ -82,12 +100,35 @@ const _printFollowFirebaseLogs = async (firebaseAuth, progressJobId, removeTimes
         }));
     };
 
+    let interval;
+
+    const statusHandler = (status) => {
+        if (_isBuildFinished(status)) {
+            clearInterval(interval);
+            resolve();
+        }
+    };
+
+    interval = setInterval(async () => {
+        try {
+            const status = await getStatus(firebaseAuth, progressJobId);
+            statusHandler(status);
+        } catch (err) {
+            process.stdout.write(`Failed to get status from firebase: ${err}`);
+            try {
+                await _connectToFirebase(firebaseAuth);
+            } catch (error) {
+                const errMsg = `Failed to connect to firebase: ${error}`;
+                process.stdout.write(errMsg);
+            }
+            const workflow = await sdk.workflows.getBuild({ buildId: workflowId, noAccount: false, firebaseFallback: 'true' });
+            statusHandler(workflow.status);
+        }
+    }, 10000);
+
     jobIdRef.child('status')
         .on('value', (snapshot) => {
-            const status = snapshot.val();
-            if (_isBuildFinished(status)) {
-                resolve();
-            }
+            statusHandler(snapshot.val());
         }, errorCallback);
 
     jobIdRef.child('steps')
@@ -103,6 +144,7 @@ const _printFollowFirebaseLogs = async (firebaseAuth, progressJobId, removeTimes
             step.ref.child('logs')
                 .on('child_added', (snapshot) => { // eslint-disable-line
                     let log = snapshot.val();
+
                     if (removeTimestamps) {
                         log = _stripTimestamps(log);
                     }
@@ -111,11 +153,11 @@ const _printFollowFirebaseLogs = async (firebaseAuth, progressJobId, removeTimes
         }, errorCallback);
 });
 
-const showWorkflowLogsFromFirebase = async (progressJobId, follow, removeTimestamps, sdk) => {
+const showWorkflowLogsFromFirebase = async (progressJobId, follow, removeTimestamps, workflowId, sdk) => {
     const firebaseAuth = await sdk.firebase.getToken({ progressId: progressJobId });
     await _connectToFirebase(firebaseAuth);
     if (follow) {
-        await _printFollowFirebaseLogs(firebaseAuth, progressJobId, removeTimestamps);
+        await _printFollowFirebaseLogs(firebaseAuth, progressJobId, removeTimestamps, workflowId, sdk);
     } else {
         await _printCurrentFirebaseLogs(firebaseAuth, progressJobId, removeTimestamps);
     }

lib/logic/logs.logic.js

@@ -8,7 +8,7 @@ class Logs extends Resource {
         const workflow = await this.sdk.workflows.getBuild({ buildId: workflowId, noAccount: false });
         const progressJob = await this.sdk.progress.get({ id: workflow.progress });
         if (progressJob.location.type === 'firebase') {
-            await showWorkflowLogsFromFirebase(progressJob._id, follow, removeTimetamps, this.sdk);
+            await showWorkflowLogsFromFirebase(progressJob._id, follow, removeTimetamps, workflowId, this.sdk);
         } else if (progressJob.location.type === 'composition') {
             await showWorkflowLogsByWebsocket(progressJob._id, progressJob.account, follow, this.sdk);
         } else {

package.json

@@ -1,6 +1,6 @@
 {
   "name": "codefresh-sdk",
-  "version": "1.11.2",
+  "version": "1.11.3",
   "description": "Codefresh_api_swagger_3_0_specification",
   "main": "index.js",
   "author": {