gamedev-the-steward/simulation/state/OpportunityStateRecord.gd

class_name OpportunityStateRecord
extends RefCounted

const SCHEMA_VERSION := 2
const LEGACY_SCHEMA_VERSION := 1
const NO_EVENT_ID := -1
const DEFAULT_TARGET_AMOUNT := 1.0
const STATUS_OPEN := SimulationIds.OPPORTUNITY_STATUS_OPEN
const STATUS_RESOLVED := SimulationIds.OPPORTUNITY_STATUS_RESOLVED
const STATUS_INVALIDATED := SimulationIds.OPPORTUNITY_STATUS_INVALIDATED

var data: Dictionary


func _init(record_data: Dictionary = {}) -> void:
	data = record_data.duplicate(true)


static func create(
	opportunity_id: int,
	created_tick: int,
	trigger_event_id: int,
	interested_npc_id: int,
	target_amount: float = DEFAULT_TARGET_AMOUNT,
	opportunity_type: StringName = SimulationIds.OPPORTUNITY_RESTOCK_EMPTY_PANTRY,
	target_id: StringName = SimulationIds.STORAGE_VILLAGE_PANTRY,
	resource_id: StringName = SimulationIds.RESOURCE_FOOD
) -> OpportunityStateRecord:
	return (
		OpportunityStateRecord
		. new(
			{
				"schema_version": SCHEMA_VERSION,
				"opportunity_id": opportunity_id,
				"opportunity_type": String(opportunity_type),
				"status": String(STATUS_OPEN),
				"created_tick": created_tick,
				"trigger_event_id": trigger_event_id,
				"interested_npc_id": interested_npc_id,
				"target_id": String(target_id),
				"resource_id": String(resource_id),
				"target_amount": target_amount,
				"resolution_event_id": NO_EVENT_ID,
				"resolved_tick": -1,
				"closed_tick": -1,
				"invalidation_reason": "",
			}
		)
	)


static func from_dictionary(record_data: Dictionary) -> OpportunityStateRecord:
	var version := int(record_data.get("schema_version", -1))
	if version not in [LEGACY_SCHEMA_VERSION, SCHEMA_VERSION]:
		return null
	var normalized := record_data.duplicate(true)
	if version == LEGACY_SCHEMA_VERSION:
		normalized["schema_version"] = SCHEMA_VERSION
		normalized["closed_tick"] = int(record_data.get("resolved_tick", -1))
		normalized["invalidation_reason"] = ""
	if not (
		normalized
		. has_all(
			[
				"opportunity_id",
				"opportunity_type",
				"status",
				"created_tick",
				"trigger_event_id",
				"interested_npc_id",
				"target_id",
				"resource_id",
				"target_amount",
				"resolution_event_id",
				"resolved_tick",
				"closed_tick",
				"invalidation_reason",
			]
		)
	):
		return null
	var opportunity_id := int(normalized["opportunity_id"])
	var opportunity_type := StringName(normalized["opportunity_type"])
	var status := StringName(normalized["status"])
	var created_tick := int(normalized["created_tick"])
	var trigger_event_id := int(normalized["trigger_event_id"])
	var interested_npc_id := int(normalized["interested_npc_id"])
	var target_id := StringName(normalized["target_id"])
	var resource_id := StringName(normalized["resource_id"])
	var target_amount := float(normalized["target_amount"])
	var resolution_event_id := int(normalized["resolution_event_id"])
	var resolved_tick := int(normalized["resolved_tick"])
	var closed_tick := int(normalized["closed_tick"])
	var invalidation_reason := StringName(normalized["invalidation_reason"])
	if opportunity_id < 0 or created_tick < 0 or trigger_event_id < 0 or interested_npc_id < 0:
		return null
	match opportunity_type:
		SimulationIds.OPPORTUNITY_RESTOCK_EMPTY_PANTRY, SimulationIds.OPPORTUNITY_FEED_WEAK_VILLAGER:
			if (
				target_id != SimulationIds.STORAGE_VILLAGE_PANTRY
				or resource_id != SimulationIds.RESOURCE_FOOD
			):
				return null
		SimulationIds.OPPORTUNITY_SUPPLY_MISSING_WOOD, SimulationIds.OPPORTUNITY_REPAIR_HOME_ROOF:
			if (
				target_id != SimulationIds.STORAGE_VILLAGE_WOODPILE
				or resource_id != SimulationIds.RESOURCE_WOOD
			):
				return null
		_:
			return null
	if not is_finite(target_amount) or not is_equal_approx(target_amount, DEFAULT_TARGET_AMOUNT):
		return null
	match status:
		STATUS_OPEN:
			if (
				resolution_event_id != NO_EVENT_ID
				or resolved_tick != -1
				or closed_tick != -1
				or not invalidation_reason.is_empty()
			):
				return null
		STATUS_RESOLVED:
			if (
				resolution_event_id <= trigger_event_id
				or resolved_tick < created_tick
				or closed_tick != resolved_tick
				or not invalidation_reason.is_empty()
			):
				return null
		STATUS_INVALIDATED:
			if (
				resolution_event_id != NO_EVENT_ID
				or resolved_tick != -1
				or closed_tick < created_tick
				or (
					invalidation_reason
					not in [
						SimulationIds.OPPORTUNITY_INVALIDATED_INTERESTED_DIED,
						SimulationIds.OPPORTUNITY_INVALIDATED_EVIDENCE_STALE,
					]
				)
			):
				return null
		_:
			return null
	var record := create(
		opportunity_id,
		created_tick,
		trigger_event_id,
		interested_npc_id,
		target_amount,
		opportunity_type,
		target_id,
		resource_id
	)
	if status == STATUS_RESOLVED:
		record.resolve(resolution_event_id, resolved_tick)
	elif status == STATUS_INVALIDATED:
		record.invalidate(invalidation_reason, closed_tick)
	return record


func resolve(resolution_event_id: int, resolved_tick: int) -> bool:
	if (
		not is_open()
		or resolution_event_id <= get_trigger_event_id()
		or resolved_tick < get_created_tick()
	):
		return false
	data["status"] = String(STATUS_RESOLVED)
	data["resolution_event_id"] = resolution_event_id
	data["resolved_tick"] = resolved_tick
	data["closed_tick"] = resolved_tick
	return true


func invalidate(reason: StringName, closed_tick: int) -> bool:
	if (
		not is_open()
		or closed_tick < get_created_tick()
		or (
			reason
			not in [
				SimulationIds.OPPORTUNITY_INVALIDATED_INTERESTED_DIED,
				SimulationIds.OPPORTUNITY_INVALIDATED_EVIDENCE_STALE,
			]
		)
	):
		return false
	data["status"] = String(STATUS_INVALIDATED)
	data["closed_tick"] = closed_tick
	data["invalidation_reason"] = String(reason)
	return true


func get_opportunity_id() -> int:
	return int(data["opportunity_id"])


func get_opportunity_type() -> StringName:
	return StringName(data["opportunity_type"])


func get_status() -> StringName:
	return StringName(data["status"])


func get_created_tick() -> int:
	return int(data["created_tick"])


func get_trigger_event_id() -> int:
	return int(data["trigger_event_id"])


func get_interested_npc_id() -> int:
	return int(data["interested_npc_id"])


func get_target_id() -> StringName:
	return StringName(data["target_id"])


func get_resource_id() -> StringName:
	return StringName(data["resource_id"])


func get_target_amount() -> float:
	return float(data["target_amount"])


func get_resolution_event_id() -> int:
	return int(data["resolution_event_id"])


func get_resolved_tick() -> int:
	return int(data["resolved_tick"])


func get_closed_tick() -> int:
	return int(data["closed_tick"])


func get_invalidation_reason() -> StringName:
	return StringName(data["invalidation_reason"])


func is_open() -> bool:
	return get_status() == STATUS_OPEN


func to_dictionary() -> Dictionary:
	return data.duplicate(true)