package org.apache.ojb.broker.metadata.torque;

 /* Copyright 2002-2005 The Apache Software Foundation

  *

  * 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.

  */

 import org.apache.ojb.broker.metadata.DescriptorRepository;

 import org.apache.ojb.broker.metadata.RepositoryPersistor;

 import java.io.FileWriter;

 import java.io.IOException;

 import java.util.Iterator;

 public class TorqueRepositoryGenerator

 {

     private TorqueTablePreprocessor tablePreprocessor;

     private TorqueTableGenerator tableGenerator;

     public TorqueRepositoryGenerator(String xmlInputFile, boolean ignoreAutoIncrement) throws Exception

     {

         RepositoryPersistor persistor = new RepositoryPersistor();

         DescriptorRepository repository = persistor.readDescriptorRepository(xmlInputFile);

         this.tablePreprocessor = new TorqueTablePreprocessor(repository);

         this.tablePreprocessor.buildStandardTables();

         this.tableGenerator = new TorqueTableGenerator(repository, ignoreAutoIncrement);

     }

     public void generateTorqueRepository(String xmlOutputFile, String databaseName, String indexTablespaceName) throws Exception

     {

         Iterator tableDescriptorIterators = this.tablePreprocessor.getStandardTables().values().iterator();

         StringBuffer buffer = new StringBuffer(4096);

         generateDatabaseHeader(databaseName, indexTablespaceName, buffer);

         while (tableDescriptorIterators.hasNext())

         {

             this.tableGenerator.generateStandardTable((TableDescriptor) tableDescriptorIterators.next(), buffer, indexTablespaceName);

         }

         this.tableGenerator.generateMappingTables(buffer, indexTablespaceName);

         buffer.append("</database>\n");

         generateOutputFile(xmlOutputFile, buffer);

     }

     private void generateDatabaseHeader(String databaseName, String indexTablespaceName, StringBuffer buffer)

     {

         buffer.append("<?xml version=\"1.0\" encoding=\"ISO-8859-1\" standalone=\"yes\" ?>\n");

         buffer.append("<!DOCTYPE database>\n\n");

         buffer.append("<database name=\"");

         buffer.append(databaseName);

         buffer.append("\" indexTablespace=\"");

         buffer.append(indexTablespaceName);

         buffer.append("\">\n");

     }

     private void generateOutputFile(String xmlOutputFile, StringBuffer buffer) throws IOException

     {

         FileWriter writer = new FileWriter(xmlOutputFile);

         writer.write(buffer.toString());

         writer.flush();

         writer.close();

     }

     public static void main(String args[]) throws Exception

     {

         if (args.length >= 4 && args.length <= 5)

         {

             boolean ignoreAutoIncrement = (args.length == 5 && args[4].equalsIgnoreCase("true"));

             TorqueRepositoryGenerator generator = new TorqueRepositoryGenerator(args[0], ignoreAutoIncrement);

             generator.generateTorqueRepository(args[1], args[2], args[3]);

         }

         else

         {

             System.out.println("[TorqueRepositoryGenerator] Usage: inputFile outputFile databaseName indexTablespaceName (ignoreAutoIncrement)");

         }

     }

 }