Skip to content

Commit ecd046d

Browse files
committed
Refactoring code
1 parent c1d42ff commit ecd046d

File tree

1 file changed

+8
-9
lines changed

1 file changed

+8
-9
lines changed

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

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -391,20 +391,19 @@ class CodeWhispererCodeScanEditorMouseMotionListener(private val project: Projec
391391
private val LOG = getLogger<CodeWhispererCodeScanEditorMouseMotionListener>()
392392
}
393393

394+
private fun extractChanges(issue: CodeWhispererCodeScanIssue): Pair<Int, List<String>> {
395+
val codeLines = issue.suggestedFixes[0].code.split("\n")
396+
val linesToDelete = codeLines.count { it.startsWith("-") }
397+
val linesToInsert = codeLines.filter { it.startsWith("+") }.map { it.removePrefix("+") }
398+
return Pair(linesToDelete, linesToInsert)
399+
}
400+
394401
private fun handleApplyFix(issue: CodeWhispererCodeScanIssue) {
395402
try {
396403
WriteCommandAction.runWriteCommandAction(issue.project) {
397404
val document = FileDocumentManager.getInstance().getDocument(issue.file) ?: return@runWriteCommandAction
398405

399-
val linesToDelete = issue.suggestedFixes[0].code
400-
.split("\n")
401-
.count { it.startsWith("-") }
402-
403-
val linesToInsert = issue.suggestedFixes[0].code
404-
.split("\n")
405-
.filter { it.startsWith("+") }
406-
.map { it.removePrefix("+") }
407-
406+
val (linesToDelete, linesToInsert) = extractChanges(issue)
408407
if (document != null) {
409408
val startLineOffset = document.getLineStartOffset(issue.startLine - 1)
410409
val endLineOffset = document.getLineEndOffset(issue.startLine + linesToDelete - 2)

0 commit comments

Comments
 (0)