#include "GpRules.hpp"
#include "Shared.hpp"
#include <set>

namespace L = LibCmo;
namespace C = LibCmo::CK2;
namespace O = LibCmo::CK2::ObjImpls;

namespace BMapInspector::Rule {

#pragma region GP1 Rule

	GpRule1::GpRule1() : IRule() {}

	GpRule1::~GpRule1() {}

	std::u8string_view GpRule1::GetRuleName() const {
		return u8"GP1";
	}

	void GpRule1::Check(Reporter::Reporter& reporter, Map::Level& level) const {}

#pragma endregion

#pragma region GP2 Rule

	constexpr char8_t GP2[] = u8"GP2";

	GpRule2::GpRule2() : IRule() {}

	GpRule2::~GpRule2() {}

	std::u8string_view GpRule2::GetRuleName() const {
		return GP2;
	}

	void GpRule2::Check(Reporter::Reporter& reporter, Map::Level& level) const {
		auto* ctx = level.GetCKContext();
		Shared::SectorNameBuilder builder;

		// We need collect all group names first,
		// becuase in following code we need frequent visit them
		std::set<std::u8string> group_names;
		for (auto* group : level.GetGroups()) {
			auto group_name = group->GetName();
			if (group_name != nullptr) {
				group_names.emplace(std::u8string(group_name));
			}
		}

		// Check the sector count of this game.
		L::CKDWORD sector_count;
		for (sector_count = Shared::MIN_SECTOR; sector_count <= Shared::MAX_SECTOR; ++sector_count) {
			// Build name first with special treat for sector 9
			if (sector_count != 9) {
				auto sector_name = builder.get_name(sector_count);
				if (!group_names.contains(sector_name)) {
					if (sector_count == Shared::MIN_SECTOR) {
						reporter.WriteError(GP2, u8R"(Can not find any reasonable sector group in your map.)");
						return;
					} else {
						break;
					}
				}
			} else {
				auto sector_names = builder.get_sector9_names();
				bool has_legacy_sector = group_names.contains(sector_names.legacy_name);
				bool has_intuitive_sector = group_names.contains(sector_names.intuitive_name);
				if (!has_legacy_sector && !has_intuitive_sector) {
					break;
				}
				if (has_legacy_sector && has_intuitive_sector) {
					reporter.FormatError(GP2, u8R"(Found "Sector_9" and "Sector_09" at the same map. This is not allowed.)");
					return;
				}
			}
		}
		// Now sector_count is the first sector which can not find,
		// so we need minus one on it.
		--sector_count;

		// Report sector count info.
		reporter.FormatInfo(GP2, u8"Your map has %" PRIuCKDWORD " sector(s).", sector_count);
		// Report special warning for map which only contain 1 sector.
		if (sector_count == 1) {
			reporter.WriteWarning(
			    GP2,
			    u8"Your map only have one sector. This is okey but not suggested because it will cause mosaic issue on the flames of checkpoint. Consider adding another sector to resolve this issue.");
		}
		// Report warning for sector greater than 8.
		if (sector_count > 8) {
			reporter.WriteWarning(
			    GP2,
			    u8"You are creating a map with more than 8 levels. This will cause vanilla Ballance freezed when loading it. Please make sure that all players of your map have properly set 999 sector loader up.");
		}
		// If there is sector 9, check its kind and report wanring if it is intuitive kind.
		if (sector_count > 8) {
			auto sector_names = builder.get_sector9_names();
			if (group_names.contains(sector_names.intuitive_name)) {
				reporter.WriteWarning(
				    GP2, u8R"(You are using intuitive sector name, "Sector_09", for sector 9. This is only accepted by new 999 sector loader.)");
			}
		}

		// We continue check following sectors to make sure that all sector is successive.
		for (L::CKDWORD i = sector_count + 1; i <= Shared::MAX_SECTOR; ++i) {
			if (i != 9) {
				auto sector_name = builder.get_name(i);
				if (group_names.contains(sector_name)) {
					reporter.FormatError(GP2,
					                     u8R"(Found group "%s" unexpected. Please check whether sector groups are successive in your map.)",
					                     sector_name.c_str());
				}
			} else {
				auto sector_names = builder.get_sector9_names();
				bool has_legacy_sector = group_names.contains(sector_names.legacy_name);
				bool has_intuitive_sector = group_names.contains(sector_names.intuitive_name);
				if (has_legacy_sector || has_intuitive_sector) {
					reporter.FormatError(
					    GP2,
					    u8R"(Found group "%s" or "%s" unexpected. Please check whether sector groups are successive in your map.)",
					    sector_names.legacy_name.c_str(),
					    sector_names.intuitive_name.c_str());
				}
			}
		}
	}

#pragma endregion

#pragma region GP3 Rule
	//
	//	Gp3Rule::Gp3Rule() : IRule() {}
	//
	//	Gp3Rule::~Gp3Rule() {}
	//
	//	std::u8string_view Gp3Rule::GetRuleName() const {
	//		return u8"GP3";
	//	}
	//
	//	void Gp3Rule::Check(Reporter::Reporter& reporter, Map::Level& level) const {
	//		// TODO: Mesh hash is not implemented.
	//	}
	//
	//#pragma endregion

} // namespace BMapInspector::Rule