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

/**
 * Represents a {@link Query} testing the site type property.
 */
public class SiteTypeQuery implements Query
{
    private Operator _operator;
    private List<String> _types;
    
    
    /**
     * Build a SiteTypeQuery to test if the site type property exits
     */
    public SiteTypeQuery()
    {
        this(Operator.EQ, "*");
    }
    
    /**
     * Build a SiteTypeQuery to test if the site type property is equals to one of the given site type
     * @param types the site type.
     */
    public SiteTypeQuery(String... types)
    {
        this(Operator.EQ, types);
    }
    
    /**
     * Build a SiteTypeQuery to test if the site type property is equals to one of the given site types 
     * @param types the site types.
     */
    public SiteTypeQuery(Collection<String> types)
    {
        this(Operator.EQ, types);
    }
    
    /**
     * Build a SiteTypeQuery to test if the site type property is equals or different to one of the given site types
     * @param operator the operator (equals ot not-equals)
     * @param types the site types.
     */
    public SiteTypeQuery(Operator operator, String... types)
    {
        this(operator, Arrays.asList(types));
    }
    
    /**
     * Build a SiteTypeQuery.
     * @param operator the operator.
     * @param types the site types.
     */
    public SiteTypeQuery(Operator operator, Collection<String> types)
    {
        if (Operator.EQ != operator && Operator.NE != operator)
        {
            throw new IllegalArgumentException("Test operator '" + operator + "' is unknown for test's expression.");
        }
        
        _operator = operator;
        _types = new ArrayList<>(types);
    }
    
    /**
     * Get the operator.
     * @return the operator
     */
    public Operator getOperator()
    {
        return _operator;
    }
    
    /**
     * Get the site types.
     * @return the site types.
     */
    public List<String> getTypes()
    {
        return Collections.unmodifiableList(_types);
    }
    
    @Override
    public String build() throws QuerySyntaxException
    {
        int count = _types.size();
        
        StringBuilder buff = new StringBuilder();
        
        if (_operator == Operator.NE)
        {
            NotQuery.appendNegation(buff);
        }
        if (count > 1)
        {
            buff.append('(');
        }
        
        boolean first = true;
        for (String id : _types)
        {
            if (!first)
            {
                buff.append(" OR ");
            }
            buff.append(SolrWebFieldNames.SITE_TYPE).append(':').append(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 + ((_types == null) ? 0 : _types.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;
        }
        SiteTypeQuery other = (SiteTypeQuery) obj;
        if (_types == null)
        {
            if (other._types != null)
            {
                return false;
            }
        }
        else if (!_types.equals(other._types))
        {
            return false;
        }
        if (_operator != other._operator)
        {
            return false;
        }
        return true;
    }
}