/*
 * 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.xml;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import com.google.common.collect.ImmutableSet;
import java.io.InputStream;
import java.net.URI;
import java.util.Collection;
import java.util.Optional;
import javax.xml.stream.XMLStreamReader;
import org.junit.Before;
import org.junit.Test;
import org.opendaylight.yangtools.odlext.model.api.YangModeledAnyxmlSchemaNode;
import org.opendaylight.yangtools.util.xml.UntrustedXML;
import org.opendaylight.yangtools.yang.common.QName;
import org.opendaylight.yangtools.yang.common.QNameModule;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
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.NormalizedNode;
import org.opendaylight.yangtools.yang.data.api.schema.YangModeledAnyXmlNode;
import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
import org.opendaylight.yangtools.yang.model.api.Module;
import org.opendaylight.yangtools.yang.model.api.MustDefinition;
import org.opendaylight.yangtools.yang.model.api.SchemaPath;
import org.opendaylight.yangtools.yang.model.api.Status;
import org.opendaylight.yangtools.yang.model.api.stmt.AnyxmlEffectiveStatement;
import org.opendaylight.yangtools.yang.test.util.YangParserTestUtils;
import org.opendaylight.yangtools.yang.xpath.api.YangXPathExpression.QualifiedBound;

public class YangModeledAnyXMLDeserializationTest {

    private QNameModule fooModuleQName;
    private QNameModule barModuleQName;
    private QName myContainer1;
    private QName myContainer2;
    private QName innerContainer;
    private QName myLeaf3;
    private QName myLeaf2;
    private QName myLeaf1;
    private QName myAnyXMLDataBar;
    private QName myAnyXMLDataFoo;
    private EffectiveModelContext schemaContext;

    @Before
    public void setUp() {
        barModuleQName = QNameModule.create(URI.create("bar"));
        myContainer1 = QName.create(barModuleQName, "my-container-1");
        myLeaf1 = QName.create(barModuleQName, "my-leaf-1");
        myAnyXMLDataBar = QName.create(barModuleQName, "my-anyxml-data");

        fooModuleQName = QNameModule.create(URI.create("foo"));
        myContainer2 = QName.create(fooModuleQName, "my-container-2");
        innerContainer = QName.create(fooModuleQName, "inner-container");
        myLeaf3 = QName.create(fooModuleQName, "my-leaf-3");
        myLeaf2 = QName.create(fooModuleQName, "my-leaf-2");
        myAnyXMLDataFoo = QName.create(fooModuleQName, "my-anyxml-data");

        schemaContext = YangParserTestUtils.parseYangResourceDirectory("/anyxml-support/yang");
    }

    @Test
    public void testRawAnyXMLFromBar() throws Exception {
        DataSchemaNode barContainer = schemaContext.findDataChildByName(QName.create(barModuleQName, "bar")).get();
        assertTrue(barContainer instanceof ContainerSchemaNode);
        final YangModeledAnyxmlSchemaNode yangModeledAnyXML = new YangModeledAnyXMLSchemaNodeImplTest(myAnyXMLDataBar,
                (ContainerSchemaNode) barContainer);

        final InputStream resourceAsStream = YangModeledAnyXMLDeserializationTest.class.getResourceAsStream(
                "/anyxml-support/xml/bar.xml");

        final XMLStreamReader reader = UntrustedXML.createXMLStreamReader(resourceAsStream);
        final NormalizedNodeResult result = new NormalizedNodeResult();

        final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);

        final XmlParserStream xmlParser = XmlParserStream.create(streamWriter, schemaContext, yangModeledAnyXML);
        xmlParser.parse(reader);

        final NormalizedNode output = result.getResult();
        assertTrue(output instanceof YangModeledAnyXmlNode);
        final YangModeledAnyXmlNode yangModeledAnyXmlNode = (YangModeledAnyXmlNode) output;

        Collection<DataContainerChild> value = yangModeledAnyXmlNode.body();
        assertEquals(2, value.size());

        DataContainerChild child = yangModeledAnyXmlNode.childByArg(new NodeIdentifier(myContainer1));
        assertTrue(child instanceof ContainerNode);
        ContainerNode myContainerNode1 = (ContainerNode) child;

        DataContainerChild child2 = myContainerNode1.childByArg(new NodeIdentifier(myLeaf1));
        assertTrue(child2 instanceof LeafNode);
        LeafNode<?> leafNode1 = (LeafNode<?>) child2;

        Object leafNode1Value = leafNode1.body();
        assertEquals("value1", leafNode1Value);
    }

    @Test
    public void testRealSchemaContextFromFoo() throws Exception {
        final InputStream resourceAsStream = YangModeledAnyXMLDeserializationTest.class.getResourceAsStream(
                "/anyxml-support/xml/foo.xml");
        final Module foo = schemaContext.findModules("foo").iterator().next();
        final YangModeledAnyxmlSchemaNode myAnyXmlData = (YangModeledAnyxmlSchemaNode) foo.findDataChildByName(
                QName.create(foo.getQNameModule(), "my-anyxml-data")).get();

        final XMLStreamReader reader = UntrustedXML.createXMLStreamReader(resourceAsStream);
        final NormalizedNodeResult result = new NormalizedNodeResult();

        final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);

        final XmlParserStream xmlParser = XmlParserStream.create(streamWriter, schemaContext, myAnyXmlData);
        xmlParser.parse(reader);

        final NormalizedNode output = result.getResult();
        assertTrue(output instanceof YangModeledAnyXmlNode);
        final YangModeledAnyXmlNode yangModeledAnyXmlNode = (YangModeledAnyXmlNode) output;

        DataSchemaNode schemaOfAnyXmlData = yangModeledAnyXmlNode.getSchemaOfAnyXmlData();
        DataSchemaNode expectedSchemaOfAnyXmlData = schemaContext.findDataChildByName(myContainer2).get();
        assertEquals(expectedSchemaOfAnyXmlData, schemaOfAnyXmlData);

        Collection<DataContainerChild> value = yangModeledAnyXmlNode.body();
        assertEquals(2, value.size());

        DataContainerChild child2 = yangModeledAnyXmlNode.childByArg(new NodeIdentifier(innerContainer));
        assertTrue(child2 instanceof ContainerNode);
        ContainerNode innerContainerNode = (ContainerNode) child2;

        DataContainerChild child3 = innerContainerNode.childByArg(new NodeIdentifier(myLeaf2));
        assertTrue(child3 instanceof LeafNode);
        LeafNode<?> leafNode2 = (LeafNode<?>) child3;

        Object leafNode2Value = leafNode2.body();
        assertEquals("any-xml-leaf-2-value", leafNode2Value);

        DataContainerChild child4 = yangModeledAnyXmlNode.childByArg(new NodeIdentifier(myLeaf3));
        assertTrue(child4 instanceof LeafNode);
        LeafNode<?> leafNode3 = (LeafNode<?>) child4;

        Object leafNode3Value = leafNode3.body();
        assertEquals("any-xml-leaf-3-value", leafNode3Value);
    }

    private static final class YangModeledAnyXMLSchemaNodeImplTest implements YangModeledAnyxmlSchemaNode {
        private final QName qname;
        private final ContainerSchemaNode contentSchema;

        YangModeledAnyXMLSchemaNodeImplTest(final QName qname, final ContainerSchemaNode contentSchema) {
            this.qname = qname;
            this.contentSchema = contentSchema;
        }

        @Override
        public boolean isAugmenting() {
            return false;
        }

        @Override
        public boolean isAddedByUses() {
            return false;
        }

        @Override
        public Optional<Boolean> effectiveConfig() {
            return Optional.of(Boolean.FALSE);
        }

        @Override
        public boolean isMandatory() {
            return false;
        }

        @Override
        public QName getQName() {
            return qname;
        }

        @Override
        @Deprecated
        public SchemaPath getPath() {
            return null;
        }

        @Override
        public Optional<String> getDescription() {
            return Optional.empty();
        }

        @Override
        public Optional<String> getReference() {
            return Optional.empty();
        }

        @Override
        public Status getStatus() {
            return Status.CURRENT;
        }

        @Override
        public ContainerSchemaNode getSchemaOfAnyXmlData() {
            return contentSchema;
        }

        @Override
        public Collection<MustDefinition> getMustConstraints() {
            return ImmutableSet.of();
        }

        @Override
        public Optional<? extends QualifiedBound> getWhenCondition() {
            return Optional.empty();
        }

        @Override
        public AnyxmlEffectiveStatement asEffectiveStatement() {
            throw new UnsupportedOperationException();
        }
    }
}