/*******************************************************************************
 * Copyright (c) 2012, 2017 Original authors and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Original authors and others - initial API and implementation
 ******************************************************************************/
package org.eclipse.nebula.widgets.nattable.search.strategy;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.regex.PatternSyntaxException;

import org.eclipse.nebula.widgets.nattable.command.VisualRefreshCommand;
import org.eclipse.nebula.widgets.nattable.config.IConfigRegistry;
import org.eclipse.nebula.widgets.nattable.coordinate.PositionCoordinate;
import org.eclipse.nebula.widgets.nattable.layer.ILayer;
import org.eclipse.nebula.widgets.nattable.search.ISearchDirection;
import org.eclipse.nebula.widgets.nattable.selection.SelectionLayer;

public class SelectionSearchStrategy extends AbstractSearchStrategy {

    private final IConfigRegistry configRegistry;

    public SelectionSearchStrategy(IConfigRegistry configRegistry) {
        this(configRegistry, true);
    }

    public SelectionSearchStrategy(IConfigRegistry configRegistry, boolean columnFirst) {
        this(configRegistry, ISearchDirection.SEARCH_FORWARD, columnFirst);
    }

    public SelectionSearchStrategy(IConfigRegistry configRegistry, String searchDirection, boolean columnFirst) {
        this.configRegistry = configRegistry;
        this.searchDirection = searchDirection;
        this.columnFirst = columnFirst;
    }

    @Override
    public PositionCoordinate executeSearch(Object valueToMatch) throws PatternSyntaxException {
        ILayer contextLayer = getContextLayer();
        if (!(contextLayer instanceof SelectionLayer)) {
            throw new RuntimeException("For the SelectionSearchStrategy to work it needs the selectionLayer to be passed as the contextLayer."); //$NON-NLS-1$
        }
        SelectionLayer selectionLayer = (SelectionLayer) contextLayer;
        @SuppressWarnings("unchecked")
        PositionCoordinate coordinate = CellDisplayValueSearchUtil.findCell(
                selectionLayer,
                this.configRegistry,
                getSelectedCells(selectionLayer),
                valueToMatch,
                (Comparator<String>) getComparator(),
                isCaseSensitive(),
                isWholeWord(),
                isRegex(),
                isIncludeCollapsed());

        if (coordinate != null) {
            selectionLayer.moveSelectionAnchor(coordinate.columnPosition, coordinate.rowPosition);
            selectionLayer.doCommand(new VisualRefreshCommand());
        }

        return coordinate;
    }

    protected PositionCoordinate[] getSelectedCells(SelectionLayer selectionLayer) {
        PositionCoordinate[] selectedCells = null;

        PositionCoordinate selectionAnchor = selectionLayer.getSelectionAnchor();
        if ((selectionAnchor.columnPosition == SelectionLayer.NO_SELECTION
                && selectionAnchor.rowPosition == SelectionLayer.NO_SELECTION)
                && !ISearchDirection.SEARCH_BACKWARDS.equals(this.searchDirection)) {
            selectedCells = selectionLayer.getSelectedCellPositions();
        } else {
            List<PositionCoordinate> coordinates = Arrays.asList(selectionLayer.getSelectedCellPositions());
            if (this.searchDirection.equals(ISearchDirection.SEARCH_BACKWARDS)) {
                Collections.reverse(coordinates);
            }

            // find the selection anchor index in the collection
            int index = coordinates.indexOf(selectionAnchor);
            // reorder to make the selection anchor the first element in the
            // list
            List<PositionCoordinate> reordered = new ArrayList<PositionCoordinate>(coordinates.subList(index + 1, coordinates.size()));
            if (this.wrapSearch) {
                reordered.addAll(coordinates.subList(0, index + 1));
            }
            selectedCells = reordered.toArray(new PositionCoordinate[0]);
        }
        return selectedCells;
    }

    @Override
    public boolean processResultInternally() {
        // we only want to move the selection but want to avoid processing a
        // selection which would clear the current selection
        return true;
    }
}