started work on right click menus

directory_tree_custom.py

@@ -0,0 +1,31 @@
+from textual.widgets import DirectoryTree, Rule
+from textual.widgets.tree import TreeNode
+from textual.events import MouseDown
+from context_menu import ContextMenu, NoSelectStatic
+
+
+class CustomDirectoryTree(DirectoryTree):
+	def __init__(self, path, *, name = None, id = None, classes = None, disabled = False):
+		super().__init__(path, name=name, id=id, classes=classes, disabled=disabled)
+		self.right_clicked_node: TreeNode | None = None
+
+	def context_menu_chosen(self, result):
+		self.right_clicked_node = None
+
+	def on_mouse_down(self, event: MouseDown):
+		if event.button != 3 or not "line" in event.style.meta:
+			return
+		selected_node = self.get_node_at_line(event.style.meta["line"])
+		self.right_clicked_node = selected_node
+
+		options = None
+		if self._safe_is_dir(self.right_clicked_node.data.path):
+			options = ["New Folder", "New File", NoSelectStatic(f'[d]{"-" * 17}[/]'), "Delete", "Rename", "Open"]
+		else:
+			options = ["Delete", "Rename", "Open"]
+		
+		self.app.push_screen(ContextMenu(
+			[NoSelectStatic(f"[b]{self.right_clicked_node.label}[/]"), NoSelectStatic(f'[d]{"-" * 17}[/]')] + options,
+			event.screen_offset
+		), self.context_menu_chosen)
+