#pragma once

#include "openvic-simulation/economy/production/Employee.hpp"
#include "openvic-simulation/economy/production/ProductionType.hpp"
#include "openvic-simulation/economy/trading/MarketInstance.hpp"
#include "openvic-simulation/modifier/ModifierEffectCache.hpp"
#include "openvic-simulation/pop/Pop.hpp"
#include "openvic-simulation/types/fixed_point/FixedPoint.hpp"
#include "openvic-simulation/utility/Getters.hpp"

namespace OpenVic {
	struct ProvinceInstance;

	struct ResourceGatheringOperation {
	private:
		MarketInstance& market_instance;
		ModifierEffectCache const& modifier_effect_cache;
		ProvinceInstance* location_ptr;
		ProductionType const* PROPERTY_RW(production_type_nullable);
		fixed_point_t PROPERTY(revenue_yesterday);
		fixed_point_t PROPERTY(output_quantity_yesterday);
		fixed_point_t PROPERTY(unsold_quantity_yesterday);
		fixed_point_t PROPERTY_RW(size_multiplier);
		std::vector<Employee> PROPERTY(employees);
		Pop::pop_size_t PROPERTY(max_employee_count_cache);
		Pop::pop_size_t PROPERTY(total_employees_count_cache);
		Pop::pop_size_t PROPERTY(total_paid_employees_count_cache);
		fixed_point_t PROPERTY(total_owner_income_cache);
		fixed_point_t PROPERTY(total_employee_income_cache);
		IndexedMap<PopType, Pop::pop_size_t> PROPERTY(employee_count_per_type_cache);

		fixed_point_t calculate_size_modifier() const;
		void hire(const Pop::pop_size_t available_worker_count);
		fixed_point_t produce(
			std::vector<Pop*> const* const owner_pops_cache,
			const Pop::pop_size_t total_owner_count_in_state_cache
		);
		void pay_employees(
			const fixed_point_t revenue,
			const Pop::pop_size_t total_worker_count_in_province,
			std::vector<Pop*> const* const owner_pops_cache,
			const Pop::pop_size_t total_owner_count_in_state_cache
		);

	public:
		ResourceGatheringOperation(
			MarketInstance& new_market_instance,
			ModifierEffectCache const& new_modifier_effect_cache,
			ProductionType const* new_production_type_nullable,
			fixed_point_t new_size_multiplier,
			fixed_point_t new_revenue_yesterday,
			fixed_point_t new_output_quantity_yesterday,
			fixed_point_t new_unsold_quantity_yesterday,
			std::vector<Employee>&& new_employees,
			decltype(employee_count_per_type_cache)::keys_t const& pop_type_keys
		);

		ResourceGatheringOperation(
			MarketInstance& new_market_instance,
			ModifierEffectCache const& new_modifier_effect_cache,
			decltype(employee_count_per_type_cache)::keys_t const& pop_type_keys
		);

		constexpr bool is_valid() const {
			return production_type_nullable != nullptr;
		}
		void setup_location_ptr(ProvinceInstance& location);
		void initialise_rgo_size_multiplier();
		void rgo_tick();
	};
}