blob: 8bf735f36992860d836d28caa88e5cade79c70bd (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
/*
* 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<DeprecatedString> get_device_paths()
{
auto device_paths = Vector<DeprecatedString>();
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));
auto app_icon = TRY(GUI::Icon::try_create_default_icon("app-partition-editor"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->set_main_widget<GUI::Widget>());
TRY(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<DeprecatedString>::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, DeprecatedString::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_command_palette_action(window)));
TRY(help_menu->try_add_action(GUI::CommonActions::make_about_action("Partition Editor", app_icon, window)));
window->show();
return app->exec();
}
|
