pollymc/launcher/pages/instance/ServersPage.cpp

#include "ServersPage.h"
#include "ui_ServersPage.h"

#include <FileSystem.h>
#include <sstream>
#include <io/stream_reader.h>
#include <tag_string.h>
#include <tag_primitive.h>
#include <tag_list.h>
#include <tag_compound.h>
#include <minecraft/MinecraftInstance.h>

#include <QFileSystemWatcher>
#include <QMenu>

static const int COLUMN_COUNT = 2; // 3 , TBD: latency and other nice things.

struct Server
{
    // Types
    enum class AcceptsTextures : int
    {
        ASK = 0,
        ALWAYS = 1,
        NEVER = 2
    };

    // Methods
    Server()
    {
        m_name = QObject::tr("Minecraft Server");
    }
    Server(const QString & name, const QString & address)
    {
        m_name = name;
        m_address = address;
    }
    Server(nbt::tag_compound& server)
    {
        std::string addressStr(server["ip"]);
        m_address = QString::fromUtf8(addressStr.c_str());

        std::string nameStr(server["name"]);
        m_name = QString::fromUtf8(nameStr.c_str());

        if(server["icon"])
        {
            std::string base64str(server["icon"]);
            m_icon = QByteArray::fromBase64(base64str.c_str());
        }

        if(server.has_key("acceptTextures", nbt::tag_type::Byte))
        {
            bool value = server["acceptTextures"].as<nbt::tag_byte>().get();
            if(value)
            {
                m_acceptsTextures = AcceptsTextures::ALWAYS;
            }
            else
            {
                m_acceptsTextures = AcceptsTextures::NEVER;
            }
        }
    }

    void serialize(nbt::tag_compound& server)
    {
        server.insert("name", m_name.trimmed().toUtf8().toStdString());
        server.insert("ip", m_address.trimmed().toUtf8().toStdString());
        if(m_icon.size())
        {
            server.insert("icon", m_icon.toBase64().toStdString());
        }
        if(m_acceptsTextures != AcceptsTextures::ASK)
        {
            server.insert("acceptTextures", nbt::tag_byte(m_acceptsTextures == AcceptsTextures::ALWAYS));
        }
    }

    // Data - persistent and user changeable
    QString m_name;
    QString m_address;
    AcceptsTextures m_acceptsTextures = AcceptsTextures::ASK;

    // Data - persistent and automatically updated
    QByteArray m_icon;

    // Data - temporary
    bool m_checked = false;
    bool m_up = false;
    QString m_motd; // https://mctools.org/motd-creator
    int m_ping = 0;
    int m_currentPlayers = 0;
    int m_maxPlayers = 0;
};

static std::unique_ptr <nbt::tag_compound> parseServersDat(const QString& filename)
{
    try
    {
        QByteArray input = FS::read(filename);
        std::istringstream foo(std::string(input.constData(), input.size()));
        auto pair = nbt::io::read_compound(foo);

        if(pair.first != "")
            return nullptr;

        if(pair.second == nullptr)
            return nullptr;

        return std::move(pair.second);
    }
    catch (...)
    {
        return nullptr;
    }
}

static bool serializeServerDat(const QString& filename, nbt::tag_compound * levelInfo)
{
    try
    {
        if(!FS::ensureFilePathExists(filename))
        {
            return false;
        }
        std::ostringstream s;
        nbt::io::write_tag("", *levelInfo, s);
        QByteArray val(s.str().data(), (int) s.str().size() );
        FS::write(filename, val);
        return true;
    }
    catch (...)
    {
        return false;
    }
}

class ServersModel: public QAbstractListModel
{
    Q_OBJECT
public:
    enum Roles
    {
        ServerPtrRole = Qt::UserRole,
    };
    explicit ServersModel(const QString &path, QObject *parent = 0)
        : QAbstractListModel(parent)
    {
        m_path = path;
        m_watcher = new QFileSystemWatcher(this);
        connect(m_watcher, &QFileSystemWatcher::fileChanged, this, &ServersModel::fileChanged);
        connect(m_watcher, &QFileSystemWatcher::directoryChanged, this, &ServersModel::dirChanged);
        m_saveTimer.setSingleShot(true);
        m_saveTimer.setInterval(5000);
        connect(&m_saveTimer, &QTimer::timeout, this, &ServersModel::save_internal);
    }
    virtual ~ServersModel() {};

    void observe()
    {
        if(m_observed)
        {
            return;
        }
        m_observed = true;

        if(!m_loaded)
        {
            load();
        }

        updateFSObserver();
    }

    void unobserve()
    {
        if(!m_observed)
        {
            return;
        }
        m_observed = false;

        updateFSObserver();
    }

    void lock()
    {
        if(m_locked)
        {
            return;
        }
        saveNow();

        m_locked = true;
        updateFSObserver();
    }

    void unlock()
    {
        if(!m_locked)
        {
            return;
        }
        m_locked = false;

        updateFSObserver();
    }

    int addEmptyRow(int position)
    {
        if(m_locked)
        {
            return -1;
        }
        if(position < 0 || position >= rowCount())
        {
            position = rowCount();
        }
        beginInsertRows(QModelIndex(), position, position);
        m_servers.insert(position, Server());
        endInsertRows();
        scheduleSave();
        return position;
    }

    bool removeRow(int row)
    {
        if(m_locked)
        {
            return false;
        }
        if(row < 0 || row >= rowCount())
        {
            return false;
        }
        beginRemoveRows(QModelIndex(), row, row);
        m_servers.removeAt(row);
        endRemoveRows(); // does absolutely nothing, the selected server stays as the next line...
        scheduleSave();
        return true;
    }

    bool moveUp(int row)
    {
        if(m_locked)
        {
            return false;
        }
        if(row <= 0)
        {
            return false;
        }
        beginMoveRows(QModelIndex(), row, row, QModelIndex(), row - 1);
        m_servers.swap(row-1, row);
        endMoveRows();
        scheduleSave();
        return true;
    }

    bool moveDown(int row)
    {
        if(m_locked)
        {
            return false;
        }
        int count = rowCount();
        if(row + 1 >= count)
        {
            return false;
        }
        beginMoveRows(QModelIndex(), row, row, QModelIndex(), row + 2);
        m_servers.swap(row+1, row);
        endMoveRows();
        scheduleSave();
        return true;
    }

    QVariant headerData(int section, Qt::Orientation orientation, int role) const override
    {
        if (section < 0 || section >= COLUMN_COUNT)
            return QVariant();

        if(role == Qt::DisplayRole)
        {
            switch(section)
            {
                case 0:
                    return tr("Name");
                case 1:
                    return tr("Address");
                case 2:
                    return tr("Latency");
            }
        }

        return QAbstractListModel::headerData(section, orientation, role);
    }

    virtual QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override
    {
        if (!index.isValid())
            return QVariant();

        int row = index.row();
        int column = index.column();
        if(column < 0 || column >= COLUMN_COUNT)
            return QVariant();

        if (row < 0 || row >= m_servers.size())
            return QVariant();

        switch(column)
        {
            case 0:
                switch (role)
                {
                case Qt::DecorationRole:
                {
                    auto & bytes = m_servers[row].m_icon;
                    if(bytes.size())
                    {
                        QPixmap px;
                        if(px.loadFromData(bytes))
                            return QIcon(px);
                    }
                    return LAUNCHER->getThemedIcon("unknown_server");
                }
                case Qt::DisplayRole:
                    return m_servers[row].m_name;
                case ServerPtrRole:
                    return QVariant::fromValue<void *>((void *)&m_servers[row]);
                default:
                    return QVariant();
                }
            case 1:
                switch (role)
                {
                case Qt::DisplayRole:
                    return m_servers[row].m_address;
                default:
                    return QVariant();
                }
            case 2:
                switch (role)
                {
                case Qt::DisplayRole:
                    return m_servers[row].m_ping;
                default:
                    return QVariant();
                }
            default:
                return QVariant();
        }
    }

    virtual int rowCount(const QModelIndex &parent = QModelIndex()) const override
    {
        return m_servers.size();
    }
    int columnCount(const QModelIndex & parent) const override
    {
        return COLUMN_COUNT;
    }

    Server * at(int index)
    {
        if(index < 0 || index >= rowCount())
        {
            return nullptr;
        }
        return &m_servers[index];
    }

    void setName(int row, const QString & name)
    {
        if(m_locked)
        {
            return;
        }
        auto server = at(row);
        if(!server || server->m_name == name)
        {
            return;
        }
        server->m_name = name;
        emit dataChanged(index(row, 0), index(row, COLUMN_COUNT - 1));
        scheduleSave();
    }

    void setAddress(int row, const QString & address)
    {
        if(m_locked)
        {
            return;
        }
        auto server = at(row);
        if(!server || server->m_address == address)
        {
            return;
        }
        server->m_address = address;
        emit dataChanged(index(row, 0), index(row, COLUMN_COUNT - 1));
        scheduleSave();
    }

    void setAcceptsTextures(int row, Server::AcceptsTextures textures)
    {
        if(m_locked)
        {
            return;
        }
        auto server = at(row);
        if(!server || server->m_acceptsTextures == textures)
        {
            return;
        }
        server->m_acceptsTextures = textures;
        emit dataChanged(index(row, 0), index(row, COLUMN_COUNT - 1));
        scheduleSave();
    }

    void load()
    {
        cancelSave();
        beginResetModel();
        QList<Server> servers;
        auto serversDat = parseServersDat(serversPath());
        if(serversDat)
        {
            auto &serversList = serversDat->at("servers").as<nbt::tag_list>();
            for(auto iter = serversList.begin(); iter != serversList.end(); iter++)
            {
                auto & serverTag = (*iter).as<nbt::tag_compound>();
                Server s(serverTag);
                servers.append(s);
            }
        }
        m_servers.swap(servers);
        m_loaded = true;
        endResetModel();
    }

    void saveNow()
    {
        if(saveIsScheduled())
        {
            save_internal();
        }
    }


public slots:
    void dirChanged(const QString& path)
    {
        qDebug() << "Changed:" << path;
        load();
    }
    void fileChanged(const QString& path)
    {
        qDebug() << "Changed:" << path;
    }

private slots:
    void save_internal()
    {
        cancelSave();
        QString path = serversPath();
        qDebug() << "Server list about to be saved to" << path;

        nbt::tag_compound out;
        nbt::tag_list list;
        for(auto & server: m_servers)
        {
            nbt::tag_compound serverNbt;
            server.serialize(serverNbt);
            list.push_back(std::move(serverNbt));
        }
        out.insert("servers", nbt::value(std::move(list)));

        if(!serializeServerDat(path, &out))
        {
            qDebug() << "Failed to save server list:" << path << "Will try again.";
            scheduleSave();
        }
    }

private:
    void scheduleSave()
    {
        if(!m_loaded)
        {
            qDebug() << "Server list should never save if it didn't successfully load, path:" << m_path;
            return;
        }
        if(!m_dirty)
        {
            m_dirty = true;
            qDebug() << "Server list save is scheduled for" << m_path;
        }
        m_saveTimer.start();
    }

    void cancelSave()
    {
        m_dirty = false;
        m_saveTimer.stop();
    }

    bool saveIsScheduled() const
    {
        return m_dirty;
    }

    void updateFSObserver()
    {
        bool observingFS = m_watcher->directories().contains(m_path);
        if(m_observed && m_locked)
        {
            if(!observingFS)
            {
                qWarning() << "Will watch" << m_path;
                if(!m_watcher->addPath(m_path))
                {
                    qWarning() << "Failed to start watching" << m_path;
                }
            }
        }
        else
        {
            if(observingFS)
            {
                qWarning() << "Will stop watching" << m_path;
                if(!m_watcher->removePath(m_path))
                {
                    qWarning() << "Failed to stop watching" << m_path;
                }
            }
        }
    }

    QString serversPath()
    {
        QFileInfo foo(FS::PathCombine(m_path, "servers.dat"));
        return foo.filePath();
    }

private:
    bool m_loaded = false;
    bool m_locked = false;
    bool m_observed = false;
    bool m_dirty = false;
    QString m_path;
    QList<Server> m_servers;
    QFileSystemWatcher *m_watcher = nullptr;
    QTimer m_saveTimer;
};

ServersPage::ServersPage(InstancePtr inst, QWidget* parent)
    : QMainWindow(parent), ui(new Ui::ServersPage)
{
    ui->setupUi(this);
    m_inst = inst;
    m_model = new ServersModel(inst->gameRoot(), this);
    ui->serversView->setIconSize(QSize(64,64));
    ui->serversView->setModel(m_model);
    ui->serversView->setContextMenuPolicy(Qt::CustomContextMenu);
    connect(ui->serversView, &QTreeView::customContextMenuRequested, this, &ServersPage::ShowContextMenu);

    auto head = ui->serversView->header();
    if(head->count())
    {
        head->setSectionResizeMode(0, QHeaderView::Stretch);
        for(int i = 1; i < head->count(); i++)
        {
            head->setSectionResizeMode(i, QHeaderView::ResizeToContents);
        }
    }

    auto selectionModel = ui->serversView->selectionModel();
    connect(selectionModel, &QItemSelectionModel::currentChanged, this, &ServersPage::currentChanged);
    connect(m_inst.get(), &MinecraftInstance::runningStatusChanged, this, &ServersPage::on_RunningState_changed);
    connect(ui->nameLine, &QLineEdit::textEdited, this, &ServersPage::nameEdited);
    connect(ui->addressLine, &QLineEdit::textEdited, this, &ServersPage::addressEdited);
    connect(ui->resourceComboBox, SIGNAL(currentIndexChanged(int)), this, SLOT(resourceIndexChanged(int)));
    connect(m_model, &QAbstractItemModel::rowsRemoved, this, &ServersPage::rowsRemoved);

    m_locked = m_inst->isRunning();
    if(m_locked)
    {
        m_model->lock();
    }

    updateState();
}

ServersPage::~ServersPage()
{
    m_model->saveNow();
    delete ui;
}

void ServersPage::ShowContextMenu(const QPoint& pos)
{
    auto menu = ui->toolBar->createContextMenu(this, tr("Context menu"));
    menu->exec(ui->serversView->mapToGlobal(pos));
    delete menu;
}

QMenu * ServersPage::createPopupMenu()
{
    QMenu* filteredMenu = QMainWindow::createPopupMenu();
    filteredMenu->removeAction( ui->toolBar->toggleViewAction() );
    return filteredMenu;
}

void ServersPage::on_RunningState_changed(bool running)
{
    if(m_locked == running)
    {
        return;
    }
    m_locked = running;
    if(m_locked)
    {
        m_model->lock();
    }
    else
    {
        m_model->unlock();
    }
    updateState();
}

void ServersPage::currentChanged(const QModelIndex &current, const QModelIndex &previous)
{
    int nextServer = -1;
    if (!current.isValid())
    {
        nextServer = -1;
    }
    else
    {
        nextServer = current.row();
    }
    currentServer = nextServer;
    updateState();
}

// WARNING: this is here because currentChanged is not accurate when removing rows. the current item needs to be fixed up after removal.
void ServersPage::rowsRemoved(const QModelIndex& parent, int first, int last)
{
    if(currentServer < first)
    {
        // current was before the removal
        return;
    }
    else if(currentServer >= first && currentServer <= last)
    {
        // current got removed...
        return;
    }
    else
    {
        // current was past the removal
        int count = last - first + 1;
        currentServer -= count;
    }
}

void ServersPage::nameEdited(const QString& name)
{
    m_model->setName(currentServer, name);
}

void ServersPage::addressEdited(const QString& address)
{
    m_model->setAddress(currentServer, address);
}

void ServersPage::resourceIndexChanged(int index)
{
    auto acceptsTextures = Server::AcceptsTextures(index);
    m_model->setAcceptsTextures(currentServer, acceptsTextures);
}

void ServersPage::updateState()
{
    auto server = m_model->at(currentServer);

    bool serverEditEnabled = server && !m_locked;
    ui->addressLine->setEnabled(serverEditEnabled);
    ui->nameLine->setEnabled(serverEditEnabled);
    ui->resourceComboBox->setEnabled(serverEditEnabled);
    ui->actionMove_Down->setEnabled(serverEditEnabled);
    ui->actionMove_Up->setEnabled(serverEditEnabled);
    ui->actionRemove->setEnabled(serverEditEnabled);
    ui->actionJoin->setEnabled(serverEditEnabled);

    if(server)
    {
        ui->addressLine->setText(server->m_address);
        ui->nameLine->setText(server->m_name);
        ui->resourceComboBox->setCurrentIndex(int(server->m_acceptsTextures));
    }
    else
    {
        ui->addressLine->setText(QString());
        ui->nameLine->setText(QString());
        ui->resourceComboBox->setCurrentIndex(0);
    }

    ui->actionAdd->setDisabled(m_locked);
}

void ServersPage::openedImpl()
{
    m_model->observe();
}

void ServersPage::closedImpl()
{
    m_model->unobserve();
}

void ServersPage::on_actionAdd_triggered()
{
    int position = m_model->addEmptyRow(currentServer + 1);
    if(position < 0)
    {
        return;
    }
    // select the new row
    ui->serversView->selectionModel()->setCurrentIndex(
        m_model->index(position),
        QItemSelectionModel::SelectCurrent | QItemSelectionModel::Clear | QItemSelectionModel::Rows
    );
    currentServer = position;
}

void ServersPage::on_actionRemove_triggered()
{
    m_model->removeRow(currentServer);
}

void ServersPage::on_actionMove_Up_triggered()
{
    if(m_model->moveUp(currentServer))
    {
        currentServer --;
    }
}

void ServersPage::on_actionMove_Down_triggered()
{
    if(m_model->moveDown(currentServer))
    {
        currentServer ++;
    }
}

void ServersPage::on_actionJoin_triggered()
{
    const auto &address = m_model->at(currentServer)->m_address;
    LAUNCHER->launch(m_inst, true, nullptr, std::make_shared<MinecraftServerTarget>(MinecraftServerTarget::parse(address)));
}

#include "ServersPage.moc"