java/java-impl/src/com/intellij/refactoring/ui/NameSuggestionsManager.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.refactoring.ui;
  17
  18 import com.intellij.psi.PsiType;
  19 import com.intellij.psi.codeStyle.SuggestedNameInfo;
  20 import com.intellij.util.containers.HashMap;
  21
  22 import javax.swing.*;
  23 import java.awt.event.ItemEvent;
  24 import java.awt.event.ItemListener;
  25
  26 /**
  27  * @author dsl
  28  */
  29 public class NameSuggestionsManager {
  30   private final TypeSelector myTypeSelector;
  31   private final NameSuggestionsField myNameField;
  32   private final NameSuggestionsGenerator myGenerator;
  33
  34   private final HashMap<PsiType, SuggestedNameInfo> myTypesToSuggestions = new HashMap<PsiType, SuggestedNameInfo>();
  35
  36   public NameSuggestionsManager(TypeSelector typeSelector, NameSuggestionsField nameField, NameSuggestionsGenerator generator) {
  37     myTypeSelector = typeSelector;
  38     myNameField = nameField;
  39     myGenerator = generator;
  40
  41     myTypeSelector.addItemListener(new ItemListener() {
  42       public void itemStateChanged(ItemEvent e) {
  43         if (e.getStateChange() == ItemEvent.SELECTED) {
  44           updateSuggestions(myTypeSelector.getSelectedType());
  45         }
  46       }
  47     });
  48     updateSuggestions(myTypeSelector.getSelectedType());
  49
  50   }
  51
  52   public void nameSelected() {
  53     SuggestedNameInfo nameInfo = myTypesToSuggestions.get(myTypeSelector.getSelectedType());
  54
  55     if (nameInfo != null) {
  56       nameInfo.nameChoosen(myNameField.getEnteredName());
  57     }
  58   }
  59
  60   private void updateSuggestions(PsiType selectedType) {
  61     SuggestedNameInfo nameInfo = myTypesToSuggestions.get(selectedType);
  62     if (nameInfo == null) {
  63       nameInfo = myGenerator.getSuggestedNameInfo(selectedType);
  64       myTypesToSuggestions.put(selectedType, nameInfo);
  65     }
  66     myNameField.setSuggestions(nameInfo.names);
  67   }
  68
  69   public void setLabelsFor(JLabel typeSelectorLabel, JLabel nameLabel) {
  70     if(myTypeSelector.getFocusableComponent() != null) {
  71       typeSelectorLabel.setLabelFor(myTypeSelector.getFocusableComponent());
  72     }
  73
  74     if(myNameField.getFocusableComponent() != null) {
  75       nameLabel.setLabelFor(myNameField.getFocusableComponent());
  76     }
  77   }
  78 }