/*
 * WorldEdit, a Minecraft world manipulation toolkit
 * Copyright (C) sk89q <http://www.sk89q.com>
 * Copyright (C) WorldEdit team and contributors
 *
 * This program 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.
 *
 * This program 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 this program.  If not, see <https://www.gnu.org/licenses/>.
 */

package com.sk89q.worldedit.internal.registry;

import com.fastasyncworldedit.core.configuration.Caption;
import com.fastasyncworldedit.core.extension.factory.parser.AliasedParser;
import com.sk89q.worldedit.WorldEdit;
import com.sk89q.worldedit.extension.input.InputParseException;
import com.sk89q.worldedit.extension.input.NoMatchException;
import com.sk89q.worldedit.extension.input.ParserContext;
import com.sk89q.worldedit.util.formatting.text.TextComponent;

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

import static com.google.common.base.Preconditions.checkNotNull;

/**
 * An abstract implementation of a factory for internal usage.
 *
 * @param <E> the element that the factory returns
 */
@SuppressWarnings("ProtectedField")
public abstract class AbstractFactory<E> {

    protected final WorldEdit worldEdit;
    private final List<InputParser<E>> parsers = new ArrayList<>();

    /**
     * Create a new factory.
     *
     * @param worldEdit     the WorldEdit instance
     * @param defaultParser the parser to fall back to
     */
    protected AbstractFactory(WorldEdit worldEdit, InputParser<E> defaultParser) {
        checkNotNull(worldEdit);
        checkNotNull(defaultParser);
        this.worldEdit = worldEdit;
        this.parsers.add(defaultParser);
    }

    /**
     * Gets an immutable list of parsers.
     *
     * <p>
     * To add parsers, use the register method.
     * </p>
     *
     * @return the parsers
     */
    public List<InputParser<E>> getParsers() {
        return Collections.unmodifiableList(parsers);
    }

    //FAWE start - javadoc

    /**
     * Parse a string and context to each {@link InputParser} added to this factory. If no result found, throws {@link InputParseException}
     *
     * @param input   input string
     * @param context input context
     * @return parsed result
     * @throws InputParseException if no result found
     */
    //FAWE end
    public E parseFromInput(String input, ParserContext context) throws InputParseException {
        E match;

        for (InputParser<E> parser : parsers) {
            match = parser.parseFromInput(input, context);

            if (match != null) {
                return match;
            }
        }

        throw new NoMatchException(Caption.of("worldedit.error.no-match", TextComponent.of(input)));
    }

    public List<String> getSuggestions(String input) {
        return parsers.stream().flatMap(
                p -> p.getSuggestions(input)
        ).collect(Collectors.toList());
    }

    /**
     * Registers an InputParser to this factory.
     *
     * @param inputParser The input parser
     */
    public void register(InputParser<E> inputParser) {
        checkNotNull(inputParser);

        parsers.add(parsers.size() - 1, inputParser);
    }

    /**
     * Test all parsers to see if alias is contained by one of them
     *
     * @param alias alias to test
     * @return if a parser contains the alias
     */
    public boolean containsAlias(String alias) {
        return parsers.stream().anyMatch(p -> {
            if (!(p instanceof AliasedParser)) {
                return false;
            }
            return ((AliasedParser) p).getMatchedAliases().contains(alias);
        });
    }

}
