1 files changed, 49 insertions, 0 deletions
diff --git a/plugins/github/src/org/jetbrains/plugins/github/pullrequest/comment/GHSuggestedChange.kt b/plugins/github/src/org/jetbrains/plugins/github/pullrequest/comment/GHSuggestedChange.kt
new file mode 100644
index 000000000000..d172a62413a0
--- /dev/null
+++ b/plugins/github/src/org/jetbrains/plugins/github/pullrequest/comment/GHSuggestedChange.kt
@@ -0,0 +1,49 @@
+// Copyright 2000-2022 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
+package org.jetbrains.plugins.github.pullrequest.comment
+
+import com.intellij.openapi.diff.impl.patch.PatchHunk
+import com.intellij.openapi.diff.impl.patch.PatchLine
+import com.intellij.openapi.diff.impl.patch.PatchReader
+import org.jetbrains.plugins.github.util.GHPatchHunkUtil
+
+data class GHSuggestedChange(
+  val commentBody: String,
+  val patchHunk: PatchHunk,
+  val filePath: String,
+  val startLine: Int,
+  val endLine: Int
+) {
+  fun cutContextContent(): List<String> = patchHunk.lines
+    .filter { it.type != PatchLine.Type.REMOVE }
+    .dropLast(endLine - startLine + 1)
+    .takeLast(3)
+    .map { it.text }
+
+  fun cutChangedContent(): List<String> = patchHunk.lines
+    .filter { it.type != PatchLine.Type.REMOVE }
+    .takeLast(endLine - startLine + 1)
+    .map { it.text }
+
+  fun cutSuggestedChangeContent(): List<String> {
+    return commentBody.lines()
+      .dropWhile { !it.startsWith("```suggestion") }
+      .drop(1)
+      .takeWhile { !it.startsWith("```") }
+  }
+
+  companion object {
+    fun create(commentBody: String, diffHunk: String, filePath: String, startLine: Int, endLine: Int): GHSuggestedChange {
+      val patchHunk = parseDiffHunk(diffHunk, filePath)
+
+      return GHSuggestedChange(commentBody, patchHunk, filePath, startLine - 1, endLine - 1)
+    }
+
+    fun containsSuggestedChange(markdownText: String): Boolean = markdownText.lines().any { it.startsWith("```suggestion") }
+
+    private fun parseDiffHunk(diffHunk: String, filePath: String): PatchHunk {
+      val patchReader = PatchReader(GHPatchHunkUtil.createPatchFromHunk(filePath, diffHunk))
+      patchReader.readTextPatches()
+      return patchReader.textPatches[0].hunks.lastOrNull() ?: PatchHunk(0, 0, 0, 0)
+    }
+  }
+}
+\ No newline at end of file