org.jboss.as.console.rebind.extensions.SubsystemExtensionProcessorGenerator Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of switchyard-console-as7-core
Bundles the core AS7 console as a GWT module. Includes minor customizations to support extensions.
There is a newer version: 0.7.0.Final
/* 
 * JBoss, Home of Professional Open Source 
 * Copyright 2012 Red Hat Inc. and/or its affiliates and other contributors
 * as indicated by the @author tags. All rights reserved. 
 * See the copyright.txt in the distribution for a 
 * full listing of individual contributors.
 *
 * This copyrighted material is made available to anyone wishing to use, 
 * modify, copy, or redistribute it subject to the terms and conditions 
 * of the GNU Lesser General Public License, v. 2.1. 
 * This program is distributed in the hope that it will be useful, but WITHOUT A 
 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A 
 * PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more details. 
 * You should have received a copy of the GNU Lesser General Public License, 
 * v.2.1 along with this distribution; if not, write to the Free Software 
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, 
 * MA  02110-1301, USA.
 */
package org.jboss.as.console.rebind.extensions;

import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import org.jboss.as.console.client.shared.SubsystemExtension;
import org.jboss.as.console.client.shared.SubsystemExtension.SubsystemGroupDefinition;
import org.jboss.as.console.client.shared.SubsystemExtension.SubsystemItemDefinition;
import org.jboss.as.console.client.shared.SubsystemExtensionProcessor;
import org.jboss.as.console.client.shared.SubsystemGroup;
import org.jboss.as.console.client.shared.SubsystemGroupItem;

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

/**
 * SubsystemExtensionProcessorGenerator
 * 
 * GWT code generator for SubsystemExtensionProcessor. A code generator is used
 * because reflection cannot be used on the client side. This generator scans
 * for classes annotated with SubsystemExtension and adds them to the manager.
 * 
 * @author Rob Cernich
 */
public class SubsystemExtensionProcessorGenerator extends Generator {

    @Override
    public String generate(TreeLogger logger, GeneratorContext context, String typeName)
            throws UnableToCompleteException {
        TypeOracle typeOracle = context.getTypeOracle();
        JClassType subsystemExtensionManager = typeOracle.findType(typeName);
        if (subsystemExtensionManager == null) {
            logger.log(TreeLogger.ERROR, "Unable to find metadata for type '" + typeName + "'", null);
            throw new UnableToCompleteException();
        }
        if (subsystemExtensionManager.isInterface() == null) {
            logger.log(TreeLogger.ERROR, subsystemExtensionManager.getQualifiedSourceName() + " is not an interface",
                    null);
            throw new UnableToCompleteException();
        }

        List subsystemExtensions = new ArrayList();
        for (JClassType type : typeOracle.getTypes()) {
            if (type.isAnnotationPresent(SubsystemExtension.class)) {
                subsystemExtensions.add(type.getAnnotation(SubsystemExtension.class));
            }
        }

        String packageName = subsystemExtensionManager.getPackage().getName();
        String className = subsystemExtensionManager.getSimpleSourceName() + "Impl";

        generateClass(logger, context, packageName, className, subsystemExtensions);

        return packageName + "." + className;
    }

    private void generateClass(TreeLogger logger, GeneratorContext context, String packageName, String className,
            List subsystemExtensions) {
        PrintWriter pw = context.tryCreate(logger, packageName, className);
        if (pw == null) {
            return;
        }

        ClassSourceFileComposerFactory composerFactory = new ClassSourceFileComposerFactory(packageName, className);

        // imports
        composerFactory.addImport(LinkedHashMap.class.getCanonicalName());
        composerFactory.addImport(Map.class.getCanonicalName());
        composerFactory.addImport(SubsystemExtensionProcessor.class.getCanonicalName());
        composerFactory.addImport(SubsystemGroup.class.getCanonicalName());
        composerFactory.addImport(SubsystemGroupItem.class.getCanonicalName());

        // interface
        composerFactory.addImplementedInterface(SubsystemExtensionProcessor.class.getCanonicalName());

        SourceWriter sw = composerFactory.createSourceWriter(context, pw);

        // begin class definition
        sw.indent();

        // fields
        sw.println("private final Map extensionGroups = new LinkedHashMap();");

        // constructor
        sw.println("public " + className + "() {");
        sw.indent();
        for (SubsystemExtension extension : subsystemExtensions) {
            for (SubsystemGroupDefinition groupDef : extension.value()) {
                sw.println("SubsystemGroup group;");
                sw.println("if (extensionGroups.containsKey(\"%s\")) {", groupDef.name());
                sw.indentln("group = extensionGroups.get(\"%s\");", groupDef.name());
                sw.println("} else {");
                sw.indent();
                sw.println("group = new SubsystemGroup(\"%s\");", groupDef.name());
                sw.println("extensionGroups.put(group.getName(), group);");
                sw.outdent();
                sw.println("}");
                for (SubsystemItemDefinition itemDef : groupDef.items()) {
                    sw.println("group.getItems().add(new SubsystemGroupItem(\"%s\", \"%s\", \"%s\"));", itemDef.name(),
                            groupDef.subsystem(), itemDef.presenter());
                }
            }
        }
        sw.outdent();
        sw.println("}");

        // methods
        // processExtensions
        sw.println("public void processExtensions(Map groups) {");
        sw.indent();
        sw.println("for (Map.Entry entry : extensionGroups.entrySet()) {");
        sw.indent();
        sw.println("if (groups.containsKey(entry.getKey())) {");
        sw.indent();
        sw.println("SubsystemGroup group = groups.get(entry.getKey());");
        sw.println("group.getItems().addAll(entry.getValue().getItems());");
        sw.outdent();
        sw.println("} else {");
        sw.indent();
        sw.println("SubsystemGroup group = entry.getValue();");
        sw.println("groups.put(group.getName(), group);");
        sw.outdent();
        sw.println("}");
        sw.outdent();
        sw.println("}");
        sw.outdent();
        sw.println("}");

        // close it out
        sw.outdent();
        sw.println("}");

        context.commit(logger, pw);
    }

}