From 9a2a18ed3e6e708e7a3d9df3a052ebd3463bc104 Mon Sep 17 00:00:00 2001
From: Ilia Permiashkin <ilia.permiashkin@jetbrains.com>
Date: Thu, 13 Nov 2025 15:42:47 +0000
Subject: [PATCH] [grazie] IJPL-218322 Cache results of
 `JsonSchemaSpellcheckerClient`

Merge-request: IJ-MR-182341
Merged-by: Ilia Permiashkin <ilia.permiashkin@jetbrains.com>

GitOrigin-RevId: 57b69cac473645ff0b26e1d52037f78d42a25e75
---
 .../json/JsonSchemaSpellcheckerClient.java    | 40 ++++++++++++-------
 .../inspection/grammar/GrazieInspection.kt    |  2 +-
 2 files changed, 26 insertions(+), 16 deletions(-)

diff --git a/json/backend/src/com/intellij/json/JsonSchemaSpellcheckerClient.java b/json/backend/src/com/intellij/json/JsonSchemaSpellcheckerClient.java
index 3c991d61cf76..3766f0050ad6 100644
--- a/json/backend/src/com/intellij/json/JsonSchemaSpellcheckerClient.java
+++ b/json/backend/src/com/intellij/json/JsonSchemaSpellcheckerClient.java
@@ -8,6 +8,7 @@ import com.intellij.openapi.project.Project;
 import com.intellij.openapi.util.text.StringUtil;
 import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.psi.PsiElement;
+import com.intellij.psi.util.CachedValuesManager;
 import com.intellij.psi.util.PsiUtilCore;
 import com.intellij.util.ObjectUtils;
 import com.intellij.util.ThreeState;
@@ -29,35 +30,44 @@ public abstract class JsonSchemaSpellcheckerClient {
   protected abstract @Nullable String getValue();
 
   public boolean matchesNameFromSchema() {
-    final VirtualFile file = PsiUtilCore.getVirtualFile(getElement());
-    if (file == null) return false;
+    PsiElement element = getElement();
+    return CachedValuesManager.getProjectPsiDependentCache(element, this::matchesNameFromSchema);
+  }
 
-    Project project = getElement().getProject();
-    final JsonSchemaService service = JsonSchemaService.Impl.get(project);
-    if (!service.isApplicableToFile(file)) return false;
-    final JsonSchemaObject rootSchema = service.getSchemaObject(getElement().getContainingFile());
-    if (rootSchema == null) return false;
-    if (isXIntellijInjection(service, rootSchema)) return true;
+  private boolean matchesNameFromSchema(PsiElement element) {
+    VirtualFile file = PsiUtilCore.getVirtualFile(element);
+    if (file == null) return false;
 
     String value = getValue();
     if (StringUtil.isEmpty(value)) return false;
 
-    JsonLikePsiWalker walker = JsonLikePsiWalker.getWalker(getElement(), rootSchema);
+    Project project = element.getProject();
+    JsonSchemaService service = JsonSchemaService.Impl.get(project);
+    if (!service.isApplicableToFile(file)) return false;
+    JsonSchemaObject rootSchema = service.getSchemaObject(element.getContainingFile());
+    if (rootSchema == null) return false;
+    if (isXIntellijInjection(service, rootSchema)) return true;
+
+    JsonLikePsiWalker walker = JsonLikePsiWalker.getWalker(element, rootSchema);
     if (walker == null) return false;
-    final PsiElement checkable = walker.findElementToCheck(getElement());
+    PsiElement checkable = walker.findElementToCheck(element);
     if (checkable == null) return false;
-    final ThreeState isName = walker.isName(checkable);
-    final JsonPointerPosition position = walker.findPosition(checkable, isName == ThreeState.NO);
+    ThreeState isName = walker.isName(checkable);
+    JsonPointerPosition position = walker.findPosition(checkable, isName == ThreeState.NO);
     if (position == null || position.isEmpty() && isName == ThreeState.NO) return false;
 
-    final Collection<JsonSchemaObject> schemas = new JsonSchemaResolver(project, rootSchema, position, walker.createValueAdapter(checkable)).resolve();
+    Collection<JsonSchemaObject> schemas =
+      new JsonSchemaResolver(project, rootSchema, position, walker.createValueAdapter(checkable)).resolve();
     if (schemas.isEmpty()) return false;
 
-    return schemas.stream().anyMatch(s -> {
+    return ContainerUtil.exists(schemas, s -> {
       if (s.getPropertyByName(value) != null || s.getMatchingPatternPropertySchema(value) != null) {
         return true;
       }
-      return ContainerUtil.notNullize(s.getEnum()).stream().anyMatch(e -> e instanceof String && StringUtil.unquoteString((String)e).equals(value));
+      return ContainerUtil.exists(
+        ContainerUtil.notNullize(s.getEnum()),
+        e -> e instanceof String && StringUtil.unquoteString((String)e).equals(value)
+      );
     });
   }
 
diff --git a/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/inspection/grammar/GrazieInspection.kt b/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/inspection/grammar/GrazieInspection.kt
index ee7eede3f5d4..10f051937a63 100644
--- a/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/inspection/grammar/GrazieInspection.kt
+++ b/plugins/grazie/src/main/kotlin/com/intellij/grazie/ide/inspection/grammar/GrazieInspection.kt
@@ -135,7 +135,7 @@ class GrazieInspection : LocalInspectionTool(), DumbAware {
         val contents = findAllTextContents(file.viewProvider, TextContent.TextDomain.ALL)
         logger<GrazieInspection>().debug("Evaluating text length of: ${TextContentRelatedData(file, contents)}")
         val length = contents.asSequence().filter { it.domain in checkedDomains }.sumOf { it.length }
-        CachedValueProvider.Result.create(length > MAX_TEXT_LENGTH_IN_FILE, service<GrazieConfig>())
+        CachedValueProvider.Result.create(length > MAX_TEXT_LENGTH_IN_FILE, service<GrazieConfig>(), file)
       }
     }