use std::sync::Arc;

use glam::Vec2;
use taffy::{AlignItems, JustifyContent, prelude::length};

use crate::{
	animation::{Animation, AnimationEasing},
	drawing::{self, Color},
	event::{EventListener, WidgetCallback},
	layout::{Layout, WidgetID},
	renderer_vk::text::{FontWeight, TextStyle},
	widget::{
		rectangle::{Rectangle, RectangleParams},
		text::{TextLabel, TextParams},
		util::WLength,
	},
};

pub struct Params<'a> {
	pub text: &'a str,
	pub color: drawing::Color,
	pub size: Vec2,
	pub text_style: TextStyle,
}

impl Default for Params<'_> {
	fn default() -> Self {
		Self {
			text: "Text",
			color: drawing::Color::new(1.0, 1.0, 1.0, 1.0),
			size: Vec2::new(128.0, 32.0),
			text_style: TextStyle::default(),
		}
	}
}

pub struct Button {
	color: drawing::Color,
	pub body: WidgetID,    // Rectangle
	pub text_id: WidgetID, // Text
	text_node: taffy::NodeId,
}

impl Button {
	pub fn set_text<'a, C>(&self, callback_data: &mut C, text: &str)
	where
		C: WidgetCallback<'a>,
	{
		callback_data.call_on_widget(self.text_id, |label: &mut TextLabel| {
			label.set_text(text);
		});
		callback_data.mark_redraw();
		callback_data.mark_dirty(self.text_node);
	}
}

fn anim_hover_in(button: Arc<Button>, widget_id: WidgetID) -> Animation {
	Animation::new(
		widget_id,
		10,
		AnimationEasing::OutQuad,
		Box::new(move |data| {
			let rect = data.obj.get_as_mut::<Rectangle>();
			let brightness = data.pos * 0.5;
			rect.params.color.r = button.color.r + brightness;
			rect.params.color.g = button.color.g + brightness;
			rect.params.color.b = button.color.b + brightness;
			rect.params.border_color = Color::new(1.0, 1.0, 1.0, 1.0);
			rect.params.border = 1.0 + data.pos;
			data.needs_redraw = true;
		}),
	)
}

fn anim_hover_out(button: Arc<Button>, widget_id: WidgetID) -> Animation {
	Animation::new(
		widget_id,
		15,
		AnimationEasing::OutQuad,
		Box::new(move |data| {
			let rect = data.obj.get_as_mut::<Rectangle>();
			let brightness = (1.0 - data.pos) * 0.5;
			rect.params.color.r = button.color.r + brightness;
			rect.params.color.g = button.color.g + brightness;
			rect.params.color.b = button.color.b + brightness;
			rect.params.border_color = Color::new(1.0, 1.0, 1.0, 1.0);
			rect.params.border = 1.0 + (1.0 - data.pos) * 2.0;
			data.needs_redraw = true;
		}),
	)
}

pub fn construct(
	layout: &mut Layout,
	parent: WidgetID,
	params: Params,
) -> anyhow::Result<Arc<Button>> {
	let (rect_id, _) = layout.add_child(
		parent,
		Rectangle::create(RectangleParams {
			color: params.color,
			round: WLength::Units(4.0),
			..Default::default()
		})?,
		taffy::Style {
			size: taffy::Size {
				width: length(params.size.x),
				height: length(params.size.y),
			},
			align_items: Some(AlignItems::Center),
			justify_content: Some(JustifyContent::Center),
			padding: length(1.0),
			..Default::default()
		},
	)?;

	let light_text = (params.color.r + params.color.g + params.color.b) < 1.5;

	let (text_id, text_node) = layout.add_child(
		rect_id,
		TextLabel::create(TextParams {
			content: String::from(params.text),
			style: TextStyle {
				weight: Some(FontWeight::Bold),
				color: Some(if light_text {
					Color::new(1.0, 1.0, 1.0, 1.0)
				} else {
					Color::new(0.0, 0.0, 0.0, 1.0)
				}),
				..params.text_style
			},
		})?,
		taffy::Style {
			..Default::default()
		},
	)?;

	let mut widget = layout.widget_map.get(rect_id).unwrap().lock().unwrap();

	let button = Arc::new(Button {
		body: rect_id,
		color: params.color,
		text_id,
		text_node,
	});

	// Highlight background on mouse enter
	{
		let button = button.clone();
		widget.add_event_listener(EventListener::MouseEnter(Box::new(move |data, _| {
			data
				.animations
				.push(anim_hover_in(button.clone(), data.widget_id));
		})));
	}

	// Bring back old color on mouse leave
	{
		let button = button.clone();
		widget.add_event_listener(EventListener::MouseLeave(Box::new(move |data, _| {
			data
				.animations
				.push(anim_hover_out(button.clone(), data.widget_id));
		})));
	}

	Ok(button)
}