// Copyright (C) 2018 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only

#include "xdgoutputv1.h"
#include "mockcompositor.h"

#include <QtGui/QRasterWindow>
#include <QtGui/QScreen>

using namespace MockCompositor;

class XdgOutputV1Compositor : public DefaultCompositor {
public:
    explicit XdgOutputV1Compositor()
    {
        exec([this] {
            int version = 3; // version 3 of of unstable-v1
            add<XdgOutputManagerV1>(version);
        });
    }
    XdgOutputV1 *xdgOutput(int i = 0) { return get<XdgOutputManagerV1>()->getXdgOutput(output(i)); }
};

class tst_xdgoutput : public QObject, private XdgOutputV1Compositor
{
    Q_OBJECT
private slots:
    void cleanup();
    void primaryScreen();
    void overrideGeometry();
    void changeGeometry();
    void outputCreateEnterRace();
};

void tst_xdgoutput::cleanup()
{
    QCOMPOSITOR_COMPARE(getAll<Output>().size(), 1); // Only the default output should be left
    QTRY_COMPARE(QGuiApplication::screens().size(), 1);
    QTRY_VERIFY2(isClean(), qPrintable(dirtyMessage()));
}

void tst_xdgoutput::primaryScreen()
{
    // Verify that the client has bound to the global
    QCOMPOSITOR_TRY_COMPARE(get<XdgOutputManagerV1>()->resourceMap().size(), 1);
    exec([&] {
        auto *resource = xdgOutput()->resourceMap().value(client());
        QCOMPARE(resource->version(), 3);
        QCOMPARE(xdgOutput()->m_logicalGeometry.size(), QSize(1920, 1080));
    });
    auto *s = QGuiApplication::primaryScreen();
    QTRY_COMPARE(s->size(), QSize(1920, 1080));
    QTRY_COMPARE(s->geometry().topLeft(), QPoint(0, 0));
    QTRY_COMPARE(s->name(), QString("WL-1"));
}

void tst_xdgoutput::overrideGeometry()
{
    exec([&] {
        auto *output = add<Output>();
        auto *xdgOutput = get<XdgOutputManagerV1>()->getXdgOutput(output);
        xdgOutput->m_logicalGeometry = QRect(10, 20, 800, 1200);
    });

    QTRY_COMPARE(QGuiApplication::screens().size(), 2);
    auto *s = QGuiApplication::screens()[1];

    QTRY_COMPARE(s->size(), QSize(800, 1200));
    QTRY_COMPARE(s->geometry().topLeft(), QPoint(10, 20));

    exec([&] { remove(output(1)); });
}

void tst_xdgoutput::changeGeometry()
{
    auto *xdgOutput = exec([&] {
        auto *output = add<Output>();
        auto *xdgOutput = get<XdgOutputManagerV1>()->getXdgOutput(output);
        xdgOutput->m_logicalGeometry = QRect(10, 20, 800, 1200);
        return xdgOutput;
    });

    QTRY_COMPARE(QGuiApplication::screens().size(), 2);
    auto *screen = QGuiApplication::screens()[1];
    QTRY_COMPARE(screen->size(), QSize(800, 1200));

    exec([&] {
        xdgOutput->sendLogicalSize(QSize(1024, 768));
    });

    // Now we want to check that the client doesn't apply the size immediately, but waits for the
    // done event. If we TRY_COMPARE immediately, we risk that the client just hasn't handled the
    // logical_size request yet, so we add a screen and verify it on the client side just to give
    // the client a chance to mess up.
    exec([&] { add<Output>(); });
    QTRY_COMPARE(QGuiApplication::screens().size(), 3);
    exec([&] { remove(output(2)); });

    // The logical_size event should have been handled by now, but state should not have been applied yet.
    QTRY_COMPARE(screen->size(), QSize(800, 1200));

    exec([&] {
        xdgOutput->m_output->sendDone();
    });

    // Finally, the size should change
    QTRY_COMPARE(screen->size(), QSize(1024, 768));

    exec([&] { remove(output(1)); });
}

void tst_xdgoutput::outputCreateEnterRace()
{
    m_config.autoConfigure = true;
    m_config.autoEnter = false;
    QRasterWindow window;
    QSignalSpy screenChanged(&window, &QWindow::screenChanged);
    window.resize(400, 320);
    window.show();
    QCOMPOSITOR_TRY_VERIFY(xdgSurface() && xdgSurface()->m_committedConfigureSerial);
    exec([&] { xdgToplevel()->surface()->sendEnter(output(0));});

    QTRY_COMPARE(QGuiApplication::screens().size(), 1);
    QScreen *primaryScreen = QGuiApplication::screens().first();
    QCOMPARE(window.screen(), primaryScreen);

     auto *out = exec([&] {
        return add<Output>();
     });

     // In Compositor Thread
     connect(out, &Output::outputBound, this, [this](QtWaylandServer::wl_output::Resource *resource){
        auto surface = xdgToplevel()->surface();
        surface->sendLeave(output(0));
        surface->QtWaylandServer::wl_surface::send_enter(surface->resource()->handle, resource->handle);
     }, Qt::DirectConnection);

    QTRY_COMPARE(QGuiApplication::screens().size(), 2);
    QTRY_COMPARE(window.screen(), QGuiApplication::screens()[1]);

    exec([&] { remove(out); });
    m_config.autoConfigure = false;
    m_config.autoEnter = true;
}

QCOMPOSITOR_TEST_MAIN(tst_xdgoutput)
#include "tst_xdgoutput.moc"
