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