/*
 *  Copyright 2020 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.comments;

import java.util.Arrays;
import java.util.Map;
import java.util.Set;

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.cocoon.xml.XMLUtils;
import org.apache.commons.lang3.StringUtils;
import org.xml.sax.ContentHandler;
import org.xml.sax.SAXException;

import org.ametys.cms.repository.Content;
import org.ametys.cms.repository.comment.Comment;
import org.ametys.core.user.User;
import org.ametys.core.user.population.PopulationContextHelper;
import org.ametys.plugins.userdirectory.UserDirectoryHelper;
import org.ametys.web.URIPrefixHandler;
import org.ametys.web.WebConstants;
import org.ametys.web.WebHelper;

/**
 * Comments DAO for workspaces
 *
 */
public class CommentsDAO extends org.ametys.web.repository.comment.CommentsDAO
{
    /** The population context helper */
    protected PopulationContextHelper _populationContextHelper;
    /** The user directory helper */
    protected UserDirectoryHelper _userDirectoryHelper;
    /** The URI prefix handler */
    protected URIPrefixHandler _prefixHandler;

    @Override
    public void service(ServiceManager smanager) throws ServiceException
    {
        super.service(smanager);
        _populationContextHelper = (PopulationContextHelper) smanager.lookup(PopulationContextHelper.ROLE);
        _userDirectoryHelper = (UserDirectoryHelper) smanager.lookup(UserDirectoryHelper.ROLE);
        _prefixHandler = (URIPrefixHandler) smanager.lookup(URIPrefixHandler.ROLE);
    }
    
    @Override
    public Map<String, Object> getComment(Comment comment, int level, Map<String, Object> contextualParameters)
    {
        Map<String, Object> comment2json = super.getComment(comment, level, contextualParameters);
        
        String siteName = (String) contextualParameters.get("siteName");
        String lang = (String) contextualParameters.get("sitemapLanguage");
        
        User user = _getUserByMail(comment.getAuthorEmail(), siteName);
        if (user != null)
        {
            // SAX additionnal user information
            Content userContent = _userDirectoryHelper.getUserContent(user.getIdentity(), lang);
            if (userContent != null)
            {
                _addAdditionalUserProperties(userContent, user, comment2json);
            }
        }
        
        return comment2json;
    }
    
    @Override
    protected void saxCommentAdditionalProperties(ContentHandler contentHandler, Comment comment, int level) throws SAXException
    {
        super.saxCommentAdditionalProperties(contentHandler, comment, level);
        
        Request request = ContextHelper.getRequest(_context);
        String siteName = WebHelper.getSiteName(request);
        String lang = (String) request.getAttribute(WebConstants.REQUEST_ATTR_SITEMAP_NAME);
        
        User user = _getUserByMail(comment.getAuthorEmail(), siteName);
        if (user != null)
        {
            // SAX additionnal user information
            Content userContent = _userDirectoryHelper.getUserContent(user.getIdentity(), lang);
            if (userContent != null)
            {
                _saxAdditionalUserProperties(contentHandler, userContent, user);
            }
            else
            {
                _saxUserImage(contentHandler, user, lang);
            }
        }
    }
    
    /**
     * Sax the user avatar
     * @param contentHandler the content handler
     * @param user the user
     * @param language the language code (if image will be taken from user content)
     * @throws SAXException if an error occurred while saxing
     */
    protected void _saxUserImage(ContentHandler contentHandler,  User user, String language) throws SAXException
    {
        Request request = ContextHelper.getRequest(_context);
        String siteName = WebHelper.getSiteName(request);
        String uriPrefix = _prefixHandler.getUriPrefix(siteName);
        
        String authorImgUrl = "/_plugins/workspaces/user/" + user.getIdentity().getPopulationId() + "/" + user.getIdentity().getLogin() + "/image_45?lang=" + language;
        XMLUtils.createElement(contentHandler, "author-img-url", uriPrefix + authorImgUrl);
    }
    
    /**
     * Add additional properties for comment's author
     * @param contentHandler the content handler
     * @param userContent the user content
     * @param user the user
     * @throws SAXException if an error occurred while saxing
     */
    protected void _saxAdditionalUserProperties(ContentHandler contentHandler, Content userContent, User user) throws SAXException
    {
        _saxUserImage(contentHandler, user, userContent.getLanguage());
        
        String function = userContent.getValue("function", false, null);
        if (StringUtils.isNotEmpty(function))
        {
            XMLUtils.createElement(contentHandler, "author-function", function);
        }
        String organisation = userContent.getValue("organisation", false, null);
        if (StringUtils.isNotEmpty(organisation))
        {
            XMLUtils.createElement(contentHandler, "author-organisation", organisation);
        }
        String organisationAccr = userContent.getValue("organisation-accronym", false, null);
        if (StringUtils.isNotEmpty(organisationAccr))
        {
            XMLUtils.createElement(contentHandler, "author-organisation-accronym", organisationAccr);
        }
    }
    
    /**
     * Add additional properties for comment's author
     * @param userContent the user content
     * @param user the user
     * @param comment2json the comment properties
     */
    protected void _addAdditionalUserProperties(Content userContent, User user, Map<String, Object> comment2json)
    {
        Request request = ContextHelper.getRequest(_context);
        String siteName = WebHelper.getSiteName(request);
        String uriPrefix = _prefixHandler.getUriPrefix(siteName);
        
        String authorImgUrl = "/_plugins/workspaces/user/" + user.getIdentity().getPopulationId() + "/" + user.getIdentity().getLogin() + "/image_45?lang=" + userContent.getLanguage();
        comment2json.put("author-img-url", uriPrefix + authorImgUrl);
        
        String function = userContent.getValue("function", false, null);
        if (StringUtils.isNotEmpty(function))
        {
            comment2json.put("author-function", function);
        }
        String organisation = userContent.getValue("organisation", false, null);
        if (StringUtils.isNotEmpty(organisation))
        {
            comment2json.put("author-organisation", function);
        }
        String organisationAccr = userContent.getValue("organisation-accronym", false, null);
        if (StringUtils.isNotEmpty(organisationAccr))
        {
            comment2json.put("author-organisation-accronym", organisationAccr);
        }
    }
    
    private User _getUserByMail(String email, String siteName)
    {
        Set<String> userPopulationsOnSite = _populationContextHelper.getUserPopulationsOnContexts(Arrays.asList("/sites/" + siteName, "/sites-fo/" + siteName), false, false);
        
        for (String populationId : userPopulationsOnSite)
        {
            User user = _userHelper.getUserByEmail(populationId, email);
            if (user != null)
            {
                return user;
            }
        }
        
        return null;
    }
}