2b07b647b2cf8463d72e73cd2f341a534730cc93
[idea/community.git] / plugins / groovy / src / org / jetbrains / plugins / groovy / annotator / intentions / dynamic / elements / DMethodElement.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 org.jetbrains.plugins.groovy.annotator.intentions.dynamic.elements;
17
18 import com.intellij.psi.PsiElement;
19 import com.intellij.psi.PsiManager;
20 import com.intellij.psi.PsiMethod;
21 import com.intellij.psi.PsiModifier;
22 import com.intellij.util.IncorrectOperationException;
23 import org.jetbrains.annotations.NotNull;
24 import org.jetbrains.plugins.groovy.annotator.intentions.QuickfixUtil;
25 import org.jetbrains.plugins.groovy.annotator.intentions.dynamic.DynamicManager;
26 import org.jetbrains.plugins.groovy.annotator.intentions.dynamic.MyPair;
27 import org.jetbrains.plugins.groovy.lang.psi.GroovyPsiElementFactory;
28 import org.jetbrains.plugins.groovy.lang.psi.api.statements.typedef.members.GrMethod;
29 import org.jetbrains.plugins.groovy.lang.psi.impl.synthetic.GrDynamicImplicitMethod;
30
31 import java.util.ArrayList;
32 import java.util.List;
33
34 /**
35  * User: Dmitry.Krasilschikov
36  * Date: 12.02.2008
37  */
38 public class DMethodElement extends DItemElement {
39   public List<MyPair> myPairs = new ArrayList<MyPair>();
40   private PsiMethod myImplicitMethod;
41
42   public DMethodElement() {
43     super(null, null, null);
44   }
45
46   public DMethodElement(Boolean isStatic, String name, String returnType, List<MyPair> pairs) {
47     super(isStatic, name, returnType);
48
49     myPairs = pairs;
50   }
51
52   public List<MyPair> getPairs() {
53     return myPairs;
54   }
55
56   public void clearCache() {
57     myImplicitMethod = null;
58   }
59
60   @NotNull
61   public PsiMethod getPsi(PsiManager manager, final String containingClassName) {
62     if (myImplicitMethod != null) return myImplicitMethod;
63
64     final String type = getType();
65
66     String staticModifier = null;
67     Boolean isStatic = isStatic();
68
69     if (isStatic != null && isStatic.booleanValue()) {
70       staticModifier = PsiModifier.STATIC;
71     }
72
73     final String[] argumentsTypes = QuickfixUtil.getArgumentsTypes(myPairs);
74     final GrMethod method = GroovyPsiElementFactory.getInstance(manager.getProject())
75         .createMethodFromText(staticModifier, getName(), type, argumentsTypes, null);
76
77     myImplicitMethod = new GrDynamicImplicitMethod(manager, method, containingClassName) {
78       @Override
79       public PsiElement setName(@NotNull String name) throws IncorrectOperationException {
80         DynamicManager.getInstance(getProject()).replaceDynamicMethodName(containingClassName, getName(), name, argumentsTypes);
81         return super.setName(name);
82       }
83     };
84     return myImplicitMethod;
85   }
86 }