ladybird/Userland/Utilities/groupadd.cpp

64 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;
}