@@ -76,12 +76,16 @@ class AgentStateService(val project: Project) {
76
76
val beforePath = patch.beforeName
77
77
val afterPath = patch.afterName
78
78
79
- val fileStatus = if (patch.isNewFile) {
80
- FileStatus .ADDED
81
- } else if (patch.isDeletedFile) {
82
- FileStatus .DELETED
83
- } else {
84
- FileStatus .MODIFIED
79
+ val fileStatus = when {
80
+ patch.isNewFile -> {
81
+ FileStatus .ADDED
82
+ }
83
+ patch.isDeletedFile -> {
84
+ FileStatus .DELETED
85
+ }
86
+ else -> {
87
+ FileStatus .MODIFIED
88
+ }
85
89
}
86
90
87
91
val beforeFilePath = VcsUtil .getFilePath(getAbsolutePath(baseDir, beforePath), false )
@@ -105,21 +109,24 @@ class AgentStateService(val project: Project) {
105
109
override fun getVirtualFile (): VirtualFile ? = afterFilePath.virtualFile
106
110
override fun getFile (): FilePath = afterFilePath
107
111
override fun getContent (): @NonNls String? {
108
- if (patch.isNewFile) {
109
- return patch.singleHunkPatchText
110
- }
111
-
112
- if (patch.isDeletedFile) {
113
- return null
114
- }
115
-
116
- val localContent: String = loadLocalContent()
117
- val appliedPatch = GenericPatchApplier .apply (localContent, patch.hunks)
118
- if (appliedPatch != null ) {
119
- return appliedPatch.patchedText
112
+ when {
113
+ patch.isNewFile -> {
114
+ return patch.singleHunkPatchText
115
+ }
116
+ patch.isDeletedFile -> {
117
+ return null
118
+ }
119
+ else -> {
120
+ val localContent: String = loadLocalContent()
121
+ val appliedPatch = GenericPatchApplier .apply (localContent, patch.hunks)
122
+ // / sometimes llm will return a wrong patch which the content is not correct
123
+ if (appliedPatch != null ) {
124
+ return appliedPatch.patchedText
125
+ }
126
+
127
+ return patch.singleHunkPatchText
128
+ }
120
129
}
121
- // / sometimes llm will return a wrong patch which the content is not correct
122
- return patch.singleHunkPatchText
123
130
}
124
131
125
132
@Throws(VcsException ::class )
0 commit comments