#include "InfrastPowerTask.h"

#include "WinMacro.h"
#include "Identify.h"
#include "Configer.h"

bool asst::InfrastPowerTask::run()
{
	if (m_controller_ptr == nullptr
		|| m_identify_ptr == nullptr)
	{
		m_callback(AsstMsg::PtrIsNull, json::value(), m_callback_arg);
		return false;
	}

	bool is_the_left = false;
	for (int i = 0; i != PowerNum; ++i) {
		enter_station({ "Power", "PowerMini"}, i);
		sleep(1000);
		if (enter_operator_selection()) {
			if (is_the_left) {
				select_operators(false);
			}
			else {
				select_operators(true);
				is_the_left = true;
			}
		}
		sleep(1000);
		click_return_button();
		sleep(1000);
	}
    return true;
}

bool asst::InfrastPowerTask::enter_operator_selection()
{
	// 有这些文字之一就说明“进驻信息”这个按钮已经点开了
	static const std::vector<std::string> info_opened_flags = {
		GbkToUtf8("当前房间入住信息"),
		GbkToUtf8("进驻人数"),
		GbkToUtf8("清空")
	};

	std::vector<TextArea> ocr_result = ocr_detect();

	bool is_info_opened =
		std::find_first_of(
			ocr_result.cbegin(), ocr_result.cend(),
			info_opened_flags.cbegin(), info_opened_flags.cend(),
			[](const TextArea& lhs, const std::string& rhs)
			-> bool { return lhs.text == rhs; })
		!= ocr_result.cend();

	static const std::string station_info = GbkToUtf8("进驻信息");
	// 如果“进驻信息”窗口没点开，那就点开
	if (!is_info_opened) {
		auto station_info_iter = std::find_if(ocr_result.cbegin(), ocr_result.cend(),
			[](const TextArea& textarea) -> bool {
				return textarea.text == station_info;
			});
		m_controller_ptr->click(station_info_iter->rect);
		sleep(1000);
	}

	cv::Mat image = m_controller_ptr->get_image();
	auto goin_result = m_identify_ptr->find_image(image, "GoIn");
	// 如果找到了GoIn（“进驻”按钮），进点进去，准备选择干员
	if (goin_result.score > Configer::TemplThresholdDefault) {
		m_controller_ptr->click(goin_result.rect);
		sleep(1000);
		return true;
	}
	else {
		return false;	// 否则说明这个发电站是有人在的，不用换班
	}
}

int asst::InfrastPowerTask::select_operators(bool need_to_the_left)
{
	if (need_to_the_left) {
		swipe_to_the_left();
	}
	// 发电站干员不用做识别，直接选择第一个即可
	click_first_operator();
	click_confirm_button();

	return 1;
}