#include "InfrastOfficeTask.h"

#include "Controller.h"

bool asst::InfrastOfficeTask::_run()
{
    m_all_available_opers.clear();

    // 办公室只能造这一个
    set_product("HR");

    swipe_to_the_right_of_main_ui();
    enter_facility();
    click_bottomleft_tab();

    for (int i = 0; i <= OperSelectRetryTimes; ++i) {
        if (need_exit()) {
            return false;
        }
        if (!opers_detect_with_swipe()) {
            return false;
        }
        swipe_to_the_left_of_operlist();

        auto find_iter = std::find_if(
            m_all_available_opers.begin(), m_all_available_opers.end(),
            [&](const infrast::Oper& info) -> bool {
                return info.selected;
            });

        bool need_shift = true;
        if (find_iter != m_all_available_opers.end()) {
            switch (m_work_mode) {
            case infrast::WorkMode::Gentle:
                // 如果之前有干员在，那就不换人，直接退出
                m_all_available_opers.erase(find_iter);
                need_shift = false;
                break;
            case infrast::WorkMode::Aggressive:
                need_shift = true;
                // TODO，这里有个bug，全部干员中的selected，和当前的，不一定是同一个页面
                // 不过目前没影响，反正滑动到最前面了，selected的一定是在最前面
                //Ctrler.click(find_iter->rect);
                //sleep(300);
                break;
            case infrast::WorkMode::Extreme: // TODO
                break;
            default:
                break;
            }
        }
        if (need_shift) {
            optimal_calc();
            bool ret = opers_choose();
            if (!ret) {
                m_all_available_opers.clear();
                swipe_to_the_left_of_operlist();
                continue;
            }
        }
        break;
    }
    click_confirm_button();
    click_return_button();

    return true;
}