// SPDX-FileCopyrightText: 2026 Contributors to Chatterino <https://chatterino.com>
//
// SPDX-License-Identifier: MIT

#include "widgets/splits/InputHighlighter.hpp"

#include "controllers/accounts/AccountController.hpp"
#include "controllers/commands/Command.hpp"
#include "controllers/commands/CommandController.hpp"
#include "controllers/highlights/HighlightController.hpp"
#include "controllers/ignores/IgnorePhrase.hpp"
#include "controllers/spellcheck/SpellChecker.hpp"
#include "messages/Emote.hpp"
#include "mocks/BaseApplication.hpp"
#include "mocks/EmoteController.hpp"
#include "mocks/TwitchIrcServer.hpp"
#include "providers/twitch/IrcMessageHandler.hpp"
#include "providers/twitch/TwitchAccount.hpp"
#include "providers/twitch/TwitchBadges.hpp"
#include "providers/twitch/TwitchChannel.hpp"
#include "Test.hpp"

#include <QDebug>
#include <QString>
#include <QStringBuilder>

using namespace chatterino;
using namespace Qt::Literals;

namespace {

class MockApplication : public mock::BaseApplication
{
public:
    MockApplication()
        : commandController(this->paths_)
    {
    }

    EmoteController *getEmotes() override
    {
        return &this->emotes;
    }

    AccountController *getAccounts() override
    {
        return &this->accounts;
    }

    ITwitchIrcServer *getTwitch() override
    {
        return &this->twitch;
    }

    BttvEmotes *getBttvEmotes() override
    {
        return &this->bttvEmotes;
    }

    FfzEmotes *getFfzEmotes() override
    {
        return &this->ffzEmotes;
    }

    SeventvEmotes *getSeventvEmotes() override
    {
        return &this->seventvEmotes;
    }

    CommandController *getCommands() override
    {
        return &this->commandController;
    }

    AccountController accounts;
    mock::EmoteController emotes;
    mock::MockTwitchIrcServer twitch;
    BttvEmotes bttvEmotes;
    FfzEmotes ffzEmotes;
    SeventvEmotes seventvEmotes;
    CommandController commandController;
};

std::pair<const EmoteName, EmotePtr> makeEmote(const QString &name)
{
    EmoteName emoteName{name};
    auto ptr = std::make_shared<Emote>(Emote{.name = emoteName});
    return {emoteName, ptr};
}

using EmoteMapPtr = std::shared_ptr<const EmoteMap>;

EmoteMapPtr makeEmotes(auto &&...emotes)
{
    auto map = std::make_shared<EmoteMap>();
    ((map->emplace(makeEmote(std::forward<decltype(emotes)>(emotes)))), ...);
    return map;
}

struct MockEmotes {
    EmoteMapPtr seventv;
    EmoteMapPtr bttv;
    EmoteMapPtr ffz;
    EmoteMapPtr twitchAccount;

    static MockEmotes channel()
    {
        return {
            .seventv = makeEmotes(u"7TVEmote"_s, u"PogChamp"_s),
            .bttv = makeEmotes(u"BTTVEmote"_s, u"Kappa"_s),
            .ffz = makeEmotes(u"FFZEmote"_s, u"Keepo"_s),
        };
    }

    static MockEmotes global()
    {
        return {
            .seventv = makeEmotes(u"7TVGlobal"_s),
            .bttv = makeEmotes(u"BTTVGlobal"_s),
            .ffz = makeEmotes(u"FFZGlobal"_s),
            .twitchAccount = makeEmotes(u"MyCoolTwitchEmote"_s),
        };
    }
};

std::shared_ptr<TwitchChannel> makeMockTwitchChannel(const QString &name)
{
    auto chan = std::make_shared<TwitchChannel>(name);
    auto mocks = MockEmotes::channel();
    chan->setSeventvEmotes(std::move(mocks.seventv));
    chan->setBttvEmotes(std::move(mocks.bttv));
    chan->setFfzEmotes(std::move(mocks.ffz));

    chan->addRecentChatter("UserChatter");
    chan->addRecentChatter("UserColor");
    chan->addRecentChatter("MyUser42");
    chan->addRecentChatter("123kappa123");

    return chan;
}

}  // namespace

class InputHighlighterTest : public ::testing::Test
{
public:
    void SetUp() override
    {
        this->mockApplication = std::make_unique<MockApplication>();
        auto mocks = MockEmotes::global();
        this->mockApplication->seventvEmotes.setGlobalEmotes(mocks.seventv);
        this->mockApplication->bttvEmotes.setEmotes(mocks.bttv);
        this->mockApplication->ffzEmotes.setEmotes(mocks.ffz);
        this->mockApplication->getAccounts()->twitch.getCurrent()->setEmotes(
            mocks.twitchAccount);
        this->channel = makeMockTwitchChannel(u"twitchdev"_s);
        this->mockApplication->commandController.items.append(
            Command("/my-command", ""));
        this->mockApplication->commandController.items.append(
            Command("command space", ""));
        this->mockApplication->commandController.items.append(
            Command("command with more than one space", ""));
    }

    void TearDown() override
    {
        this->channel.reset();
        this->mockApplication.reset();
    }

    std::shared_ptr<TwitchChannel> channel;
    std::unique_ptr<MockApplication> mockApplication;
};

TEST_F(InputHighlighterTest, getSpellCheckedWords)
{
    struct Case {
        QString input;
        std::vector<QString> words;
    };

    std::vector<Case> cases{
        {.input = "", .words = {}},
        {.input = "word", .words = {"word"}},
        {.input = "word word", .words = {"word", "word"}},
        {.input = "   word word  ", .words = {"word", "word"}},
        {.input = "word?", .words = {"word"}},
        {.input = "word?word", .words = {"word", "word"}},
        {.input = "word-word", .words = {"word-word"}},
        {
            .input = "channel emotes 7TVEmote a BTTVEmote b FFZEmote c",
            .words = {"channel", "emotes", "a", "b", "c"},
        },
        {
            .input = "global emotes 7TVGlobal a BTTVGlobal b FFZGlobal c "
                     "MyCoolTwitchEmote d",
            .words = {"global", "emotes", "a", "b", "c", "d"},
        },
        {
            .input = "/ban a user",
            .words = {"a", "user"},
        },
        {
            .input = "/my-command some text",
            .words = {"some", "text"},
        },
        {
            .input = "command space and some text",
            .words = {"and", "some", "text"},
        },
        {
            .input = "command with more than one space and some text",
            .words = {"and", "some", "text"},
        },
        {
            .input = "command with more than one and some text",
            .words =
                {
                    "command",
                    "with",
                    "more",
                    "than",
                    "one",
                    "and",
                    "some",
                    "text",
                },
        },
        {
            .input = "Hey, @userchatter, a 123kappa123 b MyUser42 c",
            .words = {"Hey", "a", "b", "c"},
        },
        {
            .input =
                "twitch.tv ignore "
                "https://wiki.chatterino.com/Help/#basic-troubleshooting links",
            .words = {"ignore", "links"},
        },
    };

    SpellChecker nullSpellChecker;
    InputHighlighter highlighter(nullSpellChecker, nullptr);
    highlighter.setChannel(this->channel);
    for (size_t i = 0; i < cases.size(); i++)
    {
        const auto &c = cases[i];
        auto got = highlighter.getSpellCheckedWords(c.input);
        ASSERT_EQ(got, c.words) << "index=" << i;
    }
}

TEST(InputHighlight, wordRegex)
{
    struct Case {
        QStringView input;
        std::vector<QStringView> words;
    };

    std::vector<Case> cases{
        {.input = u"", .words = {}},
        {.input = u"word", .words = {u"word"}},
        {.input = u"test/man", .words = {u"test", u"man"}},
        {.input = u"word?", .words = {u"word"}},
        {.input = u"word?word", .words = {u"word", u"word"}},
        {.input = u"sentence.", .words = {u"sentence"}},
        {.input = u"#hashtag", .words = {u"hashtag"}},
        {.input = u"inogre123numbers", .words = {}},
        {.input = u"under_score", .words = {}},
        {.input = u"äwördü", .words = {u"äwördü"}},
        {
            .input = u"abc!@foo#bar&(baz]",
            .words = {u"abc", u"foo", u"bar", u"baz"},
        },
        {.input = u"1234567,word/a123", .words = {u"word"}},
        {.input = u"'quotes\"", .words = {u"quotes"}},
        {.input = u"doesn't", .words = {u"doesn't"}},
        {.input = u"wouldn't've", .words = {u"wouldn't've"}},
        {
            .input = u"abc'd'efg'hij-klm'n-o-p",
            .words = {u"abc'd'efg'hij-klm'n-o-p"},
        },
        {.input = u"'abc'", .words = {u"abc"}},
        {.input = u"'abc'def", .words = {u"abc'def"}},
        {.input = u"'abc'def'", .words = {u"abc'def"}},
        {.input = u"follow-up", .words = {u"follow-up"}},
        {.input = u"abc-def", .words = {u"abc-def"}},
        {.input = u"-abc-def-", .words = {u"abc-def"}},
        {.input = u"abc--def", .words = {u"abc", u"def"}},
        {.input = u"'abc--def'", .words = {u"abc", u"def"}},
        {.input = u"abc_def", .words = {}},
        {.input = u"abc_", .words = {}},
        {.input = u"_def", .words = {}},
        {.input = u"abc'def_ghi'jkl", .words = {u"abc", u"jkl"}},
    };

    auto re = inputhighlight::detail::wordRegex();
    for (size_t i = 0; i < cases.size(); i++)
    {
        const auto &c = cases[i];
        std::vector<QStringView> got;
        auto match = re.globalMatchView(c.input);
        while (match.hasNext())
        {
            got.emplace_back(match.next().capturedView());
        }
        ASSERT_EQ(got, c.words) << "index=" << i;
    }
}

TEST_F(InputHighlighterTest, getWordAt)
{
    struct Case {
        std::pair<QStringView, qsizetype> input;
        QStringView word;
    };

    std::vector<Case> cases{
        {.input = {u"", 0}, .word = {}},
        {.input = {u"", -1}, .word = {}},
        {.input = {u"", -2}, .word = {}},
        {.input = {u"", 1}, .word = {}},
        {.input = {u"", 2}, .word = {}},
        {.input = {u"word", -1}, .word = {}},
        {.input = {u"word", 0}, .word = u"word"},
        {.input = {u"word", 1}, .word = u"word"},
        {.input = {u"word", 2}, .word = u"word"},
        {.input = {u"word", 3}, .word = u"word"},
        {.input = {u"word", 4}, .word = u"word"},
        {.input = {u"word", 5}, .word = {}},
        {.input = {u"a word", -1}, .word = {}},
        {.input = {u"a word", 0}, .word = u"a"},
        {.input = {u"a word", 1}, .word = u"a"},
        {.input = {u"a word", 2}, .word = u"word"},
        {.input = {u"a word", 3}, .word = u"word"},
        {.input = {u"a word", 4}, .word = u"word"},
        {.input = {u"a word", 5}, .word = u"word"},
        {.input = {u"a word", 6}, .word = u"word"},
        {.input = {u"a word", 7}, .word = {}},
        {.input = {u"a word!", 6}, .word = u"word"},
        {.input = {u"a word!", 7}, .word = {}},
        {.input = {u"a word!", 8}, .word = {}},
        {.input = {u"three words are", 0}, .word = u"three"},
        {.input = {u"three words are", 4}, .word = u"three"},
        {.input = {u"three words are", 5}, .word = u"three"},
        {.input = {u"three words are", 6}, .word = u"words"},
        {.input = {u"three words are", 10}, .word = u"words"},
        {.input = {u"three words are", 11}, .word = u"words"},
        {.input = {u"three words are", 12}, .word = u"are"},
        {.input = {u"three words are", 14}, .word = u"are"},
        {.input = {u"three words are", 15}, .word = u"are"},
        {.input = {u"three words are", 16}, .word = {}},
        {.input = {u"a#text#is#here!", 0}, .word = u"a"},
        {.input = {u"a#text#is#here!", 1}, .word = u"a"},
        {.input = {u"a#text#is#here!", 2}, .word = u"text"},
        {.input = {u"a#text#is#here!", 5}, .word = u"text"},
        {.input = {u"a#text#is#here!", 6}, .word = u"text"},
        {.input = {u"a#text#is#here!", 7}, .word = u"is"},
        {.input = {u"a#text#is#here!", 8}, .word = u"is"},
        {.input = {u"a#text#is#here!", 9}, .word = u"is"},
        {.input = {u"a#text#is#here!", 10}, .word = u"here"},
        {.input = {u"a#text#is#here!", 13}, .word = u"here"},
        {.input = {u"a#text#is#here!", 14}, .word = u"here"},
        {.input = {u"a#text#is#here!", 15}, .word = {}},
        {.input = {u"here! there!", 3}, .word = u"here"},
        {.input = {u"here! there!", 4}, .word = u"here"},
        {.input = {u"here! there!", 5}, .word = {}},
        {.input = {u"here! there!", 6}, .word = u"there"},
        {.input = {u"/my-command wow", 0}, .word = {}},
        {.input = {u"/my-command wow", 1}, .word = {}},
        {.input = {u"/my-command wow", 2}, .word = {}},
        {.input = {u"/my-command wow", 10}, .word = {}},
        {.input = {u"/my-command wow", 11}, .word = {}},
        {.input = {u"/my-command wow", 12}, .word = u"wow"},
        {.input = {u"a 7TVGlobal b", 0}, .word = u"a"},
        {.input = {u"a 7TVGlobal b", 1}, .word = u"a"},
        {.input = {u"a 7TVGlobal b", 2}, .word = {}},
        {.input = {u"a 7TVGlobal b", 10}, .word = {}},
        {.input = {u"a 7TVGlobal b", 11}, .word = {}},
        {.input = {u"a 7TVGlobal b", 12}, .word = u"b"},
        {.input = {u"a 7TVGlobal b", 13}, .word = u"b"},
        {.input = {u"a 7TVGlobal b", 14}, .word = {}},
    };

    SpellChecker nullSpellChecker;
    InputHighlighter highlighter(nullSpellChecker, nullptr);
    highlighter.setChannel(this->channel);
    for (size_t i = 0; i < cases.size(); i++)
    {
        const auto &c = cases[i];
        auto got = highlighter.getWordAt(c.input.first, c.input.second);
        ASSERT_EQ(got, c.word) << "index=" << i;
    }
}
