Remove too much coupling

Author: abashev

open-java-format-idea-plugin/src/main/java/com/palantir/javaformat/intellij/FormatterProvider.java

@@ -16,29 +16,31 @@
 
 package com.palantir.javaformat.intellij;
 
-import static com.intellij.formatting.service.FormattingService.EP_NAME;
-import static java.util.Optional.ofNullable;
-
 import com.github.benmanes.caffeine.cache.Caffeine;
 import com.github.benmanes.caffeine.cache.LoadingCache;
+import com.intellij.formatting.service.FormattingService;
 import com.intellij.openapi.extensions.PluginDescriptor;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.util.SystemInfo;
 import com.palantir.javaformat.bootstrap.BootstrappingFormatterService;
 import com.palantir.javaformat.bootstrap.NativeImageFormatterService;
 import com.palantir.javaformat.java.FormatterService;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 import java.io.IOException;
 import java.io.UncheckedIOException;
 import java.net.URI;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.List;
+import java.util.NoSuchElementException;
 import java.util.Objects;
 import java.util.Optional;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
+
+import static java.util.Optional.ofNullable;
 
 final class FormatterProvider {
     private static final Logger log = LoggerFactory.getLogger(FormatterProvider.class);
@@ -47,17 +49,6 @@ final class FormatterProvider {
     private final LoadingCache<FormatterCacheKey, Optional<FormatterService>> implementationCache =
             Caffeine.newBuilder().maximumSize(1).build(FormatterProvider::createFormatter);
 
-    /**
-     * The descriptor of this plugin: where its version and the directory of the bundled formatter come from. The
-     * platform hands it to the formatting service when it creates that service from plugin.xml (PluginAware), and the
-     * extension point finds the service by class. Every way of looking a plugin up by id or by class became
-     * {@code @ApiStatus.Internal} in 2026.2; PluginAware and the extension point are public API in every supported IDE.
-     */
-    static Optional<PluginDescriptor> getPluginDescriptor() {
-        return ofNullable(EP_NAME.findExtension(PalantirJavaFormatFormattingService.class))
-                .map(PalantirJavaFormatFormattingService::getPluginDescriptor);
-    }
-
     Optional<FormatterService> get(Project project, PalantirJavaFormatSettings settings) {
         return implementationCache.get(new FormatterCacheKey(
                 project,
@@ -91,9 +82,10 @@ private static List<Path> getProvidedImplementationUrls(List<URI> implementation
     @SuppressWarnings("for-rollout:Slf4jLogsafeArgs")
     private static List<Path> getBundledImplementationUrls() {
         // Load from the jars bundled with the plugin.
-        Path implDir = getPluginDescriptor()
+        Path implDir = ofNullable(FormattingService.EP_NAME.findExtension(PalantirJavaFormatFormattingService.class))
+                .flatMap(PalantirJavaFormatFormattingService::getPluginDescriptor)
                 .map(PluginDescriptor::getPluginPath)
-                .orElseThrow()
+                .orElseThrow(() -> new NoSuchElementException("The platform has not set the plugin descriptor"))
                 .resolve("impl");
 
         log.debug("Using open-java-format implementation bundled with plugin: {}", implDir);

open-java-format-idea-plugin/src/main/java/com/palantir/javaformat/intellij/PalantirJavaFormatFormattingService.java

@@ -16,9 +16,6 @@
 
 package com.palantir.javaformat.intellij;
 
-import static java.util.Comparator.comparing;
-
-import com.google.common.base.Preconditions;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Range;
 import com.intellij.formatting.service.AsyncDocumentFormattingService;
@@ -36,13 +33,17 @@
 import com.palantir.javaformat.java.FormatterException;
 import com.palantir.javaformat.java.FormatterService;
 import com.palantir.javaformat.java.Replacement;
+import org.jetbrains.annotations.NotNull;
+
+import javax.annotation.Nullable;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
 import java.util.Optional;
 import java.util.Set;
-import javax.annotation.Nullable;
-import org.jetbrains.annotations.NotNull;
+
+import static java.util.Comparator.comparing;
+import static java.util.Optional.ofNullable;
 
 class PalantirJavaFormatFormattingService extends AsyncDocumentFormattingService implements PluginAware {
     private static final Logger logger = Logger.getInstance(PalantirJavaFormatFormattingService.class);
@@ -59,8 +60,8 @@ public void setPluginDescriptor(@NotNull PluginDescriptor pluginDescriptor) {
         this.pluginDescriptor = pluginDescriptor;
     }
 
-    PluginDescriptor getPluginDescriptor() {
-        return Preconditions.checkNotNull(pluginDescriptor, "The platform has not set the plugin descriptor");
+    Optional<PluginDescriptor> getPluginDescriptor() {
+        return ofNullable(pluginDescriptor);
     }
 
     @Override
@@ -127,7 +128,7 @@ public void run() {
                 if (logger.isDebugEnabled()) {
                     logger.debug(String.format(
                             "Received request to format file=%s, length=%s with ranges=%s",
-                            Optional.ofNullable(request.getIOFile())
+                            ofNullable(request.getIOFile())
                                     .map(file -> file.toPath().toString())
                                     .orElse("null"),
                             preFormatText.length(),

open-java-format-idea-plugin/src/main/java/com/palantir/javaformat/intellij/PalantirJavaFormatSettings.java

@@ -16,15 +16,16 @@
 
 package com.palantir.javaformat.intellij;
 
-import static com.palantir.javaformat.intellij.FormatterProvider.getPluginDescriptor;
-
+import com.intellij.formatting.service.FormattingService;
 import com.intellij.openapi.components.PersistentStateComponent;
 import com.intellij.openapi.components.State;
 import com.intellij.openapi.components.Storage;
 import com.intellij.openapi.extensions.PluginDescriptor;
 import com.intellij.openapi.project.Project;
 import com.palantir.javaformat.java.FormatterService;
 import com.palantir.javaformat.java.JavaFormatterOptions;
+
+import javax.annotation.Nullable;
 import java.io.IOException;
 import java.io.UncheckedIOException;
 import java.net.URI;
@@ -33,7 +34,8 @@
 import java.util.jar.JarFile;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
-import javax.annotation.Nullable;
+
+import static java.util.Optional.ofNullable;
 
 @SuppressWarnings("for-rollout:SameNameButDifferent")
 @State(
@@ -112,7 +114,9 @@ boolean injectedVersionIsOutdated() {
     }
 
     Optional<String> getImplementationVersion() {
-        return getPluginDescriptor().map(PluginDescriptor::getVersion);
+        return ofNullable(FormattingService.EP_NAME.findExtension(PalantirJavaFormatFormattingService.class))
+                .flatMap(PalantirJavaFormatFormattingService::getPluginDescriptor)
+                .map(PluginDescriptor::getVersion);
     }
 
     Optional<String> computeFormatterVersion() {

open-java-format-idea-plugin/src/test/java/com/palantir/javaformat/intellij/PalantirJavaFormatFormattingServiceTest.java

@@ -16,10 +16,6 @@
 
 package com.palantir.javaformat.intellij;
 
-import static com.palantir.javaformat.intellij.FormatterProvider.getPluginDescriptor;
-import static org.assertj.core.api.Assertions.assertThat;
-
-import com.google.common.collect.ImmutableList;
 import com.intellij.codeInsight.actions.ReformatCodeProcessor;
 import com.intellij.formatting.service.AsyncFormattingRequest;
 import com.intellij.formatting.service.FormattingService;
@@ -40,16 +36,20 @@
 import com.intellij.testFramework.fixtures.JavaTestFixtureFactory;
 import com.intellij.testFramework.fixtures.TestFixtureBuilder;
 import com.palantir.javaformat.intellij.PalantirJavaFormatSettings.State;
+import org.jetbrains.annotations.NotNull;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
 import java.io.File;
 import java.io.IOException;
 import java.io.UncheckedIOException;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
-import org.jetbrains.annotations.NotNull;
-import org.junit.jupiter.api.AfterEach;
-import org.junit.jupiter.api.BeforeEach;
-import org.junit.jupiter.api.Test;
+import java.util.List;
+
+import static org.assertj.core.api.Assertions.assertThat;
 
 public class PalantirJavaFormatFormattingServiceTest {
     private JavaCodeInsightTestFixture fixture;
@@ -64,13 +64,12 @@ public void setUp() throws Exception {
         fixture.setUp();
 
         delegatingFormatter = new DelegatingFormatter();
-        // Only the service the platform creates from plugin.xml is handed the plugin descriptor. This one is created
-        // here, so it takes the descriptor from that service before masking it.
-        delegatingFormatter.setPluginDescriptor(getPluginDescriptor().orElseThrow());
+
         ExtensionTestUtil.maskExtensions(
-                FormattingService.EP_NAME, ImmutableList.of(delegatingFormatter), fixture.getProjectDisposable());
+                FormattingService.EP_NAME, List.of(delegatingFormatter), fixture.getProjectDisposable());
 
         settings = PalantirJavaFormatSettings.getInstance(fixture.getProject());
+
         State resetState = new State();
         resetState.setEnabled("true");
         settings.loadState(resetState);
@@ -130,6 +129,13 @@ private PsiFile createPsiFile(String path, String... contents) throws IOExceptio
     }
 
     private static final class DelegatingFormatter extends PalantirJavaFormatFormattingService {
+        @SuppressWarnings("DataFlowIssue")
+        public DelegatingFormatter() {
+            this.setPluginDescriptor(FormattingService.EP_NAME
+                    .findExtension(PalantirJavaFormatFormattingService.class)
+                    .getPluginDescriptor()
+                    .orElseThrow());
+        }
 
         private boolean invoked = false;