/*
    SPDX-FileCopyrightText: 2016 Martin Gräßlin <mgraesslin@kde.org>

    SPDX-License-Identifier: LGPL-2.1-only OR LGPL-3.0-only OR LicenseRef-KDE-Accepted-LGPL
*/
#include "testserver.h"
#include "../../server/compositor_interface.h"
#include "../../server/datadevicemanager_interface.h"
#include "../../server/display.h"
#include "../../server/fakeinput_interface.h"
#include "../../server/idle_interface.h"
#include "../../server/seat_interface.h"
#include "../../server/shell_interface.h"
#include "../../server/subcompositor_interface.h"
#include "../../server/surface_interface.h"

#include <QCoreApplication>
#include <QElapsedTimer>
#include <QProcess>
#include <QStandardPaths>
#include <QTimer>

// system
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>

using namespace KWayland::Server;

TestServer::TestServer(QObject *parent)
    : QObject(parent)
    , m_repaintTimer(new QTimer(this))
    , m_timeSinceStart(new QElapsedTimer)
    , m_cursorPos(QPointF(0, 0))
{
}

TestServer::~TestServer() = default;

void TestServer::init()
{
    Q_ASSERT(!m_display);
    m_display = new Display(this);
    m_display->start(Display::StartMode::ConnectClientsOnly);
    m_display->createShm();
    m_display->createCompositor()->create();
    m_shell = m_display->createShell(m_display);
    connect(m_shell, &ShellInterface::surfaceCreated, this, [this](ShellSurfaceInterface *surface) {
        m_shellSurfaces << surface;
        // TODO: pass keyboard/pointer/touch focus on mapped
        connect(surface, &QObject::destroyed, this, [this, surface] {
            m_shellSurfaces.removeOne(surface);
        });
    });

    m_shell->create();
    m_seat = m_display->createSeat(m_display);
    m_seat->setHasKeyboard(true);
    m_seat->setHasPointer(true);
    m_seat->setHasTouch(true);
    m_seat->create();
    m_display->createDataDeviceManager(m_display)->create();
    m_display->createIdle(m_display)->create();
    m_display->createSubCompositor(m_display)->create();
    // output
    auto output = m_display->createOutput(m_display);
    const QSize size(1280, 1024);
    output->setGlobalPosition(QPoint(0, 0));
    output->setPhysicalSize(size / 3.8);
    output->addMode(size);
    output->create();

    auto fakeInput = m_display->createFakeInput(m_display);
    fakeInput->create();
    connect(fakeInput, &FakeInputInterface::deviceCreated, this, [this](FakeInputDevice *device) {
        device->setAuthentication(true);
        connect(device, &FakeInputDevice::pointerMotionRequested, this, [this](const QSizeF &delta) {
            m_seat->setTimestamp(m_timeSinceStart->elapsed());
            m_cursorPos = m_cursorPos + QPointF(delta.width(), delta.height());
            m_seat->setPointerPos(m_cursorPos);
        });
        connect(device, &FakeInputDevice::pointerButtonPressRequested, this, [this](quint32 button) {
            m_seat->setTimestamp(m_timeSinceStart->elapsed());
            m_seat->pointerButtonPressed(button);
        });
        connect(device, &FakeInputDevice::pointerButtonReleaseRequested, this, [this](quint32 button) {
            m_seat->setTimestamp(m_timeSinceStart->elapsed());
            m_seat->pointerButtonReleased(button);
        });
        connect(device, &FakeInputDevice::pointerAxisRequested, this, [this](Qt::Orientation orientation, qreal delta) {
            m_seat->setTimestamp(m_timeSinceStart->elapsed());
            m_seat->pointerAxis(orientation, delta);
        });
        connect(device, &FakeInputDevice::touchDownRequested, this, [this](quint32 id, const QPointF &pos) {
            m_seat->setTimestamp(m_timeSinceStart->elapsed());
            m_touchIdMapper.insert(id, m_seat->touchDown(pos));
        });
        connect(device, &FakeInputDevice::touchMotionRequested, this, [this](quint32 id, const QPointF &pos) {
            m_seat->setTimestamp(m_timeSinceStart->elapsed());
            const auto it = m_touchIdMapper.constFind(id);
            if (it != m_touchIdMapper.constEnd()) {
                m_seat->touchMove(it.value(), pos);
            }
        });
        connect(device, &FakeInputDevice::touchUpRequested, this, [this](quint32 id) {
            m_seat->setTimestamp(m_timeSinceStart->elapsed());
            const auto it = m_touchIdMapper.find(id);
            if (it != m_touchIdMapper.end()) {
                m_seat->touchUp(it.value());
                m_touchIdMapper.erase(it);
            }
        });
        connect(device, &FakeInputDevice::touchCancelRequested, this, [this] {
            m_seat->setTimestamp(m_timeSinceStart->elapsed());
            m_seat->cancelTouchSequence();
        });
        connect(device, &FakeInputDevice::touchFrameRequested, this, [this] {
            m_seat->setTimestamp(m_timeSinceStart->elapsed());
            m_seat->touchFrame();
        });
    });

    m_repaintTimer->setInterval(1000 / 60);
    connect(m_repaintTimer, &QTimer::timeout, this, &TestServer::repaint);
    m_repaintTimer->start();
    m_timeSinceStart->start();
}

void TestServer::startTestApp(const QString &app, const QStringList &arguments)
{
    int sx[2];
    if (socketpair(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0, sx) < 0) {
        QCoreApplication::instance()->exit(1);
        return;
    }
    m_display->createClient(sx[0]);
    int socket = dup(sx[1]);
    if (socket == -1) {
        QCoreApplication::instance()->exit(1);
        return;
    }

    const QString exec = QStandardPaths::findExecutable(app);
    if (exec.isEmpty()) {
        qWarning() << "Couldn't find executable:" << app;
        QCoreApplication::instance()->exit(1);
        return;
    }

    QProcess *p = new QProcess(this);
    p->setProcessChannelMode(QProcess::ForwardedChannels);
    QProcessEnvironment environment = QProcessEnvironment::systemEnvironment();
    environment.insert(QStringLiteral("QT_QPA_PLATFORM"), QStringLiteral("wayland"));
    environment.insert(QStringLiteral("WAYLAND_SOCKET"), QString::fromUtf8(QByteArray::number(socket)));
    p->setProcessEnvironment(environment);
    auto finishedSignal = static_cast<void (QProcess::*)(int, QProcess::ExitStatus)>(&QProcess::finished);
    connect(p, finishedSignal, QCoreApplication::instance(), &QCoreApplication::exit);
    connect(p, &QProcess::errorOccurred, this, [] {
        QCoreApplication::instance()->exit(1);
    });
    p->start(exec, arguments);
}

void TestServer::repaint()
{
    for (auto it = m_shellSurfaces.constBegin(), end = m_shellSurfaces.constEnd(); it != end; ++it) {
        (*it)->surface()->frameRendered(m_timeSinceStart->elapsed());
    }
}
