/*
 * 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 static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.opendaylight.yangtools.yang.data.codec.gson.TestUtils.loadTextFile;
import static org.opendaylight.yangtools.yang.data.codec.gson.TestUtils.loadXmlToNormalizedNodes;
import static org.opendaylight.yangtools.yang.data.codec.gson.TestUtils.normalizedNodesToJsonString;

import com.google.gson.JsonElement;
import com.google.gson.JsonParser;
import com.google.gson.stream.JsonReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.net.URISyntaxException;
import java.nio.charset.StandardCharsets;
import javax.xml.transform.dom.DOMSource;
import org.junit.Test;
import org.opendaylight.yangtools.util.xml.UntrustedXML;
import org.opendaylight.yangtools.yang.common.QName;
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.DOMSourceAnyxmlNode;
import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
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.SchemaPath;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.Text;

public class AnyXmlSupportTest extends AbstractComplexJsonTest {

    private static final QName LF12_ANY = QName.create(CONT_1, "lf12-any");
    private static final QName LF13_ANY = QName.create(CONT_1, "lf13-any");
    private static final QName LF14_ANY = QName.create(CONT_1, "lf14-any");

    @Test
    public void anyXmlNodeWithSimpleValueInContainer() throws IOException, URISyntaxException {
        final String inputJson = loadTextFile("/complexjson/anyxml-node-with-simple-value-in-container.json");

        // deserialization
        final NormalizedNodeResult result = new NormalizedNodeResult();
        final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
        final JsonParserStream jsonParser = JsonParserStream.create(streamWriter, lhotkaCodecFactory);
        jsonParser.parse(new JsonReader(new StringReader(inputJson)));
        final NormalizedNode transformedInput = result.getResult();
        assertNotNull(transformedInput);

        // lf12-any check
        final DOMSource Lf12AnyActualValue = getParsedAnyXmlValue(transformedInput, LF12_ANY);
        final DOMSource Lf12AnyExpectedValue = createAnyXmlSimpleValue("ns:complex:json", "lf12-any", "100.5");
        verifyTransformedAnyXmlNodeValue(Lf12AnyExpectedValue, Lf12AnyActualValue);

        // lf13-any check
        final DOMSource Lf13AnyActualValue = getParsedAnyXmlValue(transformedInput, LF13_ANY);
        final DOMSource Lf13AnyExpectedValue = createAnyXmlSimpleValue("ns:complex:json", "lf13-any", "true");
        verifyTransformedAnyXmlNodeValue(Lf13AnyExpectedValue, Lf13AnyActualValue);

        // lf14-any check
        final DOMSource Lf14AnyActualValue = getParsedAnyXmlValue(transformedInput, LF14_ANY);
        final DOMSource Lf14AnyExpectedValue = createAnyXmlSimpleValue("ns:complex:json", "lf14-any", "null");
        verifyTransformedAnyXmlNodeValue(Lf14AnyExpectedValue, Lf14AnyActualValue);

        final String serializationResult = normalizedNodesToJsonString(transformedInput, schemaContext,
                SchemaPath.ROOT);

        final JsonParser parser = new JsonParser();
        final JsonElement expected = parser.parse(inputJson);
        final JsonElement actual = parser.parse(serializationResult);
        assertTrue(expected.equals(actual));
    }

    @Test
    public void anyXmlNodeWithCompositeValueInContainer() throws IOException, URISyntaxException {
        final String inputJson = loadTextFile("/complexjson/anyxml-node-with-composite-value-in-container.json");

        final NormalizedNodeResult result = new NormalizedNodeResult();
        final NormalizedNodeStreamWriter streamWriter = ImmutableNormalizedNodeStreamWriter.from(result);
        final JsonParserStream jsonParser = JsonParserStream.create(streamWriter, lhotkaCodecFactory);
        jsonParser.parse(new JsonReader(new StringReader(inputJson)));
        final NormalizedNode transformedInput = result.getResult();
        assertNotNull(transformedInput);

        // lf12-any check
        final DOMSource Lf12AnyActualValue = getParsedAnyXmlValue(transformedInput, LF12_ANY);
        final DOMSource Lf12AnyExpectedValue = createLf12AnyXmlCompositeValue("ns:complex:json", "lf12-any");
        verifyTransformedAnyXmlNodeValue(Lf12AnyExpectedValue, Lf12AnyActualValue);
        // lf13-any check
        final DOMSource Lf13AnyActualValue = getParsedAnyXmlValue(transformedInput, LF13_ANY);
        final DOMSource Lf13AnyExpectedValue = createLf13AnyXmlCompositeValue("ns:complex:json", "lf13-any");
        verifyTransformedAnyXmlNodeValue(Lf13AnyExpectedValue, Lf13AnyActualValue);

        // lf14-any check
        final DOMSource Lf14AnyActualValue = getParsedAnyXmlValue(transformedInput, LF14_ANY);
        final DOMSource Lf14AnyExpectedValue = createLf14AnyXmlCompositeValue("ns:complex:json", "lf14-any");
        verifyTransformedAnyXmlNodeValue(Lf14AnyExpectedValue, Lf14AnyActualValue);

        // serialization
        final String serializationResult = normalizedNodesToJsonString(transformedInput, schemaContext,
                SchemaPath.ROOT);

        final JsonParser parser = new JsonParser();
        final String expectedJson = loadTextFile("/bug8927/json/composite.json");
        final JsonElement expected = parser.parse(expectedJson);
        final JsonElement actual = parser.parse(serializationResult);
        assertTrue(expected.equals(actual));
    }

    @Test
    public void bug8927TestComplexArrayWithOthers() throws Exception {
        executebug8927Test("/bug8927/xml/complex_array_with_other_elements.xml",
            "/bug8927/json/complex_array_with_other_elements.json");
    }

    @Test
    public void bug8927TestComplexArray() throws Exception {
        executebug8927Test("/bug8927/xml/complex_array.xml", "/bug8927/json/complex_array.json");
    }

    @Test
    public void bug8927TestScalarArrayWithOthers() throws Exception {
        executebug8927Test("/bug8927/xml/scalar_array_with_other_elements.xml",
            "/bug8927/json/scalar_array_with_other_elements.json");
    }

    @Test
    public void bug8927TestScalarArray() throws Exception {
        executebug8927Test("/bug8927/xml/scalar_array.xml", "/bug8927/json/scalar_array.json");
    }

    private void executebug8927Test(final String inputXmlFile, final String expectedJsonFile) throws Exception {
        final InputStream resourceAsStream = AnyXmlSupportTest.class.getResourceAsStream(inputXmlFile);
        final NormalizedNodeResult result = new NormalizedNodeResult();
        loadXmlToNormalizedNodes(resourceAsStream, result, schemaContext);

        assertNotNull(result.getResult());
        assertTrue(result.getResult() instanceof ContainerNode);

        final DataContainerChild data = ((ContainerNode) result.getResult())
                .childByArg(new NodeIdentifier(QName.create("bug8927.test", "2017-01-01", "foo")));
        assertNotNull(data);
        final String jsonOutput = normalizedNodesToJsonString(data, schemaContext, SchemaPath.ROOT);
        final JsonParser parser = new JsonParser();
        final JsonElement expextedJson = parser.parse(new FileReader(
            new File(getClass().getResource(expectedJsonFile).toURI()), StandardCharsets.UTF_8));
        final JsonElement serializedJson = parser.parse(jsonOutput);
        assertEquals(expextedJson, serializedJson);
    }

    private static DOMSource getParsedAnyXmlValue(final NormalizedNode transformedInput, final QName anyxmlName) {
        assertTrue(transformedInput instanceof ContainerNode);
        final ContainerNode cont1 = (ContainerNode) transformedInput;
        final DataContainerChild child = cont1.childByArg(new NodeIdentifier(anyxmlName));
        assertTrue(child instanceof DOMSourceAnyxmlNode);
        final DOMSourceAnyxmlNode anyXmlNode = (DOMSourceAnyxmlNode) child;
        return anyXmlNode.body();
    }

    private static void verifyTransformedAnyXmlNodeValue(final DOMSource expectedValue, final DOMSource actualValue) {
        assertTrue(expectedValue.getNode().isEqualNode(actualValue.getNode()));
    }

    private static DOMSource createAnyXmlSimpleValue(final String ns, final String name, final String value) {
        final Document doc = UntrustedXML.newDocumentBuilder().newDocument();
        final Element rootElement = doc.createElementNS(ns, name);
        doc.appendChild(rootElement);
        final Text textNode = doc.createTextNode(value);
        rootElement.appendChild(textNode);
        return new DOMSource(doc.getDocumentElement());
    }

    private static DOMSource createLf12AnyXmlCompositeValue(final String ns, final String name) {
        final Document doc = UntrustedXML.newDocumentBuilder().newDocument();
        final Element rootElement = doc.createElementNS(ns, name);

        final Element arrayElement1 = doc.createElement("array-element");
        final Text arrayElement1Text = doc.createTextNode("true");
        arrayElement1.appendChild(arrayElement1Text);

        final Element arrayElement2 = doc.createElement("array-element");

        final Element arrayElement2Baz = doc.createElement("baz");
        final Element bazArrayElement1 = doc.createElement("array-element");
        final Text bazArrayElement1Text = doc.createTextNode("120");
        bazArrayElement1.appendChild(bazArrayElement1Text);
        final Element bazArrayElement2 = doc.createElement("array-element");
        final Text bazArrayElement2Text = doc.createTextNode("str-val");
        bazArrayElement2.appendChild(bazArrayElement2Text);
        final Element bazArrayElement3 = doc.createElement("array-element");
        final Text bazArrayElement3Text = doc.createTextNode("false");
        bazArrayElement3.appendChild(bazArrayElement3Text);

        arrayElement2Baz.appendChild(bazArrayElement1);
        arrayElement2Baz.appendChild(bazArrayElement2);
        arrayElement2Baz.appendChild(bazArrayElement3);

        arrayElement2.appendChild(arrayElement2Baz);

        final Element arrayElement3 = doc.createElement("array-element");
        final Element arrayElement3Foo = doc.createElement("foo");
        final Text fooText = doc.createTextNode("null");
        arrayElement3Foo.appendChild(fooText);
        arrayElement3.appendChild(arrayElement3Foo);

        rootElement.appendChild(arrayElement1);
        rootElement.appendChild(arrayElement2);
        rootElement.appendChild(arrayElement3);

        doc.appendChild(rootElement);

        return new DOMSource(doc.getDocumentElement());
    }

    private static DOMSource createLf13AnyXmlCompositeValue(final String ns, final String name) {
        final Document doc = UntrustedXML.newDocumentBuilder().newDocument();
        final Element rootElement = doc.createElementNS(ns, name);

        final Element anyXmlArrayA = doc.createElement("anyxml-array-a");

        final Element arrayAElement1 = doc.createElement("array-element");
        final Element arrayAElement1Foo = doc.createElement("foo");
        final Text fooText = doc.createTextNode("true");
        arrayAElement1Foo.appendChild(fooText);
        arrayAElement1.appendChild(arrayAElement1Foo);

        final Element arrayAElement2 = doc.createElement("array-element");
        final Text arrayAElement2Text = doc.createTextNode("10");
        arrayAElement2.appendChild(arrayAElement2Text);

        final Element arrayAElement3 = doc.createElement("array-element");
        final Element arrayAElement3Bar = doc.createElement("bar");
        final Text barText = doc.createTextNode("false");
        arrayAElement3Bar.appendChild(barText);
        arrayAElement3.appendChild(arrayAElement3Bar);

        anyXmlArrayA.appendChild(arrayAElement1);
        anyXmlArrayA.appendChild(arrayAElement2);
        anyXmlArrayA.appendChild(arrayAElement3);

        final Element anyXmlArrayB = doc.createElement("anyxml-array-b");

        final Element arrayBElement1 = doc.createElement("array-element");
        final Text arrayBElement1Text = doc.createTextNode("1");
        arrayBElement1.appendChild(arrayBElement1Text);

        final Element arrayBElement2 = doc.createElement("array-element");
        final Text arrayBElement2Text = doc.createTextNode("2");
        arrayBElement2.appendChild(arrayBElement2Text);

        final Element arrayBElement3 = doc.createElement("array-element");

        final Element arrayBElement3Element1 = doc.createElement("array-element");
        final Text arrayBElement3Element1Text = doc.createTextNode("4");
        arrayBElement3Element1.appendChild(arrayBElement3Element1Text);
        final Element arrayBElement3Element2 = doc.createElement("array-element");
        final Text arrayBElement3Element2Text = doc.createTextNode("5");
        arrayBElement3Element2.appendChild(arrayBElement3Element2Text);

        arrayBElement3.appendChild(arrayBElement3Element1);
        arrayBElement3.appendChild(arrayBElement3Element2);

        final Element arrayBElement4 = doc.createElement("array-element");
        final Text arrayBElement4Text = doc.createTextNode("7");
        arrayBElement4.appendChild(arrayBElement4Text);

        anyXmlArrayB.appendChild(arrayBElement1);
        anyXmlArrayB.appendChild(arrayBElement2);
        anyXmlArrayB.appendChild(arrayBElement3);
        anyXmlArrayB.appendChild(arrayBElement4);

        rootElement.appendChild(anyXmlArrayA);
        rootElement.appendChild(anyXmlArrayB);

        doc.appendChild(rootElement);

        return new DOMSource(doc.getDocumentElement());
    }

    private static DOMSource createLf14AnyXmlCompositeValue(final String ns, final String name) {
        final Document doc = UntrustedXML.newDocumentBuilder().newDocument();
        final Element rootElement = doc.createElementNS(ns, name);

        final Element anyXmlObjectA = doc.createElement("anyxml-object-a");
        final Element dataA1 = doc.createElement("data-a1");
        final Text dataA1Text = doc.createTextNode("10");
        dataA1.appendChild(dataA1Text);
        final Element dataA2 = doc.createElement("data-a2");
        final Text dataA2Text = doc.createTextNode("11");
        dataA2.appendChild(dataA2Text);

        anyXmlObjectA.appendChild(dataA1);
        anyXmlObjectA.appendChild(dataA2);

        final Element anyXmlObjectB = doc.createElement("anyxml-object-b");

        final Element childObjectB1 = doc.createElement("child-object-b1");
        final Element dataB1 = doc.createElement("data-b1");
        final Text dataB1Text = doc.createTextNode("5.5");
        dataB1.appendChild(dataB1Text);
        childObjectB1.appendChild(dataB1);

        final Element childObjectB2 = doc.createElement("child-object-b2");
        final Element dataB2 = doc.createElement("data-b2");
        final Text dataB2Text = doc.createTextNode("b2-val");
        dataB2.appendChild(dataB2Text);
        childObjectB2.appendChild(dataB2);

        anyXmlObjectB.appendChild(childObjectB1);
        anyXmlObjectB.appendChild(childObjectB2);

        rootElement.appendChild(anyXmlObjectA);
        rootElement.appendChild(anyXmlObjectB);

        doc.appendChild(rootElement);

        return new DOMSource(doc.getDocumentElement());
    }
}