jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
3 * Copyright (c) 2021, Mohsan Ali <mohsan0073@gmail.com>
4 *
5 * SPDX-License-Identifier: BSD-2-Clause
6 */
7
8#include "MainWidget.h"
9#include "ViewWidget.h"
10#include <AK/URL.h>
11#include <LibConfig/Client.h>
12#include <LibCore/ArgsParser.h>
13#include <LibCore/System.h>
14#include <LibDesktop/Launcher.h>
15#include <LibGUI/Action.h>
16#include <LibGUI/ActionGroup.h>
17#include <LibGUI/Application.h>
18#include <LibGUI/BoxLayout.h>
19#include <LibGUI/Clipboard.h>
20#include <LibGUI/Desktop.h>
21#include <LibGUI/FilePicker.h>
22#include <LibGUI/Label.h>
23#include <LibGUI/Menu.h>
24#include <LibGUI/Menubar.h>
25#include <LibGUI/MessageBox.h>
26#include <LibGUI/Toolbar.h>
27#include <LibGUI/ToolbarContainer.h>
28#include <LibGUI/Window.h>
29#include <LibGfx/Bitmap.h>
30#include <LibGfx/Palette.h>
31#include <LibGfx/Rect.h>
32#include <LibMain/Main.h>
33#include <serenity.h>
34#include <string.h>
35
36using namespace ImageViewer;
37
38ErrorOr<int> serenity_main(Main::Arguments arguments)
39{
40 TRY(Core::System::pledge("stdio recvfd sendfd rpath wpath cpath unix thread"));
41
42 auto app = TRY(GUI::Application::try_create(arguments));
43
44 Config::pledge_domain("ImageViewer");
45
46 app->set_config_domain(TRY("ImageViewer"_string));
47
48 TRY(Desktop::Launcher::add_allowed_handler_with_any_url("/bin/ImageViewer"));
49 TRY(Desktop::Launcher::add_allowed_handler_with_only_specific_urls("/bin/Help", { URL::create_with_file_scheme("/usr/share/man/man1/ImageViewer.md") }));
50 TRY(Desktop::Launcher::seal_allowlist());
51
52 auto app_icon = GUI::Icon::default_icon("filetype-image"sv);
53
54 StringView path;
55 Core::ArgsParser args_parser;
56 args_parser.add_positional_argument(path, "The image file to be displayed.", "file", Core::ArgsParser::Required::No);
57 args_parser.parse(arguments);
58
59 auto window = TRY(GUI::Window::try_create());
60 window->set_double_buffering_enabled(true);
61 window->resize(300, 200);
62 window->set_icon(app_icon.bitmap_for_size(16));
63 window->set_title("Image Viewer");
64
65 auto root_widget = TRY(window->set_main_widget<MainWidget>());
66
67 auto toolbar_container = TRY(root_widget->try_add<GUI::ToolbarContainer>());
68 auto main_toolbar = TRY(toolbar_container->try_add<GUI::Toolbar>());
69
70 auto widget = TRY(root_widget->try_add<ViewWidget>());
71 if (!path.is_empty()) {
72 widget->set_path(path);
73 }
74 widget->on_scale_change = [&](float scale) {
75 if (!widget->bitmap()) {
76 window->set_title("Image Viewer");
77 return;
78 }
79
80 window->set_title(DeprecatedString::formatted("{} {} {}% - Image Viewer", widget->path(), widget->bitmap()->size().to_deprecated_string(), (int)(scale * 100)));
81
82 if (!widget->scaled_for_first_image()) {
83 widget->set_scaled_for_first_image(true);
84 widget->resize_window();
85 }
86 };
87 widget->on_drop = [&](auto& event) {
88 if (!event.mime_data().has_urls())
89 return;
90
91 auto urls = event.mime_data().urls();
92
93 if (urls.is_empty())
94 return;
95
96 window->move_to_front();
97
98 auto path = urls.first().path();
99 widget->set_path(path);
100 widget->load_from_file(path);
101
102 for (size_t i = 1; i < urls.size(); ++i) {
103 Desktop::Launcher::open(URL::create_with_file_scheme(urls[i].path().characters()), "/bin/ImageViewer");
104 }
105 };
106 widget->on_doubleclick = [&] {
107 window->set_fullscreen(!window->is_fullscreen());
108 toolbar_container->set_visible(!window->is_fullscreen());
109 widget->set_frame_thickness(window->is_fullscreen() ? 0 : 2);
110 };
111
112 // Actions
113 auto open_action = GUI::CommonActions::make_open_action(
114 [&](auto&) {
115 auto path = GUI::FilePicker::get_open_filepath(window, "Open Image");
116 if (path.has_value()) {
117 widget->set_path(path.value());
118 widget->load_from_file(path.value());
119 }
120 });
121
122 auto delete_action = GUI::CommonActions::make_delete_action(
123 [&](auto&) {
124 auto path = widget->path();
125 if (path.is_empty())
126 return;
127
128 auto msgbox_result = GUI::MessageBox::show(window,
129 DeprecatedString::formatted("Are you sure you want to delete {}?", path),
130 "Confirm deletion"sv,
131 GUI::MessageBox::Type::Warning,
132 GUI::MessageBox::InputType::OKCancel);
133
134 if (msgbox_result == GUI::MessageBox::ExecResult::Cancel)
135 return;
136
137 auto unlinked_or_error = Core::System::unlink(widget->path());
138 if (unlinked_or_error.is_error()) {
139 GUI::MessageBox::show(window,
140 DeprecatedString::formatted("unlink({}) failed: {}", path, unlinked_or_error.error()),
141 "Delete failed"sv,
142 GUI::MessageBox::Type::Error);
143
144 return;
145 }
146
147 widget->clear();
148 });
149
150 auto quit_action = GUI::CommonActions::make_quit_action(
151 [&](auto&) {
152 app->quit();
153 });
154
155 auto rotate_counterclockwise_action = GUI::CommonActions::make_rotate_counterclockwise_action([&](auto&) {
156 widget->rotate(Gfx::RotationDirection::CounterClockwise);
157 });
158
159 auto rotate_clockwise_action = GUI::CommonActions::make_rotate_clockwise_action([&](auto&) {
160 widget->rotate(Gfx::RotationDirection::Clockwise);
161 });
162
163 auto vertical_flip_action = GUI::Action::create("Flip &Vertically", { Mod_None, Key_V }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/edit-flip-vertical.png"sv)),
164 [&](auto&) {
165 widget->flip(Gfx::Orientation::Vertical);
166 });
167
168 auto horizontal_flip_action = GUI::Action::create("Flip &Horizontally", { Mod_None, Key_H }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/edit-flip-horizontal.png"sv)),
169 [&](auto&) {
170 widget->flip(Gfx::Orientation::Horizontal);
171 });
172
173 auto desktop_wallpaper_action = GUI::Action::create("Set as Desktop &Wallpaper", TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/app-display-settings.png"sv)),
174 [&](auto&) {
175 if (!GUI::Desktop::the().set_wallpaper(widget->bitmap(), widget->path())) {
176 GUI::MessageBox::show(window,
177 DeprecatedString::formatted("set_wallpaper({}) failed", widget->path()),
178 "Could not set wallpaper"sv,
179 GUI::MessageBox::Type::Error);
180 }
181 });
182
183 auto go_first_action = GUI::Action::create("&Go to First", { Mod_None, Key_Home }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/go-first.png"sv)),
184 [&](auto&) {
185 widget->navigate(ViewWidget::Directions::First);
186 });
187
188 auto go_back_action = GUI::Action::create("Go to &Previous", { Mod_None, Key_Left }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/go-back.png"sv)),
189 [&](auto&) {
190 widget->navigate(ViewWidget::Directions::Back);
191 });
192
193 auto go_forward_action = GUI::Action::create("Go to &Next", { Mod_None, Key_Right }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/go-forward.png"sv)),
194 [&](auto&) {
195 widget->navigate(ViewWidget::Directions::Forward);
196 });
197
198 auto go_last_action = GUI::Action::create("Go to &Last", { Mod_None, Key_End }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/go-last.png"sv)),
199 [&](auto&) {
200 widget->navigate(ViewWidget::Directions::Last);
201 });
202
203 auto full_screen_action = GUI::CommonActions::make_fullscreen_action(
204 [&](auto&) {
205 widget->on_doubleclick();
206 });
207
208 auto zoom_in_action = GUI::CommonActions::make_zoom_in_action(
209 [&](auto&) {
210 widget->set_scale(widget->scale() * 1.44f);
211 },
212 window);
213
214 auto reset_zoom_action = GUI::CommonActions::make_reset_zoom_action(
215 [&](auto&) {
216 widget->set_scale(1.f);
217 },
218 window);
219
220 auto fit_image_to_view_action = GUI::Action::create(
221 "Fit Image To &View", TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/fit-image-to-view.png"sv)), [&](auto&) {
222 widget->fit_content_to_view();
223 });
224
225 auto zoom_out_action = GUI::CommonActions::make_zoom_out_action(
226 [&](auto&) {
227 widget->set_scale(widget->scale() / 1.44f);
228 },
229 window);
230
231 auto hide_show_toolbar_action = GUI::Action::create("Hide/Show &Toolbar", { Mod_Ctrl, Key_T },
232 [&](auto&) {
233 toolbar_container->set_visible(!toolbar_container->is_visible());
234 });
235
236 auto copy_action = GUI::CommonActions::make_copy_action([&](auto&) {
237 if (widget->bitmap())
238 GUI::Clipboard::the().set_bitmap(*widget->bitmap());
239 });
240
241 auto nearest_neighbor_action = GUI::Action::create_checkable("&Nearest Neighbor", [&](auto&) {
242 widget->set_scaling_mode(Gfx::Painter::ScalingMode::NearestNeighbor);
243 });
244 nearest_neighbor_action->set_checked(true);
245
246 auto smooth_pixels_action = GUI::Action::create_checkable("&Smooth Pixels", [&](auto&) {
247 widget->set_scaling_mode(Gfx::Painter::ScalingMode::SmoothPixels);
248 });
249
250 auto bilinear_action = GUI::Action::create_checkable("&Bilinear", [&](auto&) {
251 widget->set_scaling_mode(Gfx::Painter::ScalingMode::BilinearBlend);
252 });
253
254 widget->on_image_change = [&](Gfx::Bitmap const* bitmap) {
255 bool should_enable_image_actions = (bitmap != nullptr);
256 bool should_enable_forward_actions = (widget->is_next_available() && should_enable_image_actions);
257 bool should_enable_backward_actions = (widget->is_previous_available() && should_enable_image_actions);
258 delete_action->set_enabled(should_enable_image_actions);
259 rotate_counterclockwise_action->set_enabled(should_enable_image_actions);
260 rotate_clockwise_action->set_enabled(should_enable_image_actions);
261 vertical_flip_action->set_enabled(should_enable_image_actions);
262 horizontal_flip_action->set_enabled(should_enable_image_actions);
263 desktop_wallpaper_action->set_enabled(should_enable_image_actions);
264
265 go_first_action->set_enabled(should_enable_backward_actions);
266 go_back_action->set_enabled(should_enable_backward_actions);
267 go_forward_action->set_enabled(should_enable_forward_actions);
268 go_last_action->set_enabled(should_enable_forward_actions);
269
270 zoom_in_action->set_enabled(should_enable_image_actions);
271 reset_zoom_action->set_enabled(should_enable_image_actions);
272 zoom_out_action->set_enabled(should_enable_image_actions);
273 if (!should_enable_image_actions) {
274 window->set_title("Image Viewer");
275 }
276 };
277
278 (void)TRY(main_toolbar->try_add_action(open_action));
279 (void)TRY(main_toolbar->try_add_action(delete_action));
280 (void)TRY(main_toolbar->try_add_separator());
281 (void)TRY(main_toolbar->try_add_action(go_first_action));
282 (void)TRY(main_toolbar->try_add_action(go_back_action));
283 (void)TRY(main_toolbar->try_add_action(go_forward_action));
284 (void)TRY(main_toolbar->try_add_action(go_last_action));
285 (void)TRY(main_toolbar->try_add_separator());
286 (void)TRY(main_toolbar->try_add_action(zoom_in_action));
287 (void)TRY(main_toolbar->try_add_action(reset_zoom_action));
288 (void)TRY(main_toolbar->try_add_action(zoom_out_action));
289
290 auto file_menu = TRY(window->try_add_menu("&File"));
291 TRY(file_menu->try_add_action(open_action));
292 TRY(file_menu->try_add_action(delete_action));
293 TRY(file_menu->try_add_separator());
294
295 TRY(file_menu->add_recent_files_list([&](auto& action) {
296 auto path = action.text();
297 widget->set_path(path);
298 widget->load_from_file(path);
299 }));
300
301 TRY(file_menu->try_add_action(quit_action));
302
303 auto image_menu = TRY(window->try_add_menu("&Image"));
304 TRY(image_menu->try_add_action(rotate_counterclockwise_action));
305 TRY(image_menu->try_add_action(rotate_clockwise_action));
306 TRY(image_menu->try_add_action(vertical_flip_action));
307 TRY(image_menu->try_add_action(horizontal_flip_action));
308 TRY(image_menu->try_add_separator());
309 TRY(image_menu->try_add_action(desktop_wallpaper_action));
310
311 auto navigate_menu = TRY(window->try_add_menu("&Navigate"));
312 TRY(navigate_menu->try_add_action(go_first_action));
313 TRY(navigate_menu->try_add_action(go_back_action));
314 TRY(navigate_menu->try_add_action(go_forward_action));
315 TRY(navigate_menu->try_add_action(go_last_action));
316
317 auto view_menu = TRY(window->try_add_menu("&View"));
318 TRY(view_menu->try_add_action(full_screen_action));
319 TRY(view_menu->try_add_separator());
320 TRY(view_menu->try_add_action(zoom_in_action));
321 TRY(view_menu->try_add_action(reset_zoom_action));
322 TRY(view_menu->try_add_action(fit_image_to_view_action));
323 TRY(view_menu->try_add_action(zoom_out_action));
324 TRY(view_menu->try_add_separator());
325
326 auto scaling_mode_menu = TRY(view_menu->try_add_submenu("&Scaling Mode"));
327 scaling_mode_menu->set_icon(TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/scale.png"sv)));
328
329 auto scaling_mode_group = make<GUI::ActionGroup>();
330 scaling_mode_group->set_exclusive(true);
331 scaling_mode_group->add_action(*nearest_neighbor_action);
332 scaling_mode_group->add_action(*smooth_pixels_action);
333 scaling_mode_group->add_action(*bilinear_action);
334
335 TRY(scaling_mode_menu->try_add_action(nearest_neighbor_action));
336 TRY(scaling_mode_menu->try_add_action(smooth_pixels_action));
337 TRY(scaling_mode_menu->try_add_action(bilinear_action));
338
339 TRY(view_menu->try_add_separator());
340 TRY(view_menu->try_add_action(hide_show_toolbar_action));
341
342 auto help_menu = TRY(window->try_add_menu("&Help"));
343 TRY(help_menu->try_add_action(GUI::CommonActions::make_command_palette_action(window)));
344 TRY(help_menu->try_add_action(GUI::CommonActions::make_help_action([](auto&) {
345 Desktop::Launcher::open(URL::create_with_file_scheme("/usr/share/man/man1/ImageViewer.md"), "/bin/Help");
346 })));
347 TRY(help_menu->try_add_action(GUI::CommonActions::make_about_action("Image Viewer", app_icon, window)));
348
349 if (path != nullptr) {
350 widget->load_from_file(path);
351 } else {
352 widget->clear();
353 }
354
355 window->show();
356
357 return app->exec();
358}