RGBCube/serenity
1
Fork 0
mirror of https://github.com/RGBCube/serenity synced 2025-06-01 10:08:10 +00:00
Code Issues Activity Actions

ps: Add -p option, and distinguish it from -q

Browse source
This commit is contained in:
Sam Atkins 2023-05-15 15:32:06 +01:00 • committed by Andreas Kling
parent 45c429853a
commit 3aff3c610a
2 changed files with 22 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

22
Userland/Utilities/ps.cpp
View file

@ -78,6 +78,8 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
bool every_process_flag = false;
bool every_terminal_process_flag = false;
bool full_format_flag = false;
bool provided_pid_list = false;
bool provided_quick_pid_list = false;
Vector<pid_t> pid_list;
Core::ArgsParser args_parser;
@ -85,7 +87,15 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
args_parser.add_option(every_process_flag, "Show every process", nullptr, 'A');
args_parser.add_option(every_process_flag, "Show every process (Equivalent to -A)", nullptr, 'e');
args_parser.add_option(full_format_flag, "Full format", nullptr, 'f');
args_parser.add_option(make_list_option(pid_list, "A comma- or space-separated list of PIDs. Only processes matching those PIDs will be selected", nullptr, 'q', "pid-list", [](StringView pid_string) {
args_parser.add_option(make_list_option(pid_list, "Show processes with a matching PID. (Comma- or space-separated list)", nullptr, 'p', "pid-list", [&](StringView pid_string) {
provided_pid_list = true;
auto pid = pid_string.to_int();
if (!pid.has_value())
warnln("Could not parse '{}' as a PID.", pid_string);
return pid;
}));
args_parser.add_option(make_list_option(pid_list, "Show processes with a matching PID. (Comma- or space-separated list.) Processes will be listed in the order given.", nullptr, 'q', "pid-list", [&](StringView pid_string) {
provided_quick_pid_list = true;
auto pid = pid_string.to_int();
if (!pid.has_value())
warnln("Could not parse '{}' as a PID.", pid_string);
@ -93,6 +103,11 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
}));
args_parser.parse(arguments);
if (provided_pid_list && provided_quick_pid_list) {
warnln("`-p` and `-q` cannot be specified together.");
return 1;
}
Vector<Column> columns;
Optional<size_t> uid_column;
@ -132,9 +147,10 @@ ErrorOr<int> serenity_main(Main::Arguments arguments)
if (!pid_list.is_empty()) {
every_process_flag = true;
processes.remove_all_matching([&](auto& process) { return !pid_list.contains_slow(process.pid); });
}
processes.remove_all_matching([&](auto& a) { return !pid_list.contains_slow(a.pid); });
if (provided_quick_pid_list) {
auto processes_sort_predicate = [&pid_list](auto& a, auto& b) {
return pid_list.find_first_index(a.pid).value() < pid_list.find_first_index(b.pid).value();
};