grep for sql statements

Author: David Hasani

plugins/amazonq/codetransform/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codemodernizer/CodeModernizerManager.kt

@@ -60,7 +60,7 @@ import software.aws.toolkits.jetbrains.services.codemodernizer.toolwindow.CodeMo
 import software.aws.toolkits.jetbrains.services.codemodernizer.utils.STATES_WHERE_PLAN_EXIST
 import software.aws.toolkits.jetbrains.services.codemodernizer.utils.createFileCopy
 import software.aws.toolkits.jetbrains.services.codemodernizer.utils.findLineNumberByString
-import software.aws.toolkits.jetbrains.services.codemodernizer.utils.getJavaModules
+import software.aws.toolkits.jetbrains.services.codemodernizer.utils.getJavaModulesWithSQL
 import software.aws.toolkits.jetbrains.services.codemodernizer.utils.getMavenVersion
 import software.aws.toolkits.jetbrains.services.codemodernizer.utils.getModuleOrProjectNameForFile
 import software.aws.toolkits.jetbrains.services.codemodernizer.utils.getPathToHilArtifactPomFile
@@ -160,7 +160,7 @@ class CodeModernizerManager(private val project: Project) : PersistentStateCompo
             }
 
             if (transformationType == CodeTransformType.SQL_CONVERSION) {
-                val javaModules = project.getJavaModules()
+                val javaModules = project.getJavaModulesWithSQL()
                 return if (javaModules.isNotEmpty()) {
                     ValidationResult(
                         true,

plugins/amazonq/codetransform/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codemodernizer/controller/CodeTransformChatController.kt

@@ -95,7 +95,7 @@ import software.aws.toolkits.jetbrains.services.codemodernizer.panels.managers.C
 import software.aws.toolkits.jetbrains.services.codemodernizer.session.ChatSessionStorage
 import software.aws.toolkits.jetbrains.services.codemodernizer.session.Session
 import software.aws.toolkits.jetbrains.services.codemodernizer.state.CodeModernizerSessionState
-import software.aws.toolkits.jetbrains.services.codemodernizer.utils.getJavaModules
+import software.aws.toolkits.jetbrains.services.codemodernizer.utils.getJavaModulesWithSQL
 import software.aws.toolkits.jetbrains.services.codemodernizer.utils.getModuleOrProjectNameForFile
 import software.aws.toolkits.jetbrains.services.codemodernizer.utils.isCodeTransformAvailable
 import software.aws.toolkits.jetbrains.services.codemodernizer.utils.toVirtualFile
@@ -150,13 +150,22 @@ class CodeTransformChatController(
         // Publish a metric when transform is first initiated from chat prompt.
         telemetry.initiateTransform()
 
-        val anyModuleContainsOracleSQL = true // TODO: implement, and make sure to say "Checking for eligible modules..." here too
+        val anyModuleContainsOracleSQL = codeModernizerManager.validate(context.project, CodeTransformType.SQL_CONVERSION).valid
 
-        if (anyModuleContainsOracleSQL) {
-            this.getUserObjective(message.tabId)
-        } else {
+        if (!anyModuleContainsOracleSQL) {
             this.handleLanguageUpgrade()
+            return
+        }
+
+        val eligibleForLanguageUpgrade = codeModernizerManager.validate(context.project, CodeTransformType.LANGUAGE_UPGRADE).valid
+
+        if (!eligibleForLanguageUpgrade) {
+            this.handleSQLConversion()
+            return
         }
+
+        // eligible for both language upgrade and sql conversion, so ask user what they want to do
+        this.getUserObjective(message.tabId)
     }
 
     private suspend fun getUserObjective(tabId: String) {
@@ -170,7 +179,7 @@ class CodeTransformChatController(
             buildCheckingValidProjectChatContent()
         )
 
-        codeTransformChatHelper.chatDelayLong()
+        codeTransformChatHelper.chatDelayShort()
 
         val validationResult = codeModernizerManager.validate(context.project, transformationType)
 
@@ -337,8 +346,7 @@ class CodeTransformChatController(
                     addNewMessage(buildSQLMetadataValidationSuccessIntroChatContent())
                     addNewMessage(buildSQLMetadataValidationSuccessDetailsChatContent(metadataValidationResult))
                     addNewMessage(buildModuleSchemaFormIntroChatContent())
-                    // already validated that there are open Java modules at this point
-                    addNewMessage(buildModuleSchemaFormChatContent(context.project, context.project.getJavaModules(), metadataValidationResult.schemaOptions))
+                    addNewMessage(buildModuleSchemaFormChatContent(context.project, context.project.getJavaModulesWithSQL(), metadataValidationResult.schemaOptions))
                 }
                 val selection = CustomerSelection(
                     // for SQL conversions (no sourceJavaVersion), use dummy value of Java 8 so that startJob API can be called

plugins/amazonq/codetransform/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codemodernizer/utils/CodeTransformModuleUtils.kt

@@ -10,6 +10,8 @@ import com.intellij.openapi.projectRoots.impl.JavaSdkImpl
 import com.intellij.openapi.roots.LanguageLevelModuleExtensionImpl
 import com.intellij.openapi.roots.ModuleRootManager
 import com.intellij.openapi.roots.ProjectRootManager
+import com.intellij.openapi.vfs.VirtualFile
+import java.io.File
 
 /**
  * @description Try to get the module SDK version and/or Language level from the project settings > modules > source field.
@@ -35,5 +37,29 @@ fun Module.tryGetJdkLanguageLevelJdk(): JavaSdkVersion? {
     val moduleRootManager = ModuleRootManager.getInstance(this)
     val languageLevelModuleExtension = moduleRootManager.getModuleExtension(LanguageLevelModuleExtensionImpl::class.java)
     val languageLevel = languageLevelModuleExtension?.languageLevel
-    return languageLevel?.let { JavaSdkVersion.fromLanguageLevel(it) } ?: null
+    return languageLevel?.let { JavaSdkVersion.fromLanguageLevel(it) }
+}
+
+// search for Strings that indicate embedded Oracle SQL statements are present
+fun containsSQL(contentRoot: VirtualFile): Boolean {
+    val patterns = listOf(
+        "oracle.jdbc.OracleDriver",
+        "jdbc:oracle:thin:@//",
+        "jdbc:oracle:oci:@//"
+    )
+
+    val isWindows = System.getProperty("os.name").lowercase().startsWith("Windows")
+    val searchCommand = if (isWindows) "findstr" else "grep"
+    val searchOptions = if (isWindows) listOf("/i", "/s") else listOf("-i", "-r")
+
+    for (pattern in patterns) {
+        val command = listOf(searchCommand, *searchOptions.toTypedArray(), pattern)
+        val processBuilder = ProcessBuilder(command)
+        processBuilder.directory(File(contentRoot.path))
+        val process = processBuilder.start()
+        if (process.waitFor() == 0) {
+            return true
+        }
+    }
+    return false
 }

plugins/amazonq/codetransform/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codemodernizer/utils/CodeTransformProjectUtils.kt

@@ -79,15 +79,15 @@ fun Project.getSupportedModules(supportedJavaMappings: Map<JavaSdkVersion, Set<J
     moduleJdk in supportedJavaMappings
 }
 
-// return the first file or directory found inside each open Java module, so that user can select a Module for us to ZIP
-fun Project.getJavaModules() = this.modules.flatMap { module ->
+// return the first file or directory found inside each open Java module that contains SQL statements, so that user can select a Module for us to ZIP
+fun Project.getJavaModulesWithSQL() = this.modules.flatMap { module ->
     val rootManager = ModuleRootManager.getInstance(module)
     if (rootManager.sdk?.sdkType?.name?.lowercase()?.contains("java") == true) {
         val contentRoots = rootManager.contentRoots
         if (contentRoots.isNotEmpty()) {
-            val firstContentRoot = contentRoots.first()
-            val children = firstContentRoot.children
-            if (children.isNotEmpty()) {
+            val contentRoot = contentRoots.first()
+            val children = contentRoot.children
+            if (children.isNotEmpty() && containsSQL(contentRoot)) {
                 listOf(children.first())
             } else {
                 emptyList()