v2/src/binding/org/apache/xmlbeans/impl/binding/compile/ExplodedTylarBuilder.java

/*
* The Apache Software License, Version 1.1
*
*
* Copyright (c) 2003 The Apache Software Foundation.  All rights
* reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
*
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
*
* 2. Redistributions in binary form must reproduce the above copyright
*    notice, this list of conditions and the following disclaimer in
*    the documentation and/or other materials provided with the
*    distribution.
*
* 3. The end-user documentation included with the redistribution,
*    if any, must include the following acknowledgment:
*       "This product includes software developed by the
*        Apache Software Foundation (http://www.apache.org/)."
*    Alternately, this acknowledgment may appear in the software itself,
*    if and wherever such third-party acknowledgments normally appear.
*
* 4. The names "Apache" and "Apache Software Foundation" must
*    not be used to endorse or promote products derived from this
*    software without prior written permission. For written
*    permission, please contact apache@apache.org.
*
* 5. Products derived from this software may not be called "Apache
*    XMLBeans", nor may "Apache" appear in their name, without prior
*    written permission of the Apache Software Foundation.
*
* THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
* WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
* ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
* USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
* ====================================================================
*
* This software consists of voluntary contributions made by many
* individuals on behalf of the Apache Software Foundation and was
* originally based on software copyright (c) 2003 BEA Systems
* Inc., <http://www.bea.com/>. For more information on the Apache Software
* Foundation, please see <http://www.apache.org/>.
*/
package org.apache.xmlbeans.impl.binding.compile;

import org.w3.x2001.xmlSchema.SchemaDocument;
import org.apache.xmlbeans.impl.binding.bts.BindingFile;
import org.apache.xmlbeans.x2003.x09.bindingConfig.BindingConfigDocument;
import org.apache.xmlbeans.XmlOptions;
import org.apache.tools.ant.BuildException;

import java.io.IOException;
import java.io.File;
import java.io.FileOutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Iterator;

/**
 * Simple implementation of TylarBuilder that just dumps everything in a
 * directory.
 *
 * @author Patrick Calahan <pcal@bea.com>
 */
public class ExplodedTylarBuilder implements TylarBuilder {

  // =========================================================================
  // Constants

  private static final int XML_INDENT = 2;

  // =========================================================================
  // Variables

  private File mDir;
  private List mSchemas = new ArrayList();
  private List mBindings = new ArrayList();

  // =========================================================================
  // Constructors

  public ExplodedTylarBuilder(File dir) {
    if (dir == null) throw new IllegalArgumentException("null dir");
    if (!dir.exists()) {
      if (!dir.mkdirs()) {
        throw new IllegalArgumentException("failed to create dir "+dir);
      }
    } else {
      if (!dir.isDirectory())
        throw new IllegalArgumentException("not a directory: "+dir);
    }
    mDir = dir;
  }

  // =========================================================================
  // TylarBuilder implementation

  public void addSchema(SchemaDocument sd) {
    mSchemas.add(sd);
  }

  public void addBindingFile(BindingFile bf) {
    mBindings.add(bf);
  }

  public void complete() throws IOException {
    writeBindingFiles();
    writeSchemas();
  }

  // =========================================================================
  // Private methods

  private void writeBindingFiles() throws IOException {
    int num = 0; // FIXME need real naming rules
    for(Iterator i = mBindings.iterator(); i.hasNext(); num++) {
      BindingFile bf = (BindingFile)i.next();
      BindingConfigDocument doc = bf.write();
      File file = new File(mDir,"binding-file"+(num != 0 ? ""+num : "")+".xml");
      FileOutputStream out;
      out = new FileOutputStream(file);
      try {
        doc.save(out,
                 new XmlOptions().setSavePrettyPrint().
                 setSavePrettyPrintIndent(XML_INDENT));
      } catch(IOException e) {
        throw e;
      } finally {
        try {
          out.close();
        } catch(IOException ioe) {
          ioe.printStackTrace();
        }
      }
    }
  }

  private void writeSchemas() throws IOException {
    int num = 0; // FIXME need real naming rules
    for(Iterator i = mSchemas.iterator(); i.hasNext(); num++) {
      SchemaDocument sd = (SchemaDocument)i.next();
      File file = new File(mDir,"schema"+(num != 0 ? ""+num : "")+".xsd");
      FileOutputStream out;
      out = new FileOutputStream(file);
      try {
        sd.save(out,
                new XmlOptions().setSavePrettyPrint().
                setSavePrettyPrintIndent(XML_INDENT));
      } catch(IOException e) {
        throw e;
      } finally {
        try {
          out.close();
        } catch(IOException ioe) {
          ioe.printStackTrace();
        }
      }
    }
  }
}