cleanup
[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   @SuppressWarnings("UnusedDeclaration") //for serialization
43   public DMethodElement() {
44     super(null, null, null);
45   }
46
47   public DMethodElement(Boolean isStatic, String name, String returnType, List<MyPair> pairs) {
48     super(isStatic, name, returnType);
49
50     myPairs = pairs;
51   }
52
53   public List<MyPair> getPairs() {
54     return myPairs;
55   }
56
57   public void clearCache() {
58     myImplicitMethod = null;
59   }
60
61   @NotNull
62   public PsiMethod getPsi(PsiManager manager, final String containingClassName) {
63     if (myImplicitMethod != null) return myImplicitMethod;
64
65     final String type = getType();
66
67     String staticModifier = null;
68     Boolean isStatic = isStatic();
69
70     if (isStatic != null && isStatic.booleanValue()) {
71       staticModifier = PsiModifier.STATIC;
72     }
73
74     final String[] argumentsTypes = QuickfixUtil.getArgumentsTypes(myPairs);
75     final GroovyPsiElementFactory factory = GroovyPsiElementFactory.getInstance(manager.getProject());
76     final GrMethod method = factory.createMethodFromText(staticModifier, getName(), type, argumentsTypes, null);
77
78     myImplicitMethod = new GrDynamicImplicitMethod(manager, method, containingClassName) {
79       @Override
80       public PsiElement setName(@NotNull String name) throws IncorrectOperationException {
81         DynamicManager.getInstance(getProject()).replaceDynamicMethodName(containingClassName, getName(), name, argumentsTypes);
82         return super.setName(name);
83       }
84     };
85     return myImplicitMethod;
86   }
87 }