ASearchRequest.java

/*
 * Copyright (c) 2002-2024, City of Paris
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *  1. Redistributions of source code must retain the above copyright notice
 *     and the following disclaimer.
 *
 *  2. Redistributions in binary form must reproduce the above copyright notice
 *     and the following disclaimer in the documentation and/or other materials
 *     provided with the distribution.
 *
 *  3. Neither the name of 'Mairie de Paris' nor 'Lutece' nor the names of its
 *     contributors may be used to endorse or promote products derived from
 *     this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 * License 1.0
 */
package fr.paris.lutece.plugins.identitystore.service.indexer.elastic.search.model;

import fr.paris.lutece.plugins.identitystore.service.indexer.elastic.search.model.inner.request.InnerSearchRequest;
import fr.paris.lutece.plugins.identitystore.v3.web.rs.dto.search.SearchAttribute;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

public abstract class ASearchRequest
{
    private static final String CUSTOMER_ID = "customerId";
    protected final List<SearchAttribute> searchAttributes = new ArrayList<>( );
    protected final List<String> attributesFilter = new ArrayList<>( );

    public ASearchRequest( final List<String> attributesFilter )
    {
        this.getAttributesFilter( ).addAll( attributesFilter );
    }

    public InnerSearchRequest body( )
    {
        final InnerSearchRequest body = this.innerBody( );
        if ( this.getAttributesFilter( ) != null && !this.getAttributesFilter( ).isEmpty( ) )
        {
            body.getSourceFilters( ).add( CUSTOMER_ID );
            body.getSourceFilters( ).addAll( this.getAttributesFilter( ).stream( ).filter( attributeFilter -> !attributeFilter.equals( CUSTOMER_ID ) )
                    .map( attributeFilter -> "attributes." + attributeFilter + ".key" ).collect( Collectors.toList( ) ) );
            body.getSourceFilters( ).addAll( this.getAttributesFilter( ).stream( ).filter( attributeFilter -> !attributeFilter.equals( CUSTOMER_ID ) )
                    .map( attributeFilter -> "attributes." + attributeFilter + ".value" ).collect( Collectors.toList( ) ) );
        }
        return body;
    }

    protected abstract InnerSearchRequest innerBody( );

    public List<SearchAttribute> getSearchAttributes( )
    {
        return searchAttributes;
    }

    public List<String> getAttributesFilter( )
    {
        return attributesFilter;
    }
}