package org.apache.maven.hibernate.beans;

/* ====================================================================
 *   Copyright 2001-2004 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 java.util.Iterator;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.maven.hibernate.HibernateEntityResolver;
import org.dom4j.Document;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;

/**
 * 
 * @author <a href="as851@columbia.edu">Alex Shneyderman</a>
 * @since  $Id: MappingsAggregatorBean.java 170200 2005-05-15 06:24:19Z brett $
 */
public class MappingsAggregatorBean extends CommonOperationsBean
{

	private String aggregateOutputFile = null;
	
	private static final Log LOG = LogFactory.getLog(MappingsAggregatorBean.class);

    public void execute()
        throws Exception
    {
        String version = null;
        File files[] = getFileDescriptors();
        if(files == null || files.length <= 0)
        {
            LOG.info("nothing to process");
            return;
        }
        LOG.info("Aggregating to " + getAggregateOutputFile());
        File f = new File(getAggregateOutputFile());
        OutputFormat format = OutputFormat.createPrettyPrint();
        XMLWriter writer = new XMLWriter(new FileWriter(f), format);
        writer.setEntityResolver(new HibernateEntityResolver());
        //writer.setResolveEntityRefs(false);
        Document finalDoc = DocumentHelper.createDocument();
        Element rootHM = null;
        for(int i = 0; i < files.length; i++)
        {
            SAXReader reader = new SAXReader(false);
            reader.setEntityResolver(new HibernateEntityResolver());
            //reader.setIncludeExternalDTDDeclarations(false);
            //reader.setIncludeExternalDTDDeclarations(false);
            Document current = reader.read(files[i]);
            String currentVersion = getVersion(current);
            if(version == null)
            {
                version = currentVersion;
                finalDoc.setProcessingInstructions(current.processingInstructions());
                finalDoc.setDocType(current.getDocType());
                rootHM = finalDoc.addElement("hibernate-mapping");
            } else
            if(!version.equals(currentVersion))
            {
                LOG.warn("Mapping in " + files[i].getName() + " is not of the same mapping version as " + files[0].getName() + " mapping, so merge is impossible. Skipping");
                continue;
            }
	    for(Iterator iter = current.selectSingleNode("hibernate-mapping").selectNodes("meta").iterator(); iter.hasNext(); rootHM.add((Element)((Element)iter.next()).clone()));
	    for(Iterator iter = current.selectSingleNode("hibernate-mapping").selectNodes("import").iterator(); iter.hasNext(); rootHM.add((Element)((Element)iter.next()).clone()));
            for(Iterator iter = current.selectSingleNode("hibernate-mapping").selectNodes("class").iterator(); iter.hasNext(); rootHM.add((Element)((Element)iter.next()).clone()));
	    for(Iterator iter = current.selectSingleNode("hibernate-mapping").selectNodes("subclass").iterator(); iter.hasNext(); rootHM.add((Element)((Element)iter.next()).clone()));
	   for(Iterator iter = current.selectSingleNode("hibernate-mapping").selectNodes("joined-subclass").iterator(); iter.hasNext(); rootHM.add((Element)((Element)iter.next()).clone()));
           for(Iterator iter = current.selectSingleNode("hibernate-mapping").selectNodes("query").iterator(); iter.hasNext(); rootHM.add((Element)((Element)iter.next()).clone()));
	   for(Iterator iter = current.selectSingleNode("hibernate-mapping").selectNodes("sql-query").iterator(); iter.hasNext(); rootHM.add((Element)((Element)iter.next()).clone()));

        }

        writer.write(finalDoc);
        writer.close();
    }

    public String getAggregateOutputFile()
    {
        return aggregateOutputFile;
    }

    public void setAggregateOutputFile(String aggregateOutputFile)
    {
        this.aggregateOutputFile = aggregateOutputFile;
    }

    private String getVersion(Document current)
    {
        String docType = current.getDocType().getText();
        if(docType == null || "".equals(docType.trim()))
            return "";
        if(docType.indexOf("hibernate-mapping-2.0.dtd") > 0)
            return "2.0";
        if(docType.indexOf("hibernate-mapping-1.1.dtd") > 0)
            return "1.1";
        else
            return null;
    }

}