001/*
002 *  Copyright 2017 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.plugins.extraction.execution;
017
018import java.io.File;
019import java.util.Map;
020
021import org.apache.avalon.framework.parameters.Parameters;
022import org.apache.avalon.framework.service.ServiceException;
023import org.apache.avalon.framework.service.ServiceManager;
024import org.apache.cocoon.environment.Redirector;
025import org.apache.cocoon.environment.Request;
026import org.apache.cocoon.environment.SourceResolver;
027import org.apache.excalibur.source.TraversableSource;
028
029import org.ametys.core.right.RightManager;
030import org.ametys.core.right.RightManager.RightResult;
031import org.ametys.core.user.CurrentUserProvider;
032import org.ametys.core.util.URIUtils;
033import org.ametys.plugins.core.ui.parameter.files.AbstractGetFilesAction;
034import org.ametys.plugins.extraction.ExtractionConstants;
035import org.ametys.runtime.util.AmetysHomeHelper;
036
037/**
038 * Action for getting extraction result files
039 */
040public class GetExtractionResultFilesAction extends AbstractGetFilesAction
041{
042    private RightManager _rightManager;
043    private CurrentUserProvider _currentUserProvider;
044    
045    @Override
046    public void service(ServiceManager serviceManager) throws ServiceException
047    {
048        super.service(serviceManager);
049        _rightManager = (RightManager) serviceManager.lookup(RightManager.ROLE);
050        _currentUserProvider = (CurrentUserProvider) serviceManager.lookup(CurrentUserProvider.ROLE);
051    }
052    
053    @Override
054    public Map act(Redirector redirector, SourceResolver resolver, Map objectModel, String source, Parameters parameters) throws Exception
055    {
056        if (_rightManager.hasRight(_currentUserProvider.getUser(), ExtractionConstants.EXECUTE_EXTRACTION_RIGHT_ID, "/${WorkspaceName}") != RightResult.RIGHT_ALLOW)
057        {
058            String errorMessage = "User " + _currentUserProvider.getUser() + " try to list extraction result files with no sufficient rights"; 
059            getLogger().error(errorMessage);
060            throw new IllegalStateException(errorMessage);
061        }
062        
063        return super.act(redirector, resolver, objectModel, source, parameters);
064    }
065    
066    @Override
067    protected String getRootURI(Request request)
068    {
069        return new File(AmetysHomeHelper.getAmetysHomeData(), ExtractionConstants.RESULT_EXTRACTION_DIR_NAME).toURI().toString();
070    }
071    
072    @Override
073    protected Map<String, Object> _resource2JsonObject(TraversableSource file, TraversableSource root)
074    {
075        Map<String, Object> jsonObject = super._resource2JsonObject(file, root);
076        
077        String path = (String) jsonObject.get("path");
078        jsonObject.put("downloadUrl", URIUtils.encodePath(path));
079        
080        return jsonObject;
081    }
082}