diff --git a/plugins/InspectionGadgets/src/com/siyeh/InspectionGadgetsBundle.properties b/plugins/InspectionGadgets/src/com/siyeh/InspectionGadgetsBundle.properties
index e2adb68c4246..69f312bac08a 100644
--- a/plugins/InspectionGadgets/src/com/siyeh/InspectionGadgetsBundle.properties
+++ b/plugins/InspectionGadgets/src/com/siyeh/InspectionGadgetsBundle.properties
@@ -446,6 +446,7 @@ runtime.exec.call.display.name=Call to 'Runtime.exec()'
runtime.exec.call.problem.descriptor=Call to Runtime.#ref() is non-portable #loc
system.exit.call.display.name=Call to 'System.exit()' or related methods
system.exit.call.problem.descriptor=Call to {0}.#ref() is non-portable #loc
+system.exit.call.ignore.option=Ignore in main method
system.getenv.call.display.name=Call to 'System.getenv()'
system.getenv.call.problem.descriptor=Call to System.#ref() is non-portable #loc
use.of.awt.peer.class.display.name=Use of AWT peer class
diff --git a/plugins/InspectionGadgets/src/com/siyeh/ig/portability/SystemExitInspection.java b/plugins/InspectionGadgets/src/com/siyeh/ig/portability/SystemExitInspection.java
index 01cd8ad99d25..6ea41ef70caf 100644
--- a/plugins/InspectionGadgets/src/com/siyeh/ig/portability/SystemExitInspection.java
+++ b/plugins/InspectionGadgets/src/com/siyeh/ig/portability/SystemExitInspection.java
@@ -1,5 +1,5 @@
/*
- * Copyright 2003-2007 Dave Griffith, Bas Leijdekkers
+ * Copyright 2003-2013 Dave Griffith, Bas Leijdekkers
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
@@ -15,15 +15,24 @@
*/
package com.siyeh.ig.portability;
+import com.intellij.codeInspection.ui.SingleCheckboxOptionsPanel;
import com.intellij.psi.*;
+import com.intellij.psi.util.PsiMethodUtil;
+import com.intellij.psi.util.PsiTreeUtil;
import com.siyeh.InspectionGadgetsBundle;
import com.siyeh.ig.BaseInspection;
import com.siyeh.ig.BaseInspectionVisitor;
import org.jetbrains.annotations.NonNls;
import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import javax.swing.*;
public class SystemExitInspection extends BaseInspection {
+ @SuppressWarnings("PublicField")
+ public boolean ignoreInMainMethod = false;
+
@NotNull
public String getID() {
return "CallToSystemExit";
@@ -37,33 +46,37 @@ public class SystemExitInspection extends BaseInspection {
@NotNull
public String buildErrorString(Object... infos) {
final String className = (String)infos[0];
- return InspectionGadgetsBundle.message(
- "system.exit.call.problem.descriptor", className);
+ return InspectionGadgetsBundle.message("system.exit.call.problem.descriptor", className);
+ }
+
+ @Nullable
+ @Override
+ public JComponent createOptionsPanel() {
+ return new SingleCheckboxOptionsPanel(InspectionGadgetsBundle.message("system.exit.call.ignore.option"), this, "ignoreInMainMethod");
}
public BaseInspectionVisitor buildVisitor() {
return new SystemExitVisitor();
}
- private static class SystemExitVisitor extends BaseInspectionVisitor {
+ private class SystemExitVisitor extends BaseInspectionVisitor {
@Override
- public void visitMethodCallExpression(
- @NotNull PsiMethodCallExpression expression) {
+ public void visitMethodCallExpression(@NotNull PsiMethodCallExpression expression) {
super.visitMethodCallExpression(expression);
- final PsiReferenceExpression methodExpression =
- expression.getMethodExpression();
- final String methodName = methodExpression.getReferenceName();
- @NonNls final String exit = "exit";
- @NonNls final String halt = "halt";
- if (!exit.equals(methodName) && !halt.equals(methodName)) {
+ final PsiReferenceExpression methodExpression = expression.getMethodExpression();
+ @NonNls final String methodName = methodExpression.getReferenceName();
+ if (!"exit".equals(methodName) && !"halt".equals(methodName)) {
+ return;
+ }
+ final PsiMethod containingMethod = PsiTreeUtil.getParentOfType(expression, PsiMethod.class, true, PsiClass.class);
+ if (ignoreInMainMethod && PsiMethodUtil.isMainMethod(containingMethod)) {
return;
}
final PsiMethod method = expression.resolveMethod();
if (method == null) {
return;
}
-
final PsiParameterList parameterList = method.getParameterList();
if (parameterList.getParametersCount() != 1) {
return;
@@ -78,15 +91,10 @@ public class SystemExitInspection extends BaseInspection {
return;
}
final String className = aClass.getQualifiedName();
- if (className == null) {
+ if (!"java.lang.System".equals(className) && !"java.lang.Runtime".equals(className)) {
return;
}
- if ("java.lang.System".equals(className)) {
- registerMethodCallError(expression, "System");
- }
- else if ("java.lang.Runtime".equals(className)) {
- registerMethodCallError(expression, "Runtime");
- }
+ registerMethodCallError(expression, "System");
}
}
}
\ No newline at end of file
diff --git a/plugins/InspectionGadgets/src/inspectionDescriptions/SystemExit.html b/plugins/InspectionGadgets/src/inspectionDescriptions/SystemExit.html
index 2e78ff6fb120..81f2a5d922f3 100644
--- a/plugins/InspectionGadgets/src/inspectionDescriptions/SystemExit.html
+++ b/plugins/InspectionGadgets/src/inspectionDescriptions/SystemExit.html
@@ -5,6 +5,8 @@ Reports the calls to System.exit(),
Calls to these methods make the calling code unportable to most application servers.
+Use the checkbox below to ignore calls in main methods. +
Powered by InspectionGadgets