/*
 * Copyright (c) 2011 Tobias Markmann
 * Licensed under the simplified BSD license.
 * See Documentation/Licenses/BSD-simplified.txt for more information.
 */

/*
 * Copyright (c) 2015-2016 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#include <boost/shared_ptr.hpp>
#include <boost/smart_ptr/make_shared.hpp>

#include <cppunit/extensions/HelperMacros.h>
#include <cppunit/extensions/TestFactoryRegistry.h>

#include <Swiften/Base/DateTime.h>
#include <Swiften/Elements/JingleFileTransferDescription.h>
#include <Swiften/Elements/JingleFileTransferFileInfo.h>
#include <Swiften/Elements/JingleFileTransferHash.h>
#include <Swiften/Elements/JingleIBBTransportPayload.h>
#include <Swiften/Elements/JinglePayload.h>
#include <Swiften/Elements/JingleS5BTransportPayload.h>
#include <Swiften/Serializer/PayloadSerializers/FullPayloadSerializerCollection.h>
#include <Swiften/Serializer/PayloadSerializers/JingleFileTransferDescriptionSerializer.h>
#include <Swiften/Serializer/PayloadSerializers/JinglePayloadSerializer.h>
#include <Swiften/Serializer/PayloadSerializers/StreamInitiationFileInfoSerializer.h>
#include <Swiften/StringCodecs/Base64.h>

using namespace Swift;

class JingleSerializersTest : public CppUnit::TestFixture {
        CPPUNIT_TEST_SUITE(JingleSerializersTest);
        CPPUNIT_TEST(testSerialize_StreamInitiationFileInfo);
        CPPUNIT_TEST(testSerialize_StreamInitiationFileInfoRange);

        CPPUNIT_TEST(testSerialize_Xep0261_Example1);
        CPPUNIT_TEST(testSerialize_Xep0261_Example9);
        CPPUNIT_TEST(testSerialize_Xep0261_Example13);

        CPPUNIT_TEST(testSerialize_Xep0234_Example1);
        CPPUNIT_TEST(testSerialize_Xep0234_Example3);
        CPPUNIT_TEST(testSerialize_Xep0234_Example5);
        CPPUNIT_TEST(testSerialize_Xep0234_Example8);

        CPPUNIT_TEST(testSerialize_Xep0260_Example1);

        CPPUNIT_TEST_SUITE_END();

         boost::shared_ptr<JinglePayloadSerializer> createTestling() {
             return boost::make_shared<JinglePayloadSerializer>(&collection);
        }


    public:
        void testSerialize_StreamInitiationFileInfo() {
            std::string expected =    "<file"
                            " date=\"1969-07-21T02:56:15Z\""
                            " hash=\"552da749930852c69ae5d2141d3766b1\""
                            " name=\"test.txt\""
                            " size=\"1022\""
                            " xmlns=\"http://jabber.org/protocol/si/profile/file-transfer\">"
                            "<desc>This is a test. If this were a real file...</desc>"
                            "<range/>"
                        "</file>";

            StreamInitiationFileInfo::ref fileInfo = boost::make_shared<StreamInitiationFileInfo>();
            fileInfo->setDate(stringToDateTime("1969-07-21T02:56:15Z"));
            fileInfo->setHash("552da749930852c69ae5d2141d3766b1");
            fileInfo->setSize(1022);
            fileInfo->setName("test.txt");
            fileInfo->setDescription("This is a test. If this were a real file...");
            fileInfo->setSupportsRangeRequests(true);

            boost::shared_ptr<StreamInitiationFileInfoSerializer> serializer = boost::make_shared<StreamInitiationFileInfoSerializer>();
            CPPUNIT_ASSERT_EQUAL(expected, serializer->serializePayload(fileInfo));
        }

        void testSerialize_StreamInitiationFileInfoRange() {
            std::string expected =    "<file hash=\"552da749930852c69ae5d2141d3766b1\""
                            " xmlns=\"http://jabber.org/protocol/si/profile/file-transfer\">"
                            "<range offset=\"270336\"/>"
                        "</file>";

            StreamInitiationFileInfo::ref fileInfo = boost::make_shared<StreamInitiationFileInfo>();
            fileInfo->setHash("552da749930852c69ae5d2141d3766b1");
            fileInfo->setSupportsRangeRequests(true);
            fileInfo->setRangeOffset(270336);

            boost::shared_ptr<StreamInitiationFileInfoSerializer> serializer = boost::make_shared<StreamInitiationFileInfoSerializer>();
            CPPUNIT_ASSERT_EQUAL(expected, serializer->serializePayload(fileInfo));
        }


        // IBB Transport Method Examples

        // http://xmpp.org/extensions/xep-0261.html#example-1
        void testSerialize_Xep0261_Example1() {
            std::string expected =
                "<jingle action=\"session-initiate\""
                    " initiator=\"romeo@montague.lit/orchard\""
                    " sid=\"a73sjjvkla37jfea\""
                    " xmlns=\"urn:xmpp:jingle:1\">"
                    "<content creator=\"initiator\" name=\"ex\">"
                        "<transport block-size=\"4096\""
                            " sid=\"ch3d9s71\""
                            " xmlns=\"urn:xmpp:jingle:transports:ibb:1\"/>"
                    "</content>"
                "</jingle>";

            JinglePayload::ref payload = boost::make_shared<JinglePayload>();
            payload->setAction(JinglePayload::SessionInitiate);
            payload->setSessionID("a73sjjvkla37jfea");
            payload->setInitiator(JID("romeo@montague.lit/orchard"));

            JingleIBBTransportPayload::ref transport = boost::make_shared<JingleIBBTransportPayload>();
            transport->setBlockSize(4096);
            transport->setSessionID("ch3d9s71");

            JingleContentPayload::ref content = boost::make_shared<JingleContentPayload>();
            content->setCreator(JingleContentPayload::InitiatorCreator);
            content->setName("ex");
            content->addTransport(transport);

            payload->addPayload(content);

            CPPUNIT_ASSERT_EQUAL(expected, createTestling()->serialize(payload));
        }

        // http://xmpp.org/extensions/xep-0261.html#example-9
        void testSerialize_Xep0261_Example9() {
            std::string expected =
                "<jingle action=\"transport-info\""
                    " initiator=\"romeo@montague.lit/orchard\""
                    " sid=\"a73sjjvkla37jfea\""
                    " xmlns=\"urn:xmpp:jingle:1\">"
                    "<content creator=\"initiator\" name=\"ex\">"
                        "<transport block-size=\"2048\""
                            " sid=\"bt8a71h6\""
                            " xmlns=\"urn:xmpp:jingle:transports:ibb:1\"/>"
                    "</content>"
                "</jingle>";

            JinglePayload::ref payload = boost::make_shared<JinglePayload>();
            payload->setAction(JinglePayload::TransportInfo);
            payload->setInitiator(JID("romeo@montague.lit/orchard"));
            payload->setSessionID("a73sjjvkla37jfea");

            JingleContentPayload::ref content = boost::make_shared<JingleContentPayload>();
            content->setCreator(JingleContentPayload::InitiatorCreator);
            content->setName("ex");

            JingleIBBTransportPayload::ref transport = boost::make_shared<JingleIBBTransportPayload>();
            transport->setBlockSize(2048);
            transport->setSessionID("bt8a71h6");

            content->addTransport(transport);
            payload->addPayload(content);

            CPPUNIT_ASSERT_EQUAL(expected, createTestling()->serialize(payload));
        }

        // http://xmpp.org/extensions/xep-0261.html#example-13
        void testSerialize_Xep0261_Example13() {
            std::string expected =
                "<jingle action=\"session-terminate\""
                    " initiator=\"romeo@montague.lit/orchard\""
                    " sid=\"a73sjjvkla37jfea\""
                    " xmlns=\"urn:xmpp:jingle:1\">"
                    "<reason><success/></reason>"
                "</jingle>";

            JinglePayload::ref payload = boost::make_shared<JinglePayload>();
            payload->setAction(JinglePayload::SessionTerminate);
            payload->setInitiator(JID("romeo@montague.lit/orchard"));
            payload->setSessionID("a73sjjvkla37jfea");
            payload->setReason(JinglePayload::Reason(JinglePayload::Reason::Success));

            CPPUNIT_ASSERT_EQUAL(expected, createTestling()->serialize(payload));
        }

        // http://xmpp.org/extensions/xep-0234.html#example-1
        void testSerialize_Xep0234_Example1() {
            std::string expected =    "<description xmlns=\"urn:xmpp:jingle:apps:file-transfer:4\">"
                    "<file>"
                        "<date>1969-07-21T02:56:15Z</date>"
                        "<desc>This is a test. If this were a real file...</desc>"
                        "<name>test.txt</name>"
                        "<range/>"
                        "<size>1022</size>"
                        "<hash algo=\"sha-1\" xmlns=\"urn:xmpp:hashes:1\">VS2nSZMIUsaa5dIUHTdmsQ==</hash>"
                    "</file>"
                "</description>";
            JingleFileTransferDescription::ref desc = boost::make_shared<JingleFileTransferDescription>();
            JingleFileTransferFileInfo fileInfo;

            fileInfo.setDate(stringToDateTime("1969-07-21T02:56:15Z"));
            fileInfo.addHash(HashElement("sha-1", Base64::decode("VS2nSZMIUsaa5dIUHTdmsQ==")));
            fileInfo.setSize(1022);
            fileInfo.setName("test.txt");
            fileInfo.setDescription("This is a test. If this were a real file...");
            fileInfo.setSupportsRangeRequests(true);

            desc->setFileInfo(fileInfo);

            CPPUNIT_ASSERT_EQUAL(expected, boost::make_shared<JingleFileTransferDescriptionSerializer>()->serialize(desc));
        }

        // http://xmpp.org/extensions/xep-0234.html#example-3
        void testSerialize_Xep0234_Example3() {
            std::string expected =
                "<jingle action=\"session-accept\""
                    " initiator=\"romeo@montague.lit/orchard\""
                    " sid=\"851ba2\""
                    " xmlns=\"urn:xmpp:jingle:1\">"
                    "<content creator=\"initiator\" name=\"a-file-offer\">"
                        "<description xmlns=\"urn:xmpp:jingle:apps:file-transfer:4\">"
                            "<file>"
                                "<date>1969-07-21T02:56:15Z</date>"
                                "<desc>This is a test. If this were a real file...</desc>"
                                "<name>test.txt</name>"
                                "<range/>"
                                "<size>1022</size>"
                                "<hash algo=\"sha-1\" xmlns=\"urn:xmpp:hashes:1\">VS2nSZMIUsaa5dIUHTdmsQ==</hash>"
                            "</file>"
                        "</description>"
                        /*"<transport xmlns=\"urn:xmpp:jingle:transports:s5b:1\""
                            " mode=\"tcp\""
                            " sid=\"vj3hs98y\">"
                            "<candidate cid=\"ht567dq\""
                                " host=\"192.169.1.10\""
                                " jid=\"juliet@capulet.lit/balcony\""
                                " port=\"6539\""
                                " priority=\"8257636\""
                                " type=\"direct\"/>"
                            "<candidate cid=\"hr65dqyd\""
                                " host=\"134.102.201.180\""
                                " jid=\"juliet@capulet.lit/balcony\""
                                " port=\"16453\""
                                " priority=\"7929856\""
                                " type=\"assisted\"/>"
                            "<candidate cid=\"grt654q2\""
                                " host=\"2001:638:708:30c9:219:d1ff:fea4:a17d\""
                                " jid=\"juliet@capulet.lit/balcony\""
                                " port=\"6539\""
                                " priority=\"8257606\""
                                " type=\"direct\"/>"
                        "</transport>"*/
                    "</content>"
                "</jingle>";

            JinglePayload::ref payload = boost::make_shared<JinglePayload>();
            payload->setAction(JinglePayload::SessionAccept);
            payload->setInitiator(JID("romeo@montague.lit/orchard"));
            payload->setSessionID("851ba2");

            JingleContentPayload::ref content = boost::make_shared<JingleContentPayload>();
            content->setCreator(JingleContentPayload::InitiatorCreator);
            content->setName("a-file-offer");

            JingleFileTransferDescription::ref description = boost::make_shared<JingleFileTransferDescription>();
            JingleFileTransferFileInfo fileInfo;
            fileInfo.setName("test.txt");
            fileInfo.setSize(1022);
            fileInfo.addHash(HashElement("sha-1", Base64::decode("VS2nSZMIUsaa5dIUHTdmsQ==")));
            fileInfo.setDate(stringToDateTime("1969-07-21T02:56:15Z"));
            fileInfo.setDescription("This is a test. If this were a real file...");
            fileInfo.setSupportsRangeRequests(true);

            description->setFileInfo(fileInfo);
            content->addDescription(description);
            payload->addPayload(content);

            CPPUNIT_ASSERT_EQUAL(expected, createTestling()->serialize(payload));
        }

        // http://xmpp.org/extensions/xep-0234.html#example-5
        void testSerialize_Xep0234_Example5() {
            std::string expected =
                "<jingle"
                    " action=\"transport-info\""
                    " initiator=\"romeo@montague.lit/orchard\""
                    " sid=\"a73sjjvkla37jfea\""
                    " xmlns=\"urn:xmpp:jingle:1\">"
                    "<content creator=\"initiator\" name=\"ex\"/>"
                        /*"<transport"
                            " sid=\"vj3hs98y\""
                            " xmlns=\"urn:xmpp:jingle:transports:s5b:1\">"
                            "<candidate-used cid=\"hr65dqyd\"/>"
                        "</transport>"*/
                    //"</content>"
                "</jingle>";

            JinglePayload::ref payload = boost::make_shared<JinglePayload>();
            payload->setAction(JinglePayload::TransportInfo);
            payload->setInitiator(JID("romeo@montague.lit/orchard"));
            payload->setSessionID("a73sjjvkla37jfea");

            JingleContentPayload::ref content = boost::make_shared<JingleContentPayload>();
            content->setCreator(JingleContentPayload::InitiatorCreator);
            content->setName("ex");
            payload->addPayload(content);

            CPPUNIT_ASSERT_EQUAL(expected, createTestling()->serialize(payload));
        }

        // http://xmpp.org/extensions/xep-0234.html#example-8
        void testSerialize_Xep0234_Example8() {
            std::string expected =
                "<jingle"
                    " action=\"session-info\""
                    " initiator=\"romeo@montague.lit/orchard\""
                    " sid=\"a73sjjvkla37jfea\""
                    " xmlns=\"urn:xmpp:jingle:1\">"
                    "<checksum xmlns=\"urn:xmpp:jingle:apps:file-transfer:4\">"
                        "<file>"
                            "<hash algo=\"sha-1\" xmlns=\"urn:xmpp:hashes:1\">VS2nSZMIUsaa5dIUHTdmsQ==</hash>"
                        "</file>"
                    "</checksum>"
                "</jingle>";

            JinglePayload::ref payload = boost::make_shared<JinglePayload>();
            payload->setAction(JinglePayload::SessionInfo);
            payload->setInitiator(JID("romeo@montague.lit/orchard"));
            payload->setSessionID("a73sjjvkla37jfea");

            JingleFileTransferHash::ref hash = boost::make_shared<JingleFileTransferHash>();
            hash->getFileInfo().addHash(HashElement("sha-1", Base64::decode("VS2nSZMIUsaa5dIUHTdmsQ==")));

            payload->addPayload(hash);

            CPPUNIT_ASSERT_EQUAL(expected, createTestling()->serialize(payload));
        }

        // http://xmpp.org/extensions/xep-0260.html#example-1
        void testSerialize_Xep0260_Example1() {
            std::string expected =
                "<jingle"
                    " action=\"session-initiate\""
                    " initiator=\"romeo@montague.lit/orchard\""
                    " sid=\"a73sjjvkla37jfea\""
                    " xmlns=\"urn:xmpp:jingle:1\">"
                    "<content creator=\"initiator\" name=\"ex\">"
                        "<transport"
                            " dstaddr=\"1a12fb7bc625e55f3ed5b29a53dbe0e4aa7d80ba\""
                            " mode=\"tcp\""
                            " sid=\"vj3hs98y\""
                            " xmlns=\"urn:xmpp:jingle:transports:s5b:1\">"
                            "<candidate cid=\"hft54dqy\""
                                " host=\"192.168.4.1\""
                                " jid=\"romeo@montague.lit/orchard\""
                                " port=\"5086\""
                                " priority=\"8257636\""
                                " type=\"direct\"/>"
                            "<candidate cid=\"hutr46fe\""
                                " host=\"24.24.24.1\""
                                " jid=\"romeo@montague.lit/orchard\""
                                " port=\"5087\""
                                " priority=\"8258636\""
                                " type=\"direct\"/>"
                        "</transport>"
                    "</content>"
                "</jingle>";

            JinglePayload::ref payload = boost::make_shared<JinglePayload>();
            payload->setAction(JinglePayload::SessionInitiate);
            payload->setInitiator(JID("romeo@montague.lit/orchard"));
            payload->setSessionID("a73sjjvkla37jfea");

            JingleContentPayload::ref content = boost::make_shared<JingleContentPayload>();
            content->setCreator(JingleContentPayload::InitiatorCreator);
            content->setName("ex");

            JingleS5BTransportPayload::ref transport = boost::make_shared<JingleS5BTransportPayload>();
            transport->setMode(JingleS5BTransportPayload::TCPMode);
            transport->setDstAddr("1a12fb7bc625e55f3ed5b29a53dbe0e4aa7d80ba");
            transport->setSessionID("vj3hs98y");

            JingleS5BTransportPayload::Candidate candidate1;
            candidate1.cid = "hft54dqy";
            candidate1.hostPort = HostAddressPort(HostAddress("192.168.4.1"), 5086);
            candidate1.jid = JID("romeo@montague.lit/orchard");
            candidate1.priority = 8257636;
            candidate1.type = JingleS5BTransportPayload::Candidate::DirectType;
            transport->addCandidate(candidate1);

            JingleS5BTransportPayload::Candidate candidate2;
            candidate2.cid = "hutr46fe";
            candidate2.hostPort = HostAddressPort(HostAddress("24.24.24.1"), 5087);
            candidate2.jid = JID("romeo@montague.lit/orchard");
            candidate2.priority = 8258636;
            candidate2.type = JingleS5BTransportPayload::Candidate::DirectType;
            transport->addCandidate(candidate2);

            content->addTransport(transport);

            payload->addPayload(content);

            CPPUNIT_ASSERT_EQUAL(expected, createTestling()->serialize(payload));
        }

    private:
        FullPayloadSerializerCollection collection;
};

CPPUNIT_TEST_SUITE_REGISTRATION(JingleSerializersTest);