SpaceAnalyzer: Added context menu for nodes

Currently supports 3 actions: Open, Copy Path, and Delete.

Userland/Applications/SpaceAnalyzer/CMakeLists.txt

@@ -7,4 +7,4 @@ set(SOURCES
 )
 
 serenity_app(SpaceAnalyzer ICON app-space-analyzer)
-target_link_libraries(SpaceAnalyzer LibGfx LibGUI)
+target_link_libraries(SpaceAnalyzer LibDesktop LibGfx LibGUI)

Userland/Applications/SpaceAnalyzer/TreeMapWidget.cpp

@@ -312,6 +312,8 @@ void TreeMapWidget::mousedown_event(GUI::MouseEvent& event)
 
 void TreeMapWidget::doubleclick_event(GUI::MouseEvent& event)
 {
+ if (event.button() != GUI::MouseButton::Left)
+ return;
  const TreeMapNode* node = path_node(m_viewpoint);
  if (node && !node_is_leaf(*node)) {
  Vector<int> path = path_to_position(event.position());
@@ -341,6 +343,12 @@ void TreeMapWidget::mousewheel_event(GUI::MouseEvent& event)
  }
 }
 
+void TreeMapWidget::context_menu_event(GUI::ContextMenuEvent& context_menu_event)
+{
+ if (on_context_menu_request)
+ on_context_menu_request(context_menu_event);
+}
+
 void TreeMapWidget::set_tree(RefPtr<TreeMap> tree)
 {
  m_tree = tree;

Userland/Applications/SpaceAnalyzer/TreeMapWidget.h

@@ -50,6 +50,7 @@ class TreeMapWidget final : public GUI::Frame {
 public:
  virtual ~TreeMapWidget() override;
  Function<void()> on_path_change;
+ Function<void(GUI::ContextMenuEvent&)> on_context_menu_request;
  size_t path_size() const;
  const TreeMapNode* path_node(size_t n) const;
  size_t viewpoint() const;
@@ -62,6 +63,7 @@ class TreeMapWidget final : public GUI::Frame {
  virtual void mousedown_event(GUI::MouseEvent&) override;
  virtual void doubleclick_event(GUI::MouseEvent&) override;
  virtual void mousewheel_event(GUI::MouseEvent&) override;
+ virtual void context_menu_event(GUI::ContextMenuEvent&) override;
 
  bool rect_can_contain_children(const Gfx::IntRect& rect) const;
  bool rect_can_contain_label(const Gfx::IntRect& rect) const;

Userland/Applications/SpaceAnalyzer/main.cpp

@@ -27,15 +27,19 @@
 #include <AK/Queue.h>
 #include <AK/QuickSort.h>
 #include <AK/RefCounted.h>
+#include <AK/URL.h>
 #include <Applications/SpaceAnalyzer/SpaceAnalyzerGML.h>
 #include <LibCore/DirIterator.h>
 #include <LibCore/File.h>
+#include <LibDesktop/Launcher.h>
 #include <LibGUI/AboutDialog.h>
 #include <LibGUI/Application.h>
 #include <LibGUI/BreadcrumbBar.h>
+#include <LibGUI/Clipboard.h>
 #include <LibGUI/Icon.h>
 #include <LibGUI/Menu.h>
 #include <LibGUI/MenuBar.h>
+#include <LibGUI/MessageBox.h>
 #include <LibGUI/StatusBar.h>
 #include <sys/stat.h>
 
@@ -241,6 +245,28 @@ static void analyze(RefPtr<Tree> tree, SpaceAnalyzer::TreeMapWidget& treemapwidg
  treemapwidget.set_tree(tree);
 }
 
+static bool is_removable(const String& absolute_path)
+{
+ ASSERT(!absolute_path.is_empty());
+ int access_result = access(absolute_path.characters(), W_OK);
+ if (access_result != 0 && errno != EACCES)
+ perror("access");
+ return access_result == 0;
+}
+
+static String get_absolute_path_to_selected_node(const SpaceAnalyzer::TreeMapWidget& treemapwidget, bool include_last_node = true)
+{
+ StringBuilder path_builder;
+ for (size_t k = 0; k < treemapwidget.path_size() - (include_last_node ? 0 : 1); k++) {
+ if (k != 0) {
+ path_builder.append('/');
+ }
+ const SpaceAnalyzer::TreeMapNode* node = treemapwidget.path_node(k);
+ path_builder.append(node->name());
+ }
+ return path_builder.build();
+}
+
 int main(int argc, char* argv[])
 {
  auto app = GUI::Application::construct(argc, argv);
@@ -274,6 +300,57 @@ int main(int argc, char* argv[])
  help_menu.add_action(GUI::CommonActions::make_about_action(APP_NAME, app_icon, window));
  app->set_menubar(move(menubar));
 
+ // Configure the nodes context menu.
+ auto open_folder_action = GUI::Action::create("Open Folder", { Mod_Ctrl, Key_O }, Gfx::Bitmap::load_from_file("/res/icons/16x16/open.png"), [&](auto&) {
+ Desktop::Launcher::open(URL::create_with_file_protocol(get_absolute_path_to_selected_node(treemapwidget)));
+ });
+ auto open_containing_folder_action = GUI::Action::create("Open Containing Folder", { Mod_Ctrl, Key_O }, Gfx::Bitmap::load_from_file("/res/icons/16x16/open.png"), [&](auto&) {
+ Desktop::Launcher::open(URL::create_with_file_protocol(get_absolute_path_to_selected_node(treemapwidget, false)));
+ });
+ auto copy_path_action = GUI::Action::create("Copy Path to Clipboard", { Mod_Ctrl, Key_C }, Gfx::Bitmap::load_from_file("/res/icons/16x16/edit-copy.png"), [&](auto&) {
+ GUI::Clipboard::the().set_plain_text(get_absolute_path_to_selected_node(treemapwidget));
+ });
+ auto delete_action = GUI::CommonActions::make_delete_action([&](auto&) {
+ String selected_node_path = get_absolute_path_to_selected_node(treemapwidget);
+ bool try_again = true;
+ while (try_again) {
+ try_again = false;
+
+ auto deletion_result = Core::File::remove(selected_node_path, Core::File::RecursionMode::Allowed, true);
+ if (deletion_result.is_error()) {
+ auto retry_message_result = GUI::MessageBox::show(window,
+ String::formatted("Failed to delete \"{}\": {}. Retry?",
+ deletion_result.error().file,
+ deletion_result.error().error_code.string()),
+ "Deletion failed",
+ GUI::MessageBox::Type::Error,
+ GUI::MessageBox::InputType::YesNo);
+ if (retry_message_result == GUI::MessageBox::ExecYes) {
+ try_again = true;
+ }
+ } else {
+ GUI::MessageBox::show(window,
+ String::formatted("Successfuly deleted \"{}\".", selected_node_path),
+ "Deletion completed",
+ GUI::MessageBox::Type::Information,
+ GUI::MessageBox::InputType::OK);
+ }
+ }
+
+ // TODO: Refreshing data always causes resetting the viewport back to "/".
+ // It would be great if we found a way to preserve viewport across refreshes.
+ analyze(tree, treemapwidget, statusbar);
+ });
+ // TODO: Both these menus could've been implemented as one, but it's impossible to change action text after it's shown once.
+ auto folder_node_context_menu = GUI::Menu::construct();
+ folder_node_context_menu->add_action(*open_folder_action);
+ folder_node_context_menu->add_action(*copy_path_action);
+ folder_node_context_menu->add_action(*delete_action);
+ auto file_node_context_menu = GUI::Menu::construct();
+ file_node_context_menu->add_action(*open_containing_folder_action);
+ file_node_context_menu->add_action(*copy_path_action);
+ file_node_context_menu->add_action(*delete_action);
+
  // Configure event handlers.
  breadcrumbbar.on_segment_click = [&](size_t index) {
  ASSERT(index < treemapwidget.path_size());
@@ -291,6 +368,17 @@ int main(int argc, char* argv[])
  }
  breadcrumbbar.set_selected_segment(treemapwidget.viewpoint());
  };
+ treemapwidget.on_context_menu_request = [&](const GUI::ContextMenuEvent& event) {
+ String selected_node_path = get_absolute_path_to_selected_node(treemapwidget);
+ if (selected_node_path.is_empty())
+ return;
+ delete_action->set_enabled(is_removable(selected_node_path));
+ if (Core::File::is_directory(selected_node_path)) {
+ folder_node_context_menu->popup(event.screen_position());
+ } else {
+ file_node_context_menu->popup(event.screen_position());
+ }
+ };
 
  // At startup automatically do an analysis of root.
  analyze(tree, treemapwidget, statusbar);