#include <gtest/gtest.h>

#include <QJsonArray>
#include <QJsonObject>

#include "MessageEvent.h"
#include "MessageEventContent.h"

#include "Audio.h"
#include "Emote.h"
#include "File.h"
#include "Image.h"
#include "Location.h"
#include "Notice.h"
#include "Text.h"
#include "Video.h"

using namespace matrix::events;

TEST(MessageEvent, Audio)
{
        auto info =
          QJsonObject{ { "duration", 2140786 }, { "mimetype", "audio/mpeg" }, { "size", 1563688 } };

        auto content = QJsonObject{ { "body", "Bee Gees - Stayin' Alive" },
                                    { "msgtype", "m.audio" },
                                    { "url", "mxc://localhost/2sdfj23f33r3faad" },
                                    { "info", info } };

        auto event = QJsonObject{ { "content", content },
                                  { "event_id", "$asdfafdf8af:matrix.org" },
                                  { "room_id", "!aasdfaeae23r9:matrix.org" },
                                  { "sender", "@alice:matrix.org" },
                                  { "origin_server_ts", 1323238293289323LL },
                                  { "type", "m.room.message" } };

        MessageEvent<messages::Audio> audio;
        audio.deserialize(event);

        EXPECT_EQ(audio.msgContent().info().duration, 2140786);
        EXPECT_EQ(audio.msgContent().info().size, 1563688);
        EXPECT_EQ(audio.msgContent().info().mimetype, "audio/mpeg");
        EXPECT_EQ(audio.content().body(), "Bee Gees - Stayin' Alive");
}

TEST(MessageEvent, Emote)
{
        auto content = QJsonObject{ { "body", "emote message" }, { "msgtype", "m.emote" } };

        auto event = QJsonObject{ { "content", content },
                                  { "event_id", "$asdfafdf8af:matrix.org" },
                                  { "room_id", "!aasdfaeae23r9:matrix.org" },
                                  { "sender", "@alice:matrix.org" },
                                  { "origin_server_ts", 1323238293289323LL },
                                  { "type", "m.room.message" } };

        MessageEvent<messages::Emote> emote;
        emote.deserialize(event);

        EXPECT_EQ(emote.content().body(), "emote message");
}

TEST(MessageEvent, File)
{
        auto thumbnail_info = QJsonObject{
                { "h", 300 }, { "w", 400 }, { "size", 3432434 }, { "mimetype", "image/jpeg" }
        };

        auto file_info = QJsonObject{ { "size", 24242424 },
                                      { "mimetype", "application/msword" },
                                      { "thumbnail_url", "mxc://localhost/adfaefaFAFSDFF3" },
                                      { "thumbnail_info", thumbnail_info } };

        auto content = QJsonObject{ { "body", "something-important.doc" },
                                    { "filename", "something-important.doc" },
                                    { "url", "mxc://localhost/23d233d32r3r2r" },
                                    { "info", file_info },
                                    { "msgtype", "m.file" } };

        auto event = QJsonObject{ { "content", content },
                                  { "event_id", "$asdfafdf8af:matrix.org" },
                                  { "room_id", "!aasdfaeae23r9:matrix.org" },
                                  { "sender", "@alice:matrix.org" },
                                  { "origin_server_ts", 1323238293289323LL },
                                  { "type", "m.room.message" } };

        MessageEvent<messages::File> file;
        file.deserialize(event);

        EXPECT_EQ(file.content().body(), "something-important.doc");
        EXPECT_EQ(file.msgContent().info().thumbnail_info.h, 300);
        EXPECT_EQ(file.msgContent().info().thumbnail_info.w, 400);
        EXPECT_EQ(file.msgContent().info().thumbnail_info.mimetype, "image/jpeg");
        EXPECT_EQ(file.msgContent().info().mimetype, "application/msword");
        EXPECT_EQ(file.msgContent().info().size, 24242424);
        EXPECT_EQ(file.content().body(), "something-important.doc");
}

TEST(MessageEvent, Image)
{
        auto thumbinfo = QJsonObject{
                { "h", 11 }, { "w", 22 }, { "size", 212 }, { "mimetype", "img/jpeg" },
        };

        auto imginfo = QJsonObject{
                { "h", 110 },
                { "w", 220 },
                { "size", 2120 },
                { "mimetype", "img/jpeg" },
                { "thumbnail_url", "https://images.com/image-thumb.jpg" },
                { "thumbnail_info", thumbinfo },
        };

        auto content = QJsonObject{ { "body", "Image title" },
                                    { "msgtype", "m.image" },
                                    { "url", "https://images.com/image.jpg" },
                                    { "info", imginfo } };

        auto event = QJsonObject{ { "content", content },
                                  { "event_id", "$asdfafdf8af:matrix.org" },
                                  { "room_id", "!aasdfaeae23r9:matrix.org" },
                                  { "sender", "@alice:matrix.org" },
                                  { "origin_server_ts", 1323238293289323LL },
                                  { "type", "m.room.message" } };

        MessageEvent<messages::Image> img;
        img.deserialize(event);

        EXPECT_EQ(img.content().body(), "Image title");
        EXPECT_EQ(img.msgContent().info().h, 110);
        EXPECT_EQ(img.msgContent().info().w, 220);
        EXPECT_EQ(img.msgContent().info().thumbnail_info.w, 22);
        EXPECT_EQ(img.msgContent().info().mimetype, "img/jpeg");
        EXPECT_EQ(img.msgContent().info().thumbnail_url, "https://images.com/image-thumb.jpg");
}

TEST(MessageEvent, Location)
{
        auto thumbnail_info = QJsonObject{
                { "h", 300 }, { "w", 400 }, { "size", 3432434 }, { "mimetype", "image/jpeg" }
        };

        auto info = QJsonObject{ { "thumbnail_url", "mxc://localhost/adfaefaFAFSDFF3" },
                                 { "thumbnail_info", thumbnail_info } };

        auto content = QJsonObject{ { "body", "Big Ben, London, UK" },
                                    { "geo_uri", "geo:51.5008,0.1247" },
                                    { "info", info },
                                    { "msgtype", "m.location" } };

        auto event = QJsonObject{ { "content", content },
                                  { "event_id", "$asdfafdf8af:matrix.org" },
                                  { "room_id", "!aasdfaeae23r9:matrix.org" },
                                  { "sender", "@alice:matrix.org" },
                                  { "origin_server_ts", 1323238293289323LL },
                                  { "type", "m.room.message" } };

        MessageEvent<messages::Location> location;
        location.deserialize(event);

        EXPECT_EQ(location.msgContent().info().thumbnail_info.h, 300);
        EXPECT_EQ(location.msgContent().info().thumbnail_info.w, 400);
        EXPECT_EQ(location.msgContent().info().thumbnail_info.mimetype, "image/jpeg");
        EXPECT_EQ(location.msgContent().info().thumbnail_url, "mxc://localhost/adfaefaFAFSDFF3");
        EXPECT_EQ(location.content().body(), "Big Ben, London, UK");
}

TEST(MessageEvent, Notice)
{
        auto content = QJsonObject{ { "body", "notice message" }, { "msgtype", "m.notice" } };

        auto event = QJsonObject{ { "content", content },
                                  { "event_id", "$asdfafdf8af:matrix.org" },
                                  { "room_id", "!aasdfaeae23r9:matrix.org" },
                                  { "sender", "@alice:matrix.org" },
                                  { "origin_server_ts", 1323238293289323LL },
                                  { "type", "m.room.message" } };

        MessageEvent<messages::Notice> notice;
        notice.deserialize(event);

        EXPECT_EQ(notice.content().body(), "notice message");
}

TEST(MessageEvent, Text)
{
        auto content = QJsonObject{ { "body", "text message" }, { "msgtype", "m.text" } };

        auto event = QJsonObject{ { "content", content },
                                  { "event_id", "$asdfafdf8af:matrix.org" },
                                  { "room_id", "!aasdfaeae23r9:matrix.org" },
                                  { "sender", "@alice:matrix.org" },
                                  { "origin_server_ts", 1323238293289323LL },
                                  { "type", "m.room.message" } };

        MessageEvent<messages::Text> text;
        text.deserialize(event);

        EXPECT_EQ(text.content().body(), "text message");
}

TEST(MessageEvent, Video)
{
        auto thumbnail_info = QJsonObject{
                { "h", 300 }, { "w", 400 }, { "size", 3432434 }, { "mimetype", "image/jpeg" }
        };

        auto video_info = QJsonObject{ { "h", 222 },
                                       { "w", 333 },
                                       { "duration", 232323 },
                                       { "size", 24242424 },
                                       { "mimetype", "video/mp4" },
                                       { "thumbnail_url", "mxc://localhost/adfaefaFAFSDFF3" },
                                       { "thumbnail_info", thumbnail_info } };

        auto content = QJsonObject{ { "body", "Gangnam Style" },
                                    { "url", "mxc://localhost/23d233d32r3r2r" },
                                    { "info", video_info },
                                    { "msgtype", "m.video" } };

        auto event = QJsonObject{ { "content", content },
                                  { "event_id", "$asdfafdf8af:matrix.org" },
                                  { "room_id", "!aasdfaeae23r9:matrix.org" },
                                  { "sender", "@alice:matrix.org" },
                                  { "origin_server_ts", 1323238293289323LL },
                                  { "type", "m.room.message" } };

        MessageEvent<messages::Video> video;
        video.deserialize(event);

        EXPECT_EQ(video.msgContent().info().thumbnail_info.h, 300);
        EXPECT_EQ(video.msgContent().info().thumbnail_info.w, 400);
        EXPECT_EQ(video.msgContent().info().thumbnail_info.mimetype, "image/jpeg");
        EXPECT_EQ(video.msgContent().info().duration, 232323);
        EXPECT_EQ(video.msgContent().info().size, 24242424);
        EXPECT_EQ(video.msgContent().info().mimetype, "video/mp4");
        EXPECT_EQ(video.content().body(), "Gangnam Style");
}

TEST(MessageEvent, Types)
{
        EXPECT_EQ(
          extractMessageEventType(QJsonObject{
            { "content", QJsonObject{ { "msgtype", "m.audio" } } }, { "type", "m.room.message" },
          }),
          MessageEventType::Audio);
        EXPECT_EQ(
          extractMessageEventType(QJsonObject{
            { "content", QJsonObject{ { "msgtype", "m.emote" } } }, { "type", "m.room.message" },
          }),
          MessageEventType::Emote);
        EXPECT_EQ(
          extractMessageEventType(QJsonObject{
            { "content", QJsonObject{ { "msgtype", "m.file" } } }, { "type", "m.room.message" },
          }),
          MessageEventType::File);
        EXPECT_EQ(
          extractMessageEventType(QJsonObject{
            { "content", QJsonObject{ { "msgtype", "m.image" } } }, { "type", "m.room.message" },
          }),
          MessageEventType::Image);
        EXPECT_EQ(
          extractMessageEventType(QJsonObject{
            { "content", QJsonObject{ { "msgtype", "m.location" } } }, { "type", "m.room.message" },
          }),
          MessageEventType::Location);
        EXPECT_EQ(
          extractMessageEventType(QJsonObject{
            { "content", QJsonObject{ { "msgtype", "m.notice" } } }, { "type", "m.room.message" },
          }),
          MessageEventType::Notice);
        EXPECT_EQ(
          extractMessageEventType(QJsonObject{
            { "content", QJsonObject{ { "msgtype", "m.text" } } }, { "type", "m.room.message" },
          }),
          MessageEventType::Text);
        EXPECT_EQ(
          extractMessageEventType(QJsonObject{
            { "content", QJsonObject{ { "msgtype", "m.video" } } }, { "type", "m.room.message" },
          }),
          MessageEventType::Video);
        EXPECT_EQ(
          extractMessageEventType(QJsonObject{
            { "content", QJsonObject{ { "msgtype", "m.random" } } }, { "type", "m.room.message" },
          }),
          MessageEventType::Unknown);
}