jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2020, Andreas Kling <kling@serenityos.org>
3 *
4 * SPDX-License-Identifier: BSD-2-Clause
5 */
6
7#include <AK/DeprecatedString.h>
8#include <AK/Vector.h>
9#include <LibCore/ArgsParser.h>
10#include <LibCore/ConfigFile.h>
11#include <LibCore/System.h>
12#include <LibKeyboard/CharacterMap.h>
13#include <LibMain/Main.h>
14#include <stdio.h>
15
16int set_keymap(DeprecatedString const& keymap);
17
18int set_keymap(DeprecatedString const& keymap)
19{
20 auto character_map = Keyboard::CharacterMap::load_from_file(keymap);
21 if (character_map.is_error()) {
22 warnln("Cannot read keymap {}", keymap);
23 warnln("Hint: Must be a keymap name (e.g. 'en-us')");
24 return 1;
25 }
26
27 int rc = character_map.value().set_system_map();
28 if (rc != 0) {
29 perror("setkeymap");
30 }
31
32 return rc;
33}
34
35ErrorOr<int> serenity_main(Main::Arguments arguments)
36{
37 TRY(Core::System::pledge("stdio setkeymap getkeymap rpath wpath cpath"));
38 TRY(Core::System::unveil("/res/keymaps", "r"));
39 TRY(Core::System::unveil("/etc/Keyboard.ini", "rwc"));
40
41 DeprecatedString mapping;
42 DeprecatedString mappings;
43 Core::ArgsParser args_parser;
44 args_parser.add_option(mapping, "The mapping to be used", "set-keymap", 'm', "keymap");
45 args_parser.add_option(mappings, "Comma separated list of enabled mappings", "set-keymaps", 's', "keymaps");
46 args_parser.parse(arguments);
47
48 TRY(Core::System::unveil(nullptr, nullptr));
49
50 if (mapping.is_empty() && mappings.is_empty()) {
51 auto keymap = TRY(Keyboard::CharacterMap::fetch_system_map());
52 outln("{}", keymap.character_map_name());
53 return 0;
54 }
55
56 auto mapper_config = TRY(Core::ConfigFile::open("/etc/Keyboard.ini", Core::ConfigFile::AllowWriting::Yes));
57
58 if (!mappings.is_empty()) {
59 auto mappings_vector = mappings.split(',');
60
61 if (mappings_vector.is_empty()) {
62 warnln("Keymaps list should not be empty");
63 return 1;
64 }
65
66 // Verify that all specified keymaps are loadable
67 for (auto& keymap_name : mappings_vector) {
68 if (auto keymap = Keyboard::CharacterMap::load_from_file(keymap_name); keymap.is_error()) {
69 warnln("Cannot load keymap {}: {}({})", keymap_name, keymap.error().string_literal(), keymap.error().code());
70 return keymap.release_error();
71 }
72 }
73
74 auto keymaps = DeprecatedString::join(',', mappings_vector);
75 mapper_config->write_entry("Mapping", "Keymaps", keymaps);
76 TRY(mapper_config->sync());
77 }
78
79 auto keymaps = mapper_config->read_entry("Mapping", "Keymaps");
80 auto keymaps_vector = keymaps.split(',');
81
82 if (!mapping.is_empty()) {
83 if (keymaps_vector.is_empty()) {
84 warnln("No keymaps configured - writing default configurations (en-us)");
85 mapper_config->write_entry("Mapping", "Keymaps", "en-us");
86 TRY(mapper_config->sync());
87 keymaps_vector.append("en-us");
88 }
89
90 if (!keymaps_vector.find(mapping).is_end()) {
91 int rc = set_keymap(mapping);
92 if (rc == 0)
93 return rc;
94 } else {
95 warnln("Keymap '{}' is not in list of configured keymaps ({})", mapping, keymaps);
96 }
97 }
98
99 return set_keymap(keymaps_vector.first());
100}