Merge pull request #612 from Shan1024/package-name-inspection-fix

Remove package declaration inspection in non-Ballerina modules
ballerina-attic · Aug 23, 2017 · 36ec871 · 36ec871
2 parents ab1e300 + 924972b
commit 36ec871
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/...n/java/org/ballerinalang/plugins/idea/codeInspection/WrongPackageStatementInspection.java b/...n/java/org/ballerinalang/plugins/idea/codeInspection/WrongPackageStatementInspection.java
@@ -23,6 +23,8 @@
 import com.intellij.codeInspection.ProblemDescriptor;
 import com.intellij.codeInspection.ProblemHighlightType;
 import com.intellij.openapi.application.ApplicationManager;
+import com.intellij.openapi.module.Module;
+import com.intellij.openapi.module.ModuleUtil;
 import com.intellij.openapi.util.Comparing;
 import com.intellij.psi.PsiDirectory;
 import com.intellij.psi.PsiElement;
@@ -36,6 +38,7 @@
 import org.ballerinalang.plugins.idea.psi.FullyQualifiedPackageNameNode;
 import org.ballerinalang.plugins.idea.psi.PackageDeclarationNode;
 import org.ballerinalang.plugins.idea.psi.PackageNameNode;
+import org.ballerinalang.plugins.idea.sdk.BallerinaSdkService;
 import org.ballerinalang.plugins.idea.util.BallerinaUtil;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
@@ -58,6 +61,11 @@ public ProblemDescriptor[] checkFile(@NotNull PsiFile file, @NotNull InspectionM
         if (!(file instanceof BallerinaFile)) {
             return new ProblemDescriptor[0];
         }
+        Module module = ModuleUtil.findModuleForFile(file.getVirtualFile(), file.getProject());
+        boolean isBallerinaModule = BallerinaSdkService.isBallerinaModule(module);
+        if (!isBallerinaModule) {
+            return new ProblemDescriptor[0];
+        }
         if (FileTypeUtils.isInServerPageFile(file)) {
             return new ProblemDescriptor[0];
         }