mkxp-z/src/discordstate.cpp

#include <discord_game_sdk.h>
#include <time.h>

#include "eventthread.h"
#include "discordstate.h"
#include "exception.h"
#include "debugwriter.h"
#include "bitmap.h"
#include "config.h"

void defaultActivityCb(void *callback_data, enum EDiscordResult result)
{
    
}

struct Application {
    struct IDiscordCore* core;
    struct IDiscordImageManager* images;
    struct IDiscordUserManager* users;
    struct IDiscordActivityManager* activities;
    DiscordUserId user_id;
};

struct DiscordStatePrivate
{
    RGSSThreadData *threadData;
    
    DiscordClientId clientId;

    IDiscordCore *core;
    
    Application app;
    IDiscordActivityEvents activityEvents;
    IDiscordUserEvents userEvents;
    
    DiscordUser currentUser;
    DiscordActivity defaultActivity;
    
    bool discordInstalled;
    bool connected;
    bool userPresent;
    
    DiscordStatePrivate()
        : discordInstalled(false),
          connected(false),
          userPresent(false)
    {};
    
    ~DiscordStatePrivate()
    {
        if (core) core->destroy(core);
    }
};

void onCurrentUserUpdateCb(void *event_data)
{
    DiscordStatePrivate *p = (DiscordStatePrivate*)event_data;
    p->app.users->get_current_user(p->app.users, &p->currentUser);
    p->userPresent = true;
}

void onActivityJoinCb(void *event_data, const char *secret)
{
    
}

void onActivitySpectateCb(void *event_data, const char *secret)
{
    
}

void onActivityJoinRequestCb(void *event_data, struct DiscordUser *user)
{
    
}

void onActivityInviteRequestCb(void *event_data, enum EDiscordActivityActionType type, struct DiscordUser *user, struct DiscordActivity *activity)
{
    
}

void discordLogHook(void *hook_data, enum EDiscordLogLevel level, const char *message)
{
    Debug() << "DISCORD:" << message;
}

int discordTryConnect(DiscordStatePrivate *p)
{
    DiscordCreateParams params{};
    DiscordCreateParamsSetDefault(&params);
    
    params.client_id = p->clientId;
    params.flags = DiscordCreateFlags_NoRequireDiscord;
    params.event_data = (void*)p;
    
    p->activityEvents.on_activity_join = onActivityJoinCb;
    p->activityEvents.on_activity_spectate = onActivitySpectateCb;
    p->activityEvents.on_activity_join_request = onActivityJoinRequestCb;
    p->activityEvents.on_activity_invite = onActivityInviteRequestCb;
    params.activity_events = &p->activityEvents;
    
    
    p->userEvents.on_current_user_update = onCurrentUserUpdateCb;
    params.user_events = &p->userEvents;
    
    int rc = DiscordCreate(DISCORD_VERSION, &params, &p->core);
    
    if (rc != DiscordResult_NotInstalled)
        p->discordInstalled = true;

    if (rc != DiscordResult_Ok)
        return rc;
    
    p->core->set_log_hook(p->core, DiscordLogLevel_Error, (void*)p, discordLogHook);
    
    p->app.activities = p->core->get_activity_manager(p->core);
    p->app.users = p->core->get_user_manager(p->core);
    p->app.images = p->core->get_image_manager(p->core);
    
    p->connected = true;
    
    strncpy((char*)&p->defaultActivity.details, p->threadData->config.game.title.c_str(), 128);
    p->defaultActivity.timestamps.start = time(0);
    
    if (p->clientId == DEFAULT_CLIENT_ID)
    {
        strncpy((char*)&p->defaultActivity.assets.large_image, "default", 128);
        strncpy((char*)&p->defaultActivity.assets.large_text, "mkxp-z", 128);
    }
    
    p->app.activities->update_activity(p->app.activities, &p->defaultActivity, 0, defaultActivityCb);
    
    return rc;
}

DiscordState::DiscordState(RGSSThreadData *rtData)
{
    p = new DiscordStatePrivate();
    p->threadData = rtData;
    memset(&p->app, 0, sizeof(Application));
    p->clientId = rtData->config.discordClientId;
    discordTryConnect(p);

}

DiscordState::~DiscordState()
{
    delete p;
}

IDiscordActivityManager *DiscordState::activityManager()
{
    return p->app.activities;
}

IDiscordUserManager *DiscordState::userManager()
{
    return p->app.users;
}

IDiscordImageManager *DiscordState::imageManager()
{
    return p->app.images;
}

int DiscordState::update()
{
    if (!p->discordInstalled) return DiscordResult_NotInstalled;
    
    if (p->connected)
    {
        int rc = p->core->run_callbacks(p->core);
        if (rc == DiscordResult_NotRunning)
        {
            p->connected = false;
            memset(&p->currentUser, 0, sizeof(DiscordUser));
            memset(&p->app, 0, sizeof(Application));
            p->userPresent = false;
            return rc;
        }
        return rc;
    }
    
    return discordTryConnect(p);
}

bool DiscordState::isConnected()
{
    return p->connected;
}

std::string DiscordState::userName()
{
    std::string ret; ret.clear();
    if (p->userPresent) ret = p->currentUser.username;
    return ret;
}

std::string DiscordState::userDiscrim()
{
    std::string ret; ret.clear();
    if (p->userPresent) ret = p->currentUser.discriminator;
    return ret;
}

DiscordUserId DiscordState::userId()
{
    return (p->userPresent) ? p->currentUser.id : 0;
}

typedef struct { DiscordStatePrivate *pri; Bitmap *bmp; } AvatarCbData;
Bitmap *DiscordState::getAvatar(DiscordUserId userId, int size)
{
    if (!isConnected()) return 0;
    AvatarCbData cbData{};
    cbData.bmp = new Bitmap(size, size);
    cbData.pri = p;
    DiscordImageHandle handle{};
    handle.id = userId;
    handle.size = size;
    
    p->app.images->fetch(p->app.images, handle, true, &cbData,
                         [](void *callback_data, enum EDiscordResult result, struct DiscordImageHandle handle_result){
                             if (result == DiscordResult_Ok)
                             {
                                 AvatarCbData *data = (AvatarCbData*)callback_data;
                                 int sz = data->bmp->width()*data->bmp->height()*4;
                                 uint8_t *buf = new uint8_t[sz];
                                 data->pri->app.images->get_data(data->pri->app.images, handle_result, buf, sz);
                                 data->bmp->replaceRaw(buf, data->bmp->width(), data->bmp->height());
                                 delete[] buf;
                             }
                         });
    return cbData.bmp;
}

Bitmap *DiscordState::userAvatar()
{
    return (p->userPresent) ? getAvatar(p->currentUser.id, 256) : 0;
}