/*
 *  Copyright 2015 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.search.query;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.List;

import org.apache.solr.client.solrj.util.ClientUtils;

import org.ametys.cms.search.query.NotQuery;
import org.ametys.cms.search.query.Query;
import org.ametys.cms.search.query.QuerySyntaxException;
import org.ametys.web.indexing.solr.SolrWebFieldNames;
import org.ametys.web.search.systemprop.SiteSystemProperty;

/**
 * Represents a {@link Query} testing the site property.
 */
public class SiteQuery implements Query
{
    private Operator _operator;
    private List<String> _names;
    
    
    /**
     * Build a SiteQuery to test if the site property exists
     */
    public SiteQuery()
    {
        this(Operator.EXISTS);
    }
    
    /**
     * Build a SiteQuery to test if the site property is equals to one of the given site names
     * @param names the site names.
     */
    public SiteQuery(String... names)
    {
        this(Operator.EQ, names);
    }
    
    /**
     * Build a SiteQuery to test if the site property is equals to one of the given site names
     * @param names the site names.
     */
    public SiteQuery(Collection<String> names)
    {
        this(Operator.EQ, names);
    }
    
    /**
     * Build a SiteQuery to test if the site property is equals or different to one of the given site names
     * @param operator the operator (equals ot not-equals)
     * @param names the site names.
     */
    public SiteQuery(Operator operator, String... names)
    {
        this(operator, Arrays.asList(names));
    }
    
    /**
     * Build a SiteQuery.
     * @param operator the operator.
     * @param names the site names.
     */
    public SiteQuery(Operator operator, Collection<String> names)
    {
        if (Operator.EQ != operator && Operator.NE != operator && Operator.EXISTS != operator)
        {
            throw new IllegalArgumentException("Test operator '" + operator + "' is unknown for test's expression.");
        }
        
        _operator = operator;
        _names = new ArrayList<>(names);
    }
    
    /**
     * Get the operator.
     * @return the operator
     */
    public Operator getOperator()
    {
        return _operator;
    }
    
    /**
     * Get the site names.
     * @return the site names.
     */
    public List<String> getNames()
    {
        return Collections.unmodifiableList(_names);
    }
    
    @Override
    public String build() throws QuerySyntaxException
    {
        if (_operator == Operator.EXISTS)
        {
            return SiteSystemProperty.SOLR_FIELD_NAME + ":*";
        }
        
        int count = _names.size();
        
        StringBuilder buff = new StringBuilder();
        
        if (_operator == Operator.NE)
        {
            NotQuery.appendNegation(buff);
        }
        if (count > 1)
        {
            buff.append('(');
        }
        
        boolean first = true;
        for (String id : _names)
        {
            if (!first)
            {
                buff.append(" OR ");
            }
            buff.append(SolrWebFieldNames.SITE_NAME).append(':').append(ClientUtils.escapeQueryChars(id));
            first = false;
        }
        
        if (count > 1)
        {
            buff.append(')');
        }
        
        return buff.toString();
    }

    @Override
    public int hashCode()
    {
        final int prime = 31;
        int result = 1;
        result = prime * result + ((_names == null) ? 0 : _names.hashCode());
        result = prime * result + ((_operator == null) ? 0 : _operator.hashCode());
        return result;
    }

    @Override
    public boolean equals(Object obj)
    {
        if (this == obj)
        {
            return true;
        }
        if (obj == null)
        {
            return false;
        }
        if (getClass() != obj.getClass())
        {
            return false;
        }
        SiteQuery other = (SiteQuery) obj;
        if (_names == null)
        {
            if (other._names != null)
            {
                return false;
            }
        }
        else if (!_names.equals(other._names))
        {
            return false;
        }
        if (_operator != other._operator)
        {
            return false;
        }
        return true;
    }
}