fix(amazonqDoc): Generate README for modules that are a part of the project (#5247)

Co-authored-by: Matthew McDonald <[email protected]>

Author: mattmcdonald39

.changes/next-release/bugfix-6362e624-b4c5-4a7a-80c7-5d52fcb4d78f.json

@@ -0,0 +1,4 @@
+{
+  "type" : "bugfix",
+  "description" : "Fix doc generation for modules that are a part of the project"
+}

plugins/amazonq/chat/jetbrains-community/src/software/aws/toolkits/jetbrains/services/amazonqDoc/controller/DocController.kt

@@ -10,6 +10,9 @@ import com.intellij.diff.requests.SimpleDiffRequest
 import com.intellij.diff.util.DiffUserDataKeys
 import com.intellij.ide.BrowserUtil
 import com.intellij.openapi.application.runInEdt
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.roots.ProjectRootManager
+import com.intellij.openapi.vfs.LocalFileSystem
 import com.intellij.openapi.vfs.VfsUtil
 import com.intellij.openapi.wm.ToolWindowManager
 import kotlinx.coroutines.withContext
@@ -68,6 +71,7 @@ import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.session.NewFil
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.session.SessionStatePhase
 import software.aws.toolkits.jetbrains.services.amazonqFeatureDev.util.CancellationTokenSource
 import software.aws.toolkits.resources.message
+import java.nio.file.Paths
 import java.util.UUID
 
 enum class DocGenerationStep {
@@ -565,7 +569,7 @@ class DocController(
                     tabId = tabId,
                     followUp = listOf(
                         FollowUp(
-                            pillText = message("amazonqFeatureDev.follow_up.modify_source_folder"),
+                            pillText = message("amazonqDoc.prompt.folder.change"),
                             type = FollowUpTypes.MODIFY_DEFAULT_SOURCE_FOLDER,
                             status = FollowUpStatusType.Info,
                         )
@@ -895,30 +899,53 @@ class DocController(
         }
     }
 
+    private fun isFolderPathInProjectModules(project: Project, folderPath: String): Boolean {
+        val path = Paths.get(folderPath)
+        val virtualFile = LocalFileSystem.getInstance().findFileByIoFile(path.toFile()) ?: return false
+
+        val projectFileIndex = ProjectRootManager.getInstance(project).fileIndex
+
+        return projectFileIndex.isInProject(virtualFile)
+    }
+
     private suspend fun modifyDefaultSourceFolder(tabId: String) {
         val session = getSessionInfo(tabId)
         val currentSourceFolder = session.context.selectedSourceFolder
         val projectRoot = session.context.projectRoot
 
         withContext(EDT) {
             val selectedFolder = selectFolder(context.project, currentSourceFolder)
+
             // No folder was selected
             if (selectedFolder == null) {
                 logger.info { "Cancelled dialog and not selected any folder" }
                 return@withContext
             }
 
-            // The folder is not in the workspace
-            if (!selectedFolder.path.startsWith(projectRoot.path)) {
+            val isFolderPathInProject = isFolderPathInProjectModules(context.project, selectedFolder.path)
+
+            if (!isFolderPathInProject) {
                 logger.info { "Selected folder not in workspace: ${selectedFolder.path}" }
 
                 messenger.sendAnswer(
                     tabId = tabId,
                     messageType = DocMessageType.Answer,
                     message = message("amazonqFeatureDev.follow_up.incorrect_source_folder"),
+                    followUp = listOf(
+                        FollowUp(
+                            pillText = message("amazonqDoc.prompt.folder.change"),
+                            type = FollowUpTypes.MODIFY_DEFAULT_SOURCE_FOLDER,
+                            status = FollowUpStatusType.Info,
+                        )
+                    ),
+                    snapToTop = true
                 )
+
+                messenger.sendChatInputEnabledMessage(tabId, enabled = false)
+
                 return@withContext
             }
+
             if (selectedFolder.path == projectRoot.path) {
                 docGenerationTask.folderLevel = DocGenerationFolderLevel.ENTIRE_WORKSPACE
             } else {