@@ -76,6 +76,7 @@ import software.aws.toolkits.jetbrains.services.codemodernizer.constants.buildTr
7676import  software.aws.toolkits.jetbrains.services.codemodernizer.constants.buildTransformStoppingChatContent 
7777import  software.aws.toolkits.jetbrains.services.codemodernizer.constants.buildUserCancelledChatContent 
7878import  software.aws.toolkits.jetbrains.services.codemodernizer.constants.buildUserHilSelection 
79+ import  software.aws.toolkits.jetbrains.services.codemodernizer.constants.buildUserInputCustomDependencyVersionsChatContent 
7980import  software.aws.toolkits.jetbrains.services.codemodernizer.constants.buildUserInputLanguageUpgradeChatContent 
8081import  software.aws.toolkits.jetbrains.services.codemodernizer.constants.buildUserInputOneOrMultipleDiffsChatIntroContent 
8182import  software.aws.toolkits.jetbrains.services.codemodernizer.constants.buildUserInputOneOrMultipleDiffsFlagChatContent 
@@ -92,6 +93,7 @@ import software.aws.toolkits.jetbrains.services.codemodernizer.messages.Authenti
9293import  software.aws.toolkits.jetbrains.services.codemodernizer.messages.CodeTransformChatMessage 
9394import  software.aws.toolkits.jetbrains.services.codemodernizer.messages.CodeTransformCommandMessage 
9495import  software.aws.toolkits.jetbrains.services.codemodernizer.messages.IncomingCodeTransformMessage 
96+ import  software.aws.toolkits.jetbrains.services.codemodernizer.model.CLIENT_SIDE_BUILD 
9597import  software.aws.toolkits.jetbrains.services.codemodernizer.model.CodeModernizerArtifact 
9698import  software.aws.toolkits.jetbrains.services.codemodernizer.model.CodeModernizerJobCompletedResult 
9799import  software.aws.toolkits.jetbrains.services.codemodernizer.model.CodeTransformConversationState 
@@ -125,6 +127,10 @@ import software.aws.toolkits.jetbrains.services.codewhisperer.telemetry.broadcas
125127import  software.aws.toolkits.jetbrains.services.cwc.messages.ChatMessageType 
126128import  software.aws.toolkits.resources.message 
127129import  software.aws.toolkits.telemetry.CodeTransformPreValidationError 
130+ import  com.intellij.openapi.application.ApplicationManager 
131+ import  com.intellij.openapi.fileEditor.FileEditorManager 
132+ import  com.intellij.testFramework.LightVirtualFile 
133+ import  org.jetbrains.yaml.YAMLFileType 
128134
129135class  CodeTransformChatController (
130136    private  val  context :  AmazonQAppInitContext ,
@@ -427,23 +433,22 @@ class CodeTransformChatController(
427433
428434    override  suspend  fun  processCodeTransformOneOrMultipleDiffs (message :  IncomingCodeTransformMessage .CodeTransformConfirmOneOrMultipleDiffs ) {
429435        val  transformCapabilities =  when  (message.oneOrMultipleDiffsSelection) {
430-             //  TODO: add CLIENT_SIDE_BUILD to both below when releasing CSB
431436            message(" codemodernizer.chat.message.one_or_multiple_diffs_form.multiple_diffs" ->  listOf (
432437                EXPLAINABILITY_V1 ,
433-                 SELECTIVE_TRANSFORMATION_V1 
438+                 SELECTIVE_TRANSFORMATION_V1 ,
439+                 CLIENT_SIDE_BUILD 
434440            )
435441            else  ->  listOf (
436442                EXPLAINABILITY_V1 ,
443+                 CLIENT_SIDE_BUILD 
437444            )
438445        }
439446        telemetry.submitSelection(message.oneOrMultipleDiffsSelection)
440447        codeTransformChatHelper.addNewMessage(buildUserOneOrMultipleDiffsSelectionChatContent(message.oneOrMultipleDiffsSelection))
441448        codeModernizerManager.codeTransformationSession?.let  {
442449            it.sessionContext.transformCapabilities =  transformCapabilities
443-             codeModernizerManager.runLocalMavenBuild(context.project, it)
444450        }
445-         //  TODO: when releasing CSB, delete "runLocalMavenBuild" line above and uncomment line below
446-         //  promptForCustomYamlFile()
451+         promptForCustomYamlFile()
447452    }
448453
449454    override  suspend  fun  processCodeTransformCustomDependencyVersions (message :  IncomingCodeTransformMessage .CodeTransformConfirmCustomDependencyVersions ) {
@@ -485,14 +490,11 @@ class CodeTransformChatController(
485490        }
486491    }
487492
488-     //  TODO: uncomment when releasing CSB
489- /* 
490493    private  suspend  fun  promptForCustomYamlFile () {
491494        codeTransformChatHelper.addNewMessage(buildUserInputCustomDependencyVersionsChatContent())
492495        val  sampleYAML =  """ 
493- name: "custom- dependency-management " 
496+ name: "dependency-upgrade " 
494497description: "Custom dependency version management for Java migration from JDK 8/11/17 to JDK 17/21" 
495- 
496498dependencyManagement: 
497499  dependencies: 
498500    - identifier: "com.example:library1" 
@@ -503,18 +505,18 @@ dependencyManagement:
503505      targetVersion: "3.0.0" 
504506      originType: "THIRD_PARTY" 
505507  plugins: 
506-     - identifier: "com.example. plugin" 
508+     - identifier: "com.example: plugin" 
507509      targetVersion: "1.2.0" 
508510      versionProperty: "plugin.version"  # Optional 
509511        """  .trimIndent()
510512
511-         val virtualFile = LightVirtualFile("sample-dependency-management .yaml", YAMLFileType.YML, sampleYAML) 
513+         val  virtualFile =  LightVirtualFile (" sample-dependency-upgrade .yaml" YAMLFileType .YML , sampleYAML)
512514        virtualFile.isWritable =  true 
513515        ApplicationManager .getApplication().invokeLater {
514516            FileEditorManager .getInstance(context.project).openFile(virtualFile, true )
515517        }
516518    }
517- */ 
519+ 
518520    override  suspend  fun  processCodeTransformContinueAction (message :  IncomingCodeTransformMessage .CodeTransformContinue ) {
519521        codeTransformChatHelper.addNewMessage(buildContinueTransformationChatContent())
520522        promptForTargetJdkName(message.tabId)
0 commit comments