extends SceneTree

# A transport hub owns its logistics vessels directly. Building a hub
# provisions every researched slot immediately; Freight Yards provisions its
# extra slot on hubs that already exist.


func _initialize() -> void:
	var state: Node = root.get_node("GameState")
	state.buildings.clear()
	state.extractors.clear()
	state.transporters.clear()
	state.reset_progression()
	state.rocks = 900
	state.unlocked = ["shipyards", "logistics", "freight"]

	var hub_a: int = state.add_building(state.BUILDING_TRANSPORT_HUB, Vector2(-500, 0))
	var per_hub: int = state.hub_capacity()
	if state.transporters.size() != per_hub or _at_hub(state, hub_a) != per_hub:
		return _fail("first hub did not immediately provision %d transporters" % per_hub)
	print("first hub provisioned %d transporters" % per_hub)

	var hub_b: int = state.add_building(state.BUILDING_TRANSPORT_HUB, Vector2(-500, 600))
	if state.transporters.size() != per_hub * 2 or _at_hub(state, hub_b) != per_hub:
		return _fail("second hub did not bring its own transporters")
	print("second hub raised logistics fleet to %d" % state.transporters.size())

	var before: int = state.transporters.size()
	if not state.unlock("freight_yards"):
		return _fail("could not research Freight Yards")
	var expected: int = state.hub_capacity() * state.hub_count()
	if state.transporters.size() != expected or state.transporters.size() != before + 2:
		return _fail("Freight Yards did not provision one transporter per existing hub")
	if _at_hub(state, hub_a) != state.hub_capacity() \
			or _at_hub(state, hub_b) != state.hub_capacity():
		return _fail("researched transporters were assigned to the wrong hubs")

	print("Freight Yards provisioned existing hubs: %d -> %d" % [before, expected])
	if state.remove_buildings_of_type(state.BUILDING_TRANSPORT_HUB) != 2 \
			or not state.transporters.is_empty():
		return _fail("removing hubs left orphaned logistics transporters")
	print("TEST_HUB_CAPACITY: PASS")
	quit(0)


func _at_hub(state: Node, hub_id: int) -> int:
	var count: int = 0
	for transporter in state.transporters:
		if int(transporter.hub_id) == hub_id:
			count += 1
	return count


func _fail(message: String) -> void:
	print("TEST_HUB_CAPACITY: FAIL (%s)" % message)
	quit(1)