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, RebuildChunk},
};

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>>,
	mut commands: Commands,
	window: Query<&Window, With<PrimaryWindow>>,
	mouse: Res<ButtonInput<MouseButton>>,
	rapier_context: Res<RapierContext>,
	mut heightmap: ResMut<Map>,
	chunks: Res<PhosChunkRegistry>,
) {
	let mut multi = 0.;
	if mouse.just_pressed(MouseButton::Left) {
		multi = 1.;
	} else if mouse.just_pressed(MouseButton::Right) {
		multi = -1.;
	}

	if multi == 0. {
		return;
	}

	#[cfg(feature = "tracing")]
	let span = info_span!("Deform Mesh").entered();
	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((e, dist)) = collision {
		let contact_point = cam_ray.get_point(dist);
		let contact_coord = HexCoord::from_world_pos(contact_point);
		let modified_chunks = heightmap.create_crater(&contact_coord, 5, 5. * multi);
		for c in modified_chunks {
			commands.entity(chunks.chunks[c]).insert(RebuildChunk);
		}
		commands.entity(e).insert(RebuildChunk);
	}
}