/*
 *  Copyright 2010 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.web.editor;

import java.util.HashMap;
import java.util.Map;

import org.apache.avalon.framework.service.ServiceException;
import org.apache.avalon.framework.service.ServiceManager;
import org.apache.cocoon.components.ContextHelper;
import org.apache.cocoon.environment.Request;

import org.ametys.cms.transformation.URIResolver;
import org.ametys.core.util.FilenameUtils;
import org.ametys.core.util.URIUtils;
import org.ametys.plugins.repository.UnknownAmetysObjectException;
import org.ametys.plugins.repository.metadata.RichText;
import org.ametys.web.URIPrefixHandler;
import org.ametys.web.WebHelper;

/**
 * {@link URIResolver} for type "local".resources local to a Content
 * These links or images point resources local to a Content.
 */
public class LocalURIResolver extends org.ametys.cms.transformation.LocalURIResolver
{
    /** The URI prefix handler */
    protected URIPrefixHandler _prefixHandler;

    @Override
    public void service(ServiceManager manager) throws ServiceException
    {
        super.service(manager);
        _prefixHandler = (URIPrefixHandler) manager.lookup(URIPrefixHandler.ROLE);
    }
    
    @Override
    public String resolve(String uri, boolean download, boolean absolute, boolean internal)
    {
        String result = _resolve(uri, download, absolute, internal, "/_contents", "");
        
        if (result == null)
        {
            super.resolve(uri, download, absolute, internal);
        }
        
        return result;
    }
    
    @Override
    public String resolveImage(String uri, int height, int width, boolean download, boolean absolute, boolean internal)
    {
        if (height == 0 && width == 0)
        {
            return resolve(uri, download, absolute, internal);
        }
        
        String result = _resolve(uri, download, absolute, internal, "/_contents-images", "_" + height + "x" + width);
        
        if (result == null)
        {
            super.resolveImage(uri, height, width, download, absolute, internal);
        }
        
        return result;
    }
    
    @Override
    public String resolveBoundedImage(String uri, int maxHeight, int maxWidth, boolean download, boolean absolute, boolean internal)
    {
        if (maxHeight == 0 && maxWidth == 0)
        {
            return resolve(uri, download, absolute, internal);
        }
        
        String result = _resolve(uri, download, absolute, internal, "/_contents-images", "_max" + maxWidth + "x" + maxWidth);
        
        if (result == null)
        {
            super.resolveBoundedImage(uri, maxHeight, maxWidth, download, absolute, internal);
        }
        
        return result;
    }
    
    /**
     * Actually resolves the URI.
     * @param uri the link URI.
     * @param download true if the pointed resource is to be downloaded.
     * @param absolute true if the url must be absolute
     * @param internal true to get an internal URI.
     * @param prefix a prefix to be prepended to the generated path. Should not be null.
     * @param suffix a suffix to be appended to the generated path. Should not be null. 
     * @return the path to the resource.
     */
    protected String _resolve(String uri, boolean download, boolean absolute, boolean internal, String prefix, String suffix)
    {
        URIInfo infos = getInfos(uri, true);
        
        Request request = ContextHelper.getRequest(_context);
        
        String siteName = WebHelper.getSiteName(request, infos.getContent());
        if (siteName == null)
        {
            return null;
        }
        
        RichText richText = _getMeta(infos.getContent().getMetadataHolder(), infos.getMetadata());
        try
        {
            richText.getAdditionalDataFolder().getFile(infos.getPath());
        }
        catch (UnknownAmetysObjectException e)
        {
            getLogger().warn("Cannot resolve link " + uri);
            return "";
        }
        
        Map<String, String> params = new HashMap<>();
        
        if (download)
        {
            params.put("download", "true");
        }
        
        if (infos.getContentVersion() != null)
        {
            params.put("contentVersion", infos.getContentVersion());
        }
        
        StringBuilder resultPath = new StringBuilder();
        
        if (internal)
        {
            resultPath.append("cocoon://").append(siteName);
        }
        else if (absolute)
        {
            resultPath.append(_prefixHandler.getAbsoluteUriPrefix(siteName));
        }
        else
        {
            resultPath.append(_prefixHandler.getUriPrefix(siteName));
        }
        
        resultPath.append(prefix)
                  .append(FilenameUtils.encodePath(infos.getContent().getPath()))
                  .append("/_metadata/").append(infos.getMetadata())
                  .append("/_data/")
                  .append(FilenameUtils.encodePath(infos.getPath()))
                  .append(suffix);
        
        return URIUtils.encodeURI(resultPath.toString(), params);
    }
}