| /* |
| * Copyright (C) 2007-2010 JĂșlio Vilmar Gesser. |
| * Copyright (C) 2011, 2013-2016 The JavaParser Team. |
| * |
| * This file is part of JavaParser. |
| * |
| * JavaParser can be used either under the terms of |
| * a) the GNU Lesser General Public License as published by |
| * the Free Software Foundation, either version 3 of the License, or |
| * (at your option) any later version. |
| * b) the terms of the Apache License |
| * |
| * You should have received a copy of both licenses in LICENCE.LGPL and |
| * LICENCE.APACHE. Please refer to those files for details. |
| * |
| * JavaParser is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU Lesser General Public License for more details. |
| */ |
| |
| package com.github.javaparser.ast; |
| |
| import static com.github.javaparser.ast.expr.NameExpr.*; |
| import static com.github.javaparser.utils.Utils.ensureNotNull; |
| |
| import java.util.Arrays; |
| import java.util.EnumSet; |
| import java.util.List; |
| import java.util.stream.Collectors; |
| |
| import com.github.javaparser.ast.expr.NameExpr; |
| import com.github.javaparser.utils.ClassUtils; |
| import com.github.javaparser.Range; |
| import com.github.javaparser.ast.body.*; |
| import com.github.javaparser.ast.comments.Comment; |
| import com.github.javaparser.ast.comments.JavadocComment; |
| import com.github.javaparser.ast.expr.NameExpr; |
| import com.github.javaparser.ast.visitor.GenericVisitor; |
| import com.github.javaparser.ast.visitor.VoidVisitor; |
| import com.github.javaparser.utils.ClassUtils; |
| |
| import java.util.Arrays; |
| import java.util.EnumSet; |
| import java.util.List; |
| import java.util.stream.Collectors; |
| |
| import static com.github.javaparser.utils.Utils.ensureNotNull; |
| |
| /** |
| * <p> |
| * This class represents the entire compilation unit. Each java file denotes a |
| * compilation unit. |
| * </p> |
| * The CompilationUnit is constructed following the syntax:<br> |
| * |
| * <pre> |
| * {@code |
| * CompilationUnit ::= ( }{@link PackageDeclaration}{@code )? |
| * ( }{@link ImportDeclaration}{@code )* |
| * ( }{@link TypeDeclaration}{@code )* |
| * } |
| * </pre> |
| * |
| * @author Julio Vilmar Gesser |
| */ |
| public final class CompilationUnit extends Node { |
| |
| private PackageDeclaration pakage; |
| |
| private List<ImportDeclaration> imports; |
| |
| private List<TypeDeclaration<?>> types; |
| |
| public CompilationUnit() { |
| } |
| |
| public CompilationUnit(PackageDeclaration pakage, List<ImportDeclaration> imports, List<TypeDeclaration<?>> types) { |
| setPackage(pakage); |
| setImports(imports); |
| setTypes(types); |
| } |
| |
| public CompilationUnit(Range range, PackageDeclaration pakage, List<ImportDeclaration> imports, |
| List<TypeDeclaration<?>> types) { |
| super(range); |
| setPackage(pakage); |
| setImports(imports); |
| setTypes(types); |
| } |
| |
| @Override |
| public <R, A> R accept(GenericVisitor<R, A> v, A arg) { |
| return v.visit(this, arg); |
| } |
| |
| @Override |
| public <A> void accept(VoidVisitor<A> v, A arg) { |
| v.visit(this, arg); |
| } |
| |
| /** |
| * Return a list containing all comments declared in this compilation unit. |
| * Including javadocs, line comments and block comments of all types, |
| * inner-classes and other members.<br> |
| * If there is no comment, <code>null</code> is returned. |
| * |
| * @return list with all comments of this compilation unit or |
| * <code>null</code> |
| * @see JavadocComment |
| * @see com.github.javaparser.ast.comments.LineComment |
| * @see com.github.javaparser.ast.comments.BlockComment |
| */ |
| public List<Comment> getComments() { |
| return this.getAllContainedComments(); |
| } |
| |
| /** |
| * Retrieves the list of imports declared in this compilation unit or |
| * <code>null</code> if there is no import. |
| * |
| * @return the list of imports or <code>null</code> if there is no import |
| */ |
| public List<ImportDeclaration> getImports() { |
| imports = ensureNotNull(imports); |
| return imports; |
| } |
| |
| /** |
| * Retrieves the package declaration of this compilation unit.<br> |
| * If this compilation unit has no package declaration (default package), |
| * <code>null</code> is returned. |
| * |
| * @return the package declaration or <code>null</code> |
| */ |
| public PackageDeclaration getPackage() { |
| return pakage; |
| } |
| |
| /** |
| * Return the list of types declared in this compilation unit.<br> |
| * If there is no types declared, <code>null</code> is returned. |
| * |
| * @return the list of types or <code>null</code> null if there is no type |
| * @see AnnotationDeclaration |
| * @see ClassOrInterfaceDeclaration |
| * @see EmptyTypeDeclaration |
| * @see EnumDeclaration |
| */ |
| public List<TypeDeclaration<?>> getTypes() { |
| types = ensureNotNull(types); |
| return types; |
| } |
| |
| /** |
| * Sets the list of comments of this compilation unit. |
| * |
| * @param comments |
| * the list of comments |
| */ |
| public CompilationUnit setComments(List<Comment> comments) { |
| throw new RuntimeException("Not implemented!"); |
| } |
| |
| /** |
| * Sets the list of imports of this compilation unit. The list is initially |
| * <code>null</code>. |
| * |
| * @param imports |
| * the list of imports |
| */ |
| public CompilationUnit setImports(List<ImportDeclaration> imports) { |
| this.imports = imports; |
| setAsParentNodeOf(this.imports); |
| return this; |
| } |
| |
| /** |
| * Sets or clear the package declarations of this compilation unit. |
| * |
| * @param pakage |
| * the pakage declaration to set or <code>null</code> to default |
| * package |
| */ |
| public CompilationUnit setPackage(PackageDeclaration pakage) { |
| this.pakage = pakage; |
| setAsParentNodeOf(this.pakage); |
| return this; |
| } |
| |
| /** |
| * Sets the list of types declared in this compilation unit. |
| * |
| * @param types |
| * the lis of types |
| */ |
| public CompilationUnit setTypes(List<TypeDeclaration<?>> types) { |
| this.types = types; |
| setAsParentNodeOf(this.types); |
| return this; |
| } |
| |
| /** |
| * sets the package declaration of this compilation unit |
| * |
| * @param name the name of the package |
| * @return this, the {@link CompilationUnit} |
| */ |
| public CompilationUnit setPackageName(String name) { |
| setPackage(new PackageDeclaration(name(name))); |
| return this; |
| } |
| |
| /** |
| * Add an import to the list of {@link ImportDeclaration} of this compilation unit<br> |
| * shorthand for {@link #addImport(String, boolean, boolean)} with name,false,false |
| * |
| * @param name the import name |
| * @return this, the {@link CompilationUnit} |
| */ |
| public CompilationUnit addImport(String name) { |
| return addImport(name, false, false); |
| } |
| |
| /** |
| * Add an import to the list of {@link ImportDeclaration} of this compilation unit<br> |
| * shorthand for {@link #addImport(String)} with clazz.getName() |
| * |
| * @param clazz the class to import |
| * @return this, the {@link CompilationUnit} |
| */ |
| public CompilationUnit addImport(Class<?> clazz) { |
| if (ClassUtils.isPrimitiveOrWrapper(clazz) || clazz.getName().startsWith("java.lang")) |
| return this; |
| else if (clazz.isArray() && !ClassUtils.isPrimitiveOrWrapper(clazz.getComponentType()) |
| && !clazz.getComponentType().getName().startsWith("java.lang")) |
| return addImport(clazz.getComponentType().getName()); |
| return addImport(clazz.getName()); |
| } |
| |
| /** |
| * Add an import to the list of {@link ImportDeclaration} of this compilation unit<br> |
| * <b>This method check if no import with the same name is already in the list</b> |
| * |
| * @param name the import name |
| * @param isStatic is it an "import static" |
| * @param isAsterisk does the import end with ".*" |
| * @return this, the {@link CompilationUnit} |
| */ |
| public CompilationUnit addImport(String name, boolean isStatic, boolean isAsterisk) { |
| if (getImports().stream().anyMatch(i -> i.getName().toString().equals(name))) |
| return this; |
| else { |
| ImportDeclaration importDeclaration = new ImportDeclaration(name(name), isStatic, |
| isAsterisk); |
| getImports().add(importDeclaration); |
| importDeclaration.setParentNode(this); |
| return this; |
| } |
| } |
| |
| /** |
| * Add a public class to the types of this compilation unit |
| * |
| * @param name the class name |
| * @return the newly created class |
| */ |
| public ClassOrInterfaceDeclaration addClass(String name) { |
| return addClass(name, Modifier.PUBLIC); |
| } |
| |
| /** |
| * Add a class to the types of this compilation unit |
| * |
| * @param name the class name |
| * @param modifiers the modifiers (like Modifier.PUBLIC) |
| * @return the newly created class |
| */ |
| public ClassOrInterfaceDeclaration addClass(String name, Modifier... modifiers) { |
| ClassOrInterfaceDeclaration classOrInterfaceDeclaration = new ClassOrInterfaceDeclaration( |
| Arrays.stream(modifiers) |
| .collect(Collectors.toCollection(() -> EnumSet.noneOf(Modifier.class))), |
| false, name); |
| getTypes().add(classOrInterfaceDeclaration); |
| classOrInterfaceDeclaration.setParentNode(this); |
| return classOrInterfaceDeclaration; |
| } |
| |
| /** |
| * Add a public interface class to the types of this compilation unit |
| * |
| * @param name the interface name |
| * @return the newly created class |
| */ |
| public ClassOrInterfaceDeclaration addInterface(String name) { |
| return addInterface(name, Modifier.PUBLIC); |
| } |
| |
| /** |
| * Add an interface to the types of this compilation unit |
| * |
| * @param name the interface name |
| * @param modifiers the modifiers (like Modifier.PUBLIC) |
| * @return the newly created class |
| */ |
| public ClassOrInterfaceDeclaration addInterface(String name, Modifier... modifiers) { |
| ClassOrInterfaceDeclaration classOrInterfaceDeclaration = new ClassOrInterfaceDeclaration( |
| Arrays.stream(modifiers) |
| .collect(Collectors.toCollection(() -> EnumSet.noneOf(Modifier.class))), |
| true, name); |
| getTypes().add(classOrInterfaceDeclaration); |
| classOrInterfaceDeclaration.setParentNode(this); |
| return classOrInterfaceDeclaration; |
| } |
| |
| /** |
| * Add a public enum to the types of this compilation unit |
| * |
| * @param name the enum name |
| * @return the newly created class |
| */ |
| public EnumDeclaration addEnum(String name) { |
| return addEnum(name, Modifier.PUBLIC); |
| } |
| |
| /** |
| * Add an enum to the types of this compilation unit |
| * |
| * @param name the enum name |
| * @param modifiers the modifiers (like Modifier.PUBLIC) |
| * @return the newly created class |
| */ |
| public EnumDeclaration addEnum(String name, Modifier... modifiers) { |
| EnumDeclaration enumDeclaration = new EnumDeclaration(Arrays.stream(modifiers) |
| .collect(Collectors.toCollection(() -> EnumSet.noneOf(Modifier.class))), name); |
| getTypes().add(enumDeclaration); |
| enumDeclaration.setParentNode(this); |
| return enumDeclaration; |
| } |
| |
| /** |
| * Add a public annotation declaration to the types of this compilation unit |
| * |
| * @param name the annotation name |
| * @return the newly created class |
| */ |
| public AnnotationDeclaration addAnnotationDeclaration(String name) { |
| return addAnnotationDeclaration(name, Modifier.PUBLIC); |
| } |
| |
| /** |
| * Add an annotation declaration to the types of this compilation unit |
| * |
| * @param name the annotation name |
| * @param modifiers the modifiers (like Modifier.PUBLIC) |
| * @return the newly created class |
| */ |
| public AnnotationDeclaration addAnnotationDeclaration(String name, Modifier... modifiers) { |
| AnnotationDeclaration annotationDeclaration = new AnnotationDeclaration(Arrays.stream(modifiers) |
| .collect(Collectors.toCollection(() -> EnumSet.noneOf(Modifier.class))), name); |
| getTypes().add(annotationDeclaration); |
| annotationDeclaration.setParentNode(this); |
| return annotationDeclaration; |
| } |
| |
| /** |
| * Try to get a class by its name |
| * |
| * @param className the class name (case-sensitive) |
| * @return null if not found, the class otherwise |
| */ |
| public ClassOrInterfaceDeclaration getClassByName(String className) { |
| return (ClassOrInterfaceDeclaration) getTypes().stream().filter(type -> type.getName().equals(className) |
| && type instanceof ClassOrInterfaceDeclaration && !((ClassOrInterfaceDeclaration) type).isInterface()) |
| .findFirst().orElse(null); |
| } |
| |
| /** |
| * Try to get an interface by its name |
| * |
| * @param interfaceName the interface name (case-sensitive) |
| * @return null if not found, the interface otherwise |
| */ |
| public ClassOrInterfaceDeclaration getInterfaceByName(String interfaceName) { |
| return (ClassOrInterfaceDeclaration) getTypes().stream().filter(type -> type.getName().equals(interfaceName) |
| && type instanceof ClassOrInterfaceDeclaration && ((ClassOrInterfaceDeclaration) type).isInterface()) |
| .findFirst().orElse(null); |
| } |
| |
| /** |
| * Try to get an enum by its name |
| * |
| * @param enumName the enum name (case-sensitive) |
| * @return null if not found, the enum otherwise |
| */ |
| public EnumDeclaration getEnumByName(String enumName) { |
| return (EnumDeclaration) getTypes().stream().filter(type -> type.getName().equals(enumName) |
| && type instanceof EnumDeclaration) |
| .findFirst().orElse(null); |
| } |
| |
| /** |
| * Try to get an annotation by its name |
| * |
| * @param annotationName the annotation name (case-sensitive) |
| * @return null if not found, the annotation otherwise |
| */ |
| public AnnotationDeclaration getAnnotationDeclarationByName(String annotationName) { |
| return (AnnotationDeclaration) getTypes().stream().filter(type -> type.getName().equals(annotationName) |
| && type instanceof AnnotationDeclaration) |
| .findFirst().orElse(null); |
| } |
| } |