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.util.Collections;
020
021import org.apache.avalon.framework.service.ServiceException;
022import org.apache.avalon.framework.service.ServiceManager;
023import org.apache.cocoon.ProcessingException;
024import org.apache.cocoon.environment.ObjectModelHelper;
025import org.apache.cocoon.environment.Request;
026import org.apache.cocoon.environment.Response;
027import org.apache.cocoon.generation.ServiceableGenerator;
028import org.xml.sax.SAXException;
029
030import org.ametys.odf.program.Program;
031import org.ametys.plugins.repository.AmetysObjectResolver;
032
033/**
034 * Generates entity to LHEO
035 */
036public class ExportToLHEOGenerator extends ServiceableGenerator
037{
038    /** The ametys object resolver */
039    protected AmetysObjectResolver _resolver;
040    
041    /** The export LHEO manager */
042    protected ExportToLHEOManager _exportLHEOManager;
043    
044    @Override
045    public void service(ServiceManager smanager) throws ServiceException
046    {
047        super.service(smanager);
048        _resolver = (AmetysObjectResolver) smanager.lookup(AmetysObjectResolver.ROLE);
049        _exportLHEOManager = (ExportToLHEOManager) smanager.lookup(ExportToLHEOManager.ROLE);
050    }
051    
052    @Override
053    public void generate() throws IOException, SAXException, ProcessingException
054    {
055        Request request = ObjectModelHelper.getRequest(objectModel);
056
057        // Add the program to the request
058        String programId = request.getParameter("id");
059        Program program = _resolver.resolveById(programId);
060        
061        // Configure the response header
062        Response response = ObjectModelHelper.getResponse(objectModel);
063        response.setHeader("Content-Disposition", "attachment; filename=\"LHEO_" + program.getCode() + ".xml" + "\"");
064        
065        _exportLHEOManager.saxLHEO(contentHandler, Collections.singletonList(program));
066    }
067}