package org.apache.ojb.broker.ant;
   
   /* Copyright 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 java.io.File;
  import java.io.FileWriter;
  
  import org.apache.ddlutils.io.DataDtdWriter;
  import org.apache.ddlutils.model.Database;
  import org.apache.ojb.broker.metadata.DescriptorRepository;
  import org.apache.tools.ant.BuildException;
  import org.apache.tools.ant.Project;
  import org.apache.tools.ant.Task;
  
  /**
   * The command for creating a data DTD for a given database model.
   * 
   * @author Thomas Dudziak
   * @version $Revision: 1.1 $
   */
  public class WriteDtdToFileCommand extends Command
  {
      /** The file to output the DTD to. */
      private File _outputFile;
  
      /**
       * Sets the file to output the DTD to.
       * 
       * @param outputFile The output file
       */
      public void setOutputFile(File outputFile)
      {
          _outputFile = outputFile;
      }
  
      /**
       * {@inheritDoc}
       */
      public boolean isRequiringModel()
      {
          return true;
      }
  
      /**
       * {@inheritDoc}
       */
      public void execute(Task task, Database dbModel, DescriptorRepository objModel) throws BuildException
      {
          if (_outputFile == null)
          {
              throw new BuildException("No output file specified");
          }
          if (_outputFile.exists() && !_outputFile.canWrite())
          {
              throw new BuildException("Cannot overwrite output file "+_outputFile.getAbsolutePath());
          }
  
          try
          {
              FileWriter           outputWriter = new FileWriter(_outputFile);
              DataDtdWriter        dtdWriter    = new DataDtdWriter();
              DdlUtilsDataHandling handling     = new DdlUtilsDataHandling();
  
              handling.setModel(dbModel, objModel);
              handling.getDataDTD(outputWriter);
              outputWriter.close();
              task.log("Written DTD to "+_outputFile.getAbsolutePath(), Project.MSG_INFO);
          }
          catch (Exception ex)
          {
              throw new BuildException("Failed to write to output file "+_outputFile.getAbsolutePath(), ex);
          }
      }
  }