/*
* Copyright 2010 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.workspaces.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.TypeOracle;
import com.google.gwt.user.rebind.ClassSourceFileComposerFactory;
import com.google.gwt.user.rebind.SourceWriter;
import org.jboss.errai.bus.rebind.ScannerSingleton;
import org.jboss.errai.bus.server.service.metadata.MetaDataScanner;
import org.jboss.errai.workspaces.client.api.annotations.DefaultBundle;
import org.jboss.errai.workspaces.client.api.annotations.LoadTool;
import java.io.PrintWriter;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
public class ResourceFactoryGenerator 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 Class bundleClass = null;
private Map<String, String> tool2imageRes = new HashMap<String, String>();
private TypeOracle typeOracle;
// inherited generator method
public String generate(TreeLogger logger, GeneratorContext context,
String typeName) throws UnableToCompleteException {
typeOracle = context.getTypeOracle();
try {
// get classType and save instance variables
JClassType classType = typeOracle.getType(typeName);
packageName = classType.getPackage().getName();
className = classType.getSimpleSourceName() + "Impl";
// Generate class source code
generateClass(logger, context);
}
catch (Exception e) {
// record sendNowWith logger that Map generation threw an exception
logger.log(TreeLogger.ERROR, "Error generating icon factory", e);
}
// return the fully qualifed 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 generateClass(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;
// init composer, set class properties, create source writer
ClassSourceFileComposerFactory composer = new ClassSourceFileComposerFactory(packageName,
className);
composer.addImport("java.util.HashMap");
composer.addImport("java.util.Map");
composer.addImport("org.jboss.errai.workspaces.client.api.ResourceFactory");
composer.addImport("com.google.gwt.core.client.GWT");
composer.addImport("com.google.gwt.resources.client.ImageResource");
composer.addImplementedInterface("org.jboss.errai.workspaces.client.api.ResourceFactory");
SourceWriter sourceWriter = composer.createSourceWriter(context, printWriter);
MetaDataScanner scanner = ScannerSingleton.getOrCreateInstance();
/**
* Identify the default bundle (if exists, it's optional)
*/
Set<Class<?>> bundles = scanner.getTypesAnnotatedWith(DefaultBundle.class);
for(Class<?> clazz : bundles)
{
bundleClass = clazz.getAnnotation(DefaultBundle.class).value();
}
/**
* Fetch icon mappings from LoadTool annotation
* This will be used to reference the icon though the ResourceFactory
*/
Set<Class<?>> tools = scanner.getTypesAnnotatedWith(LoadTool.class);
for(Class<?> tool : tools)
{
LoadTool lt = tool.getAnnotation(LoadTool.class);
if (!"".equals(lt.icon()))
tool2imageRes.put(lt.name(), lt.icon());
}
// generator constructor source code
generateFactoryClass(context, logger, sourceWriter);
// close generated class
sourceWriter.outdent();
sourceWriter.println("}");
// commit generated class
context.commit(logger, printWriter);
}
private void generateFactoryClass(
GeneratorContext context,
TreeLogger logger,
SourceWriter sourceWriter) {
sourceWriter.println("private Map<String,ImageResource> mapping = new HashMap<String,ImageResource>();");
// start constructor source generation
sourceWriter.println("public " + className + "() { ");
sourceWriter.indent();
sourceWriter.println("super();");
if (bundleClass != null) // optional
{
sourceWriter.println(bundleClass.getName() + " bundle = (" + bundleClass.getName() + ") GWT.create(" + bundleClass.getName() + ".class);");
for (String tool : tool2imageRes.keySet()) {
sourceWriter.println("mapping.put(\"" + tool + "\", bundle." + tool2imageRes.get(tool) + "() );");
}
sourceWriter.outdent();
} else {
logger.log(TreeLogger.Type.WARN, "\"@DefaultBundle not found. Make sure the EntryPoint refers to a valid default resource bundle.\"");
}
sourceWriter.println("}");
sourceWriter.println("public ImageResource createImage(String name) { ");
sourceWriter.outdent();
sourceWriter.println(" return mapping.get(name);");
sourceWriter.outdent();
sourceWriter.println("}");
}
}