#include "R2Task.h"

R2Task::R2Task(const QString &cmd, bool transient)
{
    task = r_core_task_new(
        Core()->core(),
        true,
        cmd.toLocal8Bit().constData(),
        static_cast<RCoreTaskCallback>(&R2Task::taskFinishedCallback),
        this);
    if (task) {
        task->transient = transient;
        r_core_task_incref(task);
    }
}

R2Task::~R2Task()
{
    r_core_task_decref(task);
}

void R2Task::taskFinishedCallback(void *user, char *)
{
    reinterpret_cast<R2Task *>(user)->taskFinished();
}

void R2Task::taskFinished()
{
    emit finished();
}

void R2Task::startTask()
{
    r_core_task_enqueue(&Core()->core_->tasks, task);
}

void R2Task::breakTask()
{
    if (task) {
        r_core_task_break(&Core()->core_->tasks, task->id);
    }
}

void R2Task::joinTask()
{
    if (task) {
        r_core_task_join(&Core()->core_->tasks, nullptr, task->id);
    }
}

QString R2Task::getResult()
{
    if (task == nullptr) {
        return QString("");
    }
    return QString::fromUtf8(task->res);
}

QJsonDocument R2Task::getResultJson()
{
    if (task == nullptr) {
        return QJsonDocument();
    }
    return Core()->parseJson(task->res, task->cmd);
}

const char *R2Task::getResultRaw()
{
    return task != nullptr ? task->res : nullptr;
}