RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-07-27 22:07:35 +00:00
Code Issues Activity Actions

Utilities: Add ability to query specific user with id

Browse source
This commit is contained in:
Jean-Baptiste Boric 2021-07-08 22:19:09 +02:00 • committed by Andreas Kling
parent 538cc9d99a
commit 934e53079f
1 changed files with 25 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

27
Userland/Utilities/id.cpp
View file

@ -4,6 +4,7 @@
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
#include <AK/StringUtils.h>
#include <LibCore/Account.h> #include <LibCore/Account.h>
#include <LibCore/ArgsParser.h> #include <LibCore/ArgsParser.h>
#include <alloca.h> #include <alloca.h>
@ -18,6 +19,7 @@ static bool flag_print_uid = false;
static bool flag_print_gid = false; static bool flag_print_gid = false;
static bool flag_print_name = false; static bool flag_print_name = false;
static bool flag_print_gid_all = false; static bool flag_print_gid_all = false;
static String user_str;
int main(int argc, char** argv) int main(int argc, char** argv)
{ {
@ -46,6 +48,7 @@ int main(int argc, char** argv)
args_parser.add_option(flag_print_gid, "Print GID", nullptr, 'g'); args_parser.add_option(flag_print_gid, "Print GID", nullptr, 'g');
args_parser.add_option(flag_print_gid_all, "Print all GIDs", nullptr, 'G'); args_parser.add_option(flag_print_gid_all, "Print all GIDs", nullptr, 'G');
args_parser.add_option(flag_print_name, "Print name", nullptr, 'n'); args_parser.add_option(flag_print_name, "Print name", nullptr, 'n');
args_parser.add_positional_argument(user_str, "User name/UID to query", "USER", Core::ArgsParser::Required::No);
args_parser.parse(argc, argv); args_parser.parse(argc, argv);
if (flag_print_name && !(flag_print_uid || flag_print_gid || flag_print_gid_all)) { if (flag_print_name && !(flag_print_uid || flag_print_gid || flag_print_gid_all)) {
@ -58,8 +61,28 @@ int main(int argc, char** argv)
return 1; return 1;
} }
auto account = Core::Account::self(Core::Account::Read::PasswdOnly); Optional<Core::Account> account;
return print_id_objects(account); if (!user_str.is_empty()) {
if (auto user_id = user_str.to_uint(); user_id.has_value()) {
auto result = Core::Account::from_uid(user_id.value(), Core::Account::Read::PasswdOnly);
if (result.is_error()) {
warnln("Couldn't retrieve user id '{}': {}", user_str, result.error());
return 1;
}
account = result.release_value();
} else {
auto result = Core::Account::from_name(user_str.characters(), Core::Account::Read::PasswdOnly);
if (result.is_error()) {
warnln("Couldn't retrieve user name '{}': {}", user_str, result.error());
return 1;
}
account = result.release_value();
}
} else {
account = Core::Account::self(Core::Account::Read::PasswdOnly);
}
return print_id_objects(account.value());
} }
static bool print_uid_object(Core::Account const& account) static bool print_uid_object(Core::Account const& account)