mirror of
https://github.com/SerenityOS/serenity
synced 2024-07-22 02:26:11 +00:00
65 lines
2.4 KiB
C++
65 lines
2.4 KiB
C++
/*
|
|
* Copyright (c) 2019-2020, Jesse Buhagiar <jooster669@gmail.com>
|
|
* Copyright (c) 2021, Brandon Pruitt <brapru@pm.me>
|
|
* Copyright (c) 2021, Maxime Friess <M4x1me@pm.me>
|
|
* Copyright (c) 2022, Kenneth Myhra <kennethmyhra@serenityos.org>
|
|
* Copyright (c) 2023, Tim Ledbetter <timledbetter@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#include <AK/HashTable.h>
|
|
#include <LibCore/Account.h>
|
|
#include <LibCore/ArgsParser.h>
|
|
#include <LibCore/Group.h>
|
|
#include <LibCore/System.h>
|
|
#include <LibMain/Main.h>
|
|
|
|
ErrorOr<int> serenity_main(Main::Arguments arguments)
|
|
{
|
|
TRY(Core::System::pledge("stdio wpath rpath cpath chown"));
|
|
|
|
gid_t gid = 0;
|
|
StringView group_name;
|
|
Vector<DeprecatedString> group_members;
|
|
|
|
Core::ArgsParser args_parser;
|
|
args_parser.add_option(gid, "Group ID (gid) for the new group", "gid", 'g', "gid");
|
|
args_parser.add_option(Core::ArgsParser::Option {
|
|
.argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
|
|
.help_string = "A comma-separated list of usernames to add as members of the new group",
|
|
.long_name = "users",
|
|
.short_name = 'U',
|
|
.value_name = "user-list",
|
|
.accept_value = [&group_members](StringView comma_separated_users) {
|
|
auto accounts_or_error = Core::Account::all(Core::Account::Read::PasswdOnly);
|
|
if (accounts_or_error.is_error())
|
|
return false;
|
|
|
|
OrderedHashTable<DeprecatedString> unique_group_members;
|
|
auto accounts = accounts_or_error.release_value();
|
|
for (auto const& username : comma_separated_users.split_view(',')) {
|
|
auto matching_account = accounts.first_matching([&](auto const& account) { return username == account.username(); });
|
|
if (!matching_account.has_value()) {
|
|
warnln("Invalid member username: '{}'", username);
|
|
return false;
|
|
}
|
|
|
|
unique_group_members.set(matching_account->username());
|
|
}
|
|
|
|
for (auto const& member : unique_group_members)
|
|
group_members.append(member);
|
|
|
|
return true;
|
|
},
|
|
});
|
|
args_parser.add_positional_argument(group_name, "Name of the group (groupname)", "group");
|
|
args_parser.parse(arguments);
|
|
|
|
Core::Group group { group_name, gid, group_members };
|
|
TRY(Core::Group::add_group(group));
|
|
|
|
return 0;
|
|
}
|