extends Node

enum ResourceType {
	Logs,
	Labor,
	Berries,
	Food
}

const res_names = {
	#ResourceType.Logs: "Logs"
}

signal produce
signal consume
signal upgrade

signal resource_type_added
signal resource_stats_updated

class ResourceStats:
	var max: int
	var value: int
	var delta: int

var stock: Dictionary[ResourceType, int] = {}

var storage: Dictionary[ResourceType, int] = {}

func add_storage(data: ResourceAmount):
	ensure_resource(data.resource)
	storage[data.resource] += data.amount
func remove_storage(data: ResourceAmount):
	ensure_resource(data.resource)
	storage[data.resource] -= data.amount
	if stock[data.resource] > storage[data.resource]:
		stock[data.resource] = storage[data.resource]

func _ready():
	Clock.day_passed.connect(_tick_resources)

func _tick_resources():
	var stats: Dictionary[ResourceType, ResourceStats] = {}
	for resource in stock.keys():
		var stat = ResourceStats.new()
		stat.delta = -stock[resource]
		stat.max = storage[resource]
		stats[resource] = stat
	produce.emit()
	consume.emit()
	for resource in stock.keys():
		if !stats.has(resource):
			var stat = ResourceStats.new()
			stat.delta = stock[resource]
			stat.max = storage[resource]
			stats[resource] = stat
			continue
		stats[resource].delta += stock[resource]
		 # \/ this really should change here but like shrug
		stats[resource].max = storage[resource]
	for resource in stock.keys():
		var max = storage[resource]
		stock[resource] = clamp(stock[resource], 0, storage[resource])
		stats[resource].value = stock[resource]
	
	upgrade.emit()
	
	for resource in stock:
		if !stats.has(resource):
			var stat = ResourceStats.new()
			stat.value = stock[resource]
			stat.delta = stock[resource]
			stat.max = storage[resource]
			stats[resource] = stat
			continue
		stats[resource].value = stock[resource]
		stats[resource].max = storage[resource]
	
	for resource in stats.keys():
		var stat = stats[resource]
		resource_stats_updated.emit(resource, stat)

func ensure_resource(resource: ResourceType):
	if resource in stock: return
	stock[resource] = 0
	storage[resource] = 0
	resource_type_added.emit(resource)

func consume_resource(resource: ResourceType, amount: int) -> bool:
	if !stock.has(resource): return false
	if stock[resource] == 0:
		return false
	stock[resource] -= amount
	var failed = stock[resource] < 0
	if failed:
		stock[resource] = 0
	return !failed

func add_resource(resource_type: ResourceType, amount: int):
	ensure_resource(resource_type)
	stock[resource_type] += amount

func get_resource_value(resource_type: ResourceType) -> int:
	if !stock.has(resource_type): return 0
	return stock[resource_type]

func get_resource_name(resource: ResourceType) -> String:
	if resource in res_names: return res_names[resource]
	return str(ResourceType.keys()[resource])

func get_resource_storage(resource: ResourceType):
	if !storage.has(resource): return 0
	return storage[resource]