netfishing/ui/hotbar_slot.gd

class_name HotbarSlot
extends Button

signal item_hovered(slot_index: int, item_id: StringName)
signal item_hover_ended(slot_index: int)
signal item_drag_started
signal item_drag_finished

const ItemCatalogType = preload("res://items/item_catalog.gd")
const ItemDataType = preload("res://items/item_data.gd")
const PlayerBagType = preload("res://inventory/player_bag.gd")
const PlayerHotbarType = preload("res://inventory/player_hotbar.gd")

var slot_index: int = 0
var _hotbar: PlayerHotbarType
var _bag: PlayerBagType
var _catalog: ItemCatalogType
var _drag_enabled: bool = false
var _drag_in_progress: bool = false
var _slot_number_label: Label
var _quantity_label: Label


func setup(
	index: int,
	hotbar: PlayerHotbarType,
	bag: PlayerBagType,
	catalog: ItemCatalogType,
) -> void:
	slot_index = index
	_hotbar = hotbar
	_bag = bag
	_catalog = catalog
	custom_minimum_size = Vector2(50, 50)
	expand_icon = true
	focus_mode = Control.FOCUS_ALL
	_slot_number_label = _create_corner_label(str(slot_index + 1))
	_quantity_label = _create_corner_label("", true)
	pressed.connect(_select_slot)
	mouse_entered.connect(_on_mouse_entered)
	mouse_exited.connect(_on_mouse_exited)
	refresh()


func set_drag_enabled(enabled: bool) -> void:
	_drag_enabled = enabled
	mouse_filter = (
		Control.MOUSE_FILTER_STOP
		if enabled
		else Control.MOUSE_FILTER_IGNORE
	)


func refresh() -> void:
	if _hotbar == null:
		return
	var item_id: StringName = _hotbar.get_item_id(slot_index)
	var item: ItemDataType = (
		_catalog.get_item_by_id(item_id)
		if _catalog != null and not item_id.is_empty()
		else null
	)
	icon = item.icon if item != null else null
	var quantity: int = (
		_bag.get_quantity(item_id)
		if _bag != null and not item_id.is_empty()
		else 0
	)
	var quantity_text: String = (
		"×%d" % quantity
		if item != null and item.stackable and quantity > 1
		else ""
	)
	text = ""
	_quantity_label.text = quantity_text
	_quantity_label.visible = not quantity_text.is_empty()
	tooltip_text = (
		item.display_name if item != null else "Empty hotbar slot"
	)
	button_pressed = slot_index == _hotbar.get_selected_slot()


func _create_corner_label(
	label_text: String,
	bottom_right: bool = false,
) -> Label:
	var label := Label.new()
	label.text = label_text
	label.mouse_filter = Control.MOUSE_FILTER_IGNORE
	label.add_theme_font_size_override("font_size", 11)
	label.add_theme_color_override("font_color", UIPalette.TEXT)
	if bottom_right:
		label.set_anchors_preset(Control.PRESET_BOTTOM_RIGHT)
		label.position = Vector2(-22.0, -20.0)
	else:
		label.position = Vector2(5.0, 2.0)
	add_child(label)
	return label


func _gui_input(event: InputEvent) -> void:
	if (
		_drag_enabled
		and event is InputEventMouseButton
		and event.button_index == MOUSE_BUTTON_RIGHT
		and event.pressed
		and _hotbar != null
	):
		_hotbar.clear_slot(slot_index)
		accept_event()


func _get_drag_data(_at_position: Vector2) -> Variant:
	if not _drag_enabled or _hotbar == null:
		return null
	var item_id: StringName = _hotbar.get_item_id(slot_index)
	if item_id.is_empty():
		return null
	var item: ItemDataType = _catalog.get_item_by_id(item_id)
	var preview := TextureRect.new()
	preview.custom_minimum_size = Vector2(44, 44)
	preview.texture = item.icon if item != null else null
	preview.expand_mode = TextureRect.EXPAND_IGNORE_SIZE
	preview.stretch_mode = TextureRect.STRETCH_KEEP_ASPECT_CENTERED
	preview.mouse_filter = Control.MOUSE_FILTER_IGNORE
	set_drag_preview(preview)
	_drag_in_progress = true
	item_drag_started.emit()
	return {
		"kind": "hotbar_slot",
		"slot_index": slot_index,
		"item_id": String(item_id),
	}


func _can_drop_data(_at_position: Vector2, data: Variant) -> bool:
	if not _drag_enabled or typeof(data) != TYPE_DICTIONARY:
		return false
	var payload: Dictionary = data
	var kind: String = str(payload.get("kind", ""))
	if kind == "hotbar_slot":
		var source_index: int = int(payload.get("slot_index", -1))
		return (
			source_index >= 0
			and source_index < PlayerHotbarType.SLOT_COUNT
			and source_index != slot_index
		)
	if kind != "bag_item":
		return false
	var item_id: StringName = StringName(str(payload.get("item_id", "")))
	if _bag == null or not _bag.owns_item(item_id) or _catalog == null:
		return false
	var item: ItemDataType = _catalog.get_item_by_id(item_id)
	return item != null and item.is_valid() and item.hotbar_allowed


func _drop_data(_at_position: Vector2, data: Variant) -> void:
	if not _can_drop_data(Vector2.ZERO, data) or _hotbar == null:
		return
	var payload: Dictionary = data
	if str(payload.get("kind", "")) == "hotbar_slot":
		_hotbar.swap_slots(int(payload["slot_index"]), slot_index)
	else:
		_hotbar.assign_item(
			slot_index,
			StringName(str(payload["item_id"]))
		)


func _select_slot() -> void:
	if _hotbar != null:
		_hotbar.select_slot(slot_index)


func _on_mouse_entered() -> void:
	if _hotbar != null:
		item_hovered.emit(slot_index, _hotbar.get_item_id(slot_index))


func _on_mouse_exited() -> void:
	item_hover_ended.emit(slot_index)


func _notification(what: int) -> void:
	if what == NOTIFICATION_DRAG_END and _drag_in_progress:
		_drag_in_progress = false
		item_drag_finished.emit()