user/src/com/google/gwt/uibinder/rebind/BundleWriter.java

/*
 * Copyright 2009 Google 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.google.gwt.uibinder.rebind;

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.resources.client.ClientBundle;
import com.google.gwt.resources.client.DataResource;
import com.google.gwt.resources.client.ImageResource;
import com.google.gwt.resources.client.ImageResource.ImageOptions;
import com.google.gwt.resources.client.ImageResource.RepeatStyle;
import com.google.gwt.uibinder.rebind.model.ImplicitClientBundle;
import com.google.gwt.uibinder.rebind.model.ImplicitCssResource;
import com.google.gwt.uibinder.rebind.model.ImplicitDataResource;
import com.google.gwt.uibinder.rebind.model.ImplicitImageResource;

/**
 * Writes source implementing an {@link ImplicitClientBundle}.
 */
public class BundleWriter {

  private final ImplicitClientBundle bundleClass;
  private final IndentedWriter writer;
  private final PrintWriterManager writerManager;
  private final TypeOracle oracle;

  private final JClassType clientBundleType;
  private final JClassType dataResourceType;
  private final JClassType imageOptionType;
  private final JClassType imageResourceType;
  private final JClassType repeatStyleType;

  public BundleWriter(ImplicitClientBundle bundleClass,
      PrintWriterManager writerManager, TypeOracle oracle,
      PrintWriterManager printWriterProvider) {
    this.bundleClass = bundleClass;
    this.writer = new IndentedWriter(
        writerManager.makePrintWriterFor(bundleClass.getClassName()));
    this.writerManager = writerManager;
    this.oracle = oracle;

    clientBundleType = oracle.findType(ClientBundle.class.getName());
    dataResourceType = oracle.findType(DataResource.class.getCanonicalName());
    imageOptionType = oracle.findType(ImageOptions.class.getCanonicalName());
    imageResourceType = oracle.findType(ImageResource.class.getCanonicalName());
    repeatStyleType = oracle.findType(RepeatStyle.class.getCanonicalName());
  }

  public void write() throws UnableToCompleteException {
    writeBundleClass();
    for (ImplicitCssResource css : bundleClass.getCssMethods()) {
      new CssResourceWriter(css, oracle,
          writerManager.makePrintWriterFor(css.getClassName())).write();
    }
  }

  private void writeBundleClass() {
    // Package declaration
    String packageName = bundleClass.getPackageName();
    if (packageName.length() > 0) {
      writer.write("package %1$s;", packageName);
      writer.newline();
    }

    // Imports
    writer.write("import %s;", imageResourceType.getQualifiedSourceName());
    writer.write("import %s;", imageOptionType.getQualifiedSourceName());
    writer.write("import %s;", clientBundleType.getQualifiedSourceName());
    writer.write("import %s;", dataResourceType.getQualifiedSourceName());
    writer.newline();

    // Open interface
    writer.write("public interface %s extends ClientBundle {",
        bundleClass.getClassName());
    writer.indent();

    // Write css methods
    for (ImplicitCssResource css : bundleClass.getCssMethods()) {
      writer.write("@Source(\"%s\")", css.getSource());
      writer.write("%s %s();", css.getClassName(), css.getName());
      writer.newline();
    }
    
    // Write data methods
    for (ImplicitDataResource data : bundleClass.getDataMethods()) {
      writer.write("@Source(\"%s\")", data.getSource());
      writer.write("%s %s();", dataResourceType.getName(), data.getName());
      writer.newline();
    }

    writeImageMethods();

    // Close interface.
    writer.outdent();
    writer.write("}");
  }

  private void writeImageMethods() {
    for (ImplicitImageResource image : bundleClass.getImageMethods()) {
      if (null != image.getSource()) {
        writer.write("@Source(\"%s\")", image.getSource());
      }

      writeImageOptionsAnnotation(image.getFlipRtl(), image.getRepeatStyle());
      writer.write("%s %s();", imageResourceType.getName(), image.getName());
    }
  }

  private void writeImageOptionsAnnotation(Boolean flipRtl,
      RepeatStyle repeatStyle) {
    if (flipRtl != null || repeatStyle != null) {
      StringBuilder b = new StringBuilder("@ImageOptions(");
      if (null != flipRtl) {
        b.append("flipRtl=").append(flipRtl);
        if (repeatStyle != null) {
          b.append(", ");
        }
      }
      if (repeatStyle != null) {
        b.append(String.format("repeatStyle=%s.%s", repeatStyleType.getName(),
            repeatStyle.toString()));
      }
      b.append(")");
      writer.write(b.toString());
    }
  }
}