/*
 * Copyright (c) 2020 PANTHEON.tech, 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.mdsal.binding.runtime.spi;

import com.google.common.annotations.Beta;
import java.util.Arrays;
import java.util.List;
import java.util.NoSuchElementException;
import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.mdsal.binding.runtime.api.ModuleInfoSnapshot;
import org.opendaylight.yangtools.concepts.CheckedBuilder;
import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
import org.opendaylight.yangtools.yang.model.parser.api.YangParserFactory;

@Beta
public final class ModuleInfoSnapshotBuilder implements CheckedBuilder<ModuleInfoSnapshot, NoSuchElementException> {
    private final ModuleInfoSnapshotResolver registry;

    public ModuleInfoSnapshotBuilder(final String name, final YangParserFactory parserFactory) {
        registry = new ModuleInfoSnapshotResolver(name, parserFactory);
    }

    public @NonNull ModuleInfoSnapshotBuilder add(final YangModuleInfo info) {
        return add(List.of(info));
    }

    public @NonNull ModuleInfoSnapshotBuilder add(final YangModuleInfo... infos) {
        return add(Arrays.asList(infos));
    }

    public @NonNull ModuleInfoSnapshotBuilder add(final Iterable<? extends YangModuleInfo> infos) {
        registry.registerModuleInfos(infos);
        return this;
    }

    @Override
    public ModuleInfoSnapshot build() {
        return registry.takeSnapshot();
    }
}