MaaAssistantArknights/src/MaaCore/Task/Interface/ReclamationTask.cpp

#include "ReclamationTask.h"

#include "Task/ProcessTask.h"

// 通用配置及插件
#include "Task/Reclamation/ReclamationConfig.h"

// ReclamationMode::ProsperityInSave 专用配置及插件
#include "Task/Reclamation/ReclamationCraftTaskPlugin.h"

#include "Utils/Logger.hpp"

asst::ReclamationTask::ReclamationTask(const AsstCallback& callback, Assistant* inst) :
    InterfaceTask(callback, inst, TaskType),
    m_reclamation_task_ptr(std::make_shared<ProcessTask>(callback, inst, TaskType)),
    m_config_ptr(std::make_shared<ReclamationConfig>())
{
    LogTraceFunction;

    // ReclamationMode::ProsperityInSave 专用参数
    m_reclamation_task_ptr->register_plugin<ReclamationCraftTaskPlugin>(m_config_ptr);

    m_subtasks.emplace_back(m_reclamation_task_ptr);
}

bool asst::ReclamationTask::set_params(const json::value& params)
{
    LogTraceFunction;

    if (!m_config_ptr->verify_and_load_params(params)) {
        return false;
    }

    const std::string& theme = m_config_ptr->get_theme();
    const ReclamationMode& mode = m_config_ptr->get_mode();

    if (theme == ReclamationTheme::Fire) {
        Log.info(__FUNCTION__, "Reclamation Algorithm theme", theme, "is no longer available");
        m_reclamation_task_ptr->set_tasks({ "Stop" });
        return true;
    }

    switch (mode) {
    case ReclamationMode::ProsperityNoSave:
        m_reclamation_task_ptr->set_tasks({ theme + "@RA@ProsperityNoSave" });
        if (!params.get("clear_store", false)) {
            m_reclamation_task_ptr->set_times_limit("RA@Store@EnterStore", 0);
        }
        break;
    case ReclamationMode::ProsperityInSave:
        m_reclamation_task_ptr->set_tasks({ theme + "@RA@ProsperityInSave" });
        break;
    }

    // 各生息演算插件根据 params 设置插件专用参数, 停用不应被启用的插件
    for (const TaskPluginPtr& plugin : m_reclamation_task_ptr->get_plugins()) {
        if (const auto& reclamation_task_plugin = std::dynamic_pointer_cast<AbstractReclamationTaskPlugin>(plugin);
            reclamation_task_plugin != nullptr) {
            reclamation_task_plugin->set_enable(reclamation_task_plugin->load_params(params));
        }
    }

    return true;
}