SolrServerFilter.java

/*
 * Copyright (c) 2002-2014, Mairie de 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.solrserver;

import fr.paris.lutece.portal.business.user.AdminUser;
import fr.paris.lutece.portal.service.admin.AdminUserService;
import fr.paris.lutece.portal.service.util.AppPropertiesService;

import org.apache.solr.common.util.FastInputStream;
import org.apache.solr.servlet.SolrDispatchFilter;

import java.io.IOException;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletInputStream;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;


public class SolrServerFilter extends SolrDispatchFilter
{
    public static final String SOLR_DATA_DIR = "solr.data.dir";
    public static final String SOLR_HOME_LABEL = "solr.solr.home";
    public static final String SOLR_URI = AppPropertiesService.getProperty( "solrserver.solr.uri" );
    public static final String SOLR_URI_UPDATE = AppPropertiesService.getProperty( "solrserver.solr.uri.update" );
    public static final String SOLR_URI_SELECT = AppPropertiesService.getProperty( "solrserver.solr.uri.select" );
    public static final String SOLR_URI_AUTOCOMPLETE = AppPropertiesService.getProperty( "solrserver.solr.uri.autoComplete" );
    
    public static final String SOLR_HOME = AppPropertiesService.getProperty( "solrserver.solr.home" );
    public static final String SOLR_ABSOLUTE_DATA = AppPropertiesService.getProperty( "solrserver.solr.absolute.data" );
    public static final String SOLR_RELATIVE_DATA = AppPropertiesService.getProperty( "solrserver.solr.relative.data" );
    public static final String SOLR_ADMIN_CLIENT = AppPropertiesService.getProperty("solrserver.solr.host.client", "127.0.0.1" );
   // private SolrDispatchFilter solrDispatchFilter = new SolrDispatchFilter(  );

    @Override
    public  void init( FilterConfig filterConfig ) throws ServletException
    {
        String realPath = filterConfig.getServletContext(  ).getRealPath( "/" );

        System.setProperty( SOLR_HOME_LABEL, realPath + SOLR_HOME );

        if ( ( SOLR_ABSOLUTE_DATA == null ) || ( SOLR_ABSOLUTE_DATA.length(  ) == 0 ) )
        {
            System.setProperty( SOLR_DATA_DIR, realPath + SOLR_RELATIVE_DATA );
        }
        else
        {
            System.setProperty( SOLR_DATA_DIR, SOLR_ABSOLUTE_DATA );
        }

        super.init( filterConfig );
    }
    @Override
    public  void doFilter( ServletRequest request, ServletResponse response, FilterChain chain )
        throws IOException, ServletException
    {
        String strURI = ( (HttpServletRequest) request ).getRequestURI(  );
        boolean bCallSolr = false;
      
        if ( strURI.indexOf( SOLR_URI_UPDATE ) > 0 )
        {
            AdminUser adminUser = AdminUserService.getAdminUser( (HttpServletRequest) request );
            String strRemoteAddr = ( (HttpServletRequest) request ).getRemoteAddr(  );

            if ( ( adminUser != null ) || ( strRemoteAddr.compareTo( SOLR_ADMIN_CLIENT ) == 0 ) )
            {
                bCallSolr = true;
            }
        }
        else if ( strURI.indexOf( SOLR_URI_SELECT ) > 0 )
        {
            bCallSolr = true;
        }
        else if ( strURI.indexOf( SOLR_URI_AUTOCOMPLETE ) > 0 )
        {
            bCallSolr = true;
        }

        if ( bCallSolr ) {
            request = new HttpServletRequestWrapper((HttpServletRequest)request) {
                @Override public String getServletPath() {
                    String path = ((HttpServletRequest) getRequest()).getServletPath();
                    path = path.substring( SOLR_URI.length() );
                    path = "/collection1" + path;

                    return path;
                };
            };
            super.doFilter( request, response, chain );
        }
        
    }
    @Override
    public  void destroy(  )
    {
        super.destroy(  );
    }
}