/*
 * Copyright (c) 2018 Pantheon Technologies, s.r.o. 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.model.export;

import static java.util.Objects.requireNonNull;
import static org.eclipse.jdt.annotation.DefaultLocation.PARAMETER;
import static org.eclipse.jdt.annotation.DefaultLocation.RETURN_TYPE;

import com.google.common.annotations.Beta;
import java.util.Iterator;
import java.util.Set;
import java.util.Spliterator;
import java.util.Spliterators;
import java.util.stream.Collectors;
import java.util.stream.Stream;
import java.util.stream.StreamSupport;
import org.eclipse.jdt.annotation.NonNull;
import org.eclipse.jdt.annotation.NonNullByDefault;
import org.opendaylight.yangtools.concepts.Immutable;
import org.opendaylight.yangtools.yang.model.api.meta.DeclaredStatement;
import org.opendaylight.yangtools.yang.model.api.meta.StatementDefinition;

/**
 * A YANG text snippet generated from a {@link DeclaredStatement}. Generated {@link #stream()} or {@link #iterator()}
 * are guaranteed to not contain null nor empty strings. Furthermore, newlines are always emitted at the end
 * on the generated string -- which can be checked with {@link #isEolString(String)} utility method.
 *
 * <p>
 * This allows individual strings to be escaped as needed and external indentation can be accounted for by inserting
 * outer document indentation after the string which matched {@link #isEolString(String)} is emitted to the stream.
 *
 * @author Robert Varga
 */
@Beta
@NonNullByDefault({ PARAMETER, RETURN_TYPE })
public final class YangTextSnippet implements Immutable, Iterable<@NonNull String> {
    private final Set<@NonNull StatementDefinition> ignoredStatements;
    private final StatementPrefixResolver resolver;
    private final DeclaredStatement<?> statement;
    private final boolean omitDefaultStatements;

    YangTextSnippet(final DeclaredStatement<?> statement, final StatementPrefixResolver resolver,
            final Set<@NonNull StatementDefinition> ignoredStatements, final boolean omitDefaultStatements) {
        this.statement = requireNonNull(statement);
        this.resolver = requireNonNull(resolver);
        this.ignoredStatements = requireNonNull(ignoredStatements);
        this.omitDefaultStatements = omitDefaultStatements;
    }

    @Override
    public Iterator<@NonNull String> iterator() {
        return new YangTextSnippetIterator(statement, resolver, ignoredStatements, omitDefaultStatements);
    }

    @Override
    @SuppressWarnings("null")
    public Spliterator<@NonNull String> spliterator() {
        return Spliterators.spliteratorUnknownSize(iterator(),
            Spliterator.ORDERED | Spliterator.IMMUTABLE | Spliterator.NONNULL);
    }

    @SuppressWarnings("null")
    public Stream<@NonNull String> stream() {
        return StreamSupport.stream(spliterator(), false);
    }

    /**
     * Check if an emitted string contains End-Of-Line character.
     *
     * @param str String to be checked
     * @return True if the string contains end of line.
     * @throws NullPointerException if str is null
     */
    public static boolean isEolString(final String str) {
        return !str.isEmpty() && str.charAt(str.length() - 1) == '\n';
    }

    @Override
    @SuppressWarnings("null")
    public String toString() {
        return stream().collect(Collectors.joining());
    }
}