qtcreatordiscordrichpresenceplugin.cpp

#include "qtcreatordiscordrichpresenceplugin.h"
#include "qtcreatordiscordrichpresenceconstants.h"

#include <coreplugin/editormanager/editormanager.h>
#include <coreplugin/editormanager/ieditor.h>
#include <projectexplorer/project.h>
#include <projectexplorer/projecttree.h>

#include <QMessageBox>


QString getRandomAlphanumericString(int length) {
    const QString possibleCharacters("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789");

    QString randomString;
    for(int i = 0; i < length; i++)
    {
        int index = qrand() % possibleCharacters.length();
        QChar nextChar = possibleCharacters.at(index);
        randomString.append(nextChar);
    }
    return randomString;
}

//QAction* QtCreatorDiscordRichPresence::Internal::QtCreatorDiscordRichPresencePlugin::rpAction = QAction*();

namespace QtCreatorDiscordRichPresence {
    namespace Internal {

        QtCreatorDiscordRichPresencePlugin::QtCreatorDiscordRichPresencePlugin()
        {
            // Create your members
        }

        QtCreatorDiscordRichPresencePlugin::~QtCreatorDiscordRichPresencePlugin()
        {
            // Unregister objects from the plugin manager's object pool
            // Delete members
        }

        bool QtCreatorDiscordRichPresencePlugin::initialize(const QStringList &arguments, QString *errorString)
        {
            // Register objects in the plugin manager's object pool
            // Load settings
            // Add actions to menus
            // Connect to other plugins' signals
            // In the initialize function, a plugin can be sure that the plugins it
            // depends on have initialized their members.

            Q_UNUSED(arguments)
            Q_UNUSED(errorString)

            /*rpAction = new QAction(tr("Disable Rich Presence"), this);
            Core::Command *cmd = Core::ActionManager::registerAction(rpAction, Constants::ACTION_ID, Core::Context(Core::Constants::C_GLOBAL));
            connect(rpAction, &QAction::triggered, [=] {

            });

            Core::ActionContainer *menu = Core::ActionManager::createMenu(Constants::MENU_ID);
            menu->menu()->setTitle(tr("Discord Rich Presence"));
            menu->addAction(cmd);
            Core::ActionManager::actionContainer(Core::Constants::M_TOOLS)->addMenu(menu);*/

            //Register Discord Rich Presence
            initDiscord();
            DiscordRichPresence presence;
            memset(&presence, 0, sizeof(presence));
            presence.state = "Idle";
            presence.details = "Startup Screen";
            presence.largeImageKey = "qtcreator-logo";
            presence.largeImageText = "Working with Qt Creator";
            presence.instance = 1;
            Discord_UpdatePresence(&presence);

            //Connect to file editing signals
            connect(Core::EditorManager::instance(), &Core::EditorManager::currentEditorChanged, [=](Core::IEditor *editor) {
                //qDebug() << editor->document()->filePath().fileName();
                DiscordRichPresence presence;
                memset(&presence, 0, sizeof(presence));

                if (editor == nullptr) return;

                //Depending on file extension choose an icon
                QString mimeType = editor->document()->mimeType();
                qDebug() << mimeType;

                const char *smallImageKey = "", *smallImageText;
                if (mimeType == "text/x-c++hdr") { //C++ Header
                    smallImageKey = "file-cpphdr";
                    smallImageText = "C++ Header";
                } else if (mimeType == "text/x-c++src") { //C++ Source
                    smallImageKey = "file-cppsrc";
                    smallImageText = "C++ Source";
                } else if (mimeType == "text/x-chdr") { //C Header
                    smallImageKey = "file-chdr";
                    smallImageText = "C Header";
                } else if (mimeType == "text/x-csrc") { //C Source
                    smallImageKey = "file-csrc";
                    smallImageText = "C Source";
                } else if (mimeType == "application/vnd.qt.qmakeprofile" || mimeType == "application/vnd.qt.qmakeproincludefile") { //QMake Profile
                    smallImageKey = "file-qtprj";
                    smallImageText = "QMake Project Profile";
                } else if (mimeType == "application/x-designer") { //Designer
                    smallImageKey = "file-qtui";
                    smallImageText = "Qt User Interface File";
                } else if (mimeType == "application/vnd.qt.xml.resource") { //Qt Resource
                    smallImageKey = "file-qtres";
                    smallImageText = "Qt Resource File";
                } else if (mimeType == "application/x-desktop") { //Desktop File
                    smallImageKey = "file-desktop";
                    smallImageText = "Desktop File";
                } else if (mimeType == "application/json") { //JSON File
                    smallImageKey = "file-json";
                    smallImageText = "JSON File";
                } else if (mimeType == "text/plain") { //Text File
                    smallImageKey = "file-txt";
                    smallImageText = "Plain Text";
                } else if (mimeType == "application/xml") { //XML File
                    smallImageKey = "file-xml";
                    smallImageText = "XML";
                } else if (mimeType == "text/html") { //HTML File
                    smallImageKey = "file-html";
                    smallImageText = "HTML";
                } else if (mimeType == "text/vnd.qtcreator.git.submit") { //Git commit window

                }

                if (smallImageKey != nullptr && smallImageKey[0] == '\0') {
                    presence.largeImageKey = "qtcreator-logo";
                    presence.largeImageText = "Working with Qt Creator";
                } else {
                    presence.largeImageKey = smallImageKey;
                    presence.largeImageText = smallImageText;
                    presence.smallImageKey = "qtcreator-logo-small";
                    presence.smallImageText = "Working with Qt Creator";
                }

                ProjectExplorer::Project* current = ProjectExplorer::ProjectTree::currentProject();
                char stateString[256];
                if (current == nullptr) {
                    if (editor->isDesignModePreferred()) {
                        sprintf(stateString, "Designing some UI");
                    } else {
                        sprintf(stateString, "Editing a file");
                    }
                    oldProject = "";
                } else {
                    sprintf(stateString, "%s", current->displayName().prepend("Working on ").toUtf8().data());

                    if (oldProject != current->displayName()) {
                        oldProject = current->displayName();

                        presence.partySize = 1;
                        presence.partyMax = 100;

                        char partyId[256];
                        sprintf(partyId, "%s", getRandomAlphanumericString(128).toUtf8().data());
                        presence.partyId = partyId;

                        presence.joinSecret = "This is the Join secret";
                    }
                }
                presence.state = stateString;

                char detailsString[256];
                sprintf(detailsString, "%s", editor->document()->filePath().fileName().toUtf8().data());
                presence.details = detailsString;

                presence.instance = 1;
                Discord_UpdatePresence(&presence);
            });

            return true;
        }

        void QtCreatorDiscordRichPresencePlugin::extensionsInitialized()
        {
            // Retrieve objects from the plugin manager's object pool
            // In the extensionsInitialized function, a plugin can be sure that all
            // plugins that depend on it are completely initialized.
        }

        ExtensionSystem::IPlugin::ShutdownFlag QtCreatorDiscordRichPresencePlugin::aboutToShutdown()
        {
            // Save settings
            // Disconnect from signals that are not needed during shutdown
            // Hide UI (if you add UI that is not in the main window directly)

            Discord_Shutdown();
            return SynchronousShutdown;
        }
        void QtCreatorDiscordRichPresencePlugin::initDiscord() {
            DiscordEventHandlers handlers;
            memset(&handlers, 0, sizeof(handlers));
            handlers.ready = [](const DiscordUser* user) {
                qDebug() << "Discord Ready!";
            };
            handlers.errored = [](int errorCode, const char* message) {
                qDebug() << "Discord Error!";
            };
            handlers.disconnected = [](int errorCode, const char* message) {
                qDebug() << "Discord Disconnected!";
            };
            handlers.joinGame = [](const char* joinSecret) {
                //QString secret = QByteArray::fromBase64(QByteArray(joinSecret));
                QByteArray secretBytes(joinSecret);
                QString secret = QString::fromStdString(secretBytes.toStdString());
                QMessageBox::warning(nullptr, "Discord Join", QString("Discord Join Secret: ").append(secret), QMessageBox::Ok, QMessageBox::Ok);
                //qDebug() << secret;
            };
            Discord_Initialize("385624695229120519", &handlers, true, nullptr);
        }

    } // namespace Internal
} // namespace QtCreatorDiscordRichPresence