br.com.objectos.way.core.code.jdt.ASTTest.java Source code

Introduction

Here is the source code for br.com.objectos.way.core.code.jdt.ASTTest.java

Source

/*
 * Copyright 2014 Objectos, Fbrica de Software LTDA.
 *
 * 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 br.com.objectos.way.core.code.jdt;

import org.eclipse.jdt.core.dom.AST;
import org.eclipse.jdt.core.dom.Block;
import org.eclipse.jdt.core.dom.CompilationUnit;
import org.eclipse.jdt.core.dom.ExpressionStatement;
import org.eclipse.jdt.core.dom.ImportDeclaration;
import org.eclipse.jdt.core.dom.MethodDeclaration;
import org.eclipse.jdt.core.dom.MethodInvocation;
import org.eclipse.jdt.core.dom.Modifier.ModifierKeyword;
import org.eclipse.jdt.core.dom.PackageDeclaration;
import org.eclipse.jdt.core.dom.SingleVariableDeclaration;
import org.eclipse.jdt.core.dom.TypeDeclaration;
import org.eclipse.jdt.core.dom.TypeParameter;
import org.testng.annotations.Test;

/**
 * @author marcio.endo@objectos.com.br (Marcio Endo)
 */
@Test
public class ASTTest {

    @SuppressWarnings("unchecked")
    public void stackoverflow_answer() {
        AST ast = AST.newAST(AST.JLS8);
        CompilationUnit cu = ast.newCompilationUnit();

        PackageDeclaration p1 = ast.newPackageDeclaration();
        p1.setName(ast.newSimpleName("foo"));
        cu.setPackage(p1);

        ImportDeclaration id = ast.newImportDeclaration();
        id.setName(ast.newName(new String[] { "java", "util", "Set" }));
        cu.imports().add(id);

        TypeDeclaration td = ast.newTypeDeclaration();
        td.setName(ast.newSimpleName("Foo"));
        TypeParameter tp = ast.newTypeParameter();
        tp.setName(ast.newSimpleName("X"));
        td.typeParameters().add(tp);
        cu.types().add(td);

        MethodDeclaration md = ast.newMethodDeclaration();
        md.modifiers().add(ast.newModifier(ModifierKeyword.PUBLIC_KEYWORD));
        md.setName(ast.newSimpleName("bar"));

        SingleVariableDeclaration var = ast.newSingleVariableDeclaration();
        var.setType(ast.newSimpleType(ast.newSimpleName("String")));
        var.setName(ast.newSimpleName("a"));
        md.parameters().add(var);
        td.bodyDeclarations().add(md);

        Block block = ast.newBlock();
        md.setBody(block);

        MethodInvocation mi = ast.newMethodInvocation();
        mi.setName(ast.newSimpleName("x"));

        ExpressionStatement e = ast.newExpressionStatement(mi);
        block.statements().add(e);

        System.out.println(cu);
    }

}