/*
 * Copyright (C) 2015 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "flatten/XmlFlattener.h"
#include "link/Linkers.h"
#include "test/Builders.h"
#include "test/Context.h"
#include "util/BigBuffer.h"
#include "util/Util.h"

#include <androidfw/ResourceTypes.h>
#include <gtest/gtest.h>

namespace aapt {

class XmlFlattenerTest : public ::testing::Test {
public:
    void SetUp() override {
        mContext = test::ContextBuilder()
                .setCompilationPackage(u"com.app.test")
                .setNameManglerPolicy(NameManglerPolicy{ u"com.app.test" })
                .setSymbolTable(test::StaticSymbolTableBuilder()
                        .addSymbol(u"@android:attr/id", ResourceId(0x010100d0),
                                   test::AttributeBuilder().build())
                        .addSymbol(u"@com.app.test:id/id", ResourceId(0x7f020000))
                        .addSymbol(u"@android:attr/paddingStart", ResourceId(0x010103b3),
                                   test::AttributeBuilder().build())
                        .addSymbol(u"@android:attr/colorAccent", ResourceId(0x01010435),
                                   test::AttributeBuilder().build())
                        .build())
                .build();
    }

    ::testing::AssertionResult flatten(xml::XmlResource* doc, android::ResXMLTree* outTree,
                                       XmlFlattenerOptions options = {}) {
        BigBuffer buffer(1024);
        XmlFlattener flattener(&buffer, options);
        if (!flattener.consume(mContext.get(), doc)) {
            return ::testing::AssertionFailure() << "failed to flatten XML Tree";
        }

        std::unique_ptr<uint8_t[]> data = util::copy(buffer);
        if (outTree->setTo(data.get(), buffer.size(), true) != android::NO_ERROR) {
            return ::testing::AssertionFailure() << "flattened XML is corrupt";
        }
        return ::testing::AssertionSuccess();
    }

protected:
    std::unique_ptr<IAaptContext> mContext;
};

TEST_F(XmlFlattenerTest, FlattenXmlWithNoCompiledAttributes) {
    std::unique_ptr<xml::XmlResource> doc = test::buildXmlDom(R"EOF(
            <View xmlns:test="http://com.test"
                  attr="hey">
              <Layout test:hello="hi" />
              <Layout>Some text</Layout>
            </View>)EOF");


    android::ResXMLTree tree;
    ASSERT_TRUE(flatten(doc.get(), &tree));

    ASSERT_EQ(tree.next(), android::ResXMLTree::START_NAMESPACE);

    size_t len;
    const char16_t* namespacePrefix = tree.getNamespacePrefix(&len);
    EXPECT_EQ(StringPiece16(namespacePrefix, len), u"test");

    const char16_t* namespaceUri = tree.getNamespaceUri(&len);
    ASSERT_EQ(StringPiece16(namespaceUri, len), u"http://com.test");

    ASSERT_EQ(tree.next(), android::ResXMLTree::START_TAG);

    ASSERT_EQ(tree.getElementNamespace(&len), nullptr);
    const char16_t* tagName = tree.getElementName(&len);
    EXPECT_EQ(StringPiece16(tagName, len), u"View");

    ASSERT_EQ(1u, tree.getAttributeCount());
    ASSERT_EQ(tree.getAttributeNamespace(0, &len), nullptr);
    const char16_t* attrName = tree.getAttributeName(0, &len);
    EXPECT_EQ(StringPiece16(attrName, len), u"attr");

    EXPECT_EQ(0, tree.indexOfAttribute(nullptr, 0, u"attr", StringPiece16(u"attr").size()));

    ASSERT_EQ(tree.next(), android::ResXMLTree::START_TAG);

    ASSERT_EQ(tree.getElementNamespace(&len), nullptr);
    tagName = tree.getElementName(&len);
    EXPECT_EQ(StringPiece16(tagName, len), u"Layout");

    ASSERT_EQ(1u, tree.getAttributeCount());
    const char16_t* attrNamespace = tree.getAttributeNamespace(0, &len);
    EXPECT_EQ(StringPiece16(attrNamespace, len), u"http://com.test");

    attrName = tree.getAttributeName(0, &len);
    EXPECT_EQ(StringPiece16(attrName, len), u"hello");

    ASSERT_EQ(tree.next(), android::ResXMLTree::END_TAG);
    ASSERT_EQ(tree.next(), android::ResXMLTree::START_TAG);

    ASSERT_EQ(tree.getElementNamespace(&len), nullptr);
    tagName = tree.getElementName(&len);
    EXPECT_EQ(StringPiece16(tagName, len), u"Layout");
    ASSERT_EQ(0u, tree.getAttributeCount());

    ASSERT_EQ(tree.next(), android::ResXMLTree::TEXT);
    const char16_t* text = tree.getText(&len);
    EXPECT_EQ(StringPiece16(text, len), u"Some text");

    ASSERT_EQ(tree.next(), android::ResXMLTree::END_TAG);
    ASSERT_EQ(tree.getElementNamespace(&len), nullptr);
    tagName = tree.getElementName(&len);
    EXPECT_EQ(StringPiece16(tagName, len), u"Layout");

    ASSERT_EQ(tree.next(), android::ResXMLTree::END_TAG);
    ASSERT_EQ(tree.getElementNamespace(&len), nullptr);
    tagName = tree.getElementName(&len);
    EXPECT_EQ(StringPiece16(tagName, len), u"View");

    ASSERT_EQ(tree.next(), android::ResXMLTree::END_NAMESPACE);
    namespacePrefix = tree.getNamespacePrefix(&len);
    EXPECT_EQ(StringPiece16(namespacePrefix, len), u"test");

    namespaceUri = tree.getNamespaceUri(&len);
    ASSERT_EQ(StringPiece16(namespaceUri, len), u"http://com.test");

    ASSERT_EQ(tree.next(), android::ResXMLTree::END_DOCUMENT);
}

TEST_F(XmlFlattenerTest, FlattenCompiledXmlAndStripSdk21) {
    std::unique_ptr<xml::XmlResource> doc = test::buildXmlDom(R"EOF(
            <View xmlns:android="http://schemas.android.com/apk/res/android"
                android:paddingStart="1dp"
                android:colorAccent="#ffffff"/>)EOF");

    XmlReferenceLinker linker;
    ASSERT_TRUE(linker.consume(mContext.get(), doc.get()));
    ASSERT_TRUE(linker.getSdkLevels().count(17) == 1);
    ASSERT_TRUE(linker.getSdkLevels().count(21) == 1);

    android::ResXMLTree tree;
    XmlFlattenerOptions options;
    options.maxSdkLevel = 17;
    ASSERT_TRUE(flatten(doc.get(), &tree, options));

    while (tree.next() != android::ResXMLTree::START_TAG) {
        ASSERT_NE(tree.getEventType(), android::ResXMLTree::BAD_DOCUMENT);
        ASSERT_NE(tree.getEventType(), android::ResXMLTree::END_DOCUMENT);
    }

    ASSERT_EQ(1u, tree.getAttributeCount());
    EXPECT_EQ(uint32_t(0x010103b3), tree.getAttributeNameResID(0));
}

TEST_F(XmlFlattenerTest, AssignSpecialAttributeIndices) {
    std::unique_ptr<xml::XmlResource> doc = test::buildXmlDom(R"EOF(
            <View xmlns:android="http://schemas.android.com/apk/res/android"
                  android:id="@id/id"
                  class="str"
                  style="@id/id"/>)EOF");

    android::ResXMLTree tree;
    ASSERT_TRUE(flatten(doc.get(), &tree));

    while (tree.next() != android::ResXMLTree::START_TAG) {
        ASSERT_NE(tree.getEventType(), android::ResXMLTree::BAD_DOCUMENT);
        ASSERT_NE(tree.getEventType(), android::ResXMLTree::END_DOCUMENT);
    }

    EXPECT_EQ(tree.indexOfClass(), 0);
    EXPECT_EQ(tree.indexOfStyle(), 1);
}

/*
 * The device ResXMLParser in libandroidfw differentiates between empty namespace and null
 * namespace.
 */
TEST_F(XmlFlattenerTest, NoNamespaceIsNotTheSameAsEmptyNamespace) {
    std::unique_ptr<xml::XmlResource> doc = test::buildXmlDom("<View package=\"android\"/>");

    android::ResXMLTree tree;
    ASSERT_TRUE(flatten(doc.get(), &tree));

    while (tree.next() != android::ResXMLTree::START_TAG) {
        ASSERT_NE(tree.getEventType(), android::ResXMLTree::BAD_DOCUMENT);
        ASSERT_NE(tree.getEventType(), android::ResXMLTree::END_DOCUMENT);
    }

    const StringPiece16 kPackage = u"package";
    EXPECT_GE(tree.indexOfAttribute(nullptr, 0, kPackage.data(), kPackage.size()), 0);
}

} // namespace aapt