use bevy::{prelude::*, window::PrimaryWindow};
use bevy_rapier3d::{pipeline::QueryFilter, plugin::RapierContext};
use world_generation::{hex_utils::HexCoord, prelude::Map};

use crate::{camera_system::components::PhosCamera, prelude::PhosChunkRegistry};

use super::chunk_rebuild::ChunkRebuildQueue;

pub struct TerraFormingTestPlugin;

impl Plugin for TerraFormingTestPlugin {
	fn build(&self, app: &mut App) {
		app.add_systems(Update, deform);
	}
}

fn deform(
	cam_query: Query<(&GlobalTransform, &Camera), With<PhosCamera>>,
	window: Query<&Window, With<PrimaryWindow>>,
	keyboard: Res<ButtonInput<MouseButton>>,
	rapier_context: Res<RapierContext>,
	mut heightmap: ResMut<Map>,
	mut rebuild: ResMut<ChunkRebuildQueue>,
	time: Res<Time>,
) {
	if !keyboard.just_pressed(MouseButton::Left) {
		return;
	}

	let win = window.single();
	let (cam_transform, camera) = cam_query.single();
	let Some(cursor_pos) = win.cursor_position() else {
		return;
	};

	let Some(cam_ray) = camera.viewport_to_world(cam_transform, cursor_pos) else {
		return;
	};

	let collision = rapier_context.cast_ray(
		cam_ray.origin,
		cam_ray.direction.into(),
		100.,
		true,
		QueryFilter::only_fixed(),
	);

	if let Some((_, dist)) = collision {
		let contact_point = cam_ray.get_point(dist);
		let contact_coord = HexCoord::from_world_pos(contact_point);
		let cur_height = heightmap.sample_height(&contact_coord);
		heightmap.set_height(&contact_coord, cur_height + 1.);
		let cur_chunk = contact_coord.to_chunk_index(heightmap.width);

		if contact_coord.is_on_chunk_edge() {
			let neighbors = contact_coord.get_neighbors();
			let mut other_chunks: Vec<_> = neighbors
				.iter()
				.map(|c| c.to_chunk_index(heightmap.width))
				.filter(|c| c != &cur_chunk)
				.collect();
			rebuild.queue.append(&mut other_chunks);
		}
		rebuild.queue.push(cur_chunk);
	}
}