java/java-impl/src/com/intellij/codeInsight/template/macro/MethodNameMacro.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.template.macro;
  17
  18 import com.intellij.codeInsight.CodeInsightBundle;
  19 import com.intellij.codeInsight.lookup.LookupElement;
  20 import com.intellij.codeInsight.template.*;
  21 import com.intellij.openapi.project.Project;
  22 import com.intellij.psi.*;
  23 import com.intellij.lang.LangBundle;
  24 import org.jetbrains.annotations.NotNull;
  25
  26 public class MethodNameMacro implements Macro {
  27
  28   public String getName() {
  29     return "methodName";
  30   }
  31
  32   public String getDescription() {
  33     return CodeInsightBundle.message("macro.methodname");
  34   }
  35
  36   public String getDefaultValue() {
  37     return "a";
  38   }
  39
  40   public Result calculateResult(@NotNull Expression[] params, final ExpressionContext context) {
  41     Project project = context.getProject();
  42     int templateStartOffset = context.getTemplateStartOffset();
  43     final int offset = templateStartOffset > 0 ? context.getTemplateStartOffset() - 1 : context.getTemplateStartOffset();
  44
  45     PsiDocumentManager.getInstance(project).commitAllDocuments();
  46
  47     PsiFile file = PsiDocumentManager.getInstance(project).getPsiFile(context.getEditor().getDocument());
  48     PsiElement place = file.findElementAt(offset);
  49     while(place != null){
  50       if (place instanceof PsiMethod){
  51         return new TextResult(((PsiMethod)place).getName());
  52       } else if (place instanceof PsiClassInitializer) {
  53         return ((PsiClassInitializer) place).hasModifierProperty(PsiModifier.STATIC) ?
  54                new TextResult(LangBundle.message("java.terms.static.initializer")) :
  55                new TextResult(LangBundle.message("java.terms.instance.initializer"));
  56       }
  57       place = place.getParent();
  58     }
  59     return null;
  60   }
  61
  62   public Result calculateQuickResult(@NotNull Expression[] params, ExpressionContext context) {
  63     return null;
  64   }
  65
  66   public LookupElement[] calculateLookupItems(@NotNull Expression[] params, final ExpressionContext context) {
  67     return null;
  68   }
  69 }