OrHeaderChecker.java

package com.epam.wilma.stubconfig.condition.checker.general.header;

/*==========================================================================
Copyright 2013-2017 EPAM Systems

This file is part of Wilma.

Wilma is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

Wilma is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with Wilma.  If not, see <http://www.gnu.org/licenses/>.
===========================================================================*/

import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import com.epam.wilma.domain.http.WilmaHttpRequest;
import com.epam.wilma.domain.stubconfig.dialog.condition.checker.ConditionChecker;
import com.epam.wilma.domain.stubconfig.parameter.Parameter;
import com.epam.wilma.domain.stubconfig.parameter.ParameterList;
import com.epam.wilma.stubconfig.condition.checker.general.operator.ConditionCheckerOperator;

/**
 * Checks all headers in a request and if any of them contains any of the given parameter, returns true.
 * @author Tunde_Kovacs
 * @author Tamas_Bihari
 *
 */
public class OrHeaderChecker implements ConditionChecker {

    private final ConditionCheckerOperator operator;

    /**
     * Creates a new {@link OrHeaderChecker} instance with its dependency as parameter.
     * @param operator is the necessary {@link ConditionCheckerOperator} instance.
     */
    public OrHeaderChecker(final ConditionCheckerOperator operator) {
        this.operator = operator;
    }

    @Override
    public boolean checkCondition(final WilmaHttpRequest request, final ParameterList parameterList) {
        List<Parameter> params = parameterList.getAllParameters();
        Iterator<Parameter> iterator = params.iterator();
        boolean result = false;
        if (!params.isEmpty()) {
            while (iterator.hasNext() && !result) {
                result = checkInRequestHeaders(request, iterator.next());
            }
        }
        return result;
    }

    private boolean checkInRequestHeaders(final WilmaHttpRequest request, final Parameter parameter) {
        boolean result;
        if (!"".equals(parameter.getValue())) {
            result = checkInHeaders(request, parameter);
        } else {
            result = false;
        }
        return result;
    }

    private boolean checkInHeaders(final WilmaHttpRequest request, final Parameter parameter) {
        boolean result = false;
        Iterator<Entry<String, String>> headerIterator = request.getHeaders().entrySet().iterator();
        while (headerIterator.hasNext() && !result) {
            Map.Entry<String, String> header = headerIterator.next();
            result = operator.checkTarget(parameter.getValue(), header.getValue());
        }
        return result;
    }

}