package org.apache.maven.internal.aether;

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.function.Consumer;

import javax.xml.parsers.ParserConfigurationException;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerConfigurationException;
import javax.xml.transform.sax.SAXTransformerFactory;
import javax.xml.transform.sax.TransformerHandler;

import org.apache.maven.model.building.AbstractModelSourceTransformer;
import org.apache.maven.model.building.DefaultBuildPomXMLFilterFactory;
import org.apache.maven.model.building.TransformerContext;
import org.apache.maven.xml.internal.DefaultConsumerPomXMLFilterFactory;
import org.apache.maven.xml.sax.filter.AbstractSAXFilter;
import org.xml.sax.SAXException;
import org.xml.sax.ext.LexicalHandler;

class ConsumerModelSourceTransformer extends AbstractModelSourceTransformer
{
    @Override
    protected AbstractSAXFilter getSAXFilter( Path pomFile,
                                              TransformerContext context,
                                              Consumer<LexicalHandler> lexicalHandlerConsumer )
        throws TransformerConfigurationException, SAXException, ParserConfigurationException
    {
        return new DefaultConsumerPomXMLFilterFactory( new DefaultBuildPomXMLFilterFactory( context,
                                                                        lexicalHandlerConsumer, true ) ).get( pomFile );
    }
    
    /**
     * This transformer will ensure that encoding and version are kept.
     * However, it cannot prevent:
     * <ul>
     *   <li>attributes will be on one line</li>
     *   <li>Unnecessary whitespace before the rootelement will be removed</li> 
     * </ul>
     */
    @Override
    protected TransformerHandler getTransformerHandler( Path pomFile )
        throws IOException, org.apache.maven.model.building.TransformerException
    {
        final TransformerHandler transformerHandler;
        
        final SAXTransformerFactory transformerFactory = getTransformerFactory();
        
        // Keep same encoding+version
        try ( InputStream input = Files.newInputStream( pomFile ) )
        {
            XMLStreamReader streamReader =
                XMLInputFactory.newFactory().createXMLStreamReader( input );

            transformerHandler = transformerFactory.newTransformerHandler();

            final String encoding = streamReader.getCharacterEncodingScheme();
            final String version = streamReader.getVersion();
            
            Transformer transformer = transformerHandler.getTransformer();
            transformer.setOutputProperty( OutputKeys.METHOD, "xml" );
            if ( encoding == null && version == null )
            {
                transformer.setOutputProperty( OutputKeys.OMIT_XML_DECLARATION, "yes" );
            }
            else
            {
                transformer.setOutputProperty( OutputKeys.OMIT_XML_DECLARATION, "no" );

                if ( encoding != null )
                {
                    transformer.setOutputProperty( OutputKeys.ENCODING, encoding );
                }
                if ( version != null )
                {
                    transformer.setOutputProperty( OutputKeys.VERSION, version );
                }
            }
        }
        catch ( XMLStreamException | TransformerConfigurationException e )
        {
            throw new org.apache.maven.model.building.TransformerException( 
                               "Failed to detect XML encoding and version", e );
        }
        return transformerHandler;
    }

}