Introducing reformatting inside the injected DPL & JSON hosts

Author: dorian-krefft-dt

CHANGELOG.md

@@ -9,8 +9,7 @@
   - The support is also automatically added to the DQL `parse` command
   - Customizable syntax highlighting
   - Local code inspections with quick fixes if possible
-  - Customizable code style support — enforcing indents, line breaking and spaces around elements (only in separate
-    files)
+  - Customizable code style support — enforcing indents, line breaking and spaces around elements
   - Advanced code completion
   - Hover documentation
   - Useful intentions for managing DPL expressions

src/main/java/pl/thedeem/intellij/dpl/style/DQLPostFormatProcessor.java

@@ -0,0 +1,79 @@
+package pl.thedeem.intellij.dpl.style;
+
+import com.intellij.lang.injection.InjectedLanguageManager;
+import com.intellij.openapi.editor.Document;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.util.Pair;
+import com.intellij.openapi.util.TextRange;
+import com.intellij.psi.*;
+import com.intellij.psi.codeStyle.CodeStyleManager;
+import com.intellij.psi.codeStyle.CodeStyleSettings;
+import com.intellij.psi.impl.source.codeStyle.PostFormatProcessor;
+import com.intellij.psi.util.PsiTreeUtil;
+import org.jetbrains.annotations.NotNull;
+
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+
+public class DQLPostFormatProcessor implements PostFormatProcessor {
+
+    @Override
+    public @NotNull PsiElement processElement(@NotNull PsiElement element,
+                                              @NotNull CodeStyleSettings settings) {
+        return element;
+    }
+
+    @Override
+    public @NotNull TextRange processText(@NotNull PsiFile hostFile,
+                                          @NotNull TextRange range,
+                                          @NotNull CodeStyleSettings settings) {
+        Project project = hostFile.getProject();
+        InjectedLanguageManager injector = InjectedLanguageManager.getInstance(project);
+        CodeStyleManager styleManager = CodeStyleManager.getInstance(project);
+
+        Collection<PsiFile> injectedFiles = findInjectionHosts(hostFile, injector, range);
+        final PsiDocumentManager documentManager = PsiDocumentManager.getInstance(project);
+        if (!injectedFiles.isEmpty()) {
+            for (PsiFile host : injectedFiles) {
+                PsiFile updated = PsiFileFactory.getInstance(project).createFileFromText(host.getLanguage(), host.getText());
+                styleManager.reformat(updated);
+                String text = updated.getText();
+
+                Document d = documentManager.getDocument(host);
+                if (d != null) {
+                    d.setText(text);
+                }
+            }
+        }
+        return range;
+    }
+
+    private @NotNull Collection<PsiFile> findInjectionHosts(@NotNull PsiFile file, @NotNull InjectedLanguageManager injector, @NotNull TextRange range) {
+        PsiLanguageInjectionHost injectionHost = injector.getInjectionHost(file);
+        List<PsiFile> result = new ArrayList<>();
+        Collection<PsiLanguageInjectionHost> hosts;
+        if (injectionHost == null) {
+            hosts = PsiTreeUtil.findChildrenOfType(file, PsiLanguageInjectionHost.class);
+        } else {
+            hosts = List.of(injectionHost);
+        }
+        for (PsiLanguageInjectionHost host : hosts) {
+            List<Pair<PsiElement, TextRange>> files = injector.getInjectedPsiFiles(host);
+            if (files != null) {
+                for (Pair<PsiElement, TextRange> iFile : files) {
+                    if (iFile.getSecond().intersects(range)) {
+                        result.add(iFile.getFirst().getContainingFile());
+                    }
+                }
+            }
+        }
+
+        return result;
+    }
+
+    @Override
+    public boolean isWhitespaceOnly() {
+        return false;
+    }
+}

src/main/resources/META-INF/plugin.xml

@@ -367,6 +367,7 @@
                 implementation="pl.thedeem.intellij.dpl.style.DPLCodeStyleSettingsProvider"/>
         <langCodeStyleSettingsProvider
                 implementation="pl.thedeem.intellij.dpl.style.DPLLanguageCodeStyleSettingsProvider"/>
+        <postFormatProcessor implementation="pl.thedeem.intellij.dpl.style.DQLPostFormatProcessor"/>
 
         <!-- References -->
         <psi.referenceContributor