#pragma once
#include "StateManager.hpp"
#include "appstates/BaseTask.hpp"
#include "sys/sys.hpp"

#include <switch.h>

/// @brief State that spawns a task and allows updates to be printed to screen.
class TaskState final : public BaseTask
{
    public:
        /// @brief Constructs a new TaskState.
        TaskState(sys::threadpool::JobFunction function, sys::Task::TaskData taskData);

        /// @brief Constructs and returns a TaskState.
        static inline std::shared_ptr<TaskState> create(sys::threadpool::JobFunction function, sys::Task::TaskData taskData)
        {
            return std::make_shared<TaskState>(function, taskData);
        }

        /// @brief Constructs, pushes, then returns a new TaskState.
        static inline std::shared_ptr<TaskState> create_and_push(sys::threadpool::JobFunction function,
                                                                 sys::Task::TaskData taskData)
        {
            auto newState = TaskState::create(function, taskData);
            StateManager::push_state(newState);
            return newState;
        }

        /// @brief Runs update routine. Waits for thread function to signal finish and deactivates.
        void update() override;

        /// @brief Run render routine. Prints m_task's status string to screen, basically.
        /// @param
        void render() override;

    private:
        /// @brief Performs some operations and marks the state for deletion.
        void deactivate_state();
};