main.cpp 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203
  1. /*
  2. * Copyright (c) 2020, Andreas Kling <kling@serenityos.org>
  3. * All rights reserved.
  4. *
  5. * Redistribution and use in source and binary forms, with or without
  6. * modification, are permitted provided that the following conditions are met:
  7. *
  8. * 1. Redistributions of source code must retain the above copyright notice, this
  9. * list of conditions and the following disclaimer.
  10. *
  11. * 2. Redistributions in binary form must reproduce the above copyright notice,
  12. * this list of conditions and the following disclaimer in the documentation
  13. * and/or other materials provided with the distribution.
  14. *
  15. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  16. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  17. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  18. * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
  19. * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  20. * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  21. * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  22. * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  23. * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  24. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25. */
  26. #include <AK/FileSystemPath.h>
  27. #include <AK/QuickSort.h>
  28. #include <LibCore/ConfigFile.h>
  29. #include <LibCore/DirIterator.h>
  30. #include <LibGUI/Action.h>
  31. #include <LibGUI/Application.h>
  32. #include <LibGUI/Desktop.h>
  33. #include <LibGUI/Menu.h>
  34. #include <LibGUI/MenuBar.h>
  35. #include <LibGUI/MessageBox.h>
  36. #include <LibGUI/WindowServerConnection.h>
  37. #include <LibGfx/Bitmap.h>
  38. #include <stdio.h>
  39. #include <sys/utsname.h>
  40. #include <unistd.h>
  41. //#define SYSTEM_MENU_DEBUG
  42. struct AppMetadata {
  43. String executable;
  44. String name;
  45. String icon_path;
  46. String category;
  47. };
  48. Vector<AppMetadata> g_apps;
  49. HashMap<String, NonnullRefPtr<GUI::Menu>> g_app_category_menus;
  50. struct ThemeMetadata {
  51. String name;
  52. String path;
  53. };
  54. Color g_menu_selection_color;
  55. Vector<ThemeMetadata> g_themes;
  56. RefPtr<GUI::Menu> g_themes_menu;
  57. static NonnullRefPtr<GUI::Menu> build_system_menu();
  58. int main(int argc, char** argv)
  59. {
  60. GUI::Application app(argc, argv);
  61. auto menu = build_system_menu();
  62. menu->realize_menu_if_needed();
  63. GUI::WindowServerConnection::the().send_sync<Messages::WindowServer::SetSystemMenu>(menu->menu_id());
  64. if (pledge("stdio shared_buffer accept rpath proc exec", nullptr) < 0) {
  65. perror("pledge");
  66. return 1;
  67. }
  68. if (unveil("/bin", "x")) {
  69. perror("unveil");
  70. return 1;
  71. }
  72. if (unveil("/res", "r")) {
  73. perror("unveil");
  74. return 1;
  75. }
  76. unveil(nullptr, nullptr);
  77. return app.exec();
  78. }
  79. NonnullRefPtr<GUI::Menu> build_system_menu()
  80. {
  81. HashTable<String> seen_app_categories;
  82. {
  83. Core::DirIterator dt("/res/apps", Core::DirIterator::SkipDots);
  84. while (dt.has_next()) {
  85. auto af_name = dt.next_path();
  86. auto af_path = String::format("/res/apps/%s", af_name.characters());
  87. auto af = Core::ConfigFile::open(af_path);
  88. if (!af->has_key("App", "Name") || !af->has_key("App", "Executable"))
  89. continue;
  90. auto app_name = af->read_entry("App", "Name");
  91. auto app_executable = af->read_entry("App", "Executable");
  92. auto app_category = af->read_entry("App", "Category");
  93. auto app_icon_path = af->read_entry("Icons", "16x16");
  94. g_apps.append({ app_executable, app_name, app_icon_path, app_category });
  95. seen_app_categories.set(app_category);
  96. }
  97. }
  98. Vector<String> sorted_app_categories;
  99. for (auto& category : seen_app_categories)
  100. sorted_app_categories.append(category);
  101. quick_sort(sorted_app_categories.begin(), sorted_app_categories.end(), [](auto& a, auto& b) { return a < b; });
  102. u8 system_menu_name[] = { 0xc3, 0xb8, 0 };
  103. auto system_menu = GUI::Menu::construct(String((const char*)system_menu_name));
  104. // First we construct all the necessary app category submenus.
  105. for (const auto& category : sorted_app_categories) {
  106. if (g_app_category_menus.contains(category))
  107. continue;
  108. auto category_menu = GUI::Menu::construct(category);
  109. system_menu->add_submenu(category_menu);
  110. g_app_category_menus.set(category, move(category_menu));
  111. }
  112. // Then we create and insert all the app menu items into the right place.
  113. int app_identifier = 0;
  114. for (const auto& app : g_apps) {
  115. RefPtr<Gfx::Bitmap> icon;
  116. if (!app.icon_path.is_empty())
  117. icon = Gfx::Bitmap::load_from_file(app.icon_path);
  118. #ifdef SYSTEM_MENU_DEBUG
  119. if (icon)
  120. dbg() << "App " << app.name << " has icon with size " << icon->size();
  121. #endif
  122. auto parent_menu = g_app_category_menus.get(app.category).value_or(*system_menu);
  123. parent_menu->add_action(GUI::Action::create(app.name, icon.ptr(), [app_identifier](auto&) {
  124. dbg() << "Activated app with ID " << app_identifier;
  125. if (fork() == 0) {
  126. const auto& bin = g_apps[app_identifier].executable;
  127. execl(bin.characters(), bin.characters(), nullptr);
  128. ASSERT_NOT_REACHED();
  129. }
  130. }));
  131. ++app_identifier;
  132. }
  133. system_menu->add_separator();
  134. g_themes_menu = GUI::Menu::construct("Themes");
  135. system_menu->add_submenu(*g_themes_menu);
  136. {
  137. Core::DirIterator dt("/res/themes", Core::DirIterator::SkipDots);
  138. while (dt.has_next()) {
  139. auto theme_name = dt.next_path();
  140. auto theme_path = String::format("/res/themes/%s", theme_name.characters());
  141. g_themes.append({ FileSystemPath(theme_name).title(), theme_path });
  142. }
  143. quick_sort(g_themes.begin(), g_themes.end(), [](auto& a, auto& b) { return a.name < b.name; });
  144. }
  145. {
  146. int theme_identifier = 0;
  147. for (auto& theme : g_themes) {
  148. g_themes_menu->add_action(GUI::Action::create(theme.name, [theme_identifier](auto&) {
  149. auto& theme = g_themes[theme_identifier];
  150. dbg() << "Theme switched to " << theme.name << " at path " << theme.path;
  151. auto response = GUI::WindowServerConnection::the().send_sync<Messages::WindowServer::SetSystemTheme>(theme.path, theme.name);
  152. ASSERT(response->success());
  153. }));
  154. ++theme_identifier;
  155. }
  156. }
  157. system_menu->add_separator();
  158. system_menu->add_action(GUI::Action::create("About...", Gfx::Bitmap::load_from_file("/res/icons/16x16/ladybug.png"), [](auto&) {
  159. if (fork() == 0) {
  160. execl("/bin/About", "/bin/About", nullptr);
  161. ASSERT_NOT_REACHED();
  162. }
  163. }));
  164. system_menu->add_separator();
  165. system_menu->add_action(GUI::Action::create("Shutdown...", [](auto&) {
  166. if (fork() == 0) {
  167. execl("/bin/SystemDialog", "/bin/SystemDialog", "--shutdown", nullptr);
  168. ASSERT_NOT_REACHED();
  169. }
  170. }));
  171. return system_menu;
  172. }