/**
 * Copyright (C) 2021 Saturneric
 *
 * This file is part of GpgFrontend.
 *
 * GpgFrontend is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * GpgFrontend is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with GpgFrontend. If not, see <https://www.gnu.org/licenses/>.
 *
 * The initial version of the source code is inherited from
 * the gpg4usb project, which is under GPL-3.0-or-later.
 *
 * The source code version of this software was modified and released
 * by Saturneric<eric@bktus.com><eric@bktus.com> starting on May 12, 2021.
 *
 */

#include "core/thread/TaskRunner.h"

#include "core/thread/Task.h"
#include "spdlog/spdlog.h"

GpgFrontend::Thread::TaskRunner::TaskRunner() = default;

GpgFrontend::Thread::TaskRunner::~TaskRunner() = default;

void GpgFrontend::Thread::TaskRunner::PostTask(Task* task) {
  if (task == nullptr) return;

  std::string uuid = task->GetUUID();
  SPDLOG_TRACE("post task: {}", uuid);

  task->setParent(nullptr);
  task->moveToThread(this);

  connect(task, &Task::SignalTaskPostFinishedDone, this, [&, uuid]() {
    auto it = pending_tasks_.find(uuid);
    if (it == pending_tasks_.end()) {
      return;
    } else {
      it->second->deleteLater();
      pending_tasks_.erase(it);
    }
  });
  {
    std::lock_guard<std::mutex> lock(tasks_mutex_);
    tasks.push(task);
  }
  quit();
}

void GpgFrontend::Thread::TaskRunner::PostScheduleTask(Task* task,
                                                       size_t seconds) {
  if (task == nullptr) return;
  // TODO
}

[[noreturn]] void GpgFrontend::Thread::TaskRunner::run() {
  SPDLOG_TRACE("called, thread id: {}", QThread::currentThreadId());
  while (true) {
    SPDLOG_TRACE("task runner: a new cycle start");
    if (tasks.empty()) {
      SPDLOG_TRACE("task runner: no tasks to run, trapping into event loop...");
      exec();
    } else {
      SPDLOG_TRACE("task runner: task queue size:", tasks.size());

      Task* task = nullptr;
      {
        std::lock_guard<std::mutex> lock(tasks_mutex_);
        task = std::move(tasks.front());
        tasks.pop();
        pending_tasks_.insert({task->GetUUID(), task});
      }

      if (task != nullptr) {
        // Run the task
        SPDLOG_TRACE("task runner: running task {}", task->GetUUID());
        try {
          task->run();
        } catch (const std::exception& e) {
          SPDLOG_ERROR("task runner: exception in task {}, exception: {}",
                       task->GetUUID(), e.what());

          // destroy the task, remove the task from the pending tasks
          task->deleteLater();
          pending_tasks_.erase(task->GetUUID());
        } catch (...) {
          SPDLOG_ERROR("task runner: unknown exception in task: {}",
                       task->GetUUID());

          // destroy the task, remove the task from the pending tasks
          task->deleteLater();
          pending_tasks_.erase(task->GetUUID());
        }
      }
    }
  }
}