extends VBoxContainer

@onready var color_pickers_1: Array[TrainColorPicker] = [$%ColorPicker1, $%ColorPicker2, $%ColorPicker3, $%ColorPicker4, $%ColorPicker5, $%ColorPicker6]
@onready var color_pickers_2: Array[TrainColorPicker] = [$%ColorPicker7, $%ColorPicker8, $%ColorPicker9, $%ColorPicker10, $%ColorPicker11, $%ColorPicker12]
@onready var treno: Node3D = $TrainViewportContainer/SubViewport/Treno

var selected_color: Color

func _ready() -> void:
	for color_picker in color_pickers_1:
		color_picker.pressed.connect(_on_color_selected.bind(color_picker, 0))
		color_picker.deselect()
	
	var init_idx_1 = GameState.save_data.train_color_1_index
	_set_selected_color(init_idx_1, 0)

	for color_picker in color_pickers_2:
		color_picker.pressed.connect(_on_color_selected.bind(color_picker, 1))
		color_picker.deselect()
	
	var init_idx_2 = GameState.save_data.train_color_2_index
	_set_selected_color(init_idx_2, 1)

func _process(delta: float) -> void:
	if treno:
		treno.rotation_degrees.y += 30.0 * delta

func _set_selected_color(index: int, material_index: int) -> void:
	var color_pickers = color_pickers_1 if material_index == 0 else color_pickers_2
	for i in range(color_pickers.size()):
		if i == index:
			color_pickers[i].select()
			selected_color = color_pickers[i].color
			_apply_train_color(selected_color, material_index, index)
		else:
			color_pickers[i].deselect()

func _on_color_selected(selected_color_picker: TrainColorPicker, material_index: int) -> void:
	var color_pickers = color_pickers_1 if material_index == 0 else color_pickers_2
	for i in range(color_pickers.size()):
		if color_pickers[i] == selected_color_picker:
			_set_selected_color(i, material_index)
			break

func _apply_train_color(color: Color, material_index: int, color_index: int = 0) -> void:
	var path = "res://tgcc/train/Color%s_train.tres" % ("1" if material_index == 0 else "2")
	var mat: ShaderMaterial = load(path) 
	if mat:
		mat.set_shader_parameter("albedo_color", color)
		
	if material_index == 0:
		GameState.save_data.train_color_1 = color.to_html()
		GameState.save_data.train_color_1_index = color_index
	else:
		GameState.save_data.train_color_2 = color.to_html()
		GameState.save_data.train_color_2_index = color_index
	GameState.save_game()