@@ -52,19 +52,14 @@ public UctInspectionManager(final @NotNull Project project) {
5252 if (!(psiFile instanceof PhpFile )) {
5353 return null ;
5454 }
55- final PhpClass phpClass = GetFirstClassOfFile .getInstance ().execute ((PhpFile ) psiFile );
56-
57- if (phpClass == null ) {
58- return null ;
59- }
6055 final UctProblemsHolder problemsHolder = new UctProblemsHolder (
6156 InspectionManager .getInstance (project ),
6257 psiFile ,
6358 false
6459 );
6560 final List <PsiElementVisitor > visitors = SupportedIssue .getVisitors (problemsHolder );
6661
67- for (final PsiElement element : collectElements (phpClass )) {
62+ for (final PsiElement element : collectElements (psiFile )) {
6863 for (final PsiElementVisitor visitor : visitors ) {
6964 element .accept (visitor );
7065 }
@@ -76,26 +71,31 @@ public UctInspectionManager(final @NotNull Project project) {
7671 /**
7772 * Collect elements for PHP based inspections.
7873 *
79- * @param phpClass PhpClass
74+ * @param psiFile PsiFile
8075 *
8176 * @return List[PsiElement]
8277 */
83- private List <PsiElement > collectElements (final @ NotNull PhpClass phpClass ) {
78+ private List <PsiElement > collectElements (final @ NotNull PsiFile psiFile ) {
8479 final List <PsiElement > elements = new LinkedList <>();
85- elements .add (phpClass );
8680
87- final PhpPsiElement scopeForUseOperator = PhpCodeInsightUtil .findScopeForUseOperator (
88- phpClass
89- );
90- if (scopeForUseOperator != null ) {
91- elements .addAll (PhpCodeInsightUtil .collectImports (scopeForUseOperator ));
81+ final PhpClass phpClass = GetFirstClassOfFile .getInstance ().execute ((PhpFile ) psiFile );
82+
83+ if (phpClass != null ) {
84+ elements .add (phpClass );
85+ final PhpPsiElement scopeForUseOperator = PhpCodeInsightUtil .findScopeForUseOperator (
86+ phpClass
87+ );
88+
89+ if (scopeForUseOperator != null ) {
90+ elements .addAll (PhpCodeInsightUtil .collectImports (scopeForUseOperator ));
91+ }
92+ elements .addAll (Arrays .asList (phpClass .getOwnFields ()));
9293 }
93- elements .addAll (PsiTreeUtil .findChildrenOfType (phpClass , ClassConstantReference .class ));
94- elements .addAll (Arrays .asList (phpClass .getOwnFields ()));
95- elements .addAll (PsiTreeUtil .findChildrenOfType (phpClass , MethodReference .class ));
96- elements .addAll (PsiTreeUtil .findChildrenOfType (phpClass , AssignmentExpression .class ));
97- elements .addAll (PsiTreeUtil .findChildrenOfType (phpClass , ClassReference .class ));
98- elements .addAll (PsiTreeUtil .findChildrenOfType (phpClass , FieldReference .class ));
94+ elements .addAll (PsiTreeUtil .findChildrenOfType (psiFile , ClassConstantReference .class ));
95+ elements .addAll (PsiTreeUtil .findChildrenOfType (psiFile , MethodReference .class ));
96+ elements .addAll (PsiTreeUtil .findChildrenOfType (psiFile , AssignmentExpression .class ));
97+ elements .addAll (PsiTreeUtil .findChildrenOfType (psiFile , ClassReference .class ));
98+ elements .addAll (PsiTreeUtil .findChildrenOfType (psiFile , FieldReference .class ));
9999
100100 return elements ;
101101 }
0 commit comments