org.jboss.errai.ioc.rebind.IOCGenerator.java Source code

Introduction

Here is the source code for org.jboss.errai.ioc.rebind.IOCGenerator.java

Source

/*
 * Copyright 2011 JBoss, a divison Red Hat, Inc
 *
 * 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.jboss.errai.ioc.rebind;

import com.google.gwt.core.ext.Generator;
import com.google.gwt.core.ext.GeneratorContext;
import com.google.gwt.core.ext.TreeLogger;
import com.google.gwt.core.ext.UnableToCompleteException;
import com.google.gwt.core.ext.typeinfo.JClassType;
import com.google.gwt.core.ext.typeinfo.NotFoundException;
import com.google.gwt.core.ext.typeinfo.TypeOracle;
import com.google.gwt.dev.cfg.ModuleDef;
import com.google.gwt.dev.javac.StandardGeneratorContext;
import org.apache.http.impl.client.TunnelRefusedException;
import org.jboss.errai.bus.server.ErraiBootstrapFailure;
import org.jboss.errai.bus.server.service.ErraiServiceConfigurator;
import org.jboss.errai.ioc.rebind.ioc.bootstrapper.IOCBootstrapGenerator;
import org.jboss.errai.ioc.rebind.ioc.codegen.meta.MetaClass;
import org.jboss.errai.ioc.rebind.ioc.codegen.meta.MetaClassFactory;

import java.io.PrintWriter;
import java.lang.reflect.Field;
import java.util.Iterator;
import java.util.Properties;

/**
 * The main generator class for the errai-ioc framework.
 */
public class IOCGenerator extends Generator {
    /**
     * Simple name of class to be generated
     */
    private String className = null;

    /**
     * Package name of class to be generated
     */
    private String packageName = null;
    private TypeOracle typeOracle;

    private String modulePackage;

    public static final boolean isDebugCompile = Boolean.getBoolean("errai.ioc.debug");

    public IOCGenerator() {

    }

    @Override
    public String generate(TreeLogger logger, GeneratorContext context, String typeName)
            throws UnableToCompleteException {
        typeOracle = context.getTypeOracle();

        /**
         * Try to determine the module package -- hackishly
         */
        //TODO: Find a more standard way to do this.
        try {
            if (context instanceof StandardGeneratorContext) {
                StandardGeneratorContext stdContext = (StandardGeneratorContext) context;
                Field field = StandardGeneratorContext.class.getDeclaredField("module");
                field.setAccessible(true);

                ModuleDef moduleDef = (ModuleDef) field.get(stdContext);

                String moduleName = moduleDef.getName();

                for (int i = 0; i < moduleName.length(); i++) {
                    if (moduleName.charAt(i) == '.' && i < moduleName.length()
                            && Character.isUpperCase(moduleName.charAt(i + 1))) {
                        this.modulePackage = moduleName.substring(0, i);
                        break;
                    }
                }

                logger.log(TreeLogger.INFO, "will scan in package: " + modulePackage);
            }
        } catch (Exception e) {
            // could not determine package.
        }

        try {
            // get classType and save instance variables

            JClassType classType = typeOracle.getType(typeName);
            packageName = classType.getPackage().getName();
            className = classType.getSimpleSourceName() + "Impl";

            logger.log(TreeLogger.INFO, "Generating Extensions Bootstrapper...");

            // Generate class source code
            generateIOCBootstrapClass(logger, context);
        } catch (Throwable e) {
            // record sendNowWith logger that Map generation threw an exception
            e.printStackTrace();
            logger.log(TreeLogger.ERROR, "Error generating extensions", e);
        }

        // return the fully qualified name of the class generated
        return packageName + "." + className;
    }

    /**
     * Generate source code for new class. Class extends
     * <code>HashMap</code>.
     *
     * @param logger  Logger object
     * @param context Generator context
     */
    private void generateIOCBootstrapClass(TreeLogger logger, GeneratorContext context) {
        // get print writer that receives the source code
        PrintWriter printWriter = context.tryCreate(logger, packageName, className);
        // print writer if null, source code has ALREADY been generated,

        if (printWriter == null)
            return;

        IOCBootstrapGenerator iocBootstrapGenerator = new IOCBootstrapGenerator(typeOracle, context, logger);
        if (modulePackage != null && modulePackage.length() != 0) {
            iocBootstrapGenerator.setPackageFilter(modulePackage);
        }

        printWriter.append(iocBootstrapGenerator.generate(packageName, className));
        // commit generated class
        context.commit(logger, printWriter);
    }

    public MetaClass getJClassType(Class cls) {
        try {
            return MetaClassFactory.get(typeOracle.getType(cls.getName()));
        } catch (NotFoundException e) {
            return null;
        }
    }
}