Added support find usages for PHP classes in neon

Author: mesour

resources/META-INF/plugin.xml

@@ -10,7 +10,8 @@
 	<change-notes><![CDATA[
 	<h2>0.5.2</h2>
 	<ul>
-		<li>Added support find usages for PHP classes in neon</li>
+		<li>Added support find usages for PHP classes used in neon</li>
+		<li>Added inspections for used undefined classes</li>
 		<li>Increased minimum compatibility with PHPStorm 2018.3</li>
 	</ul>
 	<h2>0.5.1</h2>
@@ -74,5 +75,10 @@
 
 		<gotoDeclarationHandler implementation="cz.juzna.intellij.neon.reference.GoToClassHandler"/>
 		<annotator implementationClass="cz.juzna.intellij.neon.annotator.NeonAnnotator" language="neon"/>
+
+		<localInspection language="neon" shortName="ClassUsages" id="ClassUsages"
+						 displayName="Class usages"
+						 groupName="Neon" enabledByDefault="true"
+						 implementationClass="cz.juzna.intellij.neon.inspections.ClassUsagesInspection"/>
 	</extensions>
 </idea-plugin>

src/cz/juzna/intellij/neon/inspections/ClassUsagesInspection.java

@@ -0,0 +1,51 @@
+package cz.juzna.intellij.neon.inspections;
+
+import com.intellij.codeInspection.*;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiFile;
+import com.intellij.psi.PsiRecursiveElementWalkingVisitor;
+import cz.juzna.intellij.neon.psi.NeonFile;
+import cz.juzna.intellij.neon.psi.impl.NeonScalarImpl;
+import cz.juzna.intellij.neon.util.NeonPhpUtil;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class ClassUsagesInspection extends LocalInspectionTool {
+
+	@NotNull
+	@Override
+	public String getShortName() {
+		return "ClassUsages";
+	}
+
+	@Nullable
+	@Override
+	public ProblemDescriptor[] checkFile(@NotNull PsiFile file, @NotNull final InspectionManager manager, final boolean isOnTheFly) {
+		if (!(file instanceof NeonFile)) {
+			return null;
+		}
+
+		final List<ProblemDescriptor> problems = new ArrayList<ProblemDescriptor>();
+		file.acceptChildren(new PsiRecursiveElementWalkingVisitor() {
+			@Override
+			public void visitElement(PsiElement element) {
+				if (element instanceof NeonScalarImpl && ((NeonScalarImpl) element).isPhpScalar()) {
+					String className = ((NeonScalarImpl) element).getName();
+					if (NeonPhpUtil.getClassesByFQN(element.getProject(), className).size() == 0) {
+						String description = "Undefined class '" + className + "'";
+						ProblemDescriptor problem = manager.createProblemDescriptor(element, description, true, ProblemHighlightType.GENERIC_ERROR, isOnTheFly);
+						problems.add(problem);
+					}
+
+				} else {
+					super.visitElement(element);
+				}
+			}
+		});
+
+		return problems.toArray(new ProblemDescriptor[problems.size()]);
+	}
+}

src/cz/juzna/intellij/neon/psi/impl/NeonScalarImpl.java

@@ -6,6 +6,7 @@
 import com.intellij.psi.impl.source.tree.LeafPsiElement;
 import cz.juzna.intellij.neon.lexer.NeonTokenTypes;
 import cz.juzna.intellij.neon.psi.NeonScalar;
+import cz.juzna.intellij.neon.util.NeonPhpUtil;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
@@ -30,6 +31,10 @@ public String getValueText() {
 		return text;
 	}
 
+	public boolean isPhpScalar() {
+		return NeonPhpUtil.isPhpClassScalar(this);
+	}
+
 	@Override
 	public String getName() {
 		return getValueText();

src/cz/juzna/intellij/neon/reference/NeonReferenceContributor.java

@@ -7,7 +7,6 @@
 import cz.juzna.intellij.neon.NeonLanguage;
 import cz.juzna.intellij.neon.psi.impl.NeonScalarImpl;
 import cz.juzna.intellij.neon.reference.references.NeonPhpClassReference;
-import cz.juzna.intellij.neon.util.NeonPhpUtil;
 import org.jetbrains.annotations.NotNull;
 
 public class NeonReferenceContributor extends PsiReferenceContributor {
@@ -23,7 +22,7 @@ public PsiReference[] getReferencesByElement(@NotNull PsiElement element, @NotNu
 						}
 
 						String name = ((NeonScalarImpl) element).getName();
-						if (name != null && NeonPhpUtil.isPhpClassScalar(element)) {
+						if (name != null && ((NeonScalarImpl) element).isPhpScalar()) {
 							return new PsiReference[]{
 									new NeonPhpClassReference((NeonScalarImpl) element, new TextRange(0, name.length() + 1))};
 						}

src/cz/juzna/intellij/neon/util/NeonPhpUtil.java

@@ -1,14 +1,17 @@
 package cz.juzna.intellij.neon.util;
 
+import com.intellij.openapi.project.Project;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiRecursiveElementVisitor;
+import com.jetbrains.php.PhpIndex;
 import com.jetbrains.php.lang.psi.elements.ClassReference;
 import com.jetbrains.php.lang.psi.elements.ExtendsList;
 import com.jetbrains.php.lang.psi.elements.PhpClass;
 import cz.juzna.intellij.neon.psi.impl.NeonScalarImpl;
 import org.jetbrains.annotations.NotNull;
 
 import java.util.ArrayList;
+import java.util.Collection;
 import java.util.List;
 
 public class NeonPhpUtil {
@@ -35,6 +38,14 @@ public static void findNeonPhpClasses(List<NeonScalarImpl> properties, PsiElemen
 		}
 	}
 
+	public static Collection<PhpClass> getClassesByFQN(Project project, String className) {
+		return getPhpIndex(project).getAnyByFQN(className);
+	}
+
+	private static PhpIndex getPhpIndex(Project project) {
+		return PhpIndex.getInstance(project);
+	}
+
 	public static boolean isPhpClassScalar(@NotNull PsiElement element) {
 		if (!(element instanceof NeonScalarImpl)) {
 			return false;