#pragma once
#include <filesystem>

#include "Task/InterfaceTask.h"

namespace asst
{
    class ProcessTask;
    class InfrastInfoTask;
    class InfrastMfgTask;
    class InfrastTradeTask;
    class InfrastPowerTask;
    class InfrastControlTask;
    class InfrastReceptionTask;
    class InfrastOfficeTask;
    class InfrastDormTask;
    class ReplenishOriginiumShardTaskPlugin;
    class InfrastProcessingTask;

    class InfrastTask final : public InterfaceTask
    {
        enum class Mode
        {
            Default = 0,
            Custom = 10000,
        };

    public:
        inline static constexpr std::string_view TaskType = "Infrast";

        InfrastTask(const AsstCallback& callback, Assistant* inst);
        virtual ~InfrastTask() override = default;

        virtual bool set_params(const json::value& params) override;

    private:
        bool parse_and_set_custom_config(const std::filesystem::path& path, int index);

        std::shared_ptr<ProcessTask> m_infrast_begin_task_ptr = nullptr;
        std::shared_ptr<InfrastInfoTask> m_info_task_ptr = nullptr;
        std::shared_ptr<InfrastMfgTask> m_mfg_task_ptr = nullptr;
        std::shared_ptr<InfrastTradeTask> m_trade_task_ptr = nullptr;
        std::shared_ptr<InfrastPowerTask> m_power_task_ptr = nullptr;
        std::shared_ptr<InfrastControlTask> m_control_task_ptr = nullptr;
        std::shared_ptr<InfrastReceptionTask> m_reception_task_ptr = nullptr;
        std::shared_ptr<InfrastOfficeTask> m_office_task_ptr = nullptr;
        std::shared_ptr<InfrastProcessingTask> m_processing_task_ptr = nullptr;
        std::shared_ptr<InfrastDormTask> m_dorm_task_ptr = nullptr;
        std::shared_ptr<ReplenishOriginiumShardTaskPlugin> m_replenish_task_ptr = nullptr;
    };
}