/*
 *  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.apache.commons.lang3.StringUtils;

import org.ametys.cms.data.RichText;
import org.ametys.cms.transformation.URIResolver;
import org.ametys.core.util.FilenameUtils;
import org.ametys.core.util.URIUtils;
import org.ametys.web.URIPrefixHandler;
import org.ametys.web.WebHelper;

/**
 * {@link URIResolver} for type "local".
 * These links or images point to resources local to a rich text.
 */
public class LocalURIResolver extends org.ametys.cms.transformation.LocalURIResolver
{
    /** the {@link URIPrefixHandler} s*/
    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, "");
        
        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, "_" + 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, "_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 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 suffix)
    {
        URIInfo infos = getInfos(uri, true, null);
        
        Request request = ContextHelper.getRequest(_context);
        
        String siteName = WebHelper.getSiteName(request, infos.getContent());
        if (siteName == null)
        {
            return null;
        }
        
        return _resolve(infos, uri, siteName, download, absolute, internal, suffix);
    }
    
    /**
     * Actually resolves the URI.
     * @param infos info about parsed uri.
     * @param uri the link URI.
     * @param siteName the siteName, if any.
     * @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 suffix a suffix to be appended to the generated path. Should not be null. 
     * @return the path to the resource.
     */
    protected String _resolve(URIInfo infos, String uri, String siteName, boolean download, boolean absolute, boolean internal, String suffix)
    {
        RichText richText = infos.getContent().getValue(infos.getAttribute());
        boolean hasAttachment = richText != null && richText.hasAttachment(infos.getFilename());
        if (!hasAttachment)
        {
            getLogger().warn("Cannot resolve link " + uri);
            return StringUtils.EMPTY;
        }
        
        StringBuilder resultPath = new StringBuilder();
        
        String filename = FilenameUtils.encodeName(infos.getFilename());
        
        String baseName = org.apache.commons.io.FilenameUtils.getBaseName(filename);
        String extension = org.apache.commons.io.FilenameUtils.getExtension(filename);
        
        resultPath.append(_prefixHandler.computeUriPrefix(siteName, absolute, internal))
                  .append("/_richText-file")
                  .append(FilenameUtils.encodePath(infos.getContent().getPath()))
                  .append("/_attribute/").append(infos.getAttribute())
                  .append("/_data/")
                  .append(baseName)
                  .append(suffix)
                  .append(extension.isEmpty() ? "" : "." + extension);
        
        Map<String, String> params = new HashMap<>();
        
        if (download)
        {
            params.put("download", "true");
        }
        
        if (infos.getContentVersion() != null)
        {
            params.put("contentVersion", infos.getContentVersion());
        }
        
        return URIUtils.encodeURI(resultPath.toString(), params);
    }
}