mirror of
https://github.com/RGBCube/serenity
synced 2025-05-20 11:45:06 +00:00
88 lines
3.2 KiB
C++
88 lines
3.2 KiB
C++
/*
|
|
* Copyright (c) 2022, Samuel Bowman <sam@sambowman.tech>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <Applications/PartitionEditor/PartitionEditorWindowGML.h>
|
|
#include <Applications/PartitionEditor/PartitionModel.h>
|
|
#include <LibCore/DirIterator.h>
|
|
#include <LibCore/System.h>
|
|
#include <LibGUI/Application.h>
|
|
#include <LibGUI/ComboBox.h>
|
|
#include <LibGUI/ItemListModel.h>
|
|
#include <LibGUI/Menu.h>
|
|
#include <LibGUI/MessageBox.h>
|
|
#include <LibGUI/TableView.h>
|
|
#include <unistd.h>
|
|
|
|
static Vector<String> get_device_paths()
|
|
{
|
|
auto device_paths = Vector<String>();
|
|
Core::DirIterator iterator("/dev", Core::DirIterator::SkipParentAndBaseDir);
|
|
while (iterator.has_next()) {
|
|
auto path = iterator.next_full_path();
|
|
if (Core::File::is_block_device(path))
|
|
device_paths.append(path);
|
|
}
|
|
return device_paths;
|
|
}
|
|
|
|
ErrorOr<int> serenity_main(Main::Arguments arguments)
|
|
{
|
|
TRY(Core::System::pledge("stdio recvfd sendfd rpath unix"));
|
|
|
|
auto app = TRY(GUI::Application::try_create(arguments));
|
|
|
|
TRY(Core::System::pledge("stdio recvfd sendfd rpath"));
|
|
TRY(Core::System::unveil("/dev", "r"));
|
|
TRY(Core::System::unveil("/res", "r"));
|
|
TRY(Core::System::unveil(nullptr, nullptr));
|
|
|
|
// FIXME: PartitionEditor needs its own icon.
|
|
auto app_icon = TRY(GUI::Icon::try_create_default_icon("app-space-analyzer"sv));
|
|
|
|
auto window = TRY(GUI::Window::try_create());
|
|
window->set_title("Partition Editor");
|
|
window->resize(640, 400);
|
|
window->set_icon(app_icon.bitmap_for_size(16));
|
|
|
|
if (getuid() != 0) {
|
|
auto error_message = "PartitionEditor must be run as root in order to open raw block devices and read partition tables."sv;
|
|
GUI::MessageBox::show_error(window, error_message);
|
|
return Error::from_string_view(error_message);
|
|
}
|
|
|
|
auto widget = TRY(window->try_set_main_widget<GUI::Widget>());
|
|
widget->load_from_gml(partition_editor_window_gml);
|
|
|
|
auto device_paths = get_device_paths();
|
|
|
|
auto partition_model = PartitionEditor::PartitionModel::create();
|
|
TRY(partition_model->set_device_path(device_paths.first()));
|
|
|
|
auto& device_combobox = *widget->find_descendant_of_type_named<GUI::ComboBox>("device_combobox");
|
|
device_combobox.set_model(GUI::ItemListModel<String>::create(device_paths));
|
|
device_combobox.set_only_allow_values_from_model(true);
|
|
device_combobox.set_selected_index(0);
|
|
device_combobox.on_change = [&](auto const& path, auto const&) {
|
|
auto result = partition_model->set_device_path(path);
|
|
if (result.is_error())
|
|
GUI::MessageBox::show_error(window, String::formatted("No partition table found for device {}", path));
|
|
};
|
|
|
|
auto& partition_table_view = *widget->find_descendant_of_type_named<GUI::TableView>("partition_table_view");
|
|
partition_table_view.set_model(partition_model);
|
|
partition_table_view.set_focus(true);
|
|
|
|
auto& file_menu = window->add_menu("&File");
|
|
file_menu.add_action(GUI::CommonActions::make_quit_action([&](auto&) {
|
|
app->quit();
|
|
}));
|
|
|
|
auto help_menu = TRY(window->try_add_menu("&Help"));
|
|
TRY(help_menu->try_add_action(GUI::CommonActions::make_about_action("Partition Editor", app_icon, window)));
|
|
|
|
window->show();
|
|
return app->exec();
|
|
}
|