32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
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)
|
|
|