com.gwtplatform.mvp.rebind.ProxyGenerator.java Source code

Introduction

Here is the source code for com.gwtplatform.mvp.rebind.ProxyGenerator.java

Source

/**
 * Copyright 2011 ArcBees 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 com.gwtplatform.mvp.rebind;

import java.io.PrintWriter;

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.JPackage;
import com.google.gwt.core.ext.typeinfo.TypeOracle;
import com.google.gwt.user.rebind.ClassSourceFileComposerFactory;
import com.google.gwt.user.rebind.SourceWriter;

public class ProxyGenerator extends Generator {

    private ClassCollection classCollection;
    private GinjectorInspector ginjectorInspector;
    private PresenterInspector presenterInspector;
    private ProxyOutputterFactory proxyOutputterFactory;

    @Override
    public String generate(TreeLogger logger, GeneratorContext ctx, String requestedClass)
            throws UnableToCompleteException {
        // Initialize dependencies
        TypeOracle oracle = ctx.getTypeOracle();
        classCollection = new ClassCollection(oracle);
        ginjectorInspector = new GinjectorInspector(classCollection, ctx, logger);
        presenterInspector = new PresenterInspector(oracle, logger, classCollection, ginjectorInspector);
        proxyOutputterFactory = new ProxyOutputterFactory(oracle, logger, classCollection, ginjectorInspector,
                presenterInspector);

        // Find the requested class
        JClassType proxyInterface = oracle.findType(requestedClass);

        if (proxyInterface == null) {
            logger.log(TreeLogger.ERROR, "Unable to find metadata for type '" + requestedClass + "'", null);

            throw new UnableToCompleteException();
        }

        // If it's not an interface it's a custom user-made proxy class. Don't use generator.
        if (proxyInterface.isInterface() == null) {
            return null;
        }

        ginjectorInspector.init();
        if (!presenterInspector.init(proxyInterface)) {
            return null;
        }

        // Find the package, build the generated class name.
        JPackage interfacePackage = proxyInterface.getPackage();
        String packageName = interfacePackage == null ? "" : interfacePackage.getName();
        String implClassName = presenterInspector.getPresenterClassName() + proxyInterface.getSimpleSourceName()
                + "Impl";
        String generatedClassName = packageName + "." + implClassName;

        // Create the printWriter
        PrintWriter printWriter = ctx.tryCreate(logger, packageName, implClassName);
        if (printWriter == null) {
            // We've already created it, so nothing to do
            return generatedClassName;
        }

        try {
            ProxyOutputter proxyOutputter = proxyOutputterFactory.create(proxyInterface);

            // Start composing the class
            ClassSourceFileComposerFactory composerFactory = new ClassSourceFileComposerFactory(packageName,
                    implClassName);
            proxyOutputter.initComposerFactory(composerFactory);

            // Get a source writer
            SourceWriter writer = composerFactory.createSourceWriter(ctx, printWriter);

            proxyOutputter.writeFields(writer);
            proxyOutputter.writeInnerClasses(writer);
            proxyOutputter.writeConstructor(writer, implClassName, true);
            proxyOutputter.writeMethods(writer);

            writer.commit(logger);

            return generatedClassName;
        } finally {
            printWriter.close();
        }
    }

}