platform/lang-impl/src/com/intellij/codeInsight/intention/impl/IntentionActionWithTextCaching.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
  17 package com.intellij.codeInsight.intention.impl;
  18
  19 import com.intellij.codeInsight.daemon.impl.HighlightInfo;
  20 import com.intellij.codeInsight.intention.IntentionAction;
  21 import com.intellij.openapi.util.Comparing;
  22 import com.intellij.openapi.diagnostic.Logger;
  23
  24 import javax.swing.*;
  25 import java.util.List;
  26 import java.util.ArrayList;
  27
  28 /**
  29 * @author cdr
  30 */
  31 class IntentionActionWithTextCaching implements Comparable<IntentionActionWithTextCaching> {
  32   private static final Logger LOG = Logger.getInstance("#com.intellij.codeInsight.intention.impl.IntentionActionWithTextCaching");
  33   private final List<IntentionAction> myOptionIntentions;
  34   private final List<IntentionAction> myOptionErrorFixes;
  35   private final String myText;
  36   private final IntentionAction myAction;
  37   private final String myDisplayName;
  38   private final List<IntentionAction> myOptionInspectionFixes;
  39   private final Icon myIcon;
  40
  41   IntentionActionWithTextCaching(IntentionAction action){
  42     this(action, action.getText(), null);
  43   }
  44
  45   IntentionActionWithTextCaching(HighlightInfo.IntentionActionDescriptor action){
  46     this(action.getAction(), action.getDisplayName(), action.getIcon());
  47   }
  48
  49   private IntentionActionWithTextCaching(IntentionAction action, String displayName, Icon icon) {
  50     myIcon = icon;
  51     myOptionIntentions = new ArrayList<IntentionAction>();
  52     myOptionErrorFixes = new ArrayList<IntentionAction>();
  53     myOptionInspectionFixes = new ArrayList<IntentionAction>();
  54     myText = action.getText();
  55     // needed for checking errors in user written actions
  56     //noinspection ConstantConditions
  57     LOG.assertTrue(myText != null, "action "+action.getClass()+" text returned null");
  58     myAction = action;
  59     myDisplayName = displayName;
  60   }
  61
  62   String getText() {
  63     return myText;
  64   }
  65
  66   public void addIntention(final IntentionAction action) {
  67       myOptionIntentions.add(action);
  68   }
  69   public void addErrorFix(final IntentionAction action) {
  70     myOptionErrorFixes.add(action);
  71   }
  72   public void addInspectionFix(final IntentionAction action) {
  73     myOptionInspectionFixes.add(action);
  74   }
  75
  76   public IntentionAction getAction() {
  77     return myAction;
  78   }
  79
  80   public List<IntentionAction> getOptionIntentions() {
  81     return myOptionIntentions;
  82   }
  83
  84   public List<IntentionAction> getOptionErrorFixes() {
  85     return myOptionErrorFixes;
  86   }
  87
  88   public List<IntentionAction> getOptionInspectionFixes() {
  89     return myOptionInspectionFixes;
  90   }
  91
  92   public String getToolName() {
  93     return myDisplayName;
  94   }
  95
  96   public String toString() {
  97     return getText();
  98   }
  99
 100   public int compareTo(final IntentionActionWithTextCaching other) {
 101     if (myAction instanceof Comparable) {
 102       return ((Comparable)myAction).compareTo(other.getAction());
 103     }
 104     else if (other.getAction() instanceof Comparable) {
 105       return ((Comparable)other.getAction()).compareTo(myAction);
 106     }
 107     return Comparing.compare(getText(), other.getText());
 108   }
 109
 110   public Icon getIcon() {
 111     return myIcon;
 112   }
 113 }