#include "Good.hpp"

#include <cassert>

using namespace OpenVic;
using namespace OpenVic::NodeTools;

GoodCategory::GoodCategory(const std::string_view new_identifier) : HasIdentifier { new_identifier } {}

Good::Good(const std::string_view new_identifier, colour_t new_colour, GoodCategory const& new_category, price_t new_base_price,
	bool new_available_from_start, bool new_tradeable, bool new_money, bool new_overseas_penalty)
	: HasIdentifierAndColour { new_identifier, new_colour, true },
	  category { new_category },
	  base_price { new_base_price },
	  available_from_start { new_available_from_start },
	  tradeable { new_tradeable },
	  money { new_money },
	  overseas_penalty { new_overseas_penalty } {
	assert(base_price > NULL_PRICE);
}

GoodCategory const& Good::get_category() const {
	return category;
}

Good::price_t Good::get_base_price() const {
	return base_price;
}

Good::price_t Good::get_price() const {
	return price;
}

bool Good::get_available_from_start() const {
	return available_from_start;
}

bool Good::get_available() const {
	return available;
}

bool Good::get_tradeable() const {
	return tradeable;
}

bool Good::get_money() const {
	return money;
}

bool Good::get_overseas_penalty() {
	return overseas_penalty;
}

void Good::reset_to_defaults() {
	available = available_from_start;
	price = base_price;
}

GoodManager::GoodManager() : good_categories { "good categories" }, goods { "goods" } {}

return_t GoodManager::add_good_category(const std::string_view identifier) {
	if (identifier.empty()) {
		Logger::error("Invalid good category identifier - empty!");
		return FAILURE;
	}
	return good_categories.add_item({ identifier });
}

void GoodManager::lock_good_categories() {
	good_categories.lock();
}

GoodCategory const* GoodManager::get_good_category_by_identifier(const std::string_view identifier) const {
	return good_categories.get_item_by_identifier(identifier);
}

size_t GoodManager::get_good_category_count() const {
	return good_categories.size();
}

std::vector<GoodCategory> const& GoodManager::get_good_categories() const {
	return good_categories.get_items();
}

return_t GoodManager::add_good(const std::string_view identifier, colour_t colour, GoodCategory const* category,
	Good::price_t base_price, bool available_from_start, bool tradeable, bool money, bool overseas_penalty) {
	if (identifier.empty()) {
		Logger::error("Invalid good identifier - empty!");
		return FAILURE;
	}
	if (colour > MAX_COLOUR_RGB) {
		Logger::error("Invalid good colour for ", identifier, ": ", colour_to_hex_string(colour));
		return FAILURE;
	}
	if (category == nullptr) {
		Logger::error("Invalid good category for ", identifier, ": null");
		return FAILURE;
	}
	if (base_price <= Good::NULL_PRICE) {
		Logger::error("Invalid base price for ", identifier, ": ", base_price);
		return FAILURE;
	}
	return goods.add_item({ identifier, colour, *category, base_price, available_from_start, tradeable, money, overseas_penalty });
}

void GoodManager::lock_goods() {
	goods.lock();
}

Good const* GoodManager::get_good_by_index(size_t index) const {
	return goods.get_item_by_index(index);
}

Good const* GoodManager::get_good_by_identifier(const std::string_view identifier) const {
	return goods.get_item_by_identifier(identifier);
}

size_t GoodManager::get_good_count() const {
	return goods.size();
}

std::vector<Good> const& GoodManager::get_goods() const {
	return goods.get_items();
}

void GoodManager::reset_to_defaults() {
	for (Good& good : goods.get_items())
		good.reset_to_defaults();
}

return_t GoodManager::load_good_file(ast::NodeCPtr root) {
	size_t total_expected_goods = 0;
	return_t ret = expect_dictionary_reserve_length(
		good_categories,
		[this, &total_expected_goods](std::string_view key, ast::NodeCPtr value) -> return_t {
			return_t ret = expect_list_and_length(
				[&total_expected_goods](size_t size) -> size_t {
					total_expected_goods += size;
					return 0;
				},
				success_callback
			)(value);
			if (add_good_category(key) != SUCCESS) ret = FAILURE;
			return ret;
		}
	)(root);
	lock_good_categories();
	goods.reserve(goods.size() + total_expected_goods);
	if (expect_dictionary(
		[this](std::string_view good_category_key, ast::NodeCPtr good_category_value) -> return_t {
			GoodCategory const* good_category = get_good_category_by_identifier(good_category_key);

			return expect_dictionary(
				[this, good_category](std::string_view key, ast::NodeCPtr value) -> return_t {
					colour_t colour = NULL_COLOUR;
					Good::price_t base_price;
					bool available_from_start, tradeable = true;
					bool money, overseas_penalty = false;

					return_t ret = expect_dictionary_keys(
						"color", ONE_EXACTLY, expect_colour(assign_variable_callback(colour)),
						"cost", ONE_EXACTLY, expect_fixed_point(assign_variable_callback(base_price)),
						"available_from_start", ZERO_OR_ONE, expect_bool(assign_variable_callback(available_from_start)),
						"tradeable", ZERO_OR_ONE, expect_bool(assign_variable_callback(tradeable)),
						"money", ZERO_OR_ONE, expect_bool(assign_variable_callback(money)),
						"overseas_penalty", ZERO_OR_ONE, expect_bool(assign_variable_callback(overseas_penalty))
					)(value);
					if (add_good(key, colour, good_category, base_price, available_from_start, tradeable, money, overseas_penalty) != SUCCESS)
						ret = FAILURE;
					return ret;
				}
			)(good_category_value);
		}
	)(root) != SUCCESS)
		ret = FAILURE;
	lock_goods();
	return ret;
}