fix(auth): tail logs refresh token #702

Author: sii27858

…nds/cloudmanager/environment/tail-log.js …manager/environment/tail-log/tail-log.jssrc/commands/cloudmanager/environment/tail-log.js renamed to src/commands/cloudmanager/environment/tail-log/tail-log.js src/commands/cloudmanager/environment/tail-log.js renamed to src/commands/cloudmanager/environment/tail-log/tail-log.js

@@ -10,10 +10,11 @@ OF ANY KIND, either express or implied. See the License for the specific languag
 governing permissions and limitations under the License.
 */
 
-const BaseCommand = require('../../../base-command')
-const { initSdk, getProgramId, sanitizeEnvironmentId } = require('../../../cloudmanager-helpers')
-const commonFlags = require('../../../common-flags')
-const commonArgs = require('../../../common-args')
+const BaseCommand = require('../../../../base-command')
+const { initSdk, getProgramId, sanitizeEnvironmentId } = require('../../../../cloudmanager-helpers')
+const commonFlags = require('../../../../common-flags')
+const commonArgs = require('../../../../common-args')
+const { executeWithRetries } = require('./utils')
 
 class TailLog extends BaseCommand {
   async run () {
@@ -31,8 +32,10 @@ class TailLog extends BaseCommand {
   }
 
   async tailLog (programId, environmentId, service, logName, imsContextName = null) {
-    const sdk = await initSdk(imsContextName)
-    return sdk.tailLog(programId, environmentId, service, logName, process.stdout)
+    return executeWithRetries(async () => {
+      const sdk = await initSdk(imsContextName)
+      return sdk.tailLog(programId, environmentId, service, logName, process.stdout)
+    })
   }
 }
 

src/commands/cloudmanager/environment/tail-log/utils.js

@@ -0,0 +1,31 @@
+async function executeWithRetries (fn, maxRetries = 5) {
+  let retries = 0
+  let startTime = Date.now()
+  while (retries < maxRetries) {
+    try {
+      return await fn()
+    } catch (error) {
+      if (error.sdkDetails.status === 403) {
+        console.log('403 Forbidden error. Attempting to reauthorize...')
+      } else {
+        throw error
+      }
+      retries++
+      if (shouldResetRetires(startTime)) {
+        retries = 0
+        startTime = Date.now()
+      }
+    }
+  }
+}
+
+function shouldResetRetires (startTime, resetInterval = 3600000) {
+  const elapsedTime = Date.now() - startTime
+  if (elapsedTime > resetInterval) {
+    console.log(`Resetting retries after ${resetInterval / 1000} seconds.`)
+    return true
+  }
+  return false
+}
+
+module.exports = { executeWithRetries }

src/index.js

@@ -23,7 +23,7 @@ const GetExecutionStepDetails = require('./commands/cloudmanager/execution/get-s
 const GetExecutionStepLog = require('./commands/cloudmanager/execution/get-step-log')
 const ListAvailableLogOptions = require('./commands/cloudmanager/environment/list-available-log-options')
 const DownloadLogs = require('./commands/cloudmanager/environment/download-logs')
-const TailLog = require('./commands/cloudmanager/environment/tail-log')
+const TailLog = require('./commands/cloudmanager/environment/tail-log/tail-log')
 const DeletePipeline = require('./commands/cloudmanager/pipeline/delete')
 const UpdatePipeline = require('./commands/cloudmanager/pipeline/update')
 const OpenDeveloperConsole = require('./commands/cloudmanager/environment/open-developer-console')

test/commands/environment/tail-log.test.js

@@ -12,7 +12,7 @@ governing permissions and limitations under the License.
 
 const { resetCurrentOrgId, setCurrentOrgId } = require('@adobe/aio-lib-ims')
 const { init, mockSdk } = require('@adobe/aio-lib-cloudmanager')
-const TailLog = require('../../../src/commands/cloudmanager/environment/tail-log')
+const TailLog = require('../../../src/commands/cloudmanager/environment/tail-log/tail-log')
 
 beforeEach(() => {
   resetCurrentOrgId()