use crate::application::generate_uuid;
use crate::consts::{ROTATE_SNAP_ANGLE, SELECTION_TOLERANCE};
use crate::messages::frontend::utility_types::MouseCursorIcon;
use crate::messages::input_mapper::utility_types::input_keyboard::{Key, MouseMotion};
use crate::messages::input_mapper::utility_types::input_mouse::ViewportPosition;
use crate::messages::layout::utility_types::layout_widget::{Layout, LayoutGroup, PropertyHolder, Widget, WidgetHolder, WidgetLayout};
use crate::messages::layout::utility_types::misc::LayoutTarget;
use crate::messages::layout::utility_types::widgets::assist_widgets::{PivotAssist, PivotPosition};
use crate::messages::layout::utility_types::widgets::button_widgets::{IconButton, PopoverButton};
use crate::messages::layout::utility_types::widgets::input_widgets::{DropdownEntryData, DropdownInput};
use crate::messages::layout::utility_types::widgets::label_widgets::{Separator, SeparatorDirection, SeparatorType};
use crate::messages::portfolio::document::utility_types::misc::{AlignAggregate, AlignAxis, FlipAxis};
use crate::messages::portfolio::document::utility_types::transformation::Selected;
use crate::messages::prelude::*;
use crate::messages::tool::common_functionality::path_outline::*;
use crate::messages::tool::common_functionality::pivot::Pivot;
use crate::messages::tool::common_functionality::snapping::{self, SnapManager};
use crate::messages::tool::common_functionality::transformation_cage::*;
use crate::messages::tool::utility_types::{EventToMessageMap, Fsm, ToolActionHandlerData, ToolMetadata, ToolTransition, ToolType};
use crate::messages::tool::utility_types::{HintData, HintGroup, HintInfo};

use document_legacy::document::Document;
use document_legacy::intersection::Quad;
use document_legacy::layers::layer_info::{Layer, LayerDataType};
use document_legacy::LayerId;
use document_legacy::Operation;

use glam::{DAffine2, DVec2};
use serde::{Deserialize, Serialize};
use std::fmt;

#[derive(Default)]
pub struct SelectTool {
	fsm_state: SelectToolFsmState,
	tool_data: SelectToolData,
}

#[allow(dead_code)]
#[derive(Default)]
pub struct SelectOptions {
	nested_selection_behavior: NestedSelectionBehavior,
}

#[remain::sorted]
#[derive(PartialEq, Eq, Clone, Debug, Hash, Serialize, Deserialize, specta::Type)]
pub enum SelectOptionsUpdate {
	NestedSelectionBehavior(NestedSelectionBehavior),
}

#[derive(Default, PartialEq, Eq, Clone, Copy, Debug, Hash, Serialize, Deserialize, specta::Type)]
pub enum NestedSelectionBehavior {
	#[default]
	Deepest,
	Shallowest,
}

impl fmt::Display for NestedSelectionBehavior {
	fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
		match self {
			NestedSelectionBehavior::Deepest => write!(f, "Deep Select"),
			NestedSelectionBehavior::Shallowest => write!(f, "Shallow Select"),
		}
	}
}

#[remain::sorted]
#[impl_message(Message, ToolMessage, Select)]
#[derive(PartialEq, Eq, Clone, Debug, Serialize, Deserialize, specta::Type)]
pub enum SelectToolMessage {
	// Standard messages
	#[remain::unsorted]
	Abort,
	#[remain::unsorted]
	DocumentIsDirty,
	#[remain::unsorted]
	SelectionChanged,

	// Tool-specific messages
	Align {
		axis: AlignAxis,
		aggregate: AlignAggregate,
	},
	DragStart {
		add_to_selection: Key,
		select_deepest: Key,
	},
	DragStop {
		remove_from_selection: Key,
	},
	EditLayer,
	Enter,
	FlipHorizontal,
	FlipVertical,
	PointerMove {
		axis_align: Key,
		snap_angle: Key,
		center: Key,
		duplicate: Key,
	},
	SelectOptions(SelectOptionsUpdate),
	SetPivot {
		position: PivotPosition,
	},
}

impl ToolMetadata for SelectTool {
	fn icon_name(&self) -> String {
		"GeneralSelectTool".into()
	}
	fn tooltip(&self) -> String {
		"Select Tool".into()
	}
	fn tool_type(&self) -> crate::messages::tool::utility_types::ToolType {
		ToolType::Select
	}
}

impl PropertyHolder for SelectTool {
	fn properties(&self) -> Layout {
		let layer_selection_behavior_entries = [NestedSelectionBehavior::Deepest, NestedSelectionBehavior::Shallowest]
			.iter()
			.map(|mode| {
				DropdownEntryData::new(mode.to_string())
					.value(mode.to_string())
					.on_update(move |_| SelectToolMessage::SelectOptions(SelectOptionsUpdate::NestedSelectionBehavior(*mode)).into())
			})
			.collect();

		Layout::WidgetLayout(WidgetLayout::new(vec![LayoutGroup::Row {
			widgets: vec![
				DropdownInput::new(vec![layer_selection_behavior_entries])
					.selected_index(Some((self.tool_data.nested_selection_behavior == NestedSelectionBehavior::Shallowest) as u32))
					.tooltip("Choose if clicking nested layers directly selects the deepest, or selects the shallowest and deepens by double clicking")
					.widget_holder(),
				WidgetHolder::related_separator(),
				// We'd like this widget to hide and show itself whenever the transformation cage is active or inactive (i.e. when no layers are selected)
				PivotAssist::new(self.tool_data.pivot.to_pivot_position())
					.on_update(|pivot_assist: &PivotAssist| SelectToolMessage::SetPivot { position: pivot_assist.position }.into())
					.widget_holder(),
				Separator::new(SeparatorDirection::Horizontal, SeparatorType::Section).widget_holder(),
				IconButton::new("AlignLeft", 24)
					.tooltip("Align Left")
					.on_update(|_| {
						DocumentMessage::AlignSelectedLayers {
							axis: AlignAxis::X,
							aggregate: AlignAggregate::Min,
						}
						.into()
					})
					.widget_holder(),
				IconButton::new("AlignHorizontalCenter", 24)
					.tooltip("Align Horizontal Center")
					.on_update(|_| {
						DocumentMessage::AlignSelectedLayers {
							axis: AlignAxis::X,
							aggregate: AlignAggregate::Center,
						}
						.into()
					})
					.widget_holder(),
				IconButton::new("AlignRight", 24)
					.tooltip("Align Right")
					.on_update(|_| {
						DocumentMessage::AlignSelectedLayers {
							axis: AlignAxis::X,
							aggregate: AlignAggregate::Max,
						}
						.into()
					})
					.widget_holder(),
				WidgetHolder::unrelated_separator(),
				IconButton::new("AlignTop", 24)
					.tooltip("Align Top")
					.on_update(|_| {
						DocumentMessage::AlignSelectedLayers {
							axis: AlignAxis::Y,
							aggregate: AlignAggregate::Min,
						}
						.into()
					})
					.widget_holder(),
				IconButton::new("AlignVerticalCenter", 24)
					.tooltip("Align Vertical Center")
					.on_update(|_| {
						DocumentMessage::AlignSelectedLayers {
							axis: AlignAxis::Y,
							aggregate: AlignAggregate::Center,
						}
						.into()
					})
					.widget_holder(),
				IconButton::new("AlignBottom", 24)
					.tooltip("Align Bottom")
					.on_update(|_| {
						DocumentMessage::AlignSelectedLayers {
							axis: AlignAxis::Y,
							aggregate: AlignAggregate::Max,
						}
						.into()
					})
					.widget_holder(),
				WidgetHolder::related_separator(),
				PopoverButton::new("Align", "Coming soon").widget_holder(),
				Separator::new(SeparatorDirection::Horizontal, SeparatorType::Section).widget_holder(),
				IconButton::new("FlipHorizontal", 24)
					.tooltip("Flip Horizontal")
					.on_update(|_| SelectToolMessage::FlipHorizontal.into())
					.widget_holder(),
				IconButton::new("FlipVertical", 24)
					.tooltip("Flip Vertical")
					.on_update(|_| SelectToolMessage::FlipVertical.into())
					.widget_holder(),
				WidgetHolder::related_separator(),
				WidgetHolder::new(Widget::PopoverButton(PopoverButton {
					header: "Flip".into(),
					text: "Coming soon".into(),
					..Default::default()
				})),
				Separator::new(SeparatorDirection::Horizontal, SeparatorType::Section).widget_holder(),
				IconButton::new("BooleanUnion", 24)
					.tooltip("Boolean Union (coming soon)")
					.on_update(|_| DialogMessage::RequestComingSoonDialog { issue: Some(1091) }.into())
					.widget_holder(),
				IconButton::new("BooleanSubtractFront", 24)
					.tooltip("Boolean Subtract Front (coming soon)")
					.on_update(|_| DialogMessage::RequestComingSoonDialog { issue: Some(1091) }.into())
					.widget_holder(),
				IconButton::new("BooleanSubtractBack", 24)
					.tooltip("Boolean Subtract Back (coming soon)")
					.on_update(|_| DialogMessage::RequestComingSoonDialog { issue: Some(1091) }.into())
					.widget_holder(),
				IconButton::new("BooleanIntersect", 24)
					.tooltip("Boolean Intersect (coming soon)")
					.on_update(|_| DialogMessage::RequestComingSoonDialog { issue: Some(1091) }.into())
					.widget_holder(),
				IconButton::new("BooleanDifference", 24)
					.tooltip("Boolean Difference (coming soon)")
					.on_update(|_| DialogMessage::RequestComingSoonDialog { issue: Some(1091) }.into())
					.widget_holder(),
				WidgetHolder::related_separator(),
				PopoverButton::new("Boolean", "Coming soon").widget_holder(),
			],
		}]))
	}
}

impl<'a> MessageHandler<ToolMessage, &mut ToolActionHandlerData<'a>> for SelectTool {
	fn process_message(&mut self, message: ToolMessage, responses: &mut VecDeque<Message>, tool_data: &mut ToolActionHandlerData<'a>) {
		if let ToolMessage::Select(SelectToolMessage::SelectOptions(SelectOptionsUpdate::NestedSelectionBehavior(nested_selection_behavior))) = message {
			self.tool_data.nested_selection_behavior = nested_selection_behavior;
			responses.push_back(ToolMessage::UpdateHints.into());
		}

		self.fsm_state.process_event(message, &mut self.tool_data, tool_data, &(), responses, false);

		if self.tool_data.pivot.should_refresh_pivot_position() {
			// Notify the frontend about the updated pivot position (a bit ugly to do it here not in the fsm but that doesn't have SelectTool)
			self.register_properties(responses, LayoutTarget::ToolOptions);
		}
	}

	fn actions(&self) -> ActionList {
		use SelectToolFsmState::*;

		match self.fsm_state {
			Ready => actions!(SelectToolMessageDiscriminant;
				DragStart,
				PointerMove,
				Abort,
				EditLayer,
				Enter,
			),
			_ => actions!(SelectToolMessageDiscriminant;
				DragStop,
				PointerMove,
				Abort,
				EditLayer,
				Enter,
			),
		}
	}
}

impl ToolTransition for SelectTool {
	fn event_to_message_map(&self) -> EventToMessageMap {
		EventToMessageMap {
			document_dirty: Some(SelectToolMessage::DocumentIsDirty.into()),
			tool_abort: Some(SelectToolMessage::Abort.into()),
			selection_changed: Some(SelectToolMessage::SelectionChanged.into()),
		}
	}
}

#[derive(Clone, Copy, Debug, Default, PartialEq, Eq, Hash)]
enum SelectToolFsmState {
	#[default]
	Ready,
	Dragging,
	DrawingBox,
	ResizingBounds,
	RotatingBounds,
	DraggingPivot,
}

#[derive(Clone, Debug, Default)]
struct SelectToolData {
	drag_start: ViewportPosition,
	drag_current: ViewportPosition,
	layers_dragging: Vec<Vec<LayerId>>,
	layer_selected_on_start: Option<Vec<LayerId>>,
	is_dragging: bool,
	not_duplicated_layers: Option<Vec<Vec<LayerId>>>,
	drag_box_overlay_layer: Option<Vec<LayerId>>,
	path_outlines: PathOutline,
	bounding_box_overlays: Option<BoundingBoxOverlays>,
	snap_manager: SnapManager,
	cursor: MouseCursorIcon,
	pivot: Pivot,
	nested_selection_behavior: NestedSelectionBehavior,
}

impl SelectToolData {
	fn selection_quad(&self) -> Quad {
		let bbox = self.selection_box();
		Quad::from_box(bbox)
	}

	fn selection_box(&self) -> [DVec2; 2] {
		if self.drag_current == self.drag_start {
			let tolerance = DVec2::splat(SELECTION_TOLERANCE);
			[self.drag_start - tolerance, self.drag_start + tolerance]
		} else {
			[self.drag_start, self.drag_current]
		}
	}

	/// Duplicates the currently dragging layers. Called when Alt is pressed and the layers have not yet been duplicated.
	fn start_duplicates(&mut self, document: &DocumentMessageHandler, responses: &mut VecDeque<Message>) {
		responses.push_back(DocumentMessage::DeselectAllLayers.into());

		// Take the selected layers and store them in a separate list.
		self.not_duplicated_layers = Some(self.layers_dragging.clone());

		// Duplicate each previously selected layer and select the new ones.
		for layer_path in Document::shallowest_unique_layers(self.layers_dragging.iter_mut()) {
			// Moves the original back to its starting position.
			responses.push_front(
				GraphOperationMessage::TransformChange {
					layer: layer_path.clone(),
					transform: DAffine2::from_translation(self.drag_start - self.drag_current),
					transform_in: TransformIn::Viewport,
					skip_rerender: true,
				}
				.into(),
			);

			// Copy the layers.
			// Not using the Copy message allows us to retrieve the ids of the new layers to initialize the drag.
			let layer = match document.document_legacy.layer(layer_path) {
				Ok(layer) => layer.clone(),
				Err(e) => {
					warn!("Could not access selected layer {:?}: {:?}", layer_path, e);
					continue;
				}
			};

			let layer_metadata = *document.layer_metadata(layer_path);
			*layer_path.last_mut().unwrap() = generate_uuid();

			responses.add(Operation::InsertLayer {
				layer: Box::new(layer),
				destination_path: layer_path.clone(),
				insert_index: -1,
			});
			responses.add(DocumentMessage::UpdateLayerMetadata {
				layer_path: layer_path.clone(),
				layer_metadata,
			});
		}

		// Since the selected layers have now moved back to their original transforms before the drag began, we rerender them to be displayed as if they weren't touched.
		for layer_path in self.not_duplicated_layers.iter().flatten() {
			responses.add(DocumentMessage::NodeGraphFrameGenerate { layer_path: layer_path.clone() });
		}
	}

	/// Removes the duplicated layers. Called when Alt is released and the layers have previously been duplicated.
	fn stop_duplicates(&mut self, responses: &mut VecDeque<Message>) {
		let originals = match self.not_duplicated_layers.take() {
			Some(x) => x,
			None => return,
		};

		responses.push_back(DocumentMessage::DeselectAllLayers.into());

		// Delete the duplicated layers
		for layer_path in Document::shallowest_unique_layers(self.layers_dragging.iter()) {
			responses.push_back(Operation::DeleteLayer { path: layer_path.clone() }.into());
		}

		// Move the original to under the mouse
		for layer_path in Document::shallowest_unique_layers(originals.iter()) {
			responses.push_front(
				GraphOperationMessage::TransformChange {
					layer: layer_path.clone(),
					transform: DAffine2::from_translation(self.drag_current - self.drag_start),
					transform_in: TransformIn::Viewport,
					skip_rerender: true,
				}
				.into(),
			);
		}

		// Select the originals
		responses.push_back(
			DocumentMessage::SetSelectedLayers {
				replacement_selected_layers: originals.clone(),
			}
			.into(),
		);

		self.layers_dragging = originals;
	}
}

impl Fsm for SelectToolFsmState {
	type ToolData = SelectToolData;
	type ToolOptions = ();

	fn transition(
		self,
		event: ToolMessage,
		tool_data: &mut Self::ToolData,
		ToolActionHandlerData { document, input, render_data, .. }: &mut ToolActionHandlerData,
		_tool_options: &Self::ToolOptions,
		responses: &mut VecDeque<Message>,
	) -> Self {
		use SelectToolFsmState::*;
		use SelectToolMessage::*;

		if let ToolMessage::Select(event) = event {
			match (self, event) {
				(_, DocumentIsDirty | SelectionChanged) => {
					match (document.selected_visible_layers_bounding_box(render_data), tool_data.bounding_box_overlays.take()) {
						(None, Some(bounding_box_overlays)) => bounding_box_overlays.delete(responses),
						(Some(bounds), paths) => {
							let mut bounding_box_overlays = paths.unwrap_or_else(|| BoundingBoxOverlays::new(responses));

							bounding_box_overlays.bounds = bounds;
							bounding_box_overlays.transform = DAffine2::IDENTITY;

							bounding_box_overlays.transform(responses);

							tool_data.bounding_box_overlays = Some(bounding_box_overlays);
						}
						(_, _) => {}
					};

					tool_data.path_outlines.update_selected(document.selected_visible_layers(), document, responses, render_data);
					tool_data.path_outlines.intersect_test_hovered(input, document, responses, render_data);
					tool_data.pivot.update_pivot(document, render_data, responses);

					self
				}
				(_, EditLayer) => {
					// Setup required data for checking the clicked layer
					let mouse_pos = input.mouse.position;
					let tolerance = DVec2::splat(SELECTION_TOLERANCE);
					let quad = Quad::from_box([mouse_pos - tolerance, mouse_pos + tolerance]);

					// Check the last (topmost) intersection layer
					if let Some(intersect_layer_path) = document.document_legacy.intersects_quad_root(quad, render_data).last() {
						if let Ok(intersect) = document.document_legacy.layer(intersect_layer_path) {
							match tool_data.nested_selection_behavior {
								NestedSelectionBehavior::Shallowest => edit_layer_shallowest_manipulation(document, intersect_layer_path, tool_data, responses),
								NestedSelectionBehavior::Deepest => edit_layer_deepest_manipulation(intersect, responses),
							}
						}
					}

					self
				}
				(Ready, DragStart { add_to_selection, select_deepest }) => {
					tool_data.path_outlines.clear_hovered(responses);

					tool_data.drag_start = input.mouse.position;
					tool_data.drag_current = input.mouse.position;

					let dragging_bounds = tool_data.bounding_box_overlays.as_mut().and_then(|mut bounding_box| {
						let edges = bounding_box.check_selected_edges(input.mouse.position);

						bounding_box.selected_edges = edges.map(|(top, bottom, left, right)| {
							let selected_edges = SelectedEdges::new(top, bottom, left, right, bounding_box.bounds);
							bounding_box.opposite_pivot = selected_edges.calculate_pivot();
							selected_edges
						});

						edges
					});

					let rotating_bounds = tool_data
						.bounding_box_overlays
						.as_ref()
						.map(|bounding_box| bounding_box.check_rotate(input.mouse.position))
						.unwrap_or_default();

					let mut selected: Vec<_> = document.selected_visible_layers().map(|path| path.to_vec()).collect();
					let quad = tool_data.selection_quad();
					let mut intersection = document.document_legacy.intersects_quad_root(quad, render_data);

					// If the user is dragging the bounding box bounds, go into ResizingBounds mode.
					// If the user is dragging the rotate trigger, go into RotatingBounds mode.
					// If the user clicks on a layer that is in their current selection, go into the dragging mode.
					// If the user clicks on new shape, make that layer their new selection.
					// Otherwise enter the box select mode
					let state = if tool_data.pivot.is_over(input.mouse.position) {
						responses.push_back(DocumentMessage::StartTransaction.into());

						tool_data.snap_manager.start_snap(document, input, document.bounding_boxes(None, None, render_data), true, true);
						tool_data.snap_manager.add_all_document_handles(document, input, &[], &[], &[]);

						DraggingPivot
					} else if let Some(selected_edges) = dragging_bounds {
						responses.push_back(DocumentMessage::StartTransaction.into());

						let snap_x = selected_edges.2 || selected_edges.3;
						let snap_y = selected_edges.0 || selected_edges.1;

						tool_data
							.snap_manager
							.start_snap(document, input, document.bounding_boxes(Some(&selected), None, render_data), snap_x, snap_y);
						tool_data
							.snap_manager
							.add_all_document_handles(document, input, &[], &selected.iter().map(|x| x.as_slice()).collect::<Vec<_>>(), &[]);

						tool_data.layers_dragging = selected;

						if let Some(bounds) = &mut tool_data.bounding_box_overlays {
							let document = &document.document_legacy;

							let selected = &tool_data.layers_dragging.iter().collect::<Vec<_>>();
							let mut selected = Selected::new(&mut bounds.original_transforms, &mut bounds.center_of_transformation, selected, responses, document);
							bounds.center_of_transformation = selected.mean_average_of_pivots(render_data);
						}

						ResizingBounds
					} else if rotating_bounds {
						responses.push_back(DocumentMessage::StartTransaction.into());

						if let Some(bounds) = &mut tool_data.bounding_box_overlays {
							let selected = selected.iter().collect::<Vec<_>>();
							let mut selected = Selected::new(&mut bounds.original_transforms, &mut bounds.center_of_transformation, &selected, responses, &document.document_legacy);

							bounds.center_of_transformation = selected.mean_average_of_pivots(render_data);
						}

						tool_data.layers_dragging = selected;

						RotatingBounds
					} else if intersection.last().map(|last| selected.contains(last)).unwrap_or(false) {
						responses.push_back(DocumentMessage::StartTransaction.into());

						tool_data.layers_dragging = selected;

						tool_data
							.snap_manager
							.start_snap(document, input, document.bounding_boxes(Some(&tool_data.layers_dragging), None, render_data), true, true);

						Dragging
					} else {
						responses.push_back(DocumentMessage::StartTransaction.into());

						if !input.keyboard.get(add_to_selection as usize) && tool_data.nested_selection_behavior == NestedSelectionBehavior::Deepest {
							responses.push_back(DocumentMessage::DeselectAllLayers.into());
							tool_data.layers_dragging.clear();
						}

						if let Some(intersection) = intersection.pop() {
							tool_data.layer_selected_on_start = Some(intersection.clone());
							selected = vec![intersection.clone()];

							match tool_data.nested_selection_behavior {
								NestedSelectionBehavior::Shallowest => {
									drag_shallowest_manipulation(document, &mut selected, input, select_deepest, add_to_selection, tool_data, responses, intersection)
								}
								NestedSelectionBehavior::Deepest => drag_deepest_manipulation(responses, selected, tool_data, document, input, render_data),
							}
							Dragging
						} else {
							// Deselect all layers if using shallowest selection behavior
							// Necessary since for shallowest mode, we need to know the current selected layers to determine the next
							if tool_data.nested_selection_behavior == NestedSelectionBehavior::Shallowest {
								responses.push_back(DocumentMessage::DeselectAllLayers.into());
								tool_data.layers_dragging.clear();
							}
							tool_data.drag_box_overlay_layer = Some(add_bounding_box(responses));
							DrawingBox
						}
					};
					tool_data.not_duplicated_layers = None;

					state
				}
				(Dragging, PointerMove { axis_align, duplicate, .. }) => {
					tool_data.is_dragging = true;
					// TODO: This is a cheat. Break out the relevant functionality from the handler above and call it from there and here.
					responses.push_front(SelectToolMessage::DocumentIsDirty.into());

					let mouse_position = axis_align_drag(input.keyboard.get(axis_align as usize), input.mouse.position, tool_data.drag_start);

					let mouse_delta = mouse_position - tool_data.drag_current;

					let snap = tool_data
						.layers_dragging
						.iter()
						.filter_map(|path| document.document_legacy.viewport_bounding_box(path, render_data).ok()?)
						.flat_map(snapping::expand_bounds)
						.collect();

					let closest_move = tool_data.snap_manager.snap_layers(responses, document, snap, mouse_delta);
					// TODO: Cache the result of `shallowest_unique_layers` to avoid this heavy computation every frame of movement, see https://github.com/GraphiteEditor/Graphite/pull/481
					for path in Document::shallowest_unique_layers(tool_data.layers_dragging.iter()) {
						responses.push_front(
							GraphOperationMessage::TransformChange {
								layer: path.to_vec(),
								transform: DAffine2::from_translation(mouse_delta + closest_move),
								transform_in: TransformIn::Viewport,
								skip_rerender: true,
							}
							.into(),
						);
					}
					tool_data.drag_current = mouse_position + closest_move;

					if input.keyboard.get(duplicate as usize) && tool_data.not_duplicated_layers.is_none() {
						tool_data.start_duplicates(document, responses);
					} else if !input.keyboard.get(duplicate as usize) && tool_data.not_duplicated_layers.is_some() {
						tool_data.stop_duplicates(responses);
					}

					Dragging
				}
				(ResizingBounds, PointerMove { axis_align, center, .. }) => {
					if let Some(bounds) = &mut tool_data.bounding_box_overlays {
						if let Some(movement) = &mut bounds.selected_edges {
							let (center, axis_align) = (input.keyboard.get(center as usize), input.keyboard.get(axis_align as usize));

							let mouse_position = input.mouse.position;

							let snapped_mouse_position = tool_data.snap_manager.snap_position(responses, document, mouse_position);

							let (position, size) = movement.new_size(snapped_mouse_position, bounds.transform, center, bounds.center_of_transformation, axis_align);
							let (delta, mut pivot) = movement.bounds_to_scale_transform(position, size);

							let selected = &tool_data.layers_dragging.iter().collect::<Vec<_>>();
							let mut selected = Selected::new(&mut bounds.original_transforms, &mut pivot, selected, responses, &document.document_legacy);

							selected.update_transforms(delta);
						}
					}
					ResizingBounds
				}
				(RotatingBounds, PointerMove { snap_angle, .. }) => {
					if let Some(bounds) = &mut tool_data.bounding_box_overlays {
						let angle = {
							let start_offset = tool_data.drag_start - bounds.center_of_transformation;
							let end_offset = input.mouse.position - bounds.center_of_transformation;

							start_offset.angle_between(end_offset)
						};

						let snapped_angle = if input.keyboard.get(snap_angle as usize) {
							let snap_resolution = ROTATE_SNAP_ANGLE.to_radians();
							(angle / snap_resolution).round() * snap_resolution
						} else {
							angle
						};

						let delta = DAffine2::from_angle(snapped_angle);

						let selected = tool_data.layers_dragging.iter().collect::<Vec<_>>();
						let mut selected = Selected::new(&mut bounds.original_transforms, &mut bounds.center_of_transformation, &selected, responses, &document.document_legacy);

						selected.update_transforms(delta);
					}

					RotatingBounds
				}
				(DraggingPivot, PointerMove { .. }) => {
					let mouse_position = input.mouse.position;
					let snapped_mouse_position = tool_data.snap_manager.snap_position(responses, document, mouse_position);
					tool_data.pivot.set_viewport_position(snapped_mouse_position, document, render_data, responses);

					DraggingPivot
				}
				(DrawingBox, PointerMove { .. }) => {
					tool_data.drag_current = input.mouse.position;

					responses.push_front(
						DocumentMessage::Overlays(
							Operation::SetLayerTransformInViewport {
								path: tool_data.drag_box_overlay_layer.clone().unwrap(),
								transform: transform_from_box(tool_data.drag_start, tool_data.drag_current, DAffine2::IDENTITY).to_cols_array(),
							}
							.into(),
						)
						.into(),
					);
					DrawingBox
				}
				(Ready, PointerMove { .. }) => {
					let mut cursor = tool_data.bounding_box_overlays.as_ref().map_or(MouseCursorIcon::Default, |bounds| bounds.get_cursor(input, true));

					// Dragging the pivot overrules the other operations
					if tool_data.pivot.is_over(input.mouse.position) {
						cursor = MouseCursorIcon::Move;
					}

					// Generate the select outline (but not if the user is going to use the bound overlays)
					if cursor == MouseCursorIcon::Default {
						tool_data.path_outlines.intersect_test_hovered(input, document, responses, render_data);
					} else {
						tool_data.path_outlines.clear_hovered(responses);
					}

					if tool_data.cursor != cursor {
						tool_data.cursor = cursor;
						responses.push_back(FrontendMessage::UpdateMouseCursor { cursor }.into());
					}

					Ready
				}
				(Dragging, Enter) => {
					rerender_selected_layers(tool_data, responses);

					let response = match input.mouse.position.distance(tool_data.drag_start) < 10. * f64::EPSILON {
						true => DocumentMessage::Undo,
						false => DocumentMessage::CommitTransaction,
					};
					tool_data.snap_manager.cleanup(responses);
					responses.push_front(response.into());

					Ready
				}
				(Dragging, DragStop { remove_from_selection }) => {
					rerender_selected_layers(tool_data, responses);

					// Deselect layer if not snap dragging
					if !tool_data.is_dragging && input.keyboard.get(remove_from_selection as usize) && tool_data.layer_selected_on_start.is_none() {
						let quad = tool_data.selection_quad();
						let intersection = document.document_legacy.intersects_quad_root(quad, render_data);

						if let Some(path) = intersection.last() {
							// let folders: Vec<_> = (1..path.len() + 1).map(|i| &path[0..i]).collect();
							// let replacement_selected_layers: Vec<_> = document.selected_layers().filter(|&layer| !folders.contains(&layer)).map(|path| path.to_vec()).collect();
							let replacement_selected_layers: Vec<_> = document.selected_layers().filter(|&layer| !path.starts_with(&layer)).map(|path| path.to_vec()).collect();

							tool_data.layers_dragging.clear();
							tool_data.layers_dragging.append(replacement_selected_layers.clone().as_mut());

							responses.push_back(DocumentMessage::SetSelectedLayers { replacement_selected_layers }.into());
						}
					}

					tool_data.is_dragging = false;
					tool_data.layer_selected_on_start = None;

					responses.push_back(DocumentMessage::CommitTransaction.into());
					tool_data.snap_manager.cleanup(responses);

					Ready
				}
				(ResizingBounds, DragStop { .. } | Enter) => {
					rerender_selected_layers(tool_data, responses);

					let response = match input.mouse.position.distance(tool_data.drag_start) < 10. * f64::EPSILON {
						true => DocumentMessage::Undo,
						false => DocumentMessage::CommitTransaction,
					};
					responses.push_back(response.into());

					tool_data.snap_manager.cleanup(responses);

					if let Some(bounds) = &mut tool_data.bounding_box_overlays {
						bounds.original_transforms.clear();
					}

					Ready
				}
				(RotatingBounds, DragStop { .. } | Enter) => {
					rerender_selected_layers(tool_data, responses);

					let response = match input.mouse.position.distance(tool_data.drag_start) < 10. * f64::EPSILON {
						true => DocumentMessage::Undo,
						false => DocumentMessage::CommitTransaction,
					};
					responses.push_back(response.into());

					if let Some(bounds) = &mut tool_data.bounding_box_overlays {
						bounds.original_transforms.clear();
					}

					Ready
				}
				(DraggingPivot, DragStop { .. } | Enter) => {
					let response = match input.mouse.position.distance(tool_data.drag_start) < 10. * f64::EPSILON {
						true => DocumentMessage::Undo,
						false => DocumentMessage::CommitTransaction,
					};
					responses.push_back(response.into());

					tool_data.snap_manager.cleanup(responses);

					Ready
				}
				(DrawingBox, DragStop { .. } | Enter) => {
					let quad = tool_data.selection_quad();
					responses.push_front(
						DocumentMessage::AddSelectedLayers {
							additional_layers: document.document_legacy.intersects_quad_root(quad, render_data),
						}
						.into(),
					);
					responses.push_front(
						DocumentMessage::Overlays(
							Operation::DeleteLayer {
								path: tool_data.drag_box_overlay_layer.take().unwrap(),
							}
							.into(),
						)
						.into(),
					);
					Ready
				}
				(Ready, Enter) => {
					let mut selected_layers = document.selected_layers();

					if let Some(layer_path) = selected_layers.next() {
						// Check that only one layer is selected
						if selected_layers.next().is_none() {
							if let Ok(layer) = document.document_legacy.layer(layer_path) {
								if let LayerDataType::Text(_) = layer.data {
									responses.push_front(ToolMessage::ActivateTool { tool_type: ToolType::Text }.into());
									responses.push_back(TextToolMessage::EditSelected.into());
								}
							}
						}
					}

					Ready
				}
				(Dragging, Abort) => {
					rerender_selected_layers(tool_data, responses);

					tool_data.snap_manager.cleanup(responses);
					responses.push_back(DocumentMessage::Undo.into());

					tool_data.path_outlines.clear_selected(responses);
					tool_data.pivot.clear_overlays(responses);

					Ready
				}
				(_, Abort) => {
					if let Some(path) = tool_data.drag_box_overlay_layer.take() {
						responses.push_front(DocumentMessage::Overlays(Operation::DeleteLayer { path }.into()).into())
					};
					if let Some(mut bounding_box_overlays) = tool_data.bounding_box_overlays.take() {
						let selected = tool_data.layers_dragging.iter().collect::<Vec<_>>();
						let mut selected = Selected::new(
							&mut bounding_box_overlays.original_transforms,
							&mut bounding_box_overlays.opposite_pivot,
							&selected,
							responses,
							&document.document_legacy,
						);

						selected.revert_operation();

						bounding_box_overlays.delete(responses);
					}

					tool_data.path_outlines.clear_hovered(responses);
					tool_data.path_outlines.clear_selected(responses);
					tool_data.pivot.clear_overlays(responses);

					tool_data.snap_manager.cleanup(responses);
					Ready
				}
				(_, Align { axis, aggregate }) => {
					responses.push_back(DocumentMessage::AlignSelectedLayers { axis, aggregate }.into());

					self
				}
				(_, FlipHorizontal) => {
					responses.push_back(DocumentMessage::FlipSelectedLayers { flip_axis: FlipAxis::X }.into());

					self
				}
				(_, FlipVertical) => {
					responses.push_back(DocumentMessage::FlipSelectedLayers { flip_axis: FlipAxis::Y }.into());

					self
				}
				(_, SetPivot { position }) => {
					responses.push_back(DocumentMessage::StartTransaction.into());

					let pos: Option<DVec2> = position.into();
					tool_data.pivot.set_normalized_position(pos.unwrap(), document, render_data, responses);

					self
				}
				_ => self,
			}
		} else {
			self
		}
	}

	fn standard_tool_messages(&self, message: &ToolMessage, messages: &mut VecDeque<Message>, tool_data: &mut Self::ToolData) -> bool {
		// Check for standard hits or cursor events
		match message {
			ToolMessage::UpdateHints => {
				let hint_data = HintData(vec![
					HintGroup(vec![HintInfo::mouse(MouseMotion::LmbDrag, "Drag Selected")]),
					HintGroup(vec![HintInfo::keys([Key::KeyG, Key::KeyR, Key::KeyS], "Grab/Rotate/Scale Selected")]),
					HintGroup({
						let mut hints = vec![HintInfo::mouse(MouseMotion::Lmb, "Select Object"), HintInfo::keys([Key::Shift], "Extend Selection").prepend_plus()];
						if tool_data.nested_selection_behavior == NestedSelectionBehavior::Shallowest {
							hints.extend([HintInfo::keys([Key::Accel], "Deepest").prepend_plus(), HintInfo::mouse(MouseMotion::LmbDouble, "Deepen Selection")]);
						}
						hints
					}),
					HintGroup(vec![
						HintInfo::mouse(MouseMotion::LmbDrag, "Select Area"),
						HintInfo::keys([Key::Shift], "Extend Selection").prepend_plus(),
					]),
					HintGroup(vec![
						HintInfo::arrow_keys("Nudge Selected"),
						HintInfo::keys([Key::Shift], "10x").prepend_plus(),
						HintInfo::keys([Key::Alt], "Resize Corner").prepend_plus(),
						HintInfo::keys([Key::Control], "Opp. Corner").prepend_plus(),
					]),
					HintGroup(vec![
						HintInfo::keys_and_mouse([Key::Alt], MouseMotion::LmbDrag, "Move Duplicate"),
						HintInfo::keys([Key::Control, Key::KeyD], "Duplicate").add_mac_keys([Key::Command, Key::KeyD]),
					]),
				]);

				messages.push_back(FrontendMessage::UpdateInputHints { hint_data }.into());
				self.update_hints(messages);
				true
			}
			ToolMessage::UpdateCursor => {
				self.update_cursor(messages);
				true
			}
			_ => false,
		}
	}

	fn update_hints(&self, _responses: &mut VecDeque<Message>) {}

	fn update_cursor(&self, responses: &mut VecDeque<Message>) {
		responses.push_back(FrontendMessage::UpdateMouseCursor { cursor: MouseCursorIcon::Default }.into());
	}
}

fn rerender_selected_layers(tool_data: &mut SelectToolData, responses: &mut VecDeque<Message>) {
	for layer_path in &tool_data.layers_dragging {
		responses.add(DocumentMessage::NodeGraphFrameGenerate { layer_path: layer_path.clone() });
	}
}

fn rerender_duplicated_layers(tool_data: &mut SelectToolData, responses: &mut VecDeque<Message>) {
	for layer_path in tool_data.not_duplicated_layers.iter().flatten() {
		responses.add(DocumentMessage::NodeGraphFrameGenerate { layer_path: layer_path.clone() });
	}
}

// TODO: Majorly clean up these next five functions

fn drag_shallowest_manipulation(
	document: &DocumentMessageHandler,
	selected: &mut Vec<Vec<u64>>,
	input: &InputPreprocessorMessageHandler,
	select_deepest: Key,
	add_to_selection: Key,
	tool_data: &mut SelectToolData,
	responses: &mut VecDeque<Message>,
	intersection: Vec<u64>,
) {
	let layers: Vec<_> = document.selected_layers().collect();
	let incoming_layer_path_vector = selected.first().unwrap();
	let incoming_parent = *incoming_layer_path_vector.clone().first().unwrap();

	// Accel+Shift click adds the deepest layer to the selection
	if input.keyboard.get(select_deepest as usize) && input.keyboard.get(add_to_selection as usize) {
		// Checks if the incoming layer's root parent is already selected
		// If so we need to update the selected layer to the deeper of the two
		let mut layers_without_incoming_parent: Vec<Vec<u64>> = document.selected_layers().filter(|&layer| layer != [incoming_parent].as_slice()).map(|path| path.to_vec()).collect();
		if layers.contains(&&[incoming_parent].as_slice()) {
			// Add incoming layer
			tool_data.layers_dragging.clear();
			responses.push_back(DocumentMessage::DeselectAllLayers.into());
			layers_without_incoming_parent.append(selected.clone().as_mut());
			*selected = layers_without_incoming_parent;
		}

		tool_data.layers_dragging.append(selected.clone().as_mut());
		responses.push_back(DocumentMessage::AddSelectedLayers { additional_layers: selected.clone() }.into());
		return;
	}

	// Accel click selects the deepest layer directly
	if input.keyboard.get(select_deepest as usize) {
		tool_data.layers_dragging.clear();
		tool_data.layers_dragging.append(selected.clone().as_mut());
		responses.push_back(
			DocumentMessage::SetSelectedLayers {
				replacement_selected_layers: selected.clone(),
			}
			.into(),
		);
		return;
	}

	// Check whether a layer is selected for next selection calculations
	if let Some(previous_layer_path) = document.selected_layers().next() {
		let selected_layers_count = document.selected_layers().count();

		// Check if the intersected layer path is already selected
		let previous_parents: Vec<_> = (0..layers.len()).map(|i| &layers.get(i).unwrap()[..1]).collect();
		let already_selected_parent = previous_parents.contains(&&[incoming_parent].as_slice());

		let selected_layers: Vec<_> = document.selected_layers().collect();
		let mut search = previous_layer_path.to_vec();
		let mut recursive_found = false;

		// Only need to calculate if the incoming layer shares a parent with the selected layer
		if already_selected_parent {
			let mut selected_layer_path_parent = previous_layer_path.to_vec();
			let is_parent = selected_layer_path_parent.len() == 1;
			if selected_layer_path_parent.len() > 1 {
				selected_layer_path_parent = selected_layer_path_parent[..selected_layer_path_parent.len() - 1].to_vec();
			}

			while selected_layer_path_parent.len() > 0 && !is_parent && !recursive_found {
				let selected_children_layer_paths = document.document_legacy.folder_children_paths(&selected_layer_path_parent);
				for child in selected_children_layer_paths {
					if child == *incoming_layer_path_vector {
						search = child;
						recursive_found = true;
						break;
					} else if document.document_legacy.is_folder(child.clone()) {
						recursive_found = recursive_search(document, &child, incoming_layer_path_vector);
						if recursive_found {
							search = child;
							break;
						}
					}
				}
				selected_layer_path_parent = selected_layer_path_parent[..selected_layer_path_parent.len() - 1].to_vec();
			}

			// Check if new layer is already selected
			let mut already_selected = false;
			if selected_layers.contains(&search.clone().as_slice()) {
				already_selected = true;
			}

			// One layer is currently selected
			if selected_layers_count <= 1 {
				if input.keyboard.get(add_to_selection as usize) {
					if !already_selected {
						responses.push_back(
							DocumentMessage::AddSelectedLayers {
								additional_layers: vec![search.clone()],
							}
							.into(),
						);
					} else {
						tool_data.layer_selected_on_start = None;
					}
				} else {
					tool_data.layers_dragging.clear();
					responses.push_back(
						DocumentMessage::SetSelectedLayers {
							replacement_selected_layers: vec![search.clone()],
						}
						.into(),
					);
				}
				tool_data.layers_dragging.push(search);
			} else {
				// Previous selected layers with the intersect layer path appended to it
				let mut combined_layers = selected_layers.clone();
				let intersection_temp = intersection.clone();
				let intersection_temp_slice = intersection_temp.as_slice();
				combined_layers.push(intersection_temp_slice);
				let layers_iter = combined_layers.into_iter();
				let mut direct_child = document.document_legacy.shallowest_common_folder(layers_iter).unwrap().to_vec();
				// Append the sub layer to the base to create the deeper layer path
				for path in intersection_temp {
					if !direct_child.contains(&path) {
						direct_child.push(path);
						break;
					}
				}
				if input.keyboard.get(add_to_selection as usize) {
					if !already_selected {
						tool_data.layers_dragging.push(direct_child.clone());
						responses.push_back(
							DocumentMessage::AddSelectedLayers {
								additional_layers: vec![direct_child.clone()],
							}
							.into(),
						);
					} else {
						tool_data.layer_selected_on_start = None;
					}
				} else {
					tool_data.layers_dragging.push(direct_child.clone());
					responses.push_back(
						DocumentMessage::SetSelectedLayers {
							replacement_selected_layers: vec![direct_child.clone()],
						}
						.into(),
					);
				}
			}
		}
		// Incoming layer path has different parent, set selected layer to shallowest parent
		else {
			let parent_folder_id = selected.first().unwrap().first().unwrap();
			if input.keyboard.get(add_to_selection as usize) {
				responses.push_back(
					DocumentMessage::AddSelectedLayers {
						additional_layers: vec![vec![*parent_folder_id]],
					}
					.into(),
				);
			} else {
				tool_data.layers_dragging.clear();
				responses.push_back(
					DocumentMessage::SetSelectedLayers {
						replacement_selected_layers: vec![vec![*parent_folder_id]],
					}
					.into(),
				);
			}
			tool_data.layers_dragging.push(vec![*parent_folder_id]);
		}
	} else {
		// Check if new layer is already selected
		let parent_folder_id = selected.first().unwrap().first().unwrap();
		tool_data.layers_dragging.push(vec![*parent_folder_id]);
		responses.push_back(
			DocumentMessage::AddSelectedLayers {
				additional_layers: vec![vec![*parent_folder_id]],
			}
			.into(),
		);
	}
}

fn drag_deepest_manipulation(
	responses: &mut VecDeque<Message>,
	mut selected: Vec<Vec<u64>>,
	tool_data: &mut SelectToolData,
	document: &DocumentMessageHandler,
	input: &InputPreprocessorMessageHandler,
	render_data: &document_legacy::layers::RenderData,
) {
	responses.push_back(DocumentMessage::AddSelectedLayers { additional_layers: selected.clone() }.into());
	tool_data.layers_dragging.append(selected.as_mut());
	tool_data
		.snap_manager
		.start_snap(document, input, document.bounding_boxes(Some(&tool_data.layers_dragging), None, render_data), true, true);
}

fn edit_layer_shallowest_manipulation(document: &DocumentMessageHandler, intersect_layer_path: &Vec<u64>, tool_data: &mut SelectToolData, responses: &mut VecDeque<Message>) {
	// Double-clicking any layer within an already selected folder should select that layer
	// Add the first layer path not already included from the intersected to our new layer path
	let selected_layers: Vec<_> = document.selected_layers().collect();
	let incoming_parent = *intersect_layer_path.first().unwrap();
	let previous_parents: Vec<_> = (0..selected_layers.len()).map(|i| &selected_layers.get(i).unwrap()[..1]).collect();
	let mut incoming_parent_selected = false;
	if previous_parents.contains(&&[incoming_parent].as_slice()) {
		incoming_parent_selected = true;
	}
	if incoming_parent_selected {
		let mut intersected_layer_ancestors: Vec<Vec<u64>> = vec![];
		// Permutations of intersected layer
		for i in 1..intersect_layer_path.clone().len() + 1 {
			intersected_layer_ancestors.push(intersect_layer_path.clone()[..i].to_vec());
		}
		intersected_layer_ancestors.reverse();
		let mut new_layer_path: Vec<u64> = vec![];
		// Set the base layer path to the deepest layer that is currently selected
		for permutation in intersected_layer_ancestors {
			for layer in selected_layers.iter() {
				if permutation == *layer {
					new_layer_path.append(permutation.clone().as_mut());
				}
			}
		}
		// Append the sub layer to the base to create the deeper layer path
		for path in intersect_layer_path {
			if !new_layer_path.contains(path) {
				new_layer_path.push(*path);
				break;
			}
		}

		if !selected_layers.contains(&new_layer_path.as_slice()) {
			tool_data.layers_dragging.clear();
			tool_data.layers_dragging.push(new_layer_path.clone());
			responses.push_back(
				DocumentMessage::SetSelectedLayers {
					replacement_selected_layers: vec![new_layer_path],
				}
				.into(),
			);
		} else {
			responses.push_front(ToolMessage::ActivateTool { tool_type: ToolType::Path }.into());
		}
	}
}

fn edit_layer_deepest_manipulation(intersect: &Layer, responses: &mut VecDeque<Message>) {
	match &intersect.data {
		LayerDataType::Text(_) => {
			responses.push_front(ToolMessage::ActivateTool { tool_type: ToolType::Text }.into());
			responses.push_back(TextToolMessage::Interact.into());
		}
		LayerDataType::Shape(_) => {
			responses.push_front(ToolMessage::ActivateTool { tool_type: ToolType::Path }.into());
		}
		LayerDataType::NodeGraphFrame(frame) if frame.vector_data.is_some() => {
			responses.push_front(ToolMessage::ActivateTool { tool_type: ToolType::Path }.into());
		}
		_ => {}
	}
}

fn recursive_search(document: &DocumentMessageHandler, layer_path: &Vec<u64>, incoming_layer_path_vector: &Vec<u64>) -> bool {
	// TODO: fix below, then QA
	// DOUBLE CLICK BROKEN
	let layer_paths = document.document_legacy.folder_children_paths(layer_path);
	for path in layer_paths {
		if path == *incoming_layer_path_vector {
			return true;
		} else if document.document_legacy.is_folder(path.clone()) {
			if recursive_search(document, &path, incoming_layer_path_vector) {
				return true;
			}
		}
	}
	return false;
}