extends Node3D


var _speed_label: Label
var _state_label: Label
var _chain_label: Label
var _weapon_label: Label
var _grapple_icon: TextureRect
var _dash_icon: TextureRect
var _grapple_label: Label
var _dash_label: Label
var _fps_label: Label
var _standalone_speed_label: Label
var _player: CharacterBody3D
var _debug_ui_panel: PanelContainer


func _ready() -> void:
	# Hide mouse
	Input.mouse_mode = Input.MOUSE_MODE_CAPTURED

	# Cel-shaded presentation: stylized sky/grade + toon materials over the
	# arena's baked geometry. The scene ships its own DirectionalLight3D;
	# rename it to Sun so the helper doesn't stack extra suns.
	var existing_light := get_node_or_null("DirectionalLight3D")
	if existing_light:
		existing_light.name = "Sun"
	if not has_node("WorldEnvironment"):
		LevelEnvironment.add_to(self)
	LevelMaterials.apply_toon_recursive(self, 0.0)

	_build_hud()
	
	# Multiplayer Spawning
	var spawner = MultiplayerSpawner.new()
	spawner.name = "PlayerSpawner"
	spawner.spawn_path = "." # Spawn directly as children of the level
	spawner.spawn_function = _spawn_player
	add_child(spawner)
	
	var nm = get_node_or_null("/root/NetworkManager")
	if nm and nm.multiplayer.is_server():
		# Spawn existing players
		for pid in nm.player_stats.keys():
			spawner.spawn(pid)
		
		# Spawn late joiners
		nm.player_connected.connect(func(pid):
			spawner.spawn(pid)
		)
		
		# Despawn on disconnect
		nm.player_disconnected.connect(func(pid):
			var p = get_node_or_null(str(pid))
			if p:
				p.queue_free()
		)


# ── Player ────────────────────────────────────────────────────────────────────

func _spawn_player(pid: int) -> CharacterBody3D:
	var player := CharacterBody3D.new()
	player.name = str(pid)
	player.set_multiplayer_authority(pid)
	
	# Alternate spawn sides (Team A / Team B style)
	var spawn_z = 40.0 if (pid % 2 == 0) else -40.0
	player.position = Vector3(randf_range(-6, 6), 10.0, spawn_z + randf_range(-4, 4))
	
	# Server Synchronizer (Host is the ground truth)
	var server_sync = MultiplayerSynchronizer.new()
	server_sync.name = "ServerSynchronizer"
	server_sync.set_multiplayer_authority(1) # Host always controls these
	var server_rep_config = SceneReplicationConfig.new()
	server_rep_config.add_property(":health")
	server_rep_config.add_property(":shield")
	server_rep_config.add_property(":is_dead")
	server_sync.replication_config = server_rep_config
	player.add_child(server_sync)

	# Client Synchronizer (Client dictates their aim and loadout setup)
	var client_sync = MultiplayerSynchronizer.new()
	client_sync.name = "MultiplayerSynchronizer"
	client_sync.set_multiplayer_authority(pid)
	var client_rep_config = SceneReplicationConfig.new()
	client_rep_config.add_property(":synced_position")
	client_rep_config.add_property(":synced_velocity")
	client_rep_config.add_property(":rotation")
	client_rep_config.add_property("HeadPivot:rotation")
	client_rep_config.add_property(":synced_movement_state")
	client_rep_config.add_property(":synced_movement_speed")
	client_rep_config.add_property(":synced_is_crouching")
	client_rep_config.add_property(":synced_is_ads")
	client_rep_config.add_property(":synced_wall_side")
	client_rep_config.add_property(":synced_is_dancing")
	client_rep_config.add_property(":synced_grapple_point")
	client_rep_config.add_property(":synced_is_grapple_shooting")
	client_rep_config.add_property(":synced_skin_id")
	client_rep_config.add_property(":synced_weapon_path")
	client_rep_config.add_property(":synced_loadout_p1")
	client_rep_config.add_property(":synced_loadout_p2")
	client_rep_config.add_property(":synced_loadout_sp")
	client_rep_config.add_property(":synced_loadout_melee")
	client_rep_config.add_property(":synced_loadout_ready")
	client_rep_config.add_property(":synced_is_targeting")
	client_rep_config.add_property(":synced_homing_target_pos")
	client_sync.replication_config = client_rep_config
	player.add_child(client_sync)

	# Collision shape
	var col_shape := CollisionShape3D.new()
	col_shape.name = "CollisionShape3D"
	var capsule := CapsuleShape3D.new()
	capsule.radius = 0.4
	capsule.height = 1.8
	col_shape.shape = capsule
	player.add_child(col_shape)

	# Apply movement controller script
	var mover_script = preload("res://movement/player_movement_controller.gd")
	if mover_script:
		player.set_script(mover_script)
		
	# Humanoid Model for Player (Shadows only locally, visible to others)
	var humanoid = load("res://characters/humanoid_model.gd").new()
	humanoid.name = "HumanoidModel"
	humanoid.color = Color(0.2, 0.4, 0.8) # Blueish for player
	humanoid.shadows_only = (pid == multiplayer.get_unique_id())
	humanoid.position = Vector3(0, -0.9, 0) # Offset from center to feet
	player.add_child(humanoid)

	# Movement State Machine
	var sm := Node.new()
	sm.name = "MovementStateMachine"
	player.add_child(sm)

	var sm_script = preload("res://movement/movement_state_machine.gd")
	if sm_script:
		sm.set_script(sm_script)

	# Add all movement states
	var state_scripts := {
		"ground": "res://movement/states/state_ground.gd",
		"air": "res://movement/states/state_air.gd",
		"wall_run": "res://movement/states/state_wall_run.gd",
		"wall_cling": "res://movement/states/state_wall_cling.gd",
		"slide": "res://movement/states/state_slide.gd",
		"dash": "res://movement/states/state_dash.gd",
	}
	for state_name in state_scripts:
		var st := Node.new()
		st.name = "state_" + state_name
		sm.add_child(st)
		var st_script = load(state_scripts[state_name])
		if st_script:
			st.set_script(st_script)

	# FPS Camera Rig (HeadPivot → Camera3D)
	var head_pivot := Node3D.new()
	head_pivot.name = "HeadPivot"
	head_pivot.position = Vector3(0, 0.7, 0)  # Eye height
	player.add_child(head_pivot)

	var camera := Camera3D.new()
	camera.name = "Camera3D"
	camera.current = (pid == multiplayer.get_unique_id())
	camera.fov = 90.0
	head_pivot.add_child(camera)
	
	# ── Weapon Setup ────────────────────────────────────────────────────────────
	var wmanager_script = preload("res://weapons/weapon_manager.gd")
	if wmanager_script:
		var wman = wmanager_script.new()
		wman.name = "WeaponManager"
		wman.player = player
		wman.camera = camera
		camera.add_child(wman)

	var rig_script = preload("res://characters/player/fps_camera_rig.gd")
	if rig_script:
		head_pivot.set_script(rig_script)

	# Store original capsule height
	sm.original_capsule_height = capsule.height

	# ── Dependencies ────────────────────────────────────────────────────────
	sm.player = player
	sm.params = player.params
	head_pivot.params = player.params

	# Authority configuration
	var is_local = (pid == multiplayer.get_unique_id())
	if is_local:
		_player = player
		player.set_physics_process(true)
		player.set_process(true)
		sm.set_physics_process(true)
	else:
		player.set_physics_process(false)
		player.set_process(true)
		player.set_process_input(false)
		sm.set_physics_process(false)
		sm.set_process(false)
		sm.set_process_input(false)
		head_pivot.set_process_input(false)
		head_pivot.set_process(false)
		head_pivot.set_physics_process(false)
		var wman = camera.get_node_or_null("WeaponManager")
		if wman:
			wman.set_process_input(false)
			wman.set_process(false)
			wman.set_physics_process(false)
			wman.visible = false
			var cv = wman.get("canvas_layer")
			if cv:
				cv.visible = false

	return player


# ── HUD ───────────────────────────────────────────────────────────────────────

func _build_hud() -> void:
	var canvas := CanvasLayer.new()
	canvas.name = "UI"
	add_child(canvas)

	# ── Crosshair ─────────────────────────────────────────────────────────
	var crosshair := Control.new()
	crosshair.name = "Crosshair"
	crosshair.set_anchors_preset(Control.PRESET_CENTER)
	crosshair.custom_minimum_size = Vector2(20, 20)
	canvas.add_child(crosshair)

	var ch_dot := ColorRect.new()
	ch_dot.name = "Dot"
	ch_dot.color = Color(1, 1, 1, 0.8)
	ch_dot.size = Vector2(4, 4)
	ch_dot.position = Vector2(-2, -2)
	crosshair.add_child(ch_dot)

	# Crosshair lines
	for data in [
		{"pos": Vector2(-10, -1), "size": Vector2(6, 2)},  # Left
		{"pos": Vector2(4, -1), "size": Vector2(6, 2)},    # Right
		{"pos": Vector2(-1, -10), "size": Vector2(2, 6)},  # Top
		{"pos": Vector2(-1, 4), "size": Vector2(2, 6)},    # Bottom
	]:
		var line := ColorRect.new()
		line.color = Color(1, 1, 1, 0.6)
		line.position = data["pos"]
		line.size = data["size"]
		crosshair.add_child(line)

	# ── Info panel background ─────────────────────────────────────────────
	_fps_label = Label.new()
	_fps_label.name = "FPSLabel"
	_fps_label.text = "FPS: 0"
	_fps_label.add_theme_font_size_override("font_size", 24)
	_fps_label.add_theme_color_override("font_color", Color(0.9, 0.9, 0.2))
	_fps_label.add_theme_color_override("font_outline_color", Color(0, 0, 0))
	_fps_label.add_theme_constant_override("outline_size", 4)
	_fps_label.set_anchors_preset(Control.PRESET_TOP_LEFT)
	_fps_label.position = Vector2(12, 12)
	canvas.add_child(_fps_label)
	
	_standalone_speed_label = Label.new()
	_standalone_speed_label.name = "StandaloneSpeedLabel"
	_standalone_speed_label.text = "Speed: 0.0 m/s"
	_standalone_speed_label.add_theme_font_size_override("font_size", 24)
	_standalone_speed_label.add_theme_color_override("font_color", Color(0.3, 1.0, 0.5))
	_standalone_speed_label.add_theme_color_override("font_outline_color", Color(0, 0, 0))
	_standalone_speed_label.add_theme_constant_override("outline_size", 4)
	_standalone_speed_label.set_anchors_preset(Control.PRESET_TOP_LEFT)
	_standalone_speed_label.position = Vector2(12, 45)
	canvas.add_child(_standalone_speed_label)
	
	_debug_ui_panel = PanelContainer.new()
	_debug_ui_panel.name = "InfoPanel"
	_debug_ui_panel.offset_left = 12
	_debug_ui_panel.offset_top = 50
	_debug_ui_panel.offset_right = 400
	_debug_ui_panel.offset_bottom = 160
	var panel_style := StyleBoxFlat.new()
	panel_style.bg_color = Color(0, 0, 0, 0.55)
	panel_style.corner_radius_top_left = 8
	panel_style.corner_radius_top_right = 8
	panel_style.corner_radius_bottom_left = 8
	panel_style.corner_radius_bottom_right = 8
	panel_style.content_margin_left = 12
	panel_style.content_margin_top = 8
	panel_style.content_margin_right = 12
	panel_style.content_margin_bottom = 8
	_debug_ui_panel.add_theme_stylebox_override("panel", panel_style)
	canvas.add_child(_debug_ui_panel)

	var vbox := VBoxContainer.new()
	vbox.name = "InfoVBox"
	_debug_ui_panel.add_child(vbox)

	_speed_label = Label.new()
	_speed_label.name = "SpeedLabel"
	_speed_label.text = "Speed: 0.0 m/s"
	_speed_label.add_theme_font_size_override("font_size", 18)
	_speed_label.add_theme_color_override("font_color", Color(0.3, 1.0, 0.5))
	vbox.add_child(_speed_label)

	_state_label = Label.new()
	_state_label.name = "StateLabel"
	_state_label.text = "State: ground"
	_state_label.add_theme_font_size_override("font_size", 16)
	_state_label.add_theme_color_override("font_color", Color(0.7, 0.85, 1.0))
	vbox.add_child(_state_label)

	_chain_label = Label.new()
	_chain_label.name = "ChainLabel"
	_chain_label.text = "Chain: 0 (+0%)"
	_chain_label.add_theme_font_size_override("font_size", 16)
	_chain_label.add_theme_color_override("font_color", Color(1.0, 0.8, 0.3))
	vbox.add_child(_chain_label)


	# ── Weapon & Ammo Panel ───────────────────────────────────────────────
	var wp_panel := PanelContainer.new()
	wp_panel.name = "WeaponPanel"
	wp_panel.set_anchors_preset(Control.PRESET_BOTTOM_RIGHT)
	wp_panel.offset_left = -250
	wp_panel.offset_top = -100
	wp_panel.offset_right = -20
	wp_panel.offset_bottom = -20
	var wp_style := StyleBoxFlat.new()
	wp_style.bg_color = Color(0, 0, 0, 0.6)
	wp_style.corner_radius_top_left = 8
	wp_style.corner_radius_top_right = 8
	wp_style.corner_radius_bottom_left = 8
	wp_style.corner_radius_bottom_right = 8
	wp_style.content_margin_left = 16
	wp_style.content_margin_top = 12
	wp_style.content_margin_right = 16
	wp_style.content_margin_bottom = 12
	wp_panel.add_theme_stylebox_override("panel", wp_style)
	wp_panel.grow_horizontal = Control.GROW_DIRECTION_BEGIN
	canvas.add_child(wp_panel)
	
	_weapon_label = Label.new()
	_weapon_label.name = "WeaponLabel"
	_weapon_label.text = "Unarmed\n0 / 0"
	_weapon_label.horizontal_alignment = HORIZONTAL_ALIGNMENT_RIGHT
	_weapon_label.vertical_alignment = VERTICAL_ALIGNMENT_BOTTOM
	_weapon_label.add_theme_font_size_override("font_size", 24)
	wp_panel.add_child(_weapon_label)

	# ── Utilities Panel ───────────────────────────────────────────────────────
	var util_panel := PanelContainer.new()
	util_panel.name = "UtilPanel"
	util_panel.set_anchors_preset(Control.PRESET_BOTTOM_RIGHT)
	util_panel.offset_left = -200
	util_panel.offset_top = -180
	util_panel.offset_right = -20
	util_panel.offset_bottom = -110
	var util_style := StyleBoxFlat.new()
	util_style.bg_color = Color(0, 0, 0, 0.6)
	util_style.corner_radius_top_left = 8
	util_style.corner_radius_top_right = 8
	util_style.corner_radius_bottom_left = 8
	util_style.corner_radius_bottom_right = 8
	util_style.content_margin_left = 12
	util_style.content_margin_top = 8
	util_style.content_margin_right = 12
	util_style.content_margin_bottom = 8
	util_panel.add_theme_stylebox_override("panel", util_style)
	util_panel.grow_horizontal = Control.GROW_DIRECTION_BEGIN
	canvas.add_child(util_panel)

	var util_hbox := HBoxContainer.new()
	util_hbox.name = "UtilHBox"
	util_hbox.add_theme_constant_override("separation", 20)
	util_hbox.alignment = BoxContainer.ALIGNMENT_CENTER
	util_panel.add_child(util_hbox)

	var grapple_vbox := VBoxContainer.new()
	grapple_vbox.alignment = BoxContainer.ALIGNMENT_CENTER
	util_hbox.add_child(grapple_vbox)

	_grapple_icon = TextureRect.new()
	_grapple_icon.texture = load("res://assets/ui/grapple_icon.jpg")
	_grapple_icon.expand_mode = TextureRect.EXPAND_IGNORE_SIZE
	_grapple_icon.custom_minimum_size = Vector2(32, 32)
	_grapple_icon.stretch_mode = TextureRect.STRETCH_KEEP_ASPECT_CENTERED
	grapple_vbox.add_child(_grapple_icon)
	
	_grapple_label = Label.new()
	_grapple_label.text = "Grapple"
	_grapple_label.horizontal_alignment = HORIZONTAL_ALIGNMENT_CENTER
	_grapple_label.add_theme_font_size_override("font_size", 12)
	grapple_vbox.add_child(_grapple_label)

	var dash_vbox := VBoxContainer.new()
	dash_vbox.alignment = BoxContainer.ALIGNMENT_CENTER
	util_hbox.add_child(dash_vbox)

	_dash_icon = TextureRect.new()
	_dash_icon.texture = load("res://assets/ui/dash_icon.jpg")
	_dash_icon.expand_mode = TextureRect.EXPAND_IGNORE_SIZE
	_dash_icon.custom_minimum_size = Vector2(32, 32)
	_dash_icon.stretch_mode = TextureRect.STRETCH_KEEP_ASPECT_CENTERED
	dash_vbox.add_child(_dash_icon)
	
	_dash_label = Label.new()
	_dash_label.text = "Ready"
	_dash_label.horizontal_alignment = HORIZONTAL_ALIGNMENT_CENTER
	_dash_label.add_theme_font_size_override("font_size", 12)
	dash_vbox.add_child(_dash_label)

# ── Utility: Key Name ─────────────────────────────────────────────────────────

func _get_key_name(action: String) -> String:
	if not InputMap.has_action(action):
		return "?"
	var events = InputMap.action_get_events(action)
	for e in events:
		if e is InputEventKey:
			var code = e.physical_keycode if e.physical_keycode != 0 else e.keycode
			return OS.get_keycode_string(code)
		elif e is InputEventMouseButton:
			if e.button_index == MOUSE_BUTTON_LEFT: return "LClick"
			elif e.button_index == MOUSE_BUTTON_RIGHT: return "RClick"
			elif e.button_index == MOUSE_BUTTON_MIDDLE: return "MClick"
	return "?"

# ── HUD Update ────────────────────────────────────────────────────────────────

func _process(_delta: float) -> void:
	if not _player or not is_instance_valid(_player):
		return
		
	if _debug_ui_panel:
		_debug_ui_panel.visible = SettingsManager.show_debug_ui
		
	if _fps_label:
		_fps_label.visible = SettingsManager.show_fps
		if _fps_label.visible:
			_fps_label.text = "FPS: %d" % Engine.get_frames_per_second()

	var vel: Vector3 = _player.velocity
	var hspeed := Vector2(vel.x, vel.z).length()
	var total_speed := vel.length()

	if _speed_label:
		_speed_label.text = "Speed: %.1f m/s (total: %.1f)" % [hspeed, total_speed]

	if _standalone_speed_label:
		if SettingsManager.show_movement_speed and not SettingsManager.show_debug_ui:
			_standalone_speed_label.visible = true
			_standalone_speed_label.text = "Speed: %.1f m/s" % hspeed
			if _fps_label and _fps_label.visible:
				_standalone_speed_label.position = Vector2(12, 45)
			else:
				_standalone_speed_label.position = Vector2(12, 12)
		else:
			_standalone_speed_label.visible = false

	if _state_label:
		var sm = _player.get_node_or_null("MovementStateMachine")
		if sm:
			_state_label.text = "State: %s" % sm.current_state

	if _chain_label:
		var sm = _player.get_node_or_null("MovementStateMachine")
		if sm:
			_chain_label.text = "Chain: %d (+%d%%)" % [sm.chain_count, int(sm.current_chain_bonus * 100)]
			
			# Update utility indicators
			if sm.current_state == "grapple" or sm.is_grapple_shooting:
				_grapple_icon.modulate = Color(0.2, 1.0, 0.4)
				_grapple_label.text = "Grappling"
			else:
				_grapple_icon.modulate = Color(1.0, 1.0, 1.0)
				_grapple_label.text = "Ready"
				
			var dash_rem = sm.get_dash_cooldown_remaining()
			if dash_rem > 0.0:
				_dash_icon.modulate = Color(1.0, 0.3, 0.3)
				_dash_label.text = "%.1f" % dash_rem
			else:
				_dash_icon.modulate = Color(1.0, 1.0, 1.0)
				_dash_label.text = "Ready"

	if _weapon_label:
		var wman = _player.get_node_or_null("HeadPivot/Camera3D/WeaponManager")
		if wman and wman.weapons.has(wman.active_slot):
			var active_weapon = wman.weapons[wman.active_slot]
			var w_name = "Weapon"
			var cur_ammo = 0
			var max_ammo = 0
			
			if "weapon_name" in active_weapon:
				w_name = active_weapon.weapon_name
			elif active_weapon is DoubleBarrelShotgun:
				w_name = "Double Barrel Shotgun"
				
			if "current_ammo" in active_weapon:
				cur_ammo = active_weapon.current_ammo
				max_ammo = active_weapon.max_ammo
			elif "shells" in active_weapon:
				cur_ammo = active_weapon.shells
				max_ammo = 2
				
			if "reloading" in active_weapon and active_weapon.reloading:
				_weapon_label.text = "%s\nReloading..." % w_name
			else:
				_weapon_label.text = "%s\n%d / %d" % [w_name, cur_ammo, max_ammo]
		else:
			_weapon_label.text = "Unarmed\n0 / 0"