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