001/*
002 *  Copyright 2019 Anyware Services
003 *
004 *  Licensed under the Apache License, Version 2.0 (the "License");
005 *  you may not use this file except in compliance with the License.
006 *  You may obtain a copy of the License at
007 *
008 *      http://www.apache.org/licenses/LICENSE-2.0
009 *
010 *  Unless required by applicable law or agreed to in writing, software
011 *  distributed under the License is distributed on an "AS IS" BASIS,
012 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 *  See the License for the specific language governing permissions and
014 *  limitations under the License.
015 */
016package org.ametys.odf.lheo;
017
018import java.io.IOException;
019import java.time.LocalDateTime;
020import java.time.format.DateTimeFormatter;
021import java.util.List;
022import java.util.stream.Collectors;
023
024import org.apache.avalon.framework.service.ServiceException;
025import org.apache.avalon.framework.service.ServiceManager;
026import org.apache.cocoon.ProcessingException;
027import org.apache.cocoon.environment.ObjectModelHelper;
028import org.apache.cocoon.environment.Request;
029import org.apache.cocoon.environment.Response;
030import org.apache.cocoon.generation.ServiceableGenerator;
031import org.xml.sax.SAXException;
032
033import org.ametys.odf.catalog.CatalogsManager;
034import org.ametys.odf.program.Program;
035import org.ametys.plugins.repository.AmetysObjectIterable;
036import org.ametys.plugins.repository.AmetysObjectResolver;
037
038/**
039 * Generates catalog to LHEO
040 */
041public class GlobalExportToLHEOGenerator extends ServiceableGenerator
042{
043    /** The ametys object resolver */
044    protected AmetysObjectResolver _resolver;
045    
046    /** The export LHEO manager */
047    protected ExportToLHEOManager _exportLHEOManager;
048    
049    /** The catalogs manager */
050    protected CatalogsManager _catalogsManager;
051    
052    @Override
053    public void service(ServiceManager smanager) throws ServiceException
054    {
055        super.service(smanager);
056        _resolver = (AmetysObjectResolver) smanager.lookup(AmetysObjectResolver.ROLE);
057        _exportLHEOManager = (ExportToLHEOManager) smanager.lookup(ExportToLHEOManager.ROLE);
058        _catalogsManager = (CatalogsManager) smanager.lookup(CatalogsManager.ROLE);
059    }
060    
061    @Override
062    public void generate() throws IOException, SAXException, ProcessingException
063    {
064        Request request = ObjectModelHelper.getRequest(objectModel);
065
066        String catalog = request.getParameter("catalog");
067        String lang = request.getParameter("lang");
068        
069        AmetysObjectIterable<Program> programs = _catalogsManager.getPrograms(catalog, lang);
070        List<Program> programList = programs.stream().collect(Collectors.toList());
071        
072        // Configure the response header
073        Response response = ObjectModelHelper.getResponse(objectModel);
074        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyyMMddhhmmss");
075        response.setHeader("Content-Disposition", "attachment; filename=\"LHEO_" + catalog + "_" + LocalDateTime.now().format(formatter) + ".xml" + "\"");
076        
077        _exportLHEOManager.saxLHEO(contentHandler, programList);
078    }
079}