fix(codewhisperer): handle out of bounds endLine (#4254)

* fix(codewhisperer): handle out of bounds endLine

* reduce code duplication

Author: ctlai95

.changes/next-release/bugfix-44045a13-22c8-404e-bfb3-037c6078f680.json

@@ -0,0 +1,4 @@
+{
+  "type" : "bugfix",
+  "description" : "CodeWhisperer: handle exception when code scan service returns out of bounds line numbers"
+}

plugins/toolkit/jetbrains-core/src/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanSession.kt

@@ -55,6 +55,8 @@ import java.time.Instant
 import java.util.Base64
 import java.util.UUID
 import kotlin.coroutines.coroutineContext
+import kotlin.math.max
+import kotlin.math.min
 
 class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
     private val clientToken: UUID = UUID.randomUUID()
@@ -308,7 +310,8 @@ class CodeWhispererCodeScanSession(val sessionContext: CodeScanSessionContext) {
                     runReadAction {
                         FileDocumentManager.getInstance().getDocument(file)
                     }?.let { document ->
-                        val endCol = document.getLineEndOffset(it.endLine - 1) - document.getLineStartOffset(it.endLine - 1) + 1
+                        val endLineInDocument = min(max(0, it.endLine - 1), document.lineCount - 1)
+                        val endCol = document.getLineEndOffset(endLineInDocument) - document.getLineStartOffset(endLineInDocument) + 1
                         CodeWhispererCodeScanIssue(
                             startLine = it.startLine,
                             startCol = 1,

plugins/toolkit/jetbrains-core/tst/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanTest.kt

@@ -124,6 +124,15 @@ class CodeWhispererCodeScanTest : CodeWhispererCodeScanTestBase(PythonCodeInsigh
         assertThat(res).hasSize(2)
     }
 
+    @Test
+    fun `test mapToCodeScanIssues - handles index out of bounds`() {
+        val recommendations = listOf(
+            fakeListCodeScanFindingsOutOfBoundsIndexResponse.codeScanFindings(),
+        )
+        val res = codeScanSessionSpy.mapToCodeScanIssues(recommendations)
+        assertThat(res).hasSize(1)
+    }
+
     @Test
     fun `test run() - happypath`() {
         assertNotNull(sessionConfigSpy)

plugins/toolkit/jetbrains-core/tst/software/aws/toolkits/jetbrains/services/codewhisperer/codescan/CodeWhispererCodeScanTestBase.kt

@@ -76,6 +76,7 @@ open class CodeWhispererCodeScanTestBase(projectRule: CodeInsightTestFixtureRule
     internal lateinit var fakeCreateCodeScanResponseFailed: CreateCodeScanResponse
     internal lateinit var fakeCreateCodeScanResponsePending: CreateCodeScanResponse
     internal lateinit var fakeListCodeScanFindingsResponse: ListCodeScanFindingsResponse
+    internal lateinit var fakeListCodeScanFindingsOutOfBoundsIndexResponse: ListCodeScanFindingsResponse
     internal lateinit var fakeGetCodeScanResponse: GetCodeScanResponse
     internal lateinit var fakeGetCodeScanResponsePending: GetCodeScanResponse
     internal lateinit var fakeGetCodeScanResponseFailed: GetCodeScanResponse
@@ -108,52 +109,41 @@ open class CodeWhispererCodeScanTestBase(projectRule: CodeInsightTestFixtureRule
         )
     }
 
-    private fun setupCodeScanFindings(filePath: Path) = """
-        [
-            {
-                "filePath": "${filePath.systemIndependentPath}",
-                "startLine": 1,
-                "endLine": 2,
-                "title": "test",
-                "description": {
-                    "text": "global variable",
-                    "markdown": "### global variable"
-                },
-                "detectorId": "detectorId",
-                "detectorName": "detectorName",
-                "findingId": "findingId",
-                "relatedVulnerabilities": [],
-                "severity": "severity",
-                "remediation": {
-                    "recommendation": {
-                        "text": "recommendationText",
-                        "url": "recommendationUrl"
-                    },
-                    "suggestedFixes": []
-                }
+    private fun setupCodeScanFinding(filePath: Path, startLine: Int, endLine: Int) = """
+        {
+            "filePath": "${filePath.systemIndependentPath}",
+            "startLine": $startLine,
+            "endLine": $endLine,
+            "title": "test",
+            "description": {
+                "text": "global variable",
+                "markdown": "### global variable"
             },
-            {
-                "filePath": "${filePath.systemIndependentPath}",
-                "startLine": 1,
-                "endLine": 2,
-                "title": "test",
-                "description": {
-                    "text": "global variable",
-                    "markdown": "### global variable"
+            "detectorId": "detectorId",
+            "detectorName": "detectorName",
+            "findingId": "findingId",
+            "relatedVulnerabilities": [],
+            "severity": "severity",
+            "remediation": {
+                "recommendation": {
+                    "text": "recommendationText",
+                    "url": "recommendationUrl"
                 },
-                "detectorId": "detectorId",
-                "detectorName": "detectorName",
-                "findingId": "findingId",
-                "relatedVulnerabilities": [],
-                "severity": "severity",
-                "remediation": {
-                    "recommendation": {
-                        "text": "recommendationText",
-                        "url": "recommendationUrl"
-                    },
-                    "suggestedFixes": []
-                }
+                "suggestedFixes": []
             }
+        }
+    """.trimIndent()
+
+    private fun setupCodeScanFindings(filePath: Path) = """
+        [
+            ${setupCodeScanFinding(filePath, 1, 2)},
+            ${setupCodeScanFinding(filePath, 1, 2)}
+        ]
+    """
+
+    private fun setupCodeScanFindingsOutOfBounds(filePath: Path) = """
+        [
+            ${setupCodeScanFinding(filePath, 99999, 99999)}
         ]
     """
 
@@ -187,6 +177,11 @@ open class CodeWhispererCodeScanTestBase(projectRule: CodeInsightTestFixtureRule
             .responseMetadata(metadata)
             .build() as ListCodeScanFindingsResponse
 
+        fakeListCodeScanFindingsOutOfBoundsIndexResponse = ListCodeScanFindingsResponse.builder()
+            .codeScanFindings(setupCodeScanFindingsOutOfBounds(filePath))
+            .responseMetadata(metadata)
+            .build() as ListCodeScanFindingsResponse
+
         fakeGetCodeScanResponse = GetCodeScanResponse.builder()
             .status(CodeScanStatus.COMPLETED)
             .responseMetadata(metadata)