/*
 *  Copyright 2017 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.plugins.workspaces.documents;

import java.util.Collections;

import org.apache.avalon.framework.service.ServiceException;
import org.apache.avalon.framework.service.ServiceManager;
import org.apache.commons.lang.StringUtils;

import org.ametys.cms.transformation.URIResolver;
import org.ametys.core.authentication.token.AuthenticationTokenManager;
import org.ametys.core.user.CurrentUserProvider;
import org.ametys.core.user.UserIdentity;
import org.ametys.core.util.URIUtils;
import org.ametys.plugins.explorer.resources.Resource;
import org.ametys.plugins.explorer.resources.ResourceCollection;
import org.ametys.plugins.repository.AmetysObject;
import org.ametys.plugins.repository.UnknownAmetysObjectException;
import org.ametys.plugins.repository.version.VersionableAmetysObject;
import org.ametys.plugins.workspaces.project.objects.Project;

/**
 * {@link URIResolver} for webdav "project-resource". <br>
 * These links point to a file from the resources of a project through webdav
 */
public class WebdavProjectResourceURIResolver extends ProjectResourceURIResolver
{
    /** The authentication token manager */
    protected AuthenticationTokenManager _authenticationTokenManager;
    
    @Override
    public void service(ServiceManager manager) throws ServiceException
    {
        super.service(manager);
        _authenticationTokenManager = (AuthenticationTokenManager) manager.lookup(AuthenticationTokenManager.ROLE);
    }
    
    @Override
    public String getType()
    {
        return "webdav-project-resource";
    }
    
    // Override to single encoded instead of twice
    @Override
    protected String _resolve(String uri, boolean download, boolean absolute, boolean internal, String prefix, String suffix)
    {
        String version = null;
        String path;
        AmetysObject resource = null;
        try
        {
            String resourceId = uri;
            int i = uri.indexOf(";");
            if (i != -1)
            {
                resourceId = uri.substring(0, i);
                version = uri.substring(i + 1);
            }
            
            resource = _resolver.resolveById(resourceId);
            if (resource instanceof Resource)
            {
                path = getResourcePath((Resource) resource);
            }
            else if (resource instanceof ResourceCollection)
            {
                path = ((ResourceCollection) resource).getExplorerPath();
            }
            else
            {
                getLogger().error("Trying to access to something that is not a Resource or a ResourceCollection : " + resource.getId());
                return "";
            }
        }
        catch (UnknownAmetysObjectException e)
        {
            getLogger().warn("Link to unexisting resource " + uri);
            return "";
        }
        
        int i = path.lastIndexOf(".");
        String extension = i != -1 ? path.substring(i) : null;
        path = i != -1 ? path.substring(0, i) : path; 
        
        StringBuilder result = new StringBuilder();
        
        result.append(getUriPrefix(resource, download, absolute, internal));

        String realPrefix = null;
        if (resource instanceof Resource)
        {
            realPrefix = getRealPrefix((Resource) resource, prefix);
        }

        if (StringUtils.isNotEmpty(realPrefix))
        {
            result.append("/_").append(realPrefix);
        }
        
        if (StringUtils.isNotEmpty(version) && resource instanceof VersionableAmetysObject)
        {
            result.append("/__version").append(version);
        }
        
        result.append(path);
        
        if (suffix != null)
        {
            result.append(suffix);
        }
        
        if (extension != null)
        {
            result.append(extension);
        }
        
        String fullUri = result.toString();
        
        // the ';' char is handled by Tomcat as an URI path parameter
        // Replace it by its encoded value, which may look weird in Office, but at least it won't fail
        fullUri = fullUri.replaceAll(";", "%3B");
        
        return URIUtils.encodeURI(fullUri, download ? Collections.singletonMap("download", "true") : null);
    }
    
    @Override
    protected String getUriPrefix(AmetysObject object, boolean download, boolean absolute, boolean internal)
    {
        Project project = null;
        String projectName = null;
        String siteName = null;
        
        String token = _authenticationTokenManager.generateToken(60, "Workspaces-Webdav", null);
        
        project = _getProject(object);
        projectName = project.getName();

        siteName = project.getSites().iterator().next().getName();
        
        if (internal)
        {
            return "cocoon://_workspaces/dav/" + projectName + "/" + token;
        }
        else if (absolute)
        {
            return _webPrefixHandler.getAbsoluteUriPrefix(siteName) + "/_workspaces/dav/" + projectName + "/" + token;
        }
        else
        {
            return _webPrefixHandler.getUriPrefix(siteName) + "/_workspaces/dav/" + projectName + "/" + token;
        }
    }
    
    @Override
    protected String getRealPrefix(Resource resource, String prefix)
    {
        return null;
    }
}