From 0181c4ea8d15511c71384aab55803cc105f8b16c Mon Sep 17 00:00:00 2001 From: Johannes Mey <johannes.mey@tu-dresden.de> Date: Tue, 5 Jan 2021 10:39:17 +0100 Subject: [PATCH] fix injection target --- .../jastadd/tooling/aspect/JavaBlockLanguageInjector.java | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/main/java/org/jastadd/tooling/aspect/JavaBlockLanguageInjector.java b/src/main/java/org/jastadd/tooling/aspect/JavaBlockLanguageInjector.java index 298ea59..57c7317 100644 --- a/src/main/java/org/jastadd/tooling/aspect/JavaBlockLanguageInjector.java +++ b/src/main/java/org/jastadd/tooling/aspect/JavaBlockLanguageInjector.java @@ -6,6 +6,7 @@ import com.intellij.openapi.util.TextRange; import com.intellij.psi.InjectedLanguagePlaces; import com.intellij.psi.LanguageInjector; import com.intellij.psi.PsiLanguageInjectionHost; +import org.jastadd.tooling.aspect.psi.JastAddAspectJavaBlock; import org.jetbrains.annotations.NotNull; public class JavaBlockLanguageInjector implements LanguageInjector { @@ -20,6 +21,9 @@ public class JavaBlockLanguageInjector implements LanguageInjector { */ @Override public void getLanguagesToInject(@NotNull PsiLanguageInjectionHost host, @NotNull InjectedLanguagePlaces injectionPlacesRegistrar) { - injectionPlacesRegistrar.addPlace(JavaLanguage.INSTANCE, new TextRange(0, host.getTextLength()), "class X { public void m(){", "}"); + if (host.isValidHost() && host instanceof JastAddAspectJavaBlock) { + System.out.println(host.getText()); + injectionPlacesRegistrar.addPlace(JavaLanguage.INSTANCE, new TextRange(0, host.getTextLength()), "class X { public void m(){", "}"); + } } } -- GitLab