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.cms.search.query;
017
018import org.apache.commons.lang.StringUtils;
019import java.util.Map;
020
021import org.apache.solr.client.solrj.util.ClientUtils;
022
023/**
024 * Isolate a query by wrapping it. Used for special cases to force Solr to distinguish specific parts of the global query.
025 */
026public class IsolateQuery extends AbstractWrapperQuery
027{
028    /**
029     * Isolate a query.
030     * @param query the wrapped {@link Query}.
031     */
032    public IsolateQuery(Query query)
033    {
034        super(query);
035    }
036    
037    public String build() throws QuerySyntaxException
038    {
039        String build = getSubQuery().build();
040        if (StringUtils.isBlank(build))
041        {
042            build = new MatchAllQuery().build();
043        }
044        
045        StringBuilder sb = new StringBuilder()
046                .append("{!isolate v=\"")
047                .append(ClientUtils.escapeQueryChars(build))
048                .append("\"}");
049        return sb.toString();
050    }
051    
052    public Object buildAsJson() throws QuerySyntaxException
053    {
054        Object json = getSubQuery().buildAsJson();
055        if (json instanceof String jsonStr && StringUtils.isBlank(jsonStr))
056        {
057            json =  new MatchAllQuery().buildAsJson();
058        }
059        
060        return Map.of("isolate", Map.of("query", json));
061    }
062}