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.FileReader;

 import java.io.FileWriter;

 import java.io.PrintWriter;

 import java.io.Writer;

 import java.util.ArrayList;

 import java.util.Iterator;

 import org.apache.ddlutils.model.Database;

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

 import org.apache.tools.ant.BuildException;

 import org.apache.tools.ant.DirectoryScanner;

 import org.apache.tools.ant.Project;

 import org.apache.tools.ant.Task;

 import org.apache.tools.ant.types.FileSet;

 /**

  * Command to write the SQL used to insert data defined in terms of the repository model into database, into an XML file.

  * 

  * @author Thomas Dudziak

  * @version $Revision: 1.1 $

  */

 public class WriteDataSqlToFileCommand extends Command

 {

     /** A single data file to read. */

     private File _singleDataFile = null;

     /** The input data files. */

     private ArrayList _fileSets = new ArrayList();

     /** The file to output the data sql to. */

     private File _outputFile;

     /**

      * Adds a fileset specifying data files.

      * 

      * @param fileset The additional input files

      */

     public void addConfiguredFileset(FileSet fileset)

     {

         _fileSets.add(fileset);

     }

     /**

      * Set the xml file containing the data.

      *

      * @param schemaFile The data xml file

      */

     public void setDataFile(File dataFile)

     {

         _singleDataFile = dataFile;

     }

     /**

      * Sets the file to output the data to.

      * 

      * @param outputFile The output file

      */

     public void setOutputFile(File outputFile)

     {

         _outputFile = outputFile;

     }

     /**

      * {@inheritDoc}

      */

     public void execute(Task task, Database dbModel, DescriptorRepository objModel) throws BuildException

     {

         try

         {

             DdlUtilsDataHandling handling = new DdlUtilsDataHandling();

             PrintWriter          writer   = new PrintWriter(new FileWriter(_outputFile), true);

             handling.setModel(dbModel, objModel);

             handling.setPlatform(getPlatform());

             if (_singleDataFile != null)

             {

                 readSingleDataFile(task, handling, _singleDataFile, writer);

             }

             else

             {

                 for (Iterator it = _fileSets.iterator(); it.hasNext();)

                 {

                     FileSet          fileSet    = (FileSet)it.next();

                     File             fileSetDir = fileSet.getDir(task.getProject());

                     DirectoryScanner scanner    = fileSet.getDirectoryScanner(task.getProject());

                     String[]         files      = scanner.getIncludedFiles();

                     for (int idx = 0; (files != null) && (idx < files.length); idx++)

                     {

                         readSingleDataFile(task, handling, new File(fileSetDir, files[idx]), writer);

                     }

                 }

             }

         }

         catch (Exception ex)

         {

             if (ex instanceof BuildException)

             {

                 throw (BuildException)ex;

             }

             else

             {

                 throw new BuildException(ex);

             }

         }

     }

     /**

      * Reads a single data file.

      * 

      * @param task       The parent task

      * @param reader     The data reader

      * @param schemaFile The schema file

      */

     private void readSingleDataFile(Task task, DdlUtilsDataHandling handling, File dataFile, Writer output)

     {

         if (!dataFile.exists())

         {

             task.log("Could not find data file "+dataFile.getAbsolutePath(), Project.MSG_ERR);

         }

         else if (!dataFile.isFile())

         {

             task.log("Path "+dataFile.getAbsolutePath()+" does not denote a data file", Project.MSG_ERR);

         }

         else if (!dataFile.canRead())

         {

             task.log("Could not read data file "+dataFile.getAbsolutePath(), Project.MSG_ERR);

         }

         else

         {

             try

             {

                 FileReader reader = new FileReader(dataFile.getAbsolutePath());

                 handling.getInsertDataSql(reader, output);

                 output.flush();

                 output.close();

                 task.log("Read data file "+dataFile.getAbsolutePath(), Project.MSG_INFO);

             }

             catch (Exception ex)

             {

                 if (isFailOnError())

                 {

                     throw new BuildException("Could not read data file "+dataFile.getAbsolutePath(), ex);

                 }

             }

         }

     }

 }