plugins/InspectionGadgets/src/com/siyeh/ig/resources/DriverManagerGetConnectionInspection.java

   1 /*
   2  * Copyright 2003-2007 Dave Griffith, Bas Leijdekkers
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.siyeh.ig.resources;
  17
  18 import com.intellij.psi.PsiClass;
  19 import com.intellij.psi.PsiMethod;
  20 import com.intellij.psi.PsiMethodCallExpression;
  21 import com.intellij.psi.PsiReferenceExpression;
  22 import com.siyeh.HardcodedMethodConstants;
  23 import com.siyeh.InspectionGadgetsBundle;
  24 import com.siyeh.ig.BaseInspection;
  25 import com.siyeh.ig.BaseInspectionVisitor;
  26 import org.jetbrains.annotations.NotNull;
  27
  28 public class DriverManagerGetConnectionInspection extends BaseInspection {
  29
  30     @NotNull
  31     public String getID(){
  32         return "CallToDriverManagerGetConnection";
  33     }
  34
  35     @NotNull
  36     public String getDisplayName() {
  37         return InspectionGadgetsBundle.message(
  38                 "drivermanager.call.display.name");
  39     }
  40
  41     @NotNull
  42     public String buildErrorString(Object... infos) {
  43         return InspectionGadgetsBundle.message(
  44                 "drivermanager.call.problem.descriptor");
  45     }
  46
  47     public BaseInspectionVisitor buildVisitor() {
  48         return new DriverManagerGetConnectionVisitor();
  49     }
  50
  51     private static class DriverManagerGetConnectionVisitor
  52             extends BaseInspectionVisitor {
  53
  54         @Override public void visitMethodCallExpression(
  55                 @NotNull PsiMethodCallExpression expression) {
  56             super.visitMethodCallExpression(expression);
  57             if (!isDriverManagerGetConnection(expression)) {
  58                 return;
  59             }
  60             registerMethodCallError(expression);
  61         }
  62
  63         private static boolean isDriverManagerGetConnection(
  64                 PsiMethodCallExpression expression) {
  65             final PsiReferenceExpression methodExpression =
  66                     expression.getMethodExpression();
  67
  68             final String methodName = methodExpression.getReferenceName();
  69             if (!HardcodedMethodConstants.GET_CONNECTION.equals(methodName) ) {
  70                 return false;
  71             }
  72             final PsiMethod method = expression.resolveMethod();
  73             if (method == null) {
  74                 return false;
  75             }
  76             final PsiClass aClass = method.getContainingClass();
  77             if (aClass == null) {
  78                 return false;
  79             }
  80             final String className = aClass.getQualifiedName();
  81             if (className == null) {
  82                 return false;
  83             }
  84             return "java.sql.DriverManager".equals(className);
  85         }
  86     }
  87 }