/*
 * Copyright (c) 2023 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.restconf.nb.rfc8040;

import static org.junit.jupiter.api.Assertions.assertEquals;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.charset.StandardCharsets;
import org.eclipse.jdt.annotation.NonNull;
import org.opendaylight.restconf.api.FormatParameters;
import org.opendaylight.restconf.api.query.PrettyPrintParam;
import org.opendaylight.restconf.server.api.DatabindContext;
import org.opendaylight.yangtools.yang.common.Decimal64;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.common.Uint32;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
import org.opendaylight.yangtools.yang.data.api.schema.SystemMapNode;
import org.opendaylight.yangtools.yang.data.spi.node.ImmutableNodes;
import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;

public abstract class AbstractJukeboxTest {
    @FunctionalInterface
    public interface FormatMethod {

        void invoke(@NonNull FormatParameters format, @NonNull OutputStream out) throws IOException;
    }

    // container jukebox
    protected static final QName JUKEBOX_QNAME =
        QName.create("http://example.com/ns/example-jukebox", "2015-04-04", "jukebox");
    // container player
    protected static final QName PLAYER_QNAME = QName.create(JUKEBOX_QNAME, "player");
    // container library
    protected static final QName LIBRARY_QNAME = QName.create(JUKEBOX_QNAME, "library");
    // list song
    protected static final QName SONG_QNAME = QName.create(JUKEBOX_QNAME, "song");
    // leaf id
    protected static final QName SONG_ID_QNAME = QName.create(JUKEBOX_QNAME, "id");
    // leaf index
    protected static final QName SONG_INDEX_QNAME = QName.create(JUKEBOX_QNAME, "index");
    // list artist
    protected static final QName ARTIST_QNAME = QName.create(JUKEBOX_QNAME, "artist");
    // list album
    protected static final QName ALBUM_QNAME = QName.create(JUKEBOX_QNAME, "album");
    // leaf name
    protected static final QName NAME_QNAME = QName.create(JUKEBOX_QNAME, "name");
    // leaf gap
    protected static final QName GAP_QNAME = QName.create(JUKEBOX_QNAME, "gap");
    // leaf playlist
    protected static final QName PLAYLIST_QNAME = QName.create(JUKEBOX_QNAME, "playlist");
    // leaf description
    protected static final QName DESCRIPTION_QNAME = QName.create(JUKEBOX_QNAME, "description");

    protected static final YangInstanceIdentifier JUKEBOX_IID = YangInstanceIdentifier.of(JUKEBOX_QNAME);
    protected static final YangInstanceIdentifier PLAYLIST_IID =
        YangInstanceIdentifier.of(JUKEBOX_QNAME, PLAYLIST_QNAME);

    // instance identifier for accessing leaf node "gap"
    protected static final YangInstanceIdentifier GAP_IID =
        YangInstanceIdentifier.of(JUKEBOX_QNAME, PLAYER_QNAME, GAP_QNAME);

    protected static final LeafNode<?> GAP_LEAF = ImmutableNodes.leafNode(GAP_QNAME, Decimal64.valueOf("0.2"));
    protected static final ContainerNode CONT_PLAYER = ImmutableNodes.newContainerBuilder()
        .withNodeIdentifier(new NodeIdentifier(PLAYER_QNAME))
        .withChild(GAP_LEAF)
        .build();
    protected static final ContainerNode EMPTY_JUKEBOX = ImmutableNodes.newContainerBuilder()
        .withNodeIdentifier(new NodeIdentifier(JUKEBOX_QNAME))
        .withChild(CONT_PLAYER)
        .build();
    protected static final MapEntryNode BAND_ENTRY = ImmutableNodes.newMapEntryBuilder()
        .withNodeIdentifier(NodeIdentifierWithPredicates.of(PLAYLIST_QNAME, NAME_QNAME, "name of band"))
        .withChild(ImmutableNodes.leafNode(NAME_QNAME, "name of band"))
        .withChild(ImmutableNodes.leafNode(DESCRIPTION_QNAME, "band description"))
        .build();
    protected static final MapEntryNode SONG1 = ImmutableNodes.newMapEntryBuilder()
        .withNodeIdentifier(YangInstanceIdentifier.NodeIdentifierWithPredicates.of(SONG_QNAME, SONG_INDEX_QNAME,
            Uint32.ONE))
        .withChild(ImmutableNodes.leafNode(SONG_ID_QNAME, "A"))
        .build();
    protected static final MapEntryNode SONG2 = ImmutableNodes.newMapEntryBuilder()
        .withNodeIdentifier(YangInstanceIdentifier.NodeIdentifierWithPredicates.of(SONG_QNAME, SONG_INDEX_QNAME,
            Uint32.TWO))
        .withChild(ImmutableNodes.leafNode(SONG_ID_QNAME, "B"))
        .build();
    protected static final SystemMapNode PLAYLIST_WITH_SONGS = ImmutableNodes.newSystemMapBuilder()
        .withNodeIdentifier(new NodeIdentifier(SONG_QNAME))
            .withChild(SONG1)
            .withChild(SONG2)
        .build();

    protected static final @NonNull EffectiveModelContext JUKEBOX_SCHEMA =
        YangParserTestUtils.parseYangResourceDirectory("/jukebox");
    protected static final @NonNull DatabindContext JUKEBOX_DATABIND = DatabindContext.ofModel(JUKEBOX_SCHEMA);

    protected static final InputStream stringInputStream(final String str) {
        return new ByteArrayInputStream(str.getBytes(StandardCharsets.UTF_8));
    }

    public static void assertFormat(final String expected, final FormatMethod formatMethod,
            final boolean prettyPrint) {
        final var baos = new ByteArrayOutputStream();
        try {
            formatMethod.invoke(new FormatParameters(PrettyPrintParam.of(prettyPrint)), baos);
        } catch (IOException e) {
            throw new AssertionError(e);
        }
        assertEquals(expected, baos.toString(StandardCharsets.UTF_8));
    }
}