#include "SongChance.hpp"

using namespace OpenVic;
using namespace OpenVic::NodeTools;

SongChance::SongChance(
	size_t new_index, std::string_view new_filename, ConditionalWeight&& new_chance
) : HasIdentifier { std::to_string(new_index) }, file_name { new_filename }, chance { std::move(new_chance) } {}

bool SongChance::parse_scripts(DefinitionManager const& definition_manager) {
	return chance.parse_scripts(definition_manager);
}

bool SongChanceManager::load_songs_file(ast::NodeCPtr root) {
	bool ret = true;

	ret &= expect_dictionary_reserve_length(
		song_chances,
		[this](std::string_view key, ast::NodeCPtr value) -> bool {
			using enum scope_type_t;

			if (key != "song") {
				Logger::error("Invalid song declaration ", key);
				return false;
			}
			std::string_view name {};
			ConditionalWeight chance { COUNTRY, COUNTRY, NO_SCOPE };

			bool ret = expect_dictionary_keys(
				"name", ONE_EXACTLY, expect_string(assign_variable_callback(name)),
				"chance", ONE_EXACTLY, chance.expect_conditional_weight(ConditionalWeight::FACTOR)
			)(value);

			ret &= song_chances.add_item({ song_chances.size(), name, std::move(chance) });
			return ret;
		}
	)(root);

	if (song_chances.size() == 0) {
		Logger::error("No songs found in Songs.txt");
		return false;
	}

	return ret;
}

bool SongChanceManager::parse_scripts(DefinitionManager const& definition_manager) {
	bool ret = true;
	for (SongChance& songChance : song_chances.get_items()) {
		ret &= songChance.parse_scripts(definition_manager);
	}
	return ret;
}