001/*
002 *  Copyright 2020 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.autocompletion;
017
018import java.util.ArrayList;
019import java.util.HashMap;
020import java.util.List;
021import java.util.Map;
022
023import org.apache.avalon.framework.parameters.Parameters;
024import org.apache.avalon.framework.service.ServiceException;
025import org.apache.avalon.framework.service.ServiceManager;
026import org.apache.cocoon.ProcessingException;
027import org.apache.cocoon.acting.ServiceableAction;
028import org.apache.cocoon.environment.ObjectModelHelper;
029import org.apache.cocoon.environment.Redirector;
030import org.apache.cocoon.environment.Request;
031import org.apache.cocoon.environment.SourceResolver;
032import org.apache.commons.lang.StringUtils;
033
034import org.ametys.cms.search.solr.SearcherFactory;
035import org.ametys.cms.search.solr.SearcherFactory.Searcher;
036import org.ametys.cms.transformation.xslt.ResolveURIComponent;
037import org.ametys.core.cocoon.JSonReader;
038import org.ametys.plugins.repository.AmetysObjectIterable;
039import org.ametys.plugins.repository.provider.RequestAttributeWorkspaceSelector;
040import org.ametys.web.WebConstants;
041import org.ametys.web.repository.page.Page;
042import org.ametys.web.repository.page.PagesContainer;
043
044import com.google.common.collect.Lists;
045
046/**
047 * Get the proposed fields for auto-completion while beginning a search
048 */
049public class AutoCompletionOnTitleAction extends ServiceableAction
050{
051    private static final int __MAX_NUMBER = 12;
052
053    private SearcherFactory _searcherFactory;
054    
055    
056    @Override
057    public void service(ServiceManager serviceManager) throws ServiceException
058    {
059        super.service(serviceManager);
060        _searcherFactory = (SearcherFactory) serviceManager.lookup(SearcherFactory.ROLE);
061    }
062    
063    public Map act(Redirector redirector, SourceResolver resolver, Map objectModel, String source, Parameters parameters) throws Exception
064    {
065        Request request = ObjectModelHelper.getRequest(objectModel);
066        
067        String siteName = parameters.getParameter("siteName", "");
068        if (StringUtils.isEmpty(siteName))
069        {
070            throw new IllegalArgumentException("A site must be specified for auto completion");
071        }
072        
073        String lang = parameters.getParameter("lang", "");
074        if (StringUtils.isEmpty(lang))
075        {
076            throw new IllegalArgumentException("A language must be specified for auto completion");
077        }
078        
079        // Retrieve current workspace
080        String currentWsp = RequestAttributeWorkspaceSelector.getForcedWorkspace(request);
081        
082        try
083        {
084            // TODO Do not force the workspace, to be transmitted.
085            // Use live workspace
086            RequestAttributeWorkspaceSelector.setForcedWorkspace(request, WebConstants.LIVE_WORKSPACE);
087            
088            String prefix = request.getParameter("q");
089
090            Searcher searcher = _searcherFactory.create()
091                                                .addFilterQueryString("_documentType:page")
092                                                .addFilterQueryString("site:" + siteName)
093                                                .addFilterQueryString("sitemap:" + lang)
094                                                .addFilterQueryString("full_fr:*" + prefix + "*")
095                                                .withLimits(0, __MAX_NUMBER)
096//                                                    .withSort(new Sort("title_s_sort", Order.ASC))
097                                                .setCheckRights(true);
098            
099            AmetysObjectIterable<Page> results = searcher.search();
100            
101            List<Map<String, String>> jsonResults = new ArrayList<>();
102            
103            for (Page page : results)
104            {
105                PagesContainer cursor = page.getParent();
106                List<String> parents = new ArrayList<>();
107                while (cursor instanceof Page)
108                {
109                    parents.add(((Page) cursor).getTitle());
110                    cursor = cursor.getParent();
111                }
112                parents = Lists.reverse(parents);
113                
114                Map<String, String> pageInfo = new HashMap<>();
115                pageInfo.put("title", page.getTitle());
116                pageInfo.put("id", page.getId());
117                pageInfo.put("url", ResolveURIComponent.resolve("page", page.getId()));
118                pageInfo.put("path", String.join(" > ", parents));
119                jsonResults.add(pageInfo);
120            }
121            
122            request.setAttribute(JSonReader.OBJECT_TO_READ, jsonResults);
123            return EMPTY_MAP;
124        }
125        catch (Exception e)
126        {
127            getLogger().error("Error getting auto-complete list.", e);
128            throw new ProcessingException("Error getting auto-complete list.", e);
129        }
130        finally
131        {
132            // Restore context
133            RequestAttributeWorkspaceSelector.setForcedWorkspace(request, currentWsp);
134        }
135    }
136}