java/java-impl/src/com/intellij/codeInsight/daemon/impl/actions/AddSuppressInspectionForClassFix.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   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.intellij.codeInsight.daemon.impl.actions;
  17
  18 import com.intellij.codeInsight.daemon.HighlightDisplayKey;
  19 import com.intellij.codeInspection.InspectionsBundle;
  20 import com.intellij.psi.PsiClass;
  21 import com.intellij.psi.PsiDeclarationStatement;
  22 import com.intellij.psi.PsiDocCommentOwner;
  23 import com.intellij.psi.PsiElement;
  24 import com.intellij.psi.util.PsiTreeUtil;
  25 import org.jetbrains.annotations.NotNull;
  26 import org.jetbrains.annotations.Nullable;
  27
  28 /**
  29  * User: anna
  30  * Date: May 13, 2005
  31  */
  32 public class AddSuppressInspectionForClassFix extends AddSuppressInspectionFix {
  33
  34   public AddSuppressInspectionForClassFix(final HighlightDisplayKey key) {
  35     super(key);
  36   }
  37
  38   public AddSuppressInspectionForClassFix(final String id) {
  39    super(id);
  40   }
  41
  42   @Nullable protected PsiDocCommentOwner getContainer(final PsiElement element) {
  43     PsiDocCommentOwner container = super.getContainer(element);
  44     if (container == null || container instanceof PsiClass){
  45       return null;
  46     }
  47     while (container != null ) {
  48       final PsiClass parentClass = PsiTreeUtil.getParentOfType(container, PsiClass.class);
  49       if ((parentClass == null || container.getParent() instanceof PsiDeclarationStatement) && container instanceof PsiClass){
  50         return container;
  51       }
  52       container = parentClass;
  53     }
  54     return container;
  55   }
  56
  57   @NotNull
  58   public String getText() {
  59     return InspectionsBundle.message("suppress.inspection.class");
  60   }
  61 }