jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2021, Valtteri Koskivuori <vkoskiv@gmail.com>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#include "MagnifierWidget.h"
8#include <AK/LexicalPath.h>
9#include <LibConfig/Client.h>
10#include <LibCore/System.h>
11#include <LibDesktop/Launcher.h>
12#include <LibFileSystemAccessClient/Client.h>
13#include <LibGUI/ActionGroup.h>
14#include <LibGUI/Application.h>
15#include <LibGUI/ColorPicker.h>
16#include <LibGUI/Icon.h>
17#include <LibGUI/Menu.h>
18#include <LibGUI/Menubar.h>
19#include <LibGUI/MessageBox.h>
20#include <LibGUI/Window.h>
21#include <LibGfx/BMPWriter.h>
22#include <LibGfx/Filters/ColorBlindnessFilter.h>
23#include <LibGfx/PNGWriter.h>
24#include <LibGfx/QOIWriter.h>
25#include <LibMain/Main.h>
26
27static ErrorOr<ByteBuffer> dump_bitmap(RefPtr<Gfx::Bitmap> bitmap, AK::StringView extension)
28{
29 if (extension == "bmp") {
30 return Gfx::BMPWriter::encode(*bitmap);
31 } else if (extension == "png") {
32 return Gfx::PNGWriter::encode(*bitmap);
33 } else if (extension == "qoi") {
34 return Gfx::QOIWriter::encode(*bitmap);
35 } else {
36 return Error::from_string_literal("invalid image format");
37 }
38}
39
40ErrorOr<int> serenity_main(Main::Arguments arguments)
41{
42 TRY(Core::System::pledge("stdio cpath rpath recvfd sendfd unix"));
43 auto app = TRY(GUI::Application::try_create(arguments));
44
45 TRY(Desktop::Launcher::add_allowed_handler_with_only_specific_urls("/bin/Help", { URL::create_with_file_scheme("/usr/share/man/man1/Magnifier.md") }));
46 TRY(Desktop::Launcher::seal_allowlist());
47 Config::pledge_domain("Magnifier");
48
49 TRY(Core::System::unveil("/tmp/session/%sid/portal/filesystemaccess", "rw"));
50 TRY(Core::System::unveil("/res", "r"));
51 TRY(Core::System::unveil(nullptr, nullptr));
52
53 auto app_icon = GUI::Icon::default_icon("app-magnifier"sv);
54
55 // 4px on each side for padding
56 constexpr int window_dimensions = 240 + 4 + 4;
57 auto window = GUI::Window::construct();
58 window->set_title("Magnifier");
59 window->resize(window_dimensions, window_dimensions);
60 window->set_minimizable(false);
61 window->set_icon(app_icon.bitmap_for_size(16));
62 auto magnifier = TRY(window->set_main_widget<MagnifierWidget>());
63
64 auto file_menu = TRY(window->try_add_menu("&File"));
65 TRY(file_menu->try_add_action(GUI::CommonActions::make_save_as_action([&](auto&) {
66 AK::DeprecatedString filename = "file for saving";
67 auto do_save = [&]() -> ErrorOr<void> {
68 auto response = FileSystemAccessClient::Client::the().save_file(window, "Capture", "png");
69 if (response.is_error())
70 return {};
71 auto file = response.value().release_stream();
72 auto path = AK::LexicalPath(response.value().filename().to_deprecated_string());
73 filename = path.basename();
74 auto encoded = TRY(dump_bitmap(magnifier->current_bitmap(), path.extension()));
75
76 TRY(file->write_until_depleted(encoded));
77 return {};
78 };
79
80 auto result = do_save();
81 if (result.is_error()) {
82 GUI::MessageBox::show(window, "Unable to save file.\n"sv, "Error"sv, GUI::MessageBox::Type::Error);
83 warnln("Error saving bitmap to {}: {}", filename, result.error().string_literal());
84 }
85 })));
86 TRY(file_menu->try_add_separator());
87 TRY(file_menu->try_add_action(GUI::CommonActions::make_quit_action([&](auto&) {
88 app->quit();
89 })));
90
91 auto size_action_group = make<GUI::ActionGroup>();
92
93 auto two_x_action = GUI::Action::create_checkable(
94 "&2x", { Key_2 }, [&](auto&) {
95 magnifier->set_scale_factor(2);
96 });
97
98 auto four_x_action = GUI::Action::create_checkable(
99 "&4x", { Key_4 }, [&](auto&) {
100 magnifier->set_scale_factor(4);
101 });
102
103 auto eight_x_action = GUI::Action::create_checkable(
104 "&8x", { Key_8 }, [&](auto&) {
105 magnifier->set_scale_factor(8);
106 });
107
108 auto pause_action = GUI::Action::create_checkable(
109 "&Pause Capture", { Key_Space }, [&](auto& action) {
110 magnifier->pause_capture(action.is_checked());
111 });
112
113 auto lock_location_action = GUI::Action::create_checkable(
114 "&Lock Location", { Key_L }, [&](auto& action) {
115 magnifier->lock_location(action.is_checked());
116 });
117
118 auto show_grid_action = GUI::Action::create_checkable(
119 "Show &Grid", { Key_G }, [&](auto& action) {
120 magnifier->show_grid(action.is_checked());
121 });
122
123 auto choose_grid_color_action = GUI::Action::create(
124 "Choose Grid &Color", [&](auto& action [[maybe_unused]]) {
125 auto dialog = GUI::ColorPicker::construct(magnifier->grid_color(), window, "Magnifier: choose grid color");
126 dialog->set_color_has_alpha_channel(true);
127 if (dialog->exec() == GUI::Dialog::ExecResult::OK) {
128 Config::write_string("Magnifier"sv, "Grid"sv, "Color"sv, dialog->color().to_deprecated_string());
129 magnifier->set_grid_color(dialog->color());
130 }
131 });
132 {
133 auto color_string = Config::read_string("Magnifier"sv, "Grid"sv, "Color"sv, "#ff00ff64"sv);
134 auto maybe_color = Gfx::Color::from_string(color_string);
135 magnifier->set_grid_color(maybe_color.value_or(Gfx::Color::Magenta));
136 }
137
138 size_action_group->add_action(two_x_action);
139 size_action_group->add_action(four_x_action);
140 size_action_group->add_action(eight_x_action);
141 size_action_group->set_exclusive(true);
142
143 auto view_menu = TRY(window->try_add_menu("&View"));
144 TRY(view_menu->try_add_action(two_x_action));
145 TRY(view_menu->try_add_action(four_x_action));
146 TRY(view_menu->try_add_action(eight_x_action));
147 two_x_action->set_checked(true);
148
149 TRY(view_menu->try_add_separator());
150 TRY(view_menu->try_add_action(pause_action));
151 TRY(view_menu->try_add_action(lock_location_action));
152 TRY(view_menu->try_add_action(show_grid_action));
153 TRY(view_menu->try_add_action(choose_grid_color_action));
154
155 auto timeline_menu = TRY(window->try_add_menu("&Timeline"));
156 auto previous_frame_action = GUI::Action::create(
157 "&Previous frame", { Key_Left }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/go-back.png"sv)), [&](auto&) {
158 pause_action->set_checked(true);
159 magnifier->pause_capture(true);
160 magnifier->display_previous_frame();
161 });
162 auto next_frame_action = GUI::Action::create(
163 "&Next frame", { Key_Right }, TRY(Gfx::Bitmap::load_from_file("/res/icons/16x16/go-forward.png"sv)), [&](auto&) {
164 pause_action->set_checked(true);
165 magnifier->pause_capture(true);
166 magnifier->display_next_frame();
167 });
168 TRY(timeline_menu->try_add_action(previous_frame_action));
169 TRY(timeline_menu->try_add_action(next_frame_action));
170
171 TRY(window->try_add_menu(TRY(GUI::CommonMenus::make_accessibility_menu(magnifier))));
172
173 auto help_menu = TRY(window->try_add_menu("&Help"));
174 TRY(help_menu->try_add_action(GUI::CommonActions::make_command_palette_action(window)));
175 TRY(help_menu->try_add_action(GUI::CommonActions::make_help_action([](auto&) {
176 Desktop::Launcher::open(URL::create_with_file_scheme("/usr/share/man/man1/Magnifier.md"), "/bin/Help");
177 })));
178 TRY(help_menu->try_add_action(GUI::CommonActions::make_about_action("Magnifier", app_icon, window)));
179
180 window->show();
181 window->set_always_on_top(true);
182
183 return app->exec();
184}