From 5a8efdfc973b56ddf84fb79029f7838644163820 Mon Sep 17 00:00:00 2001
From: Denis Mukhametianov <denis.mukhametianov@jetbrains.com>
Date: Sun, 5 May 2024 12:19:02 +0200
Subject: [PATCH] [spellchecker] Added SpellCheckerQuickFixFactory to allow
 custom spellchecking quick fixes in Rider

GitOrigin-RevId: b987bcb82e84f0596e57962df55bb868b92fe23b
---
 .../src/META-INF/SpellCheckerPlugin.xml       |  1 +
 .../quickfixes/SpellCheckerQuickFixFactory.kt | 39 +++++++++++++++++++
 2 files changed, 40 insertions(+)
 create mode 100644 spellchecker/src/com/intellij/spellchecker/quickfixes/SpellCheckerQuickFixFactory.kt

diff --git a/spellchecker/src/META-INF/SpellCheckerPlugin.xml b/spellchecker/src/META-INF/SpellCheckerPlugin.xml
index d9e1321cbcb0..571cd4c8db69 100644
--- a/spellchecker/src/META-INF/SpellCheckerPlugin.xml
+++ b/spellchecker/src/META-INF/SpellCheckerPlugin.xml
@@ -9,6 +9,7 @@
     <extensionPoint name="spellchecker.dictionary.checker" interface="com.intellij.spellchecker.dictionary.DictionaryChecker" dynamic="true"/>
     <extensionPoint name="spellchecker.builtInDictionariesProvider" interface="com.intellij.spellchecker.settings.BuiltInDictionariesProvider" dynamic="true"/>
     <extensionPoint name="spellchecker.dictionaryLayersProvider" interface="com.intellij.spellchecker.DictionaryLayersProvider" dynamic="true"/>
+    <extensionPoint name="spellchecker.quickFixFactory" interface="com.intellij.spellchecker.quickfixes.SpellCheckerQuickFixFactory" dynamic="true"/>
   </extensionPoints>
 
   <extensions defaultExtensionNs="com.intellij">
diff --git a/spellchecker/src/com/intellij/spellchecker/quickfixes/SpellCheckerQuickFixFactory.kt b/spellchecker/src/com/intellij/spellchecker/quickfixes/SpellCheckerQuickFixFactory.kt
new file mode 100644
index 000000000000..5bcb02293f91
--- /dev/null
+++ b/spellchecker/src/com/intellij/spellchecker/quickfixes/SpellCheckerQuickFixFactory.kt
@@ -0,0 +1,39 @@
+// Copyright 2000-2024 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
+package com.intellij.spellchecker.quickfixes
+
+import com.intellij.codeInspection.LocalQuickFix
+import com.intellij.openapi.extensions.ExtensionPointName
+import com.intellij.openapi.util.TextRange
+import com.intellij.psi.PsiElement
+import com.intellij.spellchecker.DictionaryLayer
+import org.jetbrains.annotations.ApiStatus
+
+abstract class SpellCheckerQuickFixFactory {
+  companion object {
+    val EP_NAME = ExtensionPointName.create<SpellCheckerQuickFixFactory>("com.intellij.spellchecker.quickFixFactory")
+
+    @JvmStatic
+    fun rename(element: PsiElement): LocalQuickFix {
+      return EP_NAME.extensionList.firstNotNullOfOrNull { it.createRename(element) } ?: RenameTo()
+    }
+
+    @JvmStatic
+    fun changeToVariants(element: PsiElement, rangeInElement: TextRange, word: String): List<LocalQuickFix> {
+      return EP_NAME.extensionList.firstNotNullOfOrNull { it.createChangeToVariantsFixes(element, rangeInElement, word) } ?: ChangeTo(word, element, rangeInElement).getAllAsFixes()
+    }
+
+    @JvmStatic
+    fun saveTo(element: PsiElement, rangeInElement: TextRange, word: String): LocalQuickFix {
+      return saveTo(element, rangeInElement, word, null)
+    }
+
+    @JvmStatic
+    fun saveTo(element: PsiElement, rangeInElement: TextRange, word: String, layer: DictionaryLayer?): LocalQuickFix {
+      return EP_NAME.extensionList.firstNotNullOfOrNull { it.createSaveToFix(element, rangeInElement, word, layer) } ?: SaveTo(word, layer)
+    }
+  }
+
+  open fun createRename(element: PsiElement): LocalQuickFix? = null
+  open fun createChangeToVariantsFixes(element: PsiElement, rangeInElement: TextRange, word: String): List<LocalQuickFix>? = null
+  open fun createSaveToFix(element: PsiElement, rangeInElement: TextRange, word: String, layer: DictionaryLayer?): LocalQuickFix? = null
+}
\ No newline at end of file