class_name AudienceMember
extends Node2D

@export var mood : float = 0.
@export_enum("blue", "green", "red") var color

@export var head : Node2D
@export var face : Node2D
@export var body : Node2D
@export var body_shape : Node2D
@export var profile : AudienceProfile

@export_enum("angry", "neutral", "happy", "laugh") var expression

var is_listening : bool = false
var last_joke_heard : Joke.JokeType = -1

var known_faces : Array[String] = [
	"res://scenes/faces/face_curly.tscn",
	"res://scenes/faces/face_jenny.tscn",
	"res://scenes/faces/face_moritz.tscn",
	"res://scenes/faces/face_ronald.tscn",
	# Add more scenes as needed
]

var known_bodies = {
	"blue" : [
		"res://scenes/bodies/body_blue_1.tscn",
		"res://scenes/bodies/body_blue_2.tscn",
		"res://scenes/bodies/body_blue_3.tscn",
		# Add more scenes as needed
	],
	"green" : [
		"res://scenes/bodies/body_green_1.tscn",
		"res://scenes/bodies/body_green_2.tscn",
		"res://scenes/bodies/body_green_3.tscn",
		# Add more scenes as needed
	],
	"red" : [
		"res://scenes/bodies/body_red_1.tscn",
		"res://scenes/bodies/body_red_2.tscn",
		"res://scenes/bodies/body_red_3.tscn",
		# Add more scenes as needed
	],
}

var buh_sounds = [
	preload("res://audio/buh_1.wav"),
	preload("res://audio/buh_2.wav"),
	preload("res://audio/buh_3.wav"),
	preload("res://audio/buh_4.wav"),
]

var laugh_sounds = [
	preload("res://audio/haha_1.wav"),
	preload("res://audio/haha_2.wav"),
	preload("res://audio/haha_3.wav"),
	preload("res://audio/haha_4.wav"),
]

var cough_sounds = [
	preload("res://audio/cough_1.wav"),
	preload("res://audio/cough_2.wav"),
	preload("res://audio/cough_3.wav"),
]

const laughter_duration : float = 2. # seconds
const laughter_bobs : int = 4
var laughter_left : float = 0.

func _map_color_to_profile_data_index(color):
	match color:
		"green":
			return 0
		"blue":
			return 1
		"red":
			return 2
	return 0

static func get_random_color():
	var keys = ["blue", "green", "red"]
	return keys[randi() % keys.size()]

func get_random_cough():
	return cough_sounds[randi() % cough_sounds.size()]

func get_random_buh():
	return buh_sounds[randi() % buh_sounds.size()]

func get_random_laugh():
	return laugh_sounds[randi() % laugh_sounds.size()]

# Called when the node enters the scene tree for the first time.
func _ready():
	laughter_left = 0.
	mood = randf_range(profile.lashout_threshold, profile.happy_threshold)
	update_expression()

	var profile_index = _map_color_to_profile_data_index(color)
	var profile_data = AudienceProfile.get_profile_data(profile_index)
	profile.load_data(profile_data)

	if body_shape == null:
		set_random_body(color)

	if face == null:
		set_random_face()

# Called every frame. 'delta' is the elapsed time since the previous frame.
func _process(delta):
	if laughter_left >= 0:
		laughter_left -= delta

	if not is_listening and mood > profile.happy_threshold * .9:
		mood -= profile.happiness_decay * delta
	elif not is_listening and mood < profile.lashout_threshold * .9:
		mood += profile.lashout_decay * delta

	if mood < profile.lashout_threshold:
		throw_bottle()

	update_expression()

func _input(event):
	var just_pressed = event.is_pressed() and not event.is_echo()
	if just_pressed and OS.is_debug_build():
		if Input.is_key_pressed(KEY_SPACE):
			update_mood(1.)
		elif Input.is_key_pressed(KEY_ENTER):
			update_mood(-1.)

func update_mood(change: float):
	if laughter_left <= 0 and change > 0 and (mood + change) > profile.happy_threshold :
		laughter_left = laughter_duration
		# bob head
		var tween = get_tree().create_tween().bind_node(self).set_loops(laughter_bobs)
		var bob_duration = laughter_duration / laughter_bobs / 2
		tween.tween_property(face, "position", Vector2.UP * 20, bob_duration).set_delay(randf_range(0, bob_duration))
		tween.tween_property(face, "position", Vector2.ZERO, bob_duration)

		$Speaker.stream = get_random_laugh()
		$Speaker.play()

		tween.tween_property(face, "position", Vector2.UP * 20, bob_duration).set_delay(randf_range(0, bob_duration))
		tween.tween_property(face, "position", Vector2.ZERO, bob_duration)

	mood += change

func update_expression():
	if laughter_left > 0:
		expression = "laugh"
	elif mood > profile.happy_threshold:
		expression = "happy"
	elif mood < profile.angry_threshold:
		expression = "angry"
	else:
		expression = "neutral"

	if randi() % 25000 == 1:
		$Speaker.stream = get_random_cough()
		$Speaker.play()

func set_random_body(for_color):
	if str(for_color) not in known_bodies:
		var keys = known_bodies.keys()
		for_color = keys[randi() % keys.size()]
		print(keys.size())
	var body_path = known_bodies[for_color][randi() % known_bodies[for_color].size()]
	var body_res = load(body_path)
	body_shape = body_res.instantiate()
	body.add_child(body_shape)

func set_random_face():
	var face_res = load(known_faces[randi() % known_faces.size()])
	face = face_res.instantiate()
	head.add_child(face)

func on_joke_start():
	is_listening = true

func on_hear_joke(joke: Joke):
	var mood_change = profile.joke_mood_mapping.get(joke.type, 0) * joke.effectiveness
	if joke.type == last_joke_heard and mood_change > 0:
		mood_change *= .5
	update_mood(mood_change)
	last_joke_heard = joke.type

func on_joke_finish():
	is_listening = false

func throw_bottle():
	mood = profile.lashout_threshold * 0.99

	$Speaker.stream = get_random_buh()
	$Speaker.play()

	var bottle_scene = preload("res://scenes/objects/bottle.tscn")
	var bottle = bottle_scene.instantiate()

	add_child(bottle);