open plan periodically

Author: David Hasani

packages/core/src/codewhisperer/commands/startTransformByQ.ts

@@ -554,6 +554,11 @@ export async function pollTransformationStatusUntilPlanReady(jobId: string, prof
         // for now, no plan shown with SQL conversions. later, we may add one
         return
     }
+    jobPlanProgress['generatePlan'] = StepProgress.Succeeded
+    throwIfCancelled()
+}
+
+export async function openTransformationPlan(jobId: string, profile?: RegionProfile) {
     let plan = undefined
     try {
         plan = await getTransformationPlan(jobId, profile)
@@ -576,8 +581,6 @@ export async function pollTransformationStatusUntilPlanReady(jobId: string, prof
         transformByQState.setPlanFilePath(planFilePath)
         await setContext('gumby.isPlanAvailable', true)
     }
-    jobPlanProgress['generatePlan'] = StepProgress.Succeeded
-    throwIfCancelled()
 }
 
 export async function pollTransformationStatusUntilComplete(jobId: string, profile: RegionProfile | undefined) {

packages/core/src/codewhisperer/service/transformByQ/transformApiHandler.ts

@@ -55,6 +55,7 @@ import { AuthUtil } from '../../util/authUtil'
 import { DiffModel } from './transformationResultsViewProvider'
 import { spawnSync } from 'child_process' // eslint-disable-line no-restricted-imports
 import { isClientSideBuildEnabled } from '../../../dev/config'
+import { openTransformationPlan } from '../../commands/startTransformByQ'
 
 export function getSha256(buffer: Buffer) {
     const hasher = crypto.createHash('sha256')
@@ -785,6 +786,11 @@ export async function pollTransformationJob(jobId: string, validStates: string[]
                     `${CodeWhispererConstants.failedToCompleteJobGenericNotification} ${errorMessage}`
                 )
             }
+
+            if (CodeWhispererConstants.validStatesForPlanGenerated.includes(status)) {
+                await openTransformationPlan(jobId, profile)
+            }
+
             if (validStates.includes(status)) {
                 break
             }