Skip to content

Commit

Permalink
feat: init save and format code
Browse files Browse the repository at this point in the history
  • Loading branch information
phodal committed Jul 17, 2023
1 parent a07a56e commit 1e34279
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 1 deletion.
18 changes: 18 additions & 0 deletions src/main/kotlin/cc/unitmesh/devti/IdeHelper.kt
@@ -0,0 +1,18 @@
package cc.unitmesh.devti

import com.intellij.openapi.editor.Document
import com.intellij.openapi.project.Project
import com.intellij.openapi.util.TextRange
import com.intellij.psi.PsiDocumentManager
import com.intellij.psi.codeStyle.CodeStyleManager

object IdeHelper {
fun formatCode(project: Project, document: Document, textRange: TextRange, suggestion: String) {
PsiDocumentManager.getInstance(project).commitDocument(document)
val psiFile = PsiDocumentManager.getInstance(project).getPsiFile(document) ?: return
val reformatRange = TextRange(textRange.startOffset, textRange.startOffset + suggestion.length)
CodeStyleManager.getInstance(project).reformatText(psiFile, listOf(reformatRange))
}


}
@@ -1,14 +1,19 @@
package cc.unitmesh.devti.actions.chat

import cc.unitmesh.devti.gui.DevtiFlowToolWindowFactory
import cc.unitmesh.devti.java.prompt.JavaPromptFormatter
import cc.unitmesh.devti.gui.chat.ChatBotActionType
import cc.unitmesh.devti.gui.chat.ChatCodingComponent
import cc.unitmesh.devti.gui.chat.ChatCodingService
import cc.unitmesh.devti.java.prompt.JavaPromptFormatter
import com.intellij.openapi.actionSystem.AnAction
import com.intellij.openapi.actionSystem.AnActionEvent
import com.intellij.openapi.actionSystem.CommonDataKeys
import com.intellij.openapi.editor.Document
import com.intellij.openapi.project.Project
import com.intellij.openapi.util.TextRange
import com.intellij.openapi.wm.ToolWindowManager
import com.intellij.psi.PsiDocumentManager
import com.intellij.psi.codeStyle.CodeStyleManager

abstract class ChatBaseAction : AnAction() {
override fun actionPerformed(event: AnActionEvent) {
Expand Down

0 comments on commit 1e34279

Please sign in to comment.