262 - Support auto code formatting (#292)

src/main/java/ee/carlrobert/codegpt/settings/configuration/ConfigurationComponent.java

@@ -43,6 +43,7 @@ public class ConfigurationComponent {
   private final JBTable table;
   private final JBCheckBox openNewTabCheckBox;
   private final JBCheckBox methodNameGenerationCheckBox;
+  private final JBCheckBox autoFormattingCheckBox;
   private final JTextArea systemPromptTextArea;
   private final IntegerField maxTokensField;
   private final JBTextField temperatureField;
@@ -104,12 +105,16 @@ public class ConfigurationComponent {
     methodNameGenerationCheckBox = new JBCheckBox(
         CodeGPTBundle.get("configurationConfigurable.enableMethodNameGeneration.label"),
         configuration.isMethodRefactoringEnabled());
+    autoFormattingCheckBox = new JBCheckBox(
+        CodeGPTBundle.get("configurationConfigurable.autoFormatting.label"),
+        configuration.isAutoFormattingEnabled());
 
     mainPanel = FormBuilder.createFormBuilder()
         .addComponent(tablePanel)
         .addVerticalGap(4)
         .addComponent(openNewTabCheckBox)
         .addComponent(methodNameGenerationCheckBox)
+        .addComponent(autoFormattingCheckBox)
         .addVerticalGap(4)
         .addComponent(new TitledSeparator(
             CodeGPTBundle.get("configurationConfigurable.section.assistant.title")))
@@ -267,6 +272,14 @@ public class ConfigurationComponent {
     methodNameGenerationCheckBox.setSelected(disableMethodNameGeneration);
   }
 
+  public boolean isAutoFormattingEnabled() {
+    return autoFormattingCheckBox.isSelected();
+  }
+
+  public void setAutoFormattingEnabled(boolean enabled) {
+    autoFormattingCheckBox.setSelected(enabled);
+  }
+
   class RevertToDefaultsActionButton extends AnActionButton {
 
     RevertToDefaultsActionButton() {

src/main/java/ee/carlrobert/codegpt/settings/configuration/ConfigurationConfigurable.java

@@ -40,7 +40,9 @@ public class ConfigurationConfigurable implements Configurable {
         || configurationComponent.isCreateNewChatOnEachAction()
         != configuration.isCreateNewChatOnEachAction()
         || configurationComponent.isMethodNameGenerationEnabled()
-        != configuration.isMethodRefactoringEnabled();
+        != configuration.isMethodRefactoringEnabled()
+        || configurationComponent.isAutoFormattingEnabled()
+        != configuration.isAutoFormattingEnabled();
   }
 
   @Override
@@ -54,6 +56,7 @@ public class ConfigurationConfigurable implements Configurable {
         configurationComponent.isCreateNewChatOnEachAction());
     configuration.setMethodNameGenerationEnabled(
         configurationComponent.isMethodNameGenerationEnabled());
+    configuration.setAutoFormattingEnabled(configurationComponent.isAutoFormattingEnabled());
     EditorActionsUtil.refreshActions();
   }
 
@@ -68,6 +71,7 @@ public class ConfigurationConfigurable implements Configurable {
         configuration.isCreateNewChatOnEachAction());
     configurationComponent.setDisableMethodNameGeneration(
         configuration.isMethodRefactoringEnabled());
+    configurationComponent.setAutoFormattingEnabled(configuration.isAutoFormattingEnabled());
     EditorActionsUtil.refreshActions();
   }
 

src/main/java/ee/carlrobert/codegpt/settings/configuration/ConfigurationState.java

@@ -23,6 +23,7 @@ public class ConfigurationState implements PersistentStateComponent<Configuratio
   private boolean createNewChatOnEachAction;
   private boolean ignoreGitCommitTokenLimit;
   private boolean methodNameGenerationEnabled = true;
+  private boolean autoFormattingEnabled = true;
   private Map<String, String> tableData = EditorActionsUtil.DEFAULT_ACTIONS;
 
   public static ConfigurationState getInstance() {
@@ -95,4 +96,12 @@ public class ConfigurationState implements PersistentStateComponent<Configuratio
   public void setMethodNameGenerationEnabled(boolean methodNameGenerationEnabled) {
     this.methodNameGenerationEnabled = methodNameGenerationEnabled;
   }
+
+  public boolean isAutoFormattingEnabled() {
+    return autoFormattingEnabled;
+  }
+
+  public void setAutoFormattingEnabled(boolean autoFormattingEnabled) {
+    this.autoFormattingEnabled = autoFormattingEnabled;
+  }
 }

src/main/java/ee/carlrobert/codegpt/util/EditorUtil.java

@@ -14,7 +14,9 @@ import com.intellij.openapi.fileEditor.FileDocumentManager;
 import com.intellij.openapi.fileEditor.FileEditorManager;
 import com.intellij.openapi.project.Project;
 import com.intellij.psi.PsiDocumentManager;
+import com.intellij.psi.codeStyle.CodeStyleManager;
 import com.intellij.testFramework.LightVirtualFile;
+import ee.carlrobert.codegpt.settings.configuration.ConfigurationState;
 import java.time.LocalDateTime;
 import java.time.format.DateTimeFormatter;
 import org.jetbrains.annotations.NotNull;
@@ -88,17 +90,36 @@ public final class EditorUtil {
           var editor = getSelectedEditor(project);
           if (editor != null) {
             var selectionModel = editor.getSelectionModel();
-            editor.getDocument()
-                .replaceString(
-                    selectionModel.getSelectionStart(),
-                    selectionModel.getSelectionEnd(),
-                    text);
+            int startOffset = selectionModel.getSelectionStart();
+            int endOffset = selectionModel.getSelectionEnd();
+            var document = editor.getDocument();
+
+            document.replaceString(startOffset, endOffset, text);
+
+            if (ConfigurationState.getInstance().isAutoFormattingEnabled()) {
+              reformatDocument(project, document, startOffset, endOffset);
+            }
+
             editor.getContentComponent().requestFocus();
             selectionModel.removeSelection();
           }
         })));
   }
 
+  private static void reformatDocument(
+      @NotNull Project project,
+      @NotNull Document document,
+      int startOffset,
+      int endOffset) {
+    var psiDocumentManager = PsiDocumentManager.getInstance(project);
+    psiDocumentManager.commitDocument(document);
+    var psiFile = psiDocumentManager.getPsiFile(document);
+    if (psiFile != null) {
+      CodeStyleManager.getInstance(project)
+          .reformatText(psiFile, startOffset, endOffset);
+    }
+  }
+
   public static void disableHighlighting(@NotNull Project project, Document document) {
     var psiFile = PsiDocumentManager.getInstance(project).getPsiFile(document);
     if (psiFile != null) {

src/main/resources/messages/codegpt.properties

@@ -66,8 +66,9 @@ configurationConfigurable.table.header.actionColumnLabel=Action
 configurationConfigurable.table.header.promptColumnLabel=Prompt
 configurationConfigurable.table.action.revertToDefaults.text=Revert to Defaults
 configurationConfigurable.table.action.addKeymap.text=Add Shortcut
-configurationConfigurable.openNewTabCheckBox.label=Open new tab on each action
-configurationConfigurable.enableMethodNameGeneration.label=Enable automatic method name generation
+configurationConfigurable.openNewTabCheckBox.label=Open a new chat on each action
+configurationConfigurable.enableMethodNameGeneration.label=Enable method name lookup suggestions
+configurationConfigurable.autoFormatting.label=Enable automatic code formatting
 configurationConfigurable.section.assistant.title=Assistant Configuration
 configurationConfigurable.section.assistant.systemPromptField.label=System prompt:
 configurationConfigurable.section.assistant.systemPromptField.comment=The system message helps to set the behaviour of the assistant