/*
 * Copyright (c) 2015 Cisco Systems, Inc. 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
 */
package org.opendaylight.yangtools.yang.parser.stmt.reactor;

import com.google.common.collect.ImmutableMap;
import com.google.common.io.ByteSource;
import java.io.IOException;
import java.io.InputStream;
import java.util.Collection;
import java.util.EnumMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.model.api.SchemaContext;
import org.opendaylight.yangtools.yang.model.parser.api.YangSyntaxErrorException;
import org.opendaylight.yangtools.yang.model.repo.api.StatementParserMode;
import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
import org.opendaylight.yangtools.yang.parser.rfc6020.repo.YangStatementStreamSource;
import org.opendaylight.yangtools.yang.parser.spi.meta.ModelProcessingPhase;
import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
import org.opendaylight.yangtools.yang.parser.spi.meta.StatementSupportBundle;
import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
import org.opendaylight.yangtools.yang.parser.spi.validation.ValidationBundlesNamespace.ValidationBundleType;
import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangStatementSourceImpl;
import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.effective.EffectiveSchemaContext;

public class CrossSourceStatementReactor {

    private final Map<ModelProcessingPhase, StatementSupportBundle> supportedTerminology;
    private final Map<ValidationBundleType, Collection<?>> supportedValidation;

    CrossSourceStatementReactor(final Map<ModelProcessingPhase, StatementSupportBundle> supportedTerminology) {
        this.supportedTerminology = ImmutableMap.copyOf(supportedTerminology);
        this.supportedValidation = ImmutableMap.of();
    }

    CrossSourceStatementReactor(final Map<ModelProcessingPhase, StatementSupportBundle> supportedTerminology,
            final Map<ValidationBundleType, Collection<?>> supportedValidation) {
        this.supportedTerminology = ImmutableMap.copyOf(supportedTerminology);
        this.supportedValidation = ImmutableMap.copyOf(supportedValidation);
    }

    public static Builder builder() {
        return new Builder();
    }

    public final BuildAction newBuild() {
        return newBuild(StatementParserMode.DEFAULT_MODE);
    }

    public final BuildAction newBuild(final Set<QName> supportedFeatures) {
        return new BuildAction(StatementParserMode.DEFAULT_MODE, supportedFeatures);
    }

    public final BuildAction newBuild(final StatementParserMode statementParserMode) {
        return new BuildAction(statementParserMode, null);
    }

    public final BuildAction newBuild(final StatementParserMode statementParserMode,
            final Set<QName> supportedFeatures) {
        return new BuildAction(statementParserMode, supportedFeatures);
    }

    public static class Builder implements org.opendaylight.yangtools.concepts.Builder<CrossSourceStatementReactor> {

        final Map<ModelProcessingPhase, StatementSupportBundle> bundles = new EnumMap<>(ModelProcessingPhase.class);
        final Map<ValidationBundleType, Collection<?>> validationBundles = new EnumMap<>(ValidationBundleType.class);

        public Builder setBundle(final ModelProcessingPhase phase, final StatementSupportBundle bundle) {
            bundles.put(phase, bundle);
            return this;
        }

        public Builder setValidationBundle(final ValidationBundleType type, final Collection<?> validationBundle) {
            validationBundles.put(type, validationBundle);
            return this;
        }

        @Override
        public CrossSourceStatementReactor build() {
            return new CrossSourceStatementReactor(bundles, validationBundles);
        }
    }

    public class BuildAction {
        private final BuildGlobalContext context;

        public BuildAction() {
            this(StatementParserMode.DEFAULT_MODE);
        }

        public BuildAction(final StatementParserMode statementParserMode) {
            this(statementParserMode, null);
        }

        public BuildAction(final Set<QName> supportedFeatures) {
            this(StatementParserMode.DEFAULT_MODE, supportedFeatures);
        }

        public BuildAction(final StatementParserMode statementParserMode, final Set<QName> supportedFeatures) {
            this.context = new BuildGlobalContext(supportedTerminology, supportedValidation, statementParserMode,
                    supportedFeatures);
        }

        /**
         * Add main source. All main sources are present in resulting
         * SchemaContext.
         *
         * @param source
         *            which should be added into main sources
         */
        public void addSource(final StatementStreamSource source) {
            context.addSource(source);
        }

        /**
         * Add main sources. All main sources are present in resulting
         * SchemaContext.
         *
         * @param sources
         *            which should be added into main sources
         */
        public void addSources(final StatementStreamSource... sources) {
            for (final StatementStreamSource source : sources) {
                context.addSource(source);
            }
        }

        /**
         * Add library sources. Only library sources required by main sources
         * are present in resulting SchemaContext. Any other library sources are
         * ignored and this also applies to error reporting.
         *
         * Library sources are not supported in semantic version mode currently.
         *
         * @param libSources
         *            yang sources which should be added into library sources
         */
        public void addLibSources(final StatementStreamSource... libSources) {
            for (final StatementStreamSource libSource : libSources) {
                context.addLibSource(libSource);
            }
        }

        /**
         * @throws org.opendaylight.yangtools.yang.parser.spi.source.SourceException
         * @throws ReactorException
         */
        public EffectiveModelContext build() throws ReactorException {
            return context.build();
        }

        public EffectiveSchemaContext buildEffective() throws ReactorException {
            return context.buildEffective();
        }

        public SchemaContext buildEffective(final Collection<ByteSource> yangByteSources) throws ReactorException,
                IOException {
            for (final ByteSource source : yangByteSources) {
                if (source instanceof YangTextSchemaSource) {
                    try {
                        addSource(YangStatementStreamSource.create((YangTextSchemaSource) source));
                    } catch (YangSyntaxErrorException e) {
                        throw new IOException("Source " + source + " failed to parse", e);
                    }
                } else {
                    addSource(new YangStatementSourceImpl(source.openStream()));
                }
            }

            return buildEffective();
        }

        public SchemaContext buildEffective(final List<InputStream> yangInputStreams) throws ReactorException {
            for (final InputStream yangInputStream : yangInputStreams) {
                addSource(new YangStatementSourceImpl(yangInputStream));
            }

            return buildEffective();
        }
    }
}