use crate::{
	core::client::INTERNAL_CLIENT,
	nodes::{
		input::{tip::Tip, InputMethod, InputType},
		spatial::Spatial,
		Node,
	},
};
use color_eyre::eyre::Result;
use glam::{Mat4, Vec2, Vec3};
use serde::{Deserialize, Serialize};
use stardust_xr::values::Datamap;
use std::sync::Arc;
use stereokit::{
	named_colors::WHITE, ButtonState, Handed, Model, RenderLayer, StereoKitDraw,
	StereoKitMultiThread,
};

#[derive(Default, Deserialize, Serialize)]
struct ControllerDatamap {
	select: f32,
	grab: f32,
	scroll: Vec2,
}

pub struct SkController {
	_node: Arc<Node>,
	input: Arc<InputMethod>,
	model: Model,
	handed: Handed,
	datamap: ControllerDatamap,
}
impl SkController {
	pub fn new(sk: &impl StereoKitMultiThread, handed: Handed) -> Result<Self> {
		let _node = Node::create_parent_name(
			&INTERNAL_CLIENT,
			"",
			if handed == Handed::Left {
				"controller_left"
			} else {
				"controller_right"
			},
			false,
		)
		.add_to_scenegraph()?;
		Spatial::add_to(&_node, None, Mat4::IDENTITY, false);
		let model = sk.model_create_mem("cursor.glb", include_bytes!("cursor.glb"), None)?;
		let tip = InputType::Tip(Tip::default());
		let input = InputMethod::add_to(&_node, tip, None)?;
		Ok(SkController {
			_node,
			input,
			handed,
			model,
			datamap: Default::default(),
		})
	}
	pub fn update(&mut self, sk: &impl StereoKitDraw) {
		let controller = sk.input_controller(self.handed);
		*self.input.enabled.lock() = controller.tracked.contains(ButtonState::ACTIVE);
		if *self.input.enabled.lock() {
			let world_transform = Mat4::from_rotation_translation(
				controller.aim.orientation,
				controller.aim.position,
			);
			sk.model_draw(
				&self.model,
				world_transform * Mat4::from_scale(Vec3::ONE * 0.02),
				WHITE,
				RenderLayer::LAYER0,
			);
			self.input.spatial.set_local_transform(world_transform);
		}
		self.datamap.select = controller.trigger;
		self.datamap.grab = controller.grip;
		self.datamap.scroll = controller.stick;
		*self.input.datamap.lock() = Datamap::from_typed(&self.datamap).ok();
	}
}