lang-impl/src/com/intellij/ide/fileTemplates/actions/CreateFromTemplateActionBase.java

   1 package com.intellij.ide.fileTemplates.actions;
   2
   3 import com.intellij.ide.IdeView;
   4 import com.intellij.ide.fileTemplates.FileTemplate;
   5 import com.intellij.ide.fileTemplates.FileTemplateManager;
   6 import com.intellij.ide.fileTemplates.ui.CreateFromTemplateDialog;
   7 import com.intellij.ide.util.DirectoryChooserUtil;
   8 import com.intellij.openapi.actionSystem.*;
   9 import com.intellij.openapi.project.Project;
  10 import com.intellij.psi.PsiDirectory;
  11 import com.intellij.psi.PsiElement;
  12 import org.jetbrains.annotations.Nullable;
  13
  14 import javax.swing.*;
  15
  16 public abstract class CreateFromTemplateActionBase extends AnAction {
  17
  18   public CreateFromTemplateActionBase(final String title, final String description, final Icon icon) {
  19     super (title,description,icon);
  20   }
  21
  22   public final void actionPerformed(AnActionEvent e){
  23     DataContext dataContext = e.getDataContext();
  24
  25     IdeView view = LangDataKeys.IDE_VIEW.getData(dataContext);
  26     if (view == null) {
  27       return;
  28     }
  29     Project project = PlatformDataKeys.PROJECT.getData(dataContext);
  30
  31     PsiDirectory dir = DirectoryChooserUtil.getOrChooseDirectory(view);
  32     if (dir == null) return;
  33
  34     FileTemplate selectedTemplate = getTemplate(project, dir);
  35     if(selectedTemplate != null){
  36       AnAction action = getReplacedAction(selectedTemplate);
  37       if (action != null) {
  38         action.actionPerformed(e);
  39       }
  40       else {
  41         FileTemplateManager.getInstance().addRecentName(selectedTemplate.getName());
  42         PsiElement createdElement = new CreateFromTemplateDialog(project, dir, selectedTemplate, getDefaultFileName()).create();
  43         if (createdElement != null) {
  44           view.selectElement(createdElement);
  45         }
  46       }
  47     }
  48   }
  49
  50   @Nullable
  51   protected abstract AnAction getReplacedAction(final FileTemplate selectedTemplate);
  52
  53   protected abstract FileTemplate getTemplate(final Project project, final PsiDirectory dir);
  54
  55   @Nullable
  56   public String getDefaultFileName() {
  57     return null;
  58   }
  59 }