From c13e488ab836ad27acd058ea6f84a5b3a6419aad Mon Sep 17 00:00:00 2001
From: chrisly42 <chrisly@platon42.de>
Date: Mon, 19 Jul 2021 13:07:08 +0200
Subject: [PATCH] Added M68kSymbolReferenceElementManipulator to allow
 refactoring of local labels.

---
 .../plugins/m68k/psi/M68kElementFactory.kt      |  5 +++++
 .../M68kSymbolReferenceElementManipulator.kt    | 17 +++++++++++++++++
 src/main/resources/META-INF/plugin.xml          |  5 ++++-
 3 files changed, 26 insertions(+), 1 deletion(-)
 create mode 100644 src/main/java/de/platon42/intellij/plugins/m68k/refs/M68kSymbolReferenceElementManipulator.kt

diff --git a/src/main/java/de/platon42/intellij/plugins/m68k/psi/M68kElementFactory.kt b/src/main/java/de/platon42/intellij/plugins/m68k/psi/M68kElementFactory.kt
index f3bc6ea..16a5cbc 100644
--- a/src/main/java/de/platon42/intellij/plugins/m68k/psi/M68kElementFactory.kt
+++ b/src/main/java/de/platon42/intellij/plugins/m68k/psi/M68kElementFactory.kt
@@ -22,6 +22,11 @@ object M68kElementFactory {
         return PsiTreeUtil.findChildOfType(file, M68kSymbolDefinition::class.java)!!
     }
 
+    fun createSymbolReference(project: Project, symbol: String): M68kSymbolReference {
+        val file = createFile(project, " bra $symbol\n")
+        return PsiTreeUtil.findChildOfType(file, M68kSymbolReference::class.java)!!
+    }
+
     fun createFile(project: Project, content: String): M68kFile {
         return PsiFileFactory.getInstance(project).createFileFromText("dummy.m68k", INSTANCE, content) as M68kFile
     }
diff --git a/src/main/java/de/platon42/intellij/plugins/m68k/refs/M68kSymbolReferenceElementManipulator.kt b/src/main/java/de/platon42/intellij/plugins/m68k/refs/M68kSymbolReferenceElementManipulator.kt
new file mode 100644
index 0000000..25fa71a
--- /dev/null
+++ b/src/main/java/de/platon42/intellij/plugins/m68k/refs/M68kSymbolReferenceElementManipulator.kt
@@ -0,0 +1,17 @@
+package de.platon42.intellij.plugins.m68k.refs
+
+import com.intellij.openapi.util.TextRange
+import com.intellij.psi.AbstractElementManipulator
+import com.intellij.util.IncorrectOperationException
+import de.platon42.intellij.plugins.m68k.psi.M68kElementFactory
+import de.platon42.intellij.plugins.m68k.psi.M68kSymbolReference
+
+class M68kSymbolReferenceElementManipulator : AbstractElementManipulator<M68kSymbolReference>() {
+
+    @Throws(IncorrectOperationException::class)
+    override fun handleContentChange(element: M68kSymbolReference, range: TextRange, newContent: String): M68kSymbolReference {
+        val newSymbolReference = M68kElementFactory.createSymbolReference(element.project, newContent)
+        element.firstChild.replace(newSymbolReference.firstChild)
+        return element
+    }
+}
\ No newline at end of file
diff --git a/src/main/resources/META-INF/plugin.xml b/src/main/resources/META-INF/plugin.xml
index f74bc92..26c8df2 100644
--- a/src/main/resources/META-INF/plugin.xml
+++ b/src/main/resources/META-INF/plugin.xml
@@ -24,8 +24,11 @@
         <completion.contributor language="M68k" implementationClass="de.platon42.intellij.plugins.m68k.asm.M68kMnemonicCompletionContributor"/>
         <lang.braceMatcher language="M68k" implementationClass="de.platon42.intellij.plugins.m68k.syntax.M68kPairedBraceMatcher"/>
         <lang.quoteHandler language="M68k" implementationClass="de.platon42.intellij.plugins.m68k.M68kStringQuoteHandler"/>
-        <psi.referenceContributor implementation="de.platon42.intellij.plugins.m68k.refs.M68kReferenceContributor"/>
         <lang.findUsagesProvider language="M68k" implementationClass="de.platon42.intellij.plugins.m68k.scanner.M68kFindUsagesProvider"/>
+        <lang.elementManipulator forClass="de.platon42.intellij.plugins.m68k.psi.M68kSymbolReference"
+                                 implementationClass="de.platon42.intellij.plugins.m68k.refs.M68kSymbolReferenceElementManipulator"/>
+
+        <psi.referenceContributor implementation="de.platon42.intellij.plugins.m68k.refs.M68kReferenceContributor"/>
         <renameInputValidator implementation="de.platon42.intellij.plugins.m68k.psi.M68kRenameInputValidator"/>
     </extensions>