/*
 * Copyright (c) 2016 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.data.codec.gson;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import java.net.URI;
import java.util.Arrays;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.common.QNameModule;
import org.opendaylight.yangtools.yang.common.Revision;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.AugmentationIdentifier;
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.YangInstanceIdentifier.NodeWithValue;
import org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode;
import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
import org.opendaylight.yangtools.yang.data.api.schema.LeafSetNode;
import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListEntryNode;
import org.opendaylight.yangtools.yang.data.api.schema.UnkeyedListNode;
import org.opendaylight.yangtools.yang.data.impl.schema.Builders;

public final class TestingNormalizedNodeStructuresCreator {
    private static final QNameModule COMPLEX_JSON = QNameModule.create(URI.create("ns:complex:json"),
        Revision.of("2014-08-11"));
    private static final QNameModule COMPLEX_JSON_AUG = QNameModule.create(URI.create("ns:complex:json:augmentation"),
        Revision.of("2014-08-14"));

    private TestingNormalizedNodeStructuresCreator() {
        throw new UnsupportedOperationException();
    }

    @SafeVarargs
    static ContainerNode cont1Node(final DataContainerChild<?, ?>... children) {
        return Builders.containerBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "cont1")))
                .withValue(Arrays.asList(children))
                .build();
    }

    static ContainerNode cont2Node() {
        return Builders.containerBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "cont2")))
                .build();
    }

    private static UnkeyedListNode lst12Node() {
        return Builders.unkeyedListBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lst12")))
                .withChild(lst12Entry1Node())
                .build();
    }

    private static UnkeyedListEntryNode lst12Entry1Node() {
        return Builders.unkeyedListEntryBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lst12")))
                .withChild(Builders.leafBuilder()
                    .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf121")))
                    .withValue("lf121 value").build())
                .build();
    }

    private static ChoiceNode choc12Node() {
        return Builders.choiceBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "choc12")))
                .withChild(lf17Node())
                .build();
    }

    protected static LeafNode<Object> lf17Node() {
        return Builders.leafBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf17")))
                .withValue("lf17 value").build();
    }

    private static AugmentationNode externalAugmentC11AWithLf15_11AndLf15_12Node() {
        return Builders.augmentationBuilder()
                .withNodeIdentifier(new AugmentationIdentifier(ImmutableSet.of(
                    QName.create(COMPLEX_JSON_AUG, "lf15_11"),
                    QName.create(COMPLEX_JSON_AUG, "lf15_12"))))
                .withChild(lf15_11NodeExternal())
                .withChild(lf15_12NodeExternal())
                .build();
    }

    private static LeafNode<Object> lf15_12NodeExternal() {
        return Builders.leafBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON_AUG, "lf15_12")))
                .withValue("lf15_12 value from augmentation")
                .build();
    }

    private static LeafNode<Object> lf15_11NodeExternal() {
        return Builders.leafBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON_AUG, "lf15_11")))
                .withValue("lf15_11 value from augmentation")
                .build();
    }

    @SafeVarargs
    private static ChoiceNode choc11Node(final DataContainerChild<?, ?>... children) {
        return Builders.choiceBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "choc11")))
                .withValue(Arrays.asList(children))
                // choc11Builder.addChild(lf13Node());
                // choc11Builder.addChild(augmentChoc11_c11A_lf1511AndLf1512Children());
                // choc11Builder.addChild(augmentChoc11_c11_lf1521Children());
                .build();
    }

    private static LeafNode<Object> lf13Node() {
        return Builders.leafBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf13")))
                .withValue("lf13 value").build();
    }

    private static AugmentationNode augmentC11AWithLf15_21Node() {
        return Builders.augmentationBuilder().withNodeIdentifier(
            new AugmentationIdentifier(ImmutableSet.of(QName.create(COMPLEX_JSON, "lf15_21"))))
                .withChild(lf15_21Node()).build();
    }

    private static LeafNode<Object> lf15_21Node() {
        return Builders.leafBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf15_21")))
                .withValue("lf15_21 value").build();
    }

    private static AugmentationNode augmentC11AWithLf15_11AndLf15_12Node() {
        return Builders.augmentationBuilder()
                .withNodeIdentifier(new AugmentationIdentifier(ImmutableSet.of(
                    QName.create(COMPLEX_JSON, "lf15_11"),
                    QName.create(COMPLEX_JSON, "lf15_12"))))
                .withChild(lf15_11Node())
                .withChild(lf15_12Node())
                .build();
    }

    private static LeafNode<Object> lf15_12Node() {
        return Builders.leafBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf15_12")))
                .withValue(QName.create(COMPLEX_JSON, "ident")).build();
    }

    private static LeafNode<Object> lf15_11Node() {
        return Builders.leafBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf15_11")))
                .withValue(ImmutableSet.of("one", "two")).build();
    }

    private static AugmentationNode lf12_1Node() {
        return Builders.augmentationBuilder()
                .withNodeIdentifier(new AugmentationIdentifier(ImmutableSet.of(
                    QName.create(COMPLEX_JSON, "lf12_1"),
                    QName.create(COMPLEX_JSON, "lf12_2"))))
                .withChild(Builders.leafBuilder()
                    .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf12_1")))
                    .withValue("lf12 value").build())
                .build();
    }

    private static MapNode childLst11() {
        return Builders.mapBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lst11")))
                .withChild(Builders.mapEntryBuilder().withNodeIdentifier(
                    NodeIdentifierWithPredicates.of(QName.create(COMPLEX_JSON, "lst11"), ImmutableMap.of(
                        QName.create(COMPLEX_JSON, "key111"), "key111 value",
                        QName.create(COMPLEX_JSON, "lf111"), "lf111 value")))
                    .withChild(Builders.leafBuilder()
                        .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "key111")))
                        .withValue("key111 value").build())
                    .withChild(Builders.leafBuilder()
                        .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf112")))
                        .withValue(lf112Value()).build())
                    .withChild(Builders.leafBuilder()
                        .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf113")))
                        .withValue("lf113 value").build())
                    .withChild(Builders.leafBuilder()
                        .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf111")))
                        .withValue("lf111 value").build())
                    .build())
                .build();
    }

    private static Object lf112Value() {
        return YangInstanceIdentifier.create(
                new NodeIdentifier(QName.create(COMPLEX_JSON, "cont1")),
                new NodeIdentifier(QName.create(COMPLEX_JSON, "lflst11")),
                new NodeWithValue<>(QName.create(COMPLEX_JSON, "lflst11"),"foo")
        );
    }

    private static LeafSetNode<?> childLflst11() {
        return Builders.leafSetBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lflst11")))
                .withChild(Builders.leafSetEntryBuilder()
                    .withNodeIdentifier(new NodeWithValue<>(QName.create(COMPLEX_JSON, "lflst11"), "lflst11 value1"))
                    .withValue("lflst11 value1").build())
                .withChild(Builders.leafSetEntryBuilder()
                    .withNodeIdentifier(new NodeWithValue<>(QName.create(COMPLEX_JSON, "lflst11"), "lflst11 value2"))
                    .withValue("lflst11 value2").build())
                .build();
    }

    private static LeafSetNode<?> childLflst11Multiline() {
        return Builders.leafSetBuilder()
                .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lflst11")))
                .withChild(Builders.leafSetEntryBuilder()
                    .withNodeIdentifier(new NodeWithValue<>(QName.create(COMPLEX_JSON, "lflst11"),
                            "lflst11 value1\nanother line 1"))
                    .withValue("lflst11 value1\nanother line 1").build())
                .withChild(Builders.leafSetEntryBuilder()
                    .withNodeIdentifier(new NodeWithValue<>(QName.create(COMPLEX_JSON, "lflst11"),
                            "lflst11 value2\r\nanother line 2"))
                    .withValue("lflst11 value2\r\nanother line 2").build())
                .build();
    }

    public static ContainerNode leafNodeInContainer() {
        return cont1Node(Builders.leafBuilder()
            .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "lf11")))
            .withValue(453).build());
    }

    public static ContainerNode leafListNodeInContainer() {
        return cont1Node(childLflst11());
    }

    public static ContainerNode leafListNodeInContainerMultiline() {
        return cont1Node(childLflst11Multiline());
    }

    public static ContainerNode keyedListNodeInContainer() {
        return cont1Node(childLst11());
    }

    public static ContainerNode leafNodeViaAugmentationInContainer() {
        return cont1Node(lf12_1Node());
    }

    public static ContainerNode choiceNodeInContainer() {
        return cont1Node(choc11Node(lf13Node()));
    }

    /**
     * choc11 contains lf13, lf15_11 and lf15_12 are added via external augmentation.
     */
    public static ContainerNode caseNodeAugmentationInChoiceInContainer() {
        return cont1Node(choc11Node(augmentC11AWithLf15_11AndLf15_12Node(), lf13Node(), augmentC11AWithLf15_21Node()));
    }

    public static ContainerNode caseNodeExternalAugmentationInChoiceInContainer() {
        return cont1Node(choc11Node(lf13Node(), augmentC11AWithLf15_11AndLf15_12Node(),
            externalAugmentC11AWithLf15_11AndLf15_12Node()));
    }

    public static ContainerNode choiceNodeAugmentationInContainer() {
        return cont1Node(choc12Node());
    }

    public static ContainerNode unkeyedNodeInContainer() {
        return cont1Node(lst12Node());
    }

    public static ContainerNode topLevelContainer() {
        return cont1Node();
    }

    public static ContainerNode emptyContainerInContainer() {
        return cont1Node(Builders.augmentationBuilder()
                .withNodeIdentifier(new AugmentationIdentifier(ImmutableSet.of(QName.create(COMPLEX_JSON, "cont11"))))
                .withChild(Builders.containerBuilder()
                    .withNodeIdentifier(new NodeIdentifier(QName.create(COMPLEX_JSON, "cont11")))
                    .build())
                .build());
    }
}