Add generator buffs (additional test required)

generator_container.gd

@@ -1,5 +1,15 @@
 extends VBoxContainer
 
+const GENERATOR_BUFF_TILE_SCENE: PackedScene = preload("res://generator_buff_tile.tscn")
+
+class BuffRow extends RefCounted:
+	var buff: GeneratorBuffData
+	var tile: GeneratorBuffTile
+
+	func _init(row_buff: GeneratorBuffData, row_tile: GeneratorBuffTile) -> void:
+		buff = row_buff
+		tile = row_tile
+
 @export var _generator: CurrencyGenerator
 
 @onready var _buy_button: Button = $HBoxContainerGenerators/BuyOneButton
@@ -8,16 +18,28 @@ extends VBoxContainer
 @onready var _owned_value: Label = $GeneratorStatsGrid/OwnedValue
 @onready var _next_cost_value: Label = $GeneratorStatsGrid/NextCostValue
 @onready var _production_value: Label = $GeneratorStatsGrid/ProductionValue
+@onready var _buffs_section: VBoxContainer = $BuffsSection
+@onready var _buffs_list: VBoxContainer = $BuffsSection/BuffsList
+
+var _buff_rows: Array[BuffRow] = []
 
 func _ready() -> void:
+	if _generator == null:
+		push_warning("GeneratorContainer '%s' has no generator assigned." % String(name))
+		return
+
 	_generator.purchase_completed.connect(_on_generator_updated)
 	_generator.production_tick.connect(_on_generator_updated)
 	_generator.purchase_failed.connect(_on_generator_updated)
+	_generator.buff_purchased.connect(_on_generator_updated)
+	_generator.buff_purchase_failed.connect(_on_generator_updated)
 	GameState.currency_changed.connect(_on_currency_changed)
 
 	GameState.generator_state_changed.connect(_on_generator_state_changed)
+	GameState.generator_buff_level_changed.connect(_on_generator_buff_level_changed)
 
 	_name_label.text = _generator.data.name if _generator.data != null else "Generator"
+	_build_buff_rows()
 	_refresh_generator_ui()
 
 func _on_buy_pressed() -> void:
@@ -32,9 +54,7 @@ func _on_debug_income_pressed() -> void:
 	_generator.grant_currency(BigNumber.from_float(100.0))
 	_refresh_generator_ui()
 
-func _on_currency_changed(changed_currency_id: StringName, _new_value: BigNumber) -> void:
-	if changed_currency_id != _generator.get_currency_id():
-		return
+func _on_currency_changed(_changed_currency_id: StringName, _new_value: BigNumber) -> void:
 	_refresh_generator_ui()
 
 func _on_generator_updated(_a = null, _b = null, _c = null, _d = null) -> void:
@@ -44,6 +64,63 @@ func _on_generator_state_changed(generator_id: StringName, _state: Dictionary) -
 	if generator_id == _generator.get_generator_id():
 		visible = true
 		_refresh_generator_ui()
+
+func _on_generator_buff_level_changed(generator_id: StringName, _buff_id: StringName, _new_level: int) -> void:
+	if generator_id != _generator.get_generator_id():
+		return
+
+	_refresh_generator_ui()
+
+func _on_buy_buff_pressed(buff_id: StringName) -> void:
+	_generator.buy_buff(buff_id)
+	_refresh_generator_ui()
+
+func _build_buff_rows() -> void:
+	_buff_rows.clear()
+	for child in _buffs_list.get_children():
+		child.queue_free()
+
+	for buff in _generator.get_buffs():
+		if buff == null:
+			continue
+
+		var buff_id_text: String = String(buff.id).strip_edges()
+		if buff_id_text.is_empty():
+			continue
+
+		var tile: GeneratorBuffTile = GENERATOR_BUFF_TILE_SCENE.instantiate() as GeneratorBuffTile
+		if tile == null:
+			push_warning("Failed to instantiate GeneratorBuffTile for buff '%s'." % buff_id_text)
+			continue
+
+		_buffs_list.add_child(tile)
+		tile.set_buff(buff)
+		tile.buy_pressed.connect(_on_buy_buff_pressed)
+		_buff_rows.append(BuffRow.new(buff, tile))
+
+func _refresh_buff_rows(can_interact: bool) -> void:
+	_buffs_section.visible = not _buff_rows.is_empty()
+	if _buff_rows.is_empty():
+		return
+
+	for row in _buff_rows:
+		if row == null or row.buff == null or row.tile == null:
+			continue
+
+		var buff: GeneratorBuffData = row.buff
+		var level: int = _generator.get_buff_level(buff.id)
+		var effect_text: String = buff.get_effect_description(level)
+
+		if _generator.is_buff_maxed(buff.id):
+			row.tile.set_runtime(level, effect_text, "", false, true)
+			continue
+
+		var cost: BigNumber = _generator.get_buff_cost(buff.id)
+		var cost_currency_id: StringName = _generator.get_buff_cost_currency_id(buff.id)
+		var currency_label: String = GameState.get_currency_name(cost_currency_id)
+		var cost_text: String = "%s %s" % [cost.to_string_suffix(2), currency_label]
+		var can_buy: bool = can_interact and _generator.can_buy_buff(buff.id)
+		row.tile.set_runtime(level, effect_text, cost_text, can_buy, false)
 	
 func _refresh_generator_ui() -> void:
 	var can_interact: bool = _generator.is_available_to_player()
@@ -57,8 +134,9 @@ func _refresh_generator_ui() -> void:
 	if _generator.data != null and can_interact:
 		production_per_second = _generator.data.production_per_second(
 			_generator.owned,
-			_generator.run_multiplier,
+			_generator.run_multiplier * _generator.get_automatic_production_multiplier(),
 			_generator.purchased_count
 		)
 
 	_production_value.text = BigNumber.from_float(production_per_second).to_string_suffix(2)
+	_refresh_buff_rows(can_interact)