/*
 *  Copyright 2018 Anyware Services
 *
 *  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.
 */
package org.ametys.plugins.odfpilotage.report.impl;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.time.LocalDate;
import java.util.Set;

import javax.xml.transform.Result;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.sax.SAXTransformerFactory;
import javax.xml.transform.sax.TransformerHandler;
import javax.xml.transform.stream.StreamResult;

import org.apache.avalon.framework.service.ServiceException;
import org.apache.avalon.framework.service.ServiceManager;
import org.apache.cocoon.xml.AttributesImpl;
import org.apache.cocoon.xml.XMLUtils;
import org.apache.commons.io.FileUtils;
import org.apache.commons.lang3.StringUtils;
import org.xml.sax.SAXException;

import org.ametys.cms.FilterNameHelper;
import org.ametys.core.util.DateUtils;
import org.ametys.odf.program.Program;
import org.ametys.odf.program.SubProgram;
import org.ametys.plugins.odfpilotage.helper.PilotageStatusHelper;

/**
 * Class to generate "règlement des études" extract as DOC.
 */
public class ReglementEtudesExtract extends AbstractExtract
{
    /** The pilotage status helper */
    protected PilotageStatusHelper _pilotageStatusHelper;
    
    @Override
    public void service(ServiceManager manager) throws ServiceException
    {
        super.service(manager);
        _pilotageStatusHelper = (PilotageStatusHelper) manager.lookup(PilotageStatusHelper.ROLE);
    }
    
    @Override
    protected String getType()
    {
        return "reglementetudes";
    }
    
    @Override
    protected void _saxProgram(Program program)
    {
        Set<SubProgram> subPrograms = _odfHelper.getChildSubPrograms(program);
        _saxProgramsTree(subPrograms, program);
    }

    /**
     * Get the report filename for a given program
     * @param program The program
     * @return the file name
     */
    private String _getReportFileName(Program program)
    {
        StringBuilder sb = new StringBuilder();

        sb.append("reglement-etudes-");
        
        // Catalog
        sb.append(program.getCatalog());
        sb.append("-");
        
        // Lang
        sb.append(program.getLanguage());
        sb.append("-");

        sb.append(program.getTitle());
        
        // Code Ametys
        String code = program.getCode();
        if (StringUtils.isNotBlank(code))
        {
            sb.append("-");
            sb.append(code);
        }
        
        // Date
        sb.append("-");
        sb.append(_currentFormattedDate);
        
        return FilterNameHelper.filterName(sb.toString());
    }
    
    private void _saxProgramsTree(Set<SubProgram> subPrograms, Program program)
    {
        SAXTransformerFactory factory = (SAXTransformerFactory) TransformerFactory.newInstance();
        String fileName = _getReportFileName(program);
        
        // Delete old files
        File xmlFile = new File(_tmpFolder, fileName + ".xml");
        FileUtils.deleteQuietly(xmlFile);

        // Write XML file
        try (FileOutputStream fos = new FileOutputStream(xmlFile))
        {
            TransformerHandler handler = factory.newTransformerHandler();

            // Prepare the transformation
            Result result = new StreamResult(fos);
            handler.setResult(result);
            
            handler.startDocument();
            
            AttributesImpl attrs = new AttributesImpl();
            attrs.addCDATAAttribute("type", getType());
            attrs.addCDATAAttribute("date", _reportHelper.getReadableCurrentDate());
            XMLUtils.startElement(handler, "report", attrs);
            
            _saxReport(handler, program, subPrograms);
            
            XMLUtils.endElement(handler, "report");
            handler.endDocument();

            // Convert the report to configured output format
            convertReport(_tmpFolder, fileName, xmlFile);
        }
        catch (Exception e)
        {
            getLogger().error("An error occured while generating 'Règlements des études' extract for program '{}' ({})", program.getTitle(), program.getCode(), e);
        }
        finally
        {
            FileUtils.deleteQuietly(xmlFile);
        }
    }
    
    private void _saxReport(TransformerHandler handler, Program program, Set<SubProgram> subPrograms) throws SAXException
    {
        _saxProgram(handler, program);
        
        for (SubProgram subProgram : subPrograms)
        {
            _saxSubProgram(handler, subProgram);
        }
    }
    
    private void _saxProgram(TransformerHandler handler, Program program) throws SAXException
    {
        AttributesImpl attr = new AttributesImpl();
        attr.addCDATAAttribute("title", program.getTitle());
        
        LocalDate date = _pilotageStatusHelper.getCFVUMCCValidationDate(program);
        if (date != null)
        {
            attr.addCDATAAttribute("CFVUDate", DateUtils.localDateToString(date));
        }
        
        XMLUtils.startElement(handler, "program", attr);
        
        // Catalog
        XMLUtils.createElement(handler, "catalog", program.getCatalog());
        
        // Degree
        String degreeId = program.getDegree();
        if (StringUtils.isNotEmpty(degreeId))
        {
            XMLUtils.createElement(handler, "degree", _refTableHelper.getItemLabel(degreeId, program.getLanguage()));
        }

        // Mention
        String mentionId = program.getMention();
        if (StringUtils.isNotEmpty(mentionId))
        {
            XMLUtils.createElement(handler, "mention", _refTableHelper.getItemLabel(mentionId, program.getLanguage()));
        }
        
        // Domain
        String[] domains = program.getDomain();
        for (String domain : domains)
        {
            AttributesImpl attrs = new AttributesImpl();
            attrs.addCDATAAttribute("code", _refTableHelper.getItemCode(domain));
            XMLUtils.createElement(handler, "domain", attrs, _refTableHelper.getItemLabel(domain, program.getLanguage()));
        }
        
        // Règlement des études
        try
        {
            program.dataToSAX(handler, "reglementEtudes");
        }
        catch (IOException e)
        {
            getLogger().error("Unable to SAX rich-text metadata 'reglementEtudes'.", e);
        }
        
        XMLUtils.endElement(handler, "program");
    }
    
    private void _saxSubProgram(TransformerHandler handler, SubProgram subprogram) throws SAXException
    {
        AttributesImpl attr = new AttributesImpl();
        attr.addCDATAAttribute("title", subprogram.getTitle());
        XMLUtils.startElement(handler, "subprogram", attr);
        
        try
        {
            subprogram.dataToSAX(handler, "reglementEtudes");
        }
        catch (IOException e)
        {
            getLogger().error("Unable to SAX rich-text metadata 'reglementEtudes'.", e);
        }
        
        XMLUtils.endElement(handler, "subprogram");
    }
}