From 4b9ffa0b027f9c3ea1a56d44b4f2b033185cd3f1 Mon Sep 17 00:00:00 2001 From: Maxim Medvedev Date: Fri, 17 Jun 2011 15:48:27 +0400 Subject: [PATCH] api --- .../statements/typedef/GrTypeDefinition.java | 4 + .../GroovyTopLevelElementFactoryImpl.java | 100 ++++++++++++++++++ .../typedef/GrTypeDefinitionImpl.java | 2 +- 3 files changed, 105 insertions(+), 1 deletion(-) create mode 100644 plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/GroovyTopLevelElementFactoryImpl.java diff --git a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/api/statements/typedef/GrTypeDefinition.java b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/api/statements/typedef/GrTypeDefinition.java index 3c3bae5a443d..136683a16c42 100644 --- a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/api/statements/typedef/GrTypeDefinition.java +++ b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/api/statements/typedef/GrTypeDefinition.java @@ -24,6 +24,7 @@ import org.jetbrains.annotations.NonNls; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; import org.jetbrains.plugins.groovy.lang.groovydoc.psi.api.GrDocCommentOwner; +import org.jetbrains.plugins.groovy.lang.psi.api.statements.GrClassInitializer; import org.jetbrains.plugins.groovy.lang.psi.api.statements.GrField; import org.jetbrains.plugins.groovy.lang.psi.api.statements.GrTopLevelDefinition; import org.jetbrains.plugins.groovy.lang.psi.api.statements.typedef.members.GrMember; @@ -52,6 +53,9 @@ public interface GrTypeDefinition extends GrTopStatement, NavigatablePsiElement, @NotNull GrField[] getFields(); + @NotNull + GrClassInitializer[] getInitializers(); + @NotNull GrMembersDeclaration[] getMemberDeclarations(); diff --git a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/GroovyTopLevelElementFactoryImpl.java b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/GroovyTopLevelElementFactoryImpl.java new file mode 100644 index 000000000000..aee4f1406b50 --- /dev/null +++ b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/GroovyTopLevelElementFactoryImpl.java @@ -0,0 +1,100 @@ +/* + * Copyright 2000-2011 JetBrains s.r.o. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.jetbrains.plugins.groovy.lang.psi.impl; + +import com.intellij.openapi.project.Project; +import com.intellij.psi.*; +import com.intellij.util.IncorrectOperationException; +import com.intellij.util.StringBuilderSpinAllocator; +import org.jetbrains.annotations.NonNls; +import org.jetbrains.annotations.NotNull; +import org.jetbrains.plugins.groovy.lang.psi.GroovyPsiElementFactory; +import org.jetbrains.plugins.groovy.lang.psi.api.statements.GrVariableDeclaration; +import org.jetbrains.plugins.groovy.lang.psi.api.statements.typedef.GrTypeDefinition; + +/** + * @author Medvedev Max + */ +public class GroovyTopLevelElementFactoryImpl implements PsiTopLevelElementFactory { + private final GroovyPsiElementFactory factory; + + public GroovyTopLevelElementFactoryImpl(Project project) { + factory = GroovyPsiElementFactory.getInstance(project); + } + + @NotNull + @Override + public PsiClass createClass(@NonNls @NotNull String name) throws IncorrectOperationException { + return factory.createTypeDefinition("class " + name + "{}"); + } + + @NotNull + @Override + public PsiClass createInterface(@NonNls @NotNull String name) throws IncorrectOperationException { + return factory.createTypeDefinition("interface " + name + "{}"); + } + + @NotNull + @Override + public PsiClass createEnum(@NotNull @NonNls String name) throws IncorrectOperationException { + return factory.createTypeDefinition("enum " + name + "{}"); + } + + @NotNull + @Override + public PsiField createField(@NotNull @NonNls String name, @NotNull PsiType type) throws IncorrectOperationException { + final GrVariableDeclaration fieldDeclaration = factory.createFieldDeclaration(new String[]{PsiModifier.PRIVATE}, name, null, type); + return (PsiField)fieldDeclaration.getVariables()[0]; + } + + @NotNull + @Override + public PsiMethod createMethod(@NotNull @NonNls String name, PsiType returnType) throws IncorrectOperationException { + StringBuilder builder = StringBuilderSpinAllocator.alloc(); + builder.append("public"); + if (returnType != null) { + builder.append(' '); + builder.append(returnType.getCanonicalText()); + } + builder.append(' ').append(name).append("(){}"); + return factory.createMethodFromText(builder.toString()); + } + + @NotNull + @Override + public PsiMethod createConstructor() { + return factory.createConstructorFromText("Foo", "", null); + } + + @NotNull + @Override + public PsiClassInitializer createClassInitializer() throws IncorrectOperationException { + final GrTypeDefinition typeDefinition = factory.createTypeDefinition("class X {{}}"); + return typeDefinition.getInitializers()[0]; + } + + @NotNull + @Override + public PsiParameter createParameter(@NotNull @NonNls String name, @NotNull PsiType type) throws IncorrectOperationException { + return null; //To change body of implemented methods use File | Settings | File Templates. + } + + @NotNull + @Override + public PsiParameterList createParameterList(@NotNull @NonNls String[] names, @NotNull PsiType[] types) throws IncorrectOperationException { + return null; //To change body of implemented methods use File | Settings | File Templates. + } +} diff --git a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/typedef/GrTypeDefinitionImpl.java b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/typedef/GrTypeDefinitionImpl.java index a8dffa251e99..178743441643 100644 --- a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/typedef/GrTypeDefinitionImpl.java +++ b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/impl/statements/typedef/GrTypeDefinitionImpl.java @@ -400,7 +400,7 @@ public abstract class GrTypeDefinitionImpl extends GrStubElementBase