RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-05-31 13:48:12 +00:00
Code Issues Activity Actions

SpaceAnalyzer: Make TreeMapWidget responsible for filesystem analysis

Browse source 
Also, the Tree object was only ever used by the TreeMapWidget, so its
creation can happen inside `analyze()`, fixing the memory leak issue.
Plus, it doesn't have to be RefCounted.
This commit is contained in:
Sam Atkins 2023-02-03 14:00:00 +00:00 • committed by Andreas Kling
parent 98e9ee07e3
commit 1ec59cc52a
4 changed files with 82 additions and 93 deletions
Download patch file Download diff file Expand all files Collapse all files

87
Userland/Applications/SpaceAnalyzer/main.cpp
View file

@ -5,20 +5,13 @@
* SPDX-License-Identifier: BSD-2-Clause
*/
#include "ProgressWindow.h"
#include "Tree.h"
#include "TreeMapWidget.h"
#include <AK/Error.h>
#include <AK/LexicalPath.h>
#include <AK/Queue.h>
#include <AK/QuickSort.h>
#include <AK/String.h>
#include <AK/StringView.h>
#include <AK/URL.h>
#include <Applications/SpaceAnalyzer/SpaceAnalyzerGML.h>
#include <LibCore/File.h>
#include <LibCore/IODevice.h>
#include <LibCore/Stream.h>
#include <LibDesktop/Launcher.h>
#include <LibGUI/Application.h>
#include <LibGUI/BoxLayout.h>
@ -26,7 +19,6 @@
#include <LibGUI/Clipboard.h>
#include <LibGUI/FileIconProvider.h>
#include <LibGUI/Icon.h>
#include <LibGUI/Label.h>
#include <LibGUI/Menu.h>
#include <LibGUI/Menubar.h>
#include <LibGUI/MessageBox.h>
@ -36,73 +28,6 @@
static constexpr auto APP_NAME = "Space Analyzer"sv;
static ErrorOr<void> fill_mounts(Vector<MountInfo>& output)
{
// Output info about currently mounted filesystems.
auto file = TRY(Core::Stream::File::open("/sys/kernel/df"sv, Core::Stream::OpenMode::Read));
auto content = TRY(file->read_until_eof());
auto json = TRY(JsonValue::from_string(content));
TRY(json.as_array().try_for_each([&output](JsonValue const& value) -> ErrorOr<void> {
auto& filesystem_object = value.as_object();
MountInfo mount_info;
mount_info.mount_point = filesystem_object.get_deprecated_string("mount_point"sv).value_or({});
mount_info.source = filesystem_object.get_deprecated_string("source"sv).value_or("none");
TRY(output.try_append(mount_info));
return {};
}));
return {};
}
static ErrorOr<void> analyze(RefPtr<Tree> tree, SpaceAnalyzer::TreeMapWidget& treemapwidget, GUI::Statusbar& statusbar)
{
statusbar.set_text("");
auto progress_window = TRY(ProgressWindow::try_create(APP_NAME));
progress_window->show();
// Build an in-memory tree mirroring the filesystem and for each node
// calculate the sum of the file size for all its descendants.
Vector<MountInfo> mounts;
TRY(fill_mounts(mounts));
auto errors = tree->root().populate_filesize_tree(mounts, [&](size_t processed_file_count) {
progress_window->update_progress_label(processed_file_count);
});
progress_window->close();
// Display an error summary in the statusbar.
if (!errors.is_empty()) {
StringBuilder builder;
bool first = true;
builder.append("Some directories were not analyzed: "sv);
for (auto& key : errors.keys()) {
if (!first) {
builder.append(", "sv);
}
auto const* error = strerror(key);
builder.append({ error, strlen(error) });
builder.append(" ("sv);
int value = errors.get(key).value();
builder.append(DeprecatedString::number(value));
if (value == 1) {
builder.append(" time"sv);
} else {
builder.append(" times"sv);
}
builder.append(')');
first = false;
}
statusbar.set_text(builder.to_deprecated_string());
} else {
statusbar.set_text("No errors");
}
treemapwidget.set_tree(tree);
return {};
}
static DeprecatedString get_absolute_path_to_selected_node(SpaceAnalyzer::TreeMapWidget const& treemapwidget, bool include_last_node = true)
{
StringBuilder path_builder;
@ -120,8 +45,6 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
{
auto app = TRY(GUI::Application::try_create(arguments));
RefPtr<Tree> tree = adopt_ref(*new Tree(""));
// Configure application window.
auto app_icon = GUI::Icon::default_icon("app-space-analyzer"sv);
auto window = TRY(GUI::Window::try_create());
@ -141,9 +64,7 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
auto file_menu = TRY(window->try_add_menu("&File"));
TRY(file_menu->try_add_action(GUI::Action::create("&Analyze", [&](auto&) {
// FIXME: Just modify the tree in memory instead of traversing the entire file system
// FIXME: Dispose of the old tree
auto new_tree = adopt_ref(*new Tree(""));
if (auto result = analyze(new_tree, treemapwidget, statusbar); result.is_error()) {
if (auto result = treemapwidget.analyze(statusbar); result.is_error()) {
GUI::MessageBox::show_error(window, DeprecatedString::formatted("{}", result.error()));
}
})));
@ -197,9 +118,7 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
}
}
// FIXME: Dispose of the old tree
auto new_tree = adopt_ref(*new Tree(""));
if (auto result = analyze(new_tree, treemapwidget, statusbar); result.is_error()) {
if (auto result = treemapwidget.analyze(statusbar); result.is_error()) {
GUI::MessageBox::show_error(window, DeprecatedString::formatted("{}", result.error()));
}
});
@ -257,7 +176,7 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
};
// At startup automatically do an analysis of root.
TRY(analyze(tree, treemapwidget, statusbar));
TRY(treemapwidget.analyze(statusbar));
window->show();
return app->exec();