12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799 |
- /*
- * Copyright (c) 2020-2021, the SerenityOS developers.
- *
- * SPDX-License-Identifier: BSD-2-Clause
- */
- #include "AST.h"
- #include "Formatter.h"
- #include "PosixParser.h"
- #include "Shell.h"
- #include <AK/DeprecatedString.h>
- #include <AK/LexicalPath.h>
- #include <AK/ScopeGuard.h>
- #include <AK/Statistics.h>
- #include <LibCore/ArgsParser.h>
- #include <LibCore/DeprecatedFile.h>
- #include <LibCore/EventLoop.h>
- #include <errno.h>
- #include <inttypes.h>
- #include <limits.h>
- #include <signal.h>
- #include <sys/wait.h>
- #include <unistd.h>
- extern char** environ;
- namespace Shell {
- ErrorOr<int> Shell::builtin_noop(Main::Arguments)
- {
- return 0;
- }
- ErrorOr<int> Shell::builtin_dump(Main::Arguments arguments)
- {
- bool posix = false;
- StringView source;
- Core::ArgsParser parser;
- parser.add_positional_argument(source, "Shell code to parse and dump", "source");
- parser.add_option(posix, "Use the POSIX parser", "posix", 'p');
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- TRY((posix ? Posix::Parser { source }.parse() : Parser { source }.parse())->dump(0));
- return 0;
- }
- enum FollowSymlinks {
- Yes,
- No
- };
- static Vector<DeprecatedString> find_matching_executables_in_path(StringView filename, FollowSymlinks follow_symlinks = FollowSymlinks::No)
- {
- // Edge cases in which there are guaranteed no solutions
- if (filename.is_empty() || filename.contains('/'))
- return {};
- char const* path_str = getenv("PATH");
- auto path = DEFAULT_PATH_SV;
- if (path_str != nullptr) // maybe && *path_str
- path = { path_str, strlen(path_str) };
- Vector<DeprecatedString> executables;
- auto directories = path.split_view(':');
- for (auto directory : directories) {
- auto file = DeprecatedString::formatted("{}/{}", directory, filename);
- if (follow_symlinks == FollowSymlinks::Yes) {
- auto path_or_error = Core::DeprecatedFile::read_link(file);
- if (!path_or_error.is_error())
- file = path_or_error.release_value();
- }
- if (access(file.characters(), X_OK) == 0)
- executables.append(move(file));
- }
- return executables;
- }
- ErrorOr<int> Shell::builtin_where(Main::Arguments arguments)
- {
- Vector<StringView> values_to_lookup;
- bool do_only_path_search { false };
- bool do_follow_symlinks { false };
- bool do_print_only_type { false };
- Core::ArgsParser parser;
- parser.add_positional_argument(values_to_lookup, "List of shell builtins, aliases or executables", "arguments");
- parser.add_option(do_only_path_search, "Search only for executables in the PATH environment variable", "path-only", 'p');
- parser.add_option(do_follow_symlinks, "Follow symlinks and print the symlink free path", "follow-symlink", 's');
- parser.add_option(do_print_only_type, "Print the argument type instead of a human readable description", "type", 'w');
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- auto const lookup_alias = [do_only_path_search, &m_aliases = this->m_aliases](StringView alias) -> Optional<DeprecatedString> {
- if (do_only_path_search)
- return {};
- return m_aliases.get(alias);
- };
- auto const lookup_builtin = [do_only_path_search](StringView builtin) -> Optional<DeprecatedString> {
- if (do_only_path_search)
- return {};
- for (auto const& _builtin : builtin_names) {
- if (_builtin == builtin) {
- return builtin;
- }
- }
- return {};
- };
- bool at_least_one_succeded { false };
- for (auto const& argument : values_to_lookup) {
- auto const alias = lookup_alias(argument);
- if (alias.has_value()) {
- if (do_print_only_type)
- outln("{}: alias", argument);
- else
- outln("{}: aliased to {}", argument, alias.value());
- at_least_one_succeded = true;
- }
- auto const builtin = lookup_builtin(argument);
- if (builtin.has_value()) {
- if (do_print_only_type)
- outln("{}: builtin", builtin.value());
- else
- outln("{}: shell built-in command", builtin.value());
- at_least_one_succeded = true;
- }
- auto const executables = find_matching_executables_in_path(argument, do_follow_symlinks ? FollowSymlinks::Yes : FollowSymlinks::No);
- for (auto const& path : executables) {
- if (do_print_only_type)
- outln("{}: command", argument);
- else
- outln(path);
- at_least_one_succeded = true;
- }
- if (!at_least_one_succeded)
- warnln("{} not found", argument);
- }
- return at_least_one_succeded ? 0 : 1;
- }
- ErrorOr<int> Shell::builtin_alias(Main::Arguments arguments)
- {
- Vector<DeprecatedString> aliases;
- Core::ArgsParser parser;
- parser.add_positional_argument(aliases, "List of name[=values]'s", "name[=value]", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (aliases.is_empty()) {
- for (auto& alias : m_aliases)
- printf("%s=%s\n", escape_token(alias.key).characters(), escape_token(alias.value).characters());
- return 0;
- }
- bool fail = false;
- for (auto& argument : aliases) {
- auto parts = argument.split_limit('=', 2, SplitBehavior::KeepEmpty);
- if (parts.size() == 1) {
- auto alias = m_aliases.get(parts[0]);
- if (alias.has_value()) {
- printf("%s=%s\n", escape_token(parts[0]).characters(), escape_token(alias.value()).characters());
- } else {
- fail = true;
- }
- } else {
- m_aliases.set(parts[0], parts[1]);
- add_entry_to_cache({ RunnablePath::Kind::Alias, parts[0] });
- }
- }
- return fail ? 1 : 0;
- }
- ErrorOr<int> Shell::builtin_unalias(Main::Arguments arguments)
- {
- bool remove_all { false };
- Vector<DeprecatedString> aliases;
- Core::ArgsParser parser;
- parser.set_general_help("Remove alias from the list of aliases");
- parser.add_option(remove_all, "Remove all aliases", nullptr, 'a');
- parser.add_positional_argument(aliases, "List of aliases to remove", "alias", Core::ArgsParser::Required::Yes);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (remove_all) {
- m_aliases.clear();
- cache_path();
- return 0;
- }
- bool failed { false };
- for (auto& argument : aliases) {
- if (!m_aliases.contains(argument)) {
- warnln("unalias: {}: alias not found", argument);
- failed = true;
- continue;
- }
- m_aliases.remove(argument);
- remove_entry_from_cache(argument);
- }
- return failed ? 1 : 0;
- }
- ErrorOr<int> Shell::builtin_break(Main::Arguments arguments)
- {
- if (!m_in_posix_mode) {
- raise_error(ShellError::EvaluatedSyntaxError, "break: Invalid use of builtin break in non-POSIX mode");
- return 1;
- }
- unsigned count = 1;
- Core::ArgsParser parser;
- parser.add_positional_argument(count, "Number of loops to 'break' out of", "count", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (count != 1) {
- raise_error(ShellError::EvaluatedSyntaxError, "break: count must be equal to 1 (NYI)");
- return 1;
- }
- raise_error(ShellError::InternalControlFlowBreak, "POSIX break");
- return 0;
- }
- ErrorOr<int> Shell::builtin_continue(Main::Arguments arguments)
- {
- if (!m_in_posix_mode) {
- raise_error(ShellError::EvaluatedSyntaxError, "break: Invalid use of builtin continue in non-POSIX mode");
- return 1;
- }
- unsigned count = 1;
- Core::ArgsParser parser;
- parser.add_positional_argument(count, "Number of loops to 'continue' out of", "count", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (count != 0) {
- raise_error(ShellError::EvaluatedSyntaxError, "continue: count must be equal to 1 (NYI)");
- return 1;
- }
- raise_error(ShellError::InternalControlFlowContinue, "POSIX continue");
- return 0;
- }
- ErrorOr<int> Shell::builtin_bg(Main::Arguments arguments)
- {
- int job_id = -1;
- bool is_pid = false;
- Core::ArgsParser parser;
- parser.add_positional_argument(Core::ArgsParser::Arg {
- .help_string = "Job ID or Jobspec to run in background",
- .name = "job-id",
- .min_values = 0,
- .max_values = 1,
- .accept_value = [&](StringView value) -> bool {
- // Check if it's a pid (i.e. literal integer)
- if (auto number = value.to_uint(); number.has_value()) {
- job_id = number.value();
- is_pid = true;
- return true;
- }
- // Check if it's a jobspec
- if (auto id = resolve_job_spec(value); id.has_value()) {
- job_id = id.value();
- is_pid = false;
- return true;
- }
- return false;
- } });
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (job_id == -1 && !jobs.is_empty())
- job_id = find_last_job_id();
- auto* job = const_cast<Job*>(find_job(job_id, is_pid));
- if (!job) {
- if (job_id == -1) {
- warnln("bg: No current job");
- } else {
- warnln("bg: Job with id/pid {} not found", job_id);
- }
- return 1;
- }
- job->set_running_in_background(true);
- job->set_should_announce_exit(true);
- job->set_shell_did_continue(true);
- dbgln("Resuming {} ({})", job->pid(), job->cmd());
- warnln("Resuming job {} - {}", job->job_id(), job->cmd());
- // Try using the PGID, but if that fails, just use the PID.
- if (killpg(job->pgid(), SIGCONT) < 0) {
- if (kill(job->pid(), SIGCONT) < 0) {
- perror("kill");
- return 1;
- }
- }
- return 0;
- }
- ErrorOr<int> Shell::builtin_type(Main::Arguments arguments)
- {
- Vector<DeprecatedString> commands;
- bool dont_show_function_source = false;
- Core::ArgsParser parser;
- parser.set_general_help("Display information about commands.");
- parser.add_positional_argument(commands, "Command(s) to list info about", "command");
- parser.add_option(dont_show_function_source, "Do not show functions source.", "no-fn-source", 'f');
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- bool something_not_found = false;
- for (auto& command : commands) {
- // check if it is an alias
- if (auto alias = m_aliases.get(command); alias.has_value()) {
- printf("%s is aliased to `%s`\n", escape_token(command).characters(), escape_token(alias.value()).characters());
- continue;
- }
- // check if it is a function
- if (auto function = m_functions.get(command); function.has_value()) {
- auto fn = function.value();
- printf("%s is a function\n", command.characters());
- if (!dont_show_function_source) {
- StringBuilder builder;
- builder.append(fn.name);
- builder.append('(');
- for (size_t i = 0; i < fn.arguments.size(); i++) {
- builder.append(fn.arguments[i]);
- if (!(i == fn.arguments.size() - 1))
- builder.append(' ');
- }
- builder.append(") {\n"sv);
- if (fn.body) {
- auto formatter = Formatter(*fn.body);
- builder.append(formatter.format());
- printf("%s\n}\n", builder.to_deprecated_string().characters());
- } else {
- printf("%s\n}\n", builder.to_deprecated_string().characters());
- }
- }
- continue;
- }
- // check if its a builtin
- if (has_builtin(command)) {
- printf("%s is a shell builtin\n", command.characters());
- continue;
- }
- // check if its an executable in PATH
- auto fullpath = Core::DeprecatedFile::resolve_executable_from_environment(command);
- if (fullpath.has_value()) {
- printf("%s is %s\n", command.characters(), escape_token(fullpath.release_value()).characters());
- continue;
- }
- something_not_found = true;
- printf("type: %s not found\n", command.characters());
- }
- if (something_not_found)
- return 1;
- else
- return 0;
- }
- ErrorOr<int> Shell::builtin_cd(Main::Arguments arguments)
- {
- StringView arg_path;
- Core::ArgsParser parser;
- parser.add_positional_argument(arg_path, "Path to change to", "path", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- DeprecatedString new_path;
- if (arg_path.is_empty()) {
- new_path = home;
- } else {
- if (arg_path == "-"sv) {
- char* oldpwd = getenv("OLDPWD");
- if (oldpwd == nullptr)
- return 1;
- new_path = oldpwd;
- } else {
- new_path = arg_path;
- }
- }
- auto real_path = Core::DeprecatedFile::real_path_for(new_path);
- if (real_path.is_empty()) {
- warnln("Invalid path '{}'", new_path);
- return 1;
- }
- if (cd_history.is_empty() || cd_history.last() != real_path)
- cd_history.enqueue(real_path);
- auto path_relative_to_current_directory = LexicalPath::relative_path(real_path, cwd);
- if (path_relative_to_current_directory.is_empty())
- path_relative_to_current_directory = real_path;
- char const* path = path_relative_to_current_directory.characters();
- int rc = chdir(path);
- if (rc < 0) {
- if (errno == ENOTDIR) {
- warnln("Not a directory: {}", path);
- } else {
- warnln("chdir({}) failed: {}", path, strerror(errno));
- }
- return 1;
- }
- setenv("OLDPWD", cwd.characters(), 1);
- cwd = move(real_path);
- setenv("PWD", cwd.characters(), 1);
- return 0;
- }
- ErrorOr<int> Shell::builtin_cdh(Main::Arguments arguments)
- {
- int index = -1;
- Core::ArgsParser parser;
- parser.add_positional_argument(index, "Index of the cd history entry (leave out for a list)", "index", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (index == -1) {
- if (cd_history.is_empty()) {
- warnln("cdh: no history available");
- return 0;
- }
- for (ssize_t i = cd_history.size() - 1; i >= 0; --i)
- printf("%zu: %s\n", cd_history.size() - i, cd_history.at(i).characters());
- return 0;
- }
- if (index < 1 || (size_t)index > cd_history.size()) {
- warnln("cdh: history index out of bounds: {} not in (0, {})", index, cd_history.size());
- return 1;
- }
- StringView path = cd_history.at(cd_history.size() - index);
- StringView cd_args[] = { "cd"sv, path };
- return Shell::builtin_cd({ .argc = 0, .argv = 0, .strings = cd_args });
- }
- ErrorOr<int> Shell::builtin_dirs(Main::Arguments arguments)
- {
- // The first directory in the stack is ALWAYS the current directory
- directory_stack.at(0) = cwd.characters();
- bool clear = false;
- bool print = false;
- bool number_when_printing = false;
- char separator = ' ';
- Vector<DeprecatedString> paths;
- Core::ArgsParser parser;
- parser.add_option(clear, "Clear the directory stack", "clear", 'c');
- parser.add_option(print, "Print directory entries one per line", "print", 'p');
- parser.add_option(number_when_printing, "Number the directories in the stack when printing", "number", 'v');
- parser.add_positional_argument(paths, "Extra paths to put on the stack", "path", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- // -v implies -p
- print = print || number_when_printing;
- if (print) {
- if (!paths.is_empty()) {
- warnln("dirs: 'print' and 'number' are not allowed when any path is specified");
- return 1;
- }
- separator = '\n';
- }
- if (clear) {
- for (size_t i = 1; i < directory_stack.size(); i++)
- directory_stack.remove(i);
- }
- for (auto& path : paths)
- directory_stack.append(path);
- if (print || (!clear && paths.is_empty())) {
- int index = 0;
- for (auto& directory : directory_stack) {
- if (number_when_printing)
- printf("%d ", index++);
- print_path(directory);
- fputc(separator, stdout);
- }
- }
- return 0;
- }
- ErrorOr<int> Shell::builtin_exec(Main::Arguments arguments)
- {
- if (arguments.strings.size() < 2) {
- warnln("Shell: No command given to exec");
- return 1;
- }
- TRY(execute_process(arguments.strings.slice(1)));
- // NOTE: Won't get here.
- return 0;
- }
- ErrorOr<int> Shell::builtin_exit(Main::Arguments arguments)
- {
- int exit_code = 0;
- Core::ArgsParser parser;
- parser.add_positional_argument(exit_code, "Exit code", "code", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (m_is_interactive) {
- if (!jobs.is_empty()) {
- if (!m_should_ignore_jobs_on_next_exit) {
- warnln("Shell: You have {} active job{}, run 'exit' again to really exit.", jobs.size(), jobs.size() > 1 ? "s" : "");
- m_should_ignore_jobs_on_next_exit = true;
- return 1;
- }
- }
- }
- stop_all_jobs();
- if (m_is_interactive) {
- m_editor->save_history(get_history_path());
- printf("Good-bye!\n");
- }
- exit(exit_code);
- }
- ErrorOr<int> Shell::builtin_export(Main::Arguments arguments)
- {
- Vector<DeprecatedString> vars;
- Core::ArgsParser parser;
- parser.add_positional_argument(vars, "List of variable[=value]'s", "values", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (vars.is_empty()) {
- for (size_t i = 0; environ[i]; ++i)
- puts(environ[i]);
- return 0;
- }
- for (auto& value : vars) {
- auto parts = value.split_limit('=', 2);
- if (parts.is_empty()) {
- warnln("Shell: Invalid export spec '{}', expected `variable=value' or `variable'", value);
- return 1;
- }
- if (parts.size() == 1) {
- auto value = TRY(lookup_local_variable(parts[0]));
- if (value) {
- auto values = TRY(const_cast<AST::Value&>(*value).resolve_as_list(*this));
- StringBuilder builder;
- builder.join(' ', values);
- parts.append(builder.to_deprecated_string());
- } else {
- // Ignore the export.
- continue;
- }
- }
- int setenv_return = setenv(parts[0].characters(), parts[1].characters(), 1);
- if (setenv_return != 0) {
- perror("setenv");
- return 1;
- }
- if (parts[0] == "PATH")
- cache_path();
- }
- return 0;
- }
- ErrorOr<int> Shell::builtin_glob(Main::Arguments arguments)
- {
- Vector<DeprecatedString> globs;
- Core::ArgsParser parser;
- parser.add_positional_argument(globs, "Globs to resolve", "glob");
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- for (auto& glob : globs) {
- for (auto& expanded : expand_globs(glob, cwd))
- outln("{}", expanded);
- }
- return 0;
- }
- ErrorOr<int> Shell::builtin_fg(Main::Arguments arguments)
- {
- int job_id = -1;
- bool is_pid = false;
- Core::ArgsParser parser;
- parser.add_positional_argument(Core::ArgsParser::Arg {
- .help_string = "Job ID or Jobspec to bring to foreground",
- .name = "job-id",
- .min_values = 0,
- .max_values = 1,
- .accept_value = [&](StringView value) -> bool {
- // Check if it's a pid (i.e. literal integer)
- if (auto number = value.to_uint(); number.has_value()) {
- job_id = number.value();
- is_pid = true;
- return true;
- }
- // Check if it's a jobspec
- if (auto id = resolve_job_spec(value); id.has_value()) {
- job_id = id.value();
- is_pid = false;
- return true;
- }
- return false;
- } });
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (job_id == -1 && !jobs.is_empty())
- job_id = find_last_job_id();
- RefPtr<Job> job = find_job(job_id, is_pid);
- if (!job) {
- if (job_id == -1) {
- warnln("fg: No current job");
- } else {
- warnln("fg: Job with id/pid {} not found", job_id);
- }
- return 1;
- }
- job->set_running_in_background(false);
- job->set_shell_did_continue(true);
- dbgln("Resuming {} ({})", job->pid(), job->cmd());
- warnln("Resuming job {} - {}", job->job_id(), job->cmd());
- tcsetpgrp(STDOUT_FILENO, job->pgid());
- tcsetpgrp(STDIN_FILENO, job->pgid());
- // Try using the PGID, but if that fails, just use the PID.
- if (killpg(job->pgid(), SIGCONT) < 0) {
- if (kill(job->pid(), SIGCONT) < 0) {
- perror("kill");
- return 1;
- }
- }
- block_on_job(job);
- if (job->exited())
- return job->exit_code();
- else
- return 0;
- }
- ErrorOr<int> Shell::builtin_disown(Main::Arguments arguments)
- {
- Vector<int> job_ids;
- Vector<bool> id_is_pid;
- Core::ArgsParser parser;
- parser.add_positional_argument(Core::ArgsParser::Arg {
- .help_string = "Job IDs or Jobspecs to disown",
- .name = "job-id",
- .min_values = 0,
- .max_values = INT_MAX,
- .accept_value = [&](StringView value) -> bool {
- // Check if it's a pid (i.e. literal integer)
- if (auto number = value.to_uint(); number.has_value()) {
- job_ids.append(number.value());
- id_is_pid.append(true);
- return true;
- }
- // Check if it's a jobspec
- if (auto id = resolve_job_spec(value); id.has_value()) {
- job_ids.append(id.value());
- id_is_pid.append(false);
- return true;
- }
- return false;
- } });
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (job_ids.is_empty()) {
- job_ids.append(find_last_job_id());
- id_is_pid.append(false);
- }
- Vector<Job const*> jobs_to_disown;
- for (size_t i = 0; i < job_ids.size(); ++i) {
- auto id = job_ids[i];
- auto is_pid = id_is_pid[i];
- auto job = find_job(id, is_pid);
- if (!job)
- warnln("disown: Job with id/pid {} not found", id);
- else
- jobs_to_disown.append(job);
- }
- if (jobs_to_disown.is_empty()) {
- if (job_ids.is_empty())
- warnln("disown: No current job");
- // An error message has already been printed about the nonexistence of each listed job.
- return 1;
- }
- for (auto job : jobs_to_disown) {
- job->deactivate();
- if (!job->is_running_in_background())
- warnln("disown warning: Job {} is currently not running, 'kill -{} {}' to make it continue", job->job_id(), SIGCONT, job->pid());
- jobs.remove(job->pid());
- }
- return 0;
- }
- ErrorOr<int> Shell::builtin_history(Main::Arguments)
- {
- for (size_t i = 0; i < m_editor->history().size(); ++i) {
- printf("%6zu %s\n", i + 1, m_editor->history()[i].entry.characters());
- }
- return 0;
- }
- ErrorOr<int> Shell::builtin_jobs(Main::Arguments arguments)
- {
- bool list = false, show_pid = false;
- Core::ArgsParser parser;
- parser.add_option(list, "List all information about jobs", "list", 'l');
- parser.add_option(show_pid, "Display the PID of the jobs", "pid", 'p');
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- Job::PrintStatusMode mode = Job::PrintStatusMode::Basic;
- if (show_pid)
- mode = Job::PrintStatusMode::OnlyPID;
- if (list)
- mode = Job::PrintStatusMode::ListAll;
- for (auto& it : jobs) {
- if (!it.value->print_status(mode))
- return 1;
- }
- return 0;
- }
- ErrorOr<int> Shell::builtin_popd(Main::Arguments arguments)
- {
- if (directory_stack.size() <= 1) {
- warnln("Shell: popd: directory stack empty");
- return 1;
- }
- bool should_not_switch = false;
- Core::ArgsParser parser;
- parser.add_option(should_not_switch, "Do not switch dirs", "no-switch", 'n');
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- auto popped_path = directory_stack.take_last();
- if (should_not_switch)
- return 0;
- auto new_path = LexicalPath::canonicalized_path(popped_path);
- if (chdir(new_path.characters()) < 0) {
- warnln("chdir({}) failed: {}", new_path, strerror(errno));
- return 1;
- }
- cwd = new_path;
- return 0;
- }
- ErrorOr<int> Shell::builtin_pushd(Main::Arguments arguments)
- {
- StringBuilder path_builder;
- bool should_switch = true;
- // From the BASH reference manual: https://www.gnu.org/software/bash/manual/html_node/Directory-Stack-Builtins.html
- // With no arguments, pushd exchanges the top two directories and makes the new top the current directory.
- if (arguments.strings.size() == 1) {
- if (directory_stack.size() < 2) {
- warnln("pushd: no other directory");
- return 1;
- }
- DeprecatedString dir1 = directory_stack.take_first();
- DeprecatedString dir2 = directory_stack.take_first();
- directory_stack.insert(0, dir2);
- directory_stack.insert(1, dir1);
- int rc = chdir(dir2.characters());
- if (rc < 0) {
- warnln("chdir({}) failed: {}", dir2, strerror(errno));
- return 1;
- }
- cwd = dir2;
- return 0;
- }
- // Let's assume the user's typed in 'pushd <dir>'
- if (arguments.strings.size() == 2) {
- directory_stack.append(cwd.characters());
- if (arguments.strings[1].starts_with('/')) {
- path_builder.append(arguments.strings[1]);
- } else {
- path_builder.appendff("{}/{}", cwd, arguments.strings[1]);
- }
- } else if (arguments.strings.size() == 3) {
- directory_stack.append(cwd.characters());
- for (size_t i = 1; i < arguments.strings.size(); i++) {
- auto arg = arguments.strings[i];
- if (arg.starts_with('-')) {
- if (arg.starts_with('/')) {
- path_builder.append(arg);
- } else {
- path_builder.appendff("{}/{}", cwd, arg);
- }
- }
- if (arg == "-n"sv)
- should_switch = false;
- }
- }
- auto real_path = LexicalPath::canonicalized_path(path_builder.to_deprecated_string());
- struct stat st;
- int rc = stat(real_path.characters(), &st);
- if (rc < 0) {
- warnln("stat({}) failed: {}", real_path, strerror(errno));
- return 1;
- }
- if (!S_ISDIR(st.st_mode)) {
- warnln("Not a directory: {}", real_path);
- return 1;
- }
- if (should_switch) {
- int rc = chdir(real_path.characters());
- if (rc < 0) {
- warnln("chdir({}) failed: {}", real_path, strerror(errno));
- return 1;
- }
- cwd = real_path;
- }
- return 0;
- }
- ErrorOr<int> Shell::builtin_pwd(Main::Arguments)
- {
- print_path(cwd);
- fputc('\n', stdout);
- return 0;
- }
- ErrorOr<int> Shell::builtin_setopt(Main::Arguments arguments)
- {
- if (arguments.strings.size() == 1) {
- #define __ENUMERATE_SHELL_OPTION(name, default_, description) \
- if (options.name) \
- warnln("{}", #name);
- ENUMERATE_SHELL_OPTIONS();
- #undef __ENUMERATE_SHELL_OPTION
- }
- Core::ArgsParser parser;
- #define __ENUMERATE_SHELL_OPTION(name, default_, description) \
- bool name = false; \
- bool not_##name = false; \
- parser.add_option(name, "Enable: " description, #name, '\0'); \
- parser.add_option(not_##name, "Disable: " description, "no_" #name, '\0');
- ENUMERATE_SHELL_OPTIONS();
- #undef __ENUMERATE_SHELL_OPTION
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- #define __ENUMERATE_SHELL_OPTION(name, default_, description) \
- if (name) \
- options.name = true; \
- if (not_##name) \
- options.name = false;
- ENUMERATE_SHELL_OPTIONS();
- #undef __ENUMERATE_SHELL_OPTION
- return 0;
- }
- ErrorOr<int> Shell::builtin_shift(Main::Arguments arguments)
- {
- int count = 1;
- Core::ArgsParser parser;
- parser.add_positional_argument(count, "Shift count", "count", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (count < 1)
- return 0;
- auto argv_ = TRY(lookup_local_variable("ARGV"sv));
- if (!argv_) {
- warnln("shift: ARGV is unset");
- return 1;
- }
- if (!argv_->is_list())
- argv_ = adopt_ref(*new AST::ListValue({ const_cast<AST::Value&>(*argv_) }));
- auto& values = const_cast<AST::ListValue*>(static_cast<AST::ListValue const*>(argv_.ptr()))->values();
- if ((size_t)count > values.size()) {
- warnln("shift: shift count must not be greater than {}", values.size());
- return 1;
- }
- for (auto i = 0; i < count; ++i)
- (void)values.take_first();
- return 0;
- }
- ErrorOr<int> Shell::builtin_source(Main::Arguments arguments)
- {
- StringView file_to_source;
- Vector<StringView> args;
- Core::ArgsParser parser;
- parser.add_positional_argument(file_to_source, "File to read commands from", "path");
- parser.add_positional_argument(args, "ARGV for the sourced file", "args", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments))
- return 1;
- auto previous_argv = TRY(lookup_local_variable("ARGV"sv));
- ScopeGuard guard { [&] {
- if (!args.is_empty())
- set_local_variable("ARGV", const_cast<AST::Value&>(*previous_argv));
- } };
- if (!args.is_empty()) {
- Vector<String> arguments;
- arguments.ensure_capacity(args.size());
- for (auto& arg : args)
- arguments.append(TRY(String::from_utf8(arg)));
- set_local_variable("ARGV", AST::make_ref_counted<AST::ListValue>(move(arguments)));
- }
- if (!run_file(file_to_source, true))
- return 126;
- return 0;
- }
- ErrorOr<int> Shell::builtin_time(Main::Arguments arguments)
- {
- Vector<StringView> args;
- int number_of_iterations = 1;
- Core::ArgsParser parser;
- parser.add_option(number_of_iterations, "Number of iterations", "iterations", 'n', "iterations");
- parser.set_stop_on_first_non_option(true);
- parser.add_positional_argument(args, "Command to execute with arguments", "command", Core::ArgsParser::Required::Yes);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (number_of_iterations < 1)
- return 1;
- AST::Command command;
- TRY(command.argv.try_ensure_capacity(args.size()));
- for (auto& arg : args)
- command.argv.append(TRY(String::from_utf8(arg)));
- auto commands = TRY(expand_aliases({ move(command) }));
- AK::Statistics iteration_times;
- int exit_code = 1;
- for (int i = 0; i < number_of_iterations; ++i) {
- auto timer = Core::ElapsedTimer::start_new();
- for (auto& job : run_commands(commands)) {
- block_on_job(job);
- exit_code = job->exit_code();
- }
- iteration_times.add(static_cast<float>(timer.elapsed()));
- }
- if (number_of_iterations == 1) {
- warnln("Time: {} ms", iteration_times.values().first());
- } else {
- AK::Statistics iteration_times_excluding_first;
- for (size_t i = 1; i < iteration_times.size(); i++)
- iteration_times_excluding_first.add(iteration_times.values()[i]);
- warnln("Timing report: {} ms", iteration_times.sum());
- warnln("==============");
- warnln("Command: {}", DeprecatedString::join(' ', arguments.strings));
- warnln("Average time: {:.2} ms (median: {}, stddev: {:.2}, min: {}, max:{})",
- iteration_times.average(), iteration_times.median(),
- iteration_times.standard_deviation(),
- iteration_times.min(), iteration_times.max());
- warnln("Excluding first: {:.2} ms (median: {}, stddev: {:.2}, min: {}, max:{})",
- iteration_times_excluding_first.average(), iteration_times_excluding_first.median(),
- iteration_times_excluding_first.standard_deviation(),
- iteration_times_excluding_first.min(), iteration_times_excluding_first.max());
- }
- return exit_code;
- }
- ErrorOr<int> Shell::builtin_umask(Main::Arguments arguments)
- {
- StringView mask_text;
- Core::ArgsParser parser;
- parser.add_positional_argument(mask_text, "New mask (omit to get current mask)", "octal-mask", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- if (mask_text.is_empty()) {
- mode_t old_mask = umask(0);
- printf("%#o\n", old_mask);
- umask(old_mask);
- return 0;
- }
- unsigned mask = 0;
- auto matches = true;
- // FIXME: There's currently no way to parse an StringView as an octal integer,
- // when that becomes available, replace this code.
- for (auto byte : mask_text.bytes()) {
- if (isspace(byte))
- continue;
- if (!is_ascii_octal_digit(byte)) {
- matches = false;
- break;
- }
- mask = (mask << 3) + (byte - '0');
- }
- if (matches) {
- umask(mask);
- return 0;
- }
- warnln("umask: Invalid mask '{}'", mask_text);
- return 1;
- }
- ErrorOr<int> Shell::builtin_wait(Main::Arguments arguments)
- {
- Vector<int> job_ids;
- Vector<bool> id_is_pid;
- Core::ArgsParser parser;
- parser.add_positional_argument(Core::ArgsParser::Arg {
- .help_string = "Job IDs or Jobspecs to wait for",
- .name = "job-id",
- .min_values = 0,
- .max_values = INT_MAX,
- .accept_value = [&](StringView value) -> bool {
- // Check if it's a pid (i.e. literal integer)
- if (auto number = value.to_uint(); number.has_value()) {
- job_ids.append(number.value());
- id_is_pid.append(true);
- return true;
- }
- // Check if it's a jobspec
- if (auto id = resolve_job_spec(value); id.has_value()) {
- job_ids.append(id.value());
- id_is_pid.append(false);
- return true;
- }
- return false;
- } });
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- Vector<NonnullRefPtr<Job>> jobs_to_wait_for;
- for (size_t i = 0; i < job_ids.size(); ++i) {
- auto id = job_ids[i];
- auto is_pid = id_is_pid[i];
- auto job = find_job(id, is_pid);
- if (!job)
- warnln("wait: Job with id/pid {} not found", id);
- else
- jobs_to_wait_for.append(*job);
- }
- if (job_ids.is_empty()) {
- for (auto const& it : jobs)
- jobs_to_wait_for.append(it.value);
- }
- for (auto& job : jobs_to_wait_for) {
- job->set_running_in_background(false);
- block_on_job(job);
- }
- return 0;
- }
- ErrorOr<int> Shell::builtin_unset(Main::Arguments arguments)
- {
- Vector<DeprecatedString> vars;
- Core::ArgsParser parser;
- parser.add_positional_argument(vars, "List of variables", "variables", Core::ArgsParser::Required::Yes);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::PrintUsage))
- return 1;
- bool did_touch_path = false;
- for (auto& value : vars) {
- if (!did_touch_path && value == "PATH"sv)
- did_touch_path = true;
- if (TRY(lookup_local_variable(value)) != nullptr) {
- unset_local_variable(value);
- } else {
- unsetenv(value.characters());
- }
- }
- if (did_touch_path)
- cache_path();
- return 0;
- }
- ErrorOr<int> Shell::builtin_not(Main::Arguments arguments)
- {
- Vector<StringView> args;
- Core::ArgsParser parser;
- parser.set_stop_on_first_non_option(true);
- parser.add_positional_argument(args, "Command to run followed by its arguments", "string");
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::Ignore))
- return 1;
- AST::Command command;
- TRY(command.argv.try_ensure_capacity(args.size()));
- for (auto& arg : args)
- command.argv.unchecked_append(TRY(String::from_utf8(arg)));
- auto commands = TRY(expand_aliases({ move(command) }));
- int exit_code = 1;
- auto found_a_job = false;
- for (auto& job : run_commands(commands)) {
- found_a_job = true;
- block_on_job(job);
- exit_code = job->exit_code();
- }
- // In case it was a function.
- if (!found_a_job)
- exit_code = last_return_code.value_or(0);
- return exit_code == 0 ? 1 : 0;
- }
- ErrorOr<int> Shell::builtin_kill(Main::Arguments arguments)
- {
- // Simply translate the arguments and pass them to `kill'
- Vector<String> replaced_values;
- auto kill_path = Core::DeprecatedFile::resolve_executable_from_environment("kill"sv);
- if (!kill_path.has_value()) {
- warnln("kill: `kill' not found in PATH");
- return 126;
- }
- replaced_values.append(TRY(String::from_deprecated_string(kill_path.release_value())));
- for (size_t i = 1; i < arguments.strings.size(); ++i) {
- if (auto job_id = resolve_job_spec(arguments.strings[i]); job_id.has_value()) {
- auto job = find_job(job_id.value());
- if (job) {
- replaced_values.append(TRY(String::number(job->pid())));
- } else {
- warnln("kill: Job with pid {} not found", job_id.value());
- return 1;
- }
- } else {
- replaced_values.append(TRY(String::from_utf8(arguments.strings[i])));
- }
- }
- // Now just run `kill'
- AST::Command command;
- command.argv = move(replaced_values);
- command.position = m_source_position.has_value() ? m_source_position->position : Optional<AST::Position> {};
- auto exit_code = 1;
- auto job_result = run_command(command);
- if (job_result.is_error()) {
- warnln("kill: Failed to run {}: {}", command.argv.first(), job_result.error());
- return exit_code;
- }
- if (auto job = job_result.release_value()) {
- block_on_job(job);
- exit_code = job->exit_code();
- }
- return exit_code;
- }
- ErrorOr<bool> Shell::run_builtin(const AST::Command& command, Vector<NonnullRefPtr<AST::Rewiring>> const& rewirings, int& retval)
- {
- if (command.argv.is_empty())
- return false;
- if (!has_builtin(command.argv.first()))
- return false;
- Vector<StringView> arguments;
- TRY(arguments.try_ensure_capacity(command.argv.size()));
- for (auto& arg : command.argv)
- arguments.unchecked_append(arg);
- Main::Arguments arguments_object {
- .argc = 0,
- .argv = nullptr,
- .strings = arguments
- };
- StringView name = command.argv.first();
- SavedFileDescriptors fds { rewirings };
- for (auto& rewiring : rewirings) {
- int rc = dup2(rewiring->old_fd, rewiring->new_fd);
- if (rc < 0) {
- perror("dup2(run)");
- return false;
- }
- }
- Core::EventLoop loop;
- setup_signals();
- if (name == ":"sv)
- name = "noop"sv;
- #define __ENUMERATE_SHELL_BUILTIN(builtin) \
- if (name == #builtin) { \
- retval = TRY(builtin_##builtin(arguments_object)); \
- if (!has_error(ShellError::None)) \
- raise_error(m_error, m_error_description, command.position); \
- fflush(stdout); \
- fflush(stderr); \
- return true; \
- }
- ENUMERATE_SHELL_BUILTINS();
- #undef __ENUMERATE_SHELL_BUILTIN
- return false;
- }
- ErrorOr<int> Shell::builtin_argsparser_parse(Main::Arguments arguments)
- {
- // argsparser_parse
- // --add-option variable [--type (bool | string | i32 | u32 | double | size)] --help-string "" --long-name "" --short-name "" [--value-name "" <if not --type bool>] --list
- // --add-positional-argument variable [--type (bool | string | i32 | u32 | double | size)] ([--min n] [--max n] | [--required]) --help-string "" --value-name ""
- // [--general-help ""]
- // [--stop-on-first-non-option]
- // --
- // $args_to_parse
- Core::ArgsParser parser;
- Core::ArgsParser user_parser;
- Vector<StringView> descriptors;
- Variant<Core::ArgsParser::Option, Core::ArgsParser::Arg, Empty> current;
- DeprecatedString help_string_storage;
- DeprecatedString long_name_storage;
- DeprecatedString value_name_storage;
- DeprecatedString name_storage;
- DeprecatedString current_variable;
- // if max > 1 or min < 1, or explicit `--list`.
- bool treat_arg_as_list = false;
- enum class Type {
- Bool,
- String,
- I32,
- U32,
- Double,
- Size,
- };
- auto type = Type::String;
- auto try_convert = [](StringView value, Type type) -> ErrorOr<Optional<RefPtr<AST::Value>>> {
- switch (type) {
- case Type::Bool:
- return AST::make_ref_counted<AST::StringValue>(TRY("true"_string));
- case Type::String:
- return AST::make_ref_counted<AST::StringValue>(TRY(String::from_utf8(value)));
- case Type::I32:
- if (auto number = value.to_int(); number.has_value())
- return AST::make_ref_counted<AST::StringValue>(TRY(String::number(*number)));
- warnln("Invalid value for type i32: {}", value);
- return OptionalNone {};
- case Type::U32:
- case Type::Size:
- if (auto number = value.to_uint(); number.has_value())
- return AST::make_ref_counted<AST::StringValue>(TRY(String::number(*number)));
- warnln("Invalid value for type u32|size: {}", value);
- return OptionalNone {};
- case Type::Double: {
- DeprecatedString string = value;
- char* endptr = nullptr;
- auto number = strtod(string.characters(), &endptr);
- if (endptr != string.characters() + string.length()) {
- warnln("Invalid value for type double: {}", value);
- return OptionalNone {};
- }
- return AST::make_ref_counted<AST::StringValue>(TRY(String::number(number)));
- }
- default:
- VERIFY_NOT_REACHED();
- }
- };
- auto enlist = [&](auto name, auto value) -> ErrorOr<NonnullRefPtr<AST::Value>> {
- auto variable = TRY(lookup_local_variable(name));
- if (variable) {
- auto list = TRY(const_cast<AST::Value&>(*variable).resolve_as_list(*this));
- auto new_value = TRY(value->resolve_as_string(*this));
- list.append(move(new_value));
- return try_make_ref_counted<AST::ListValue>(move(list));
- }
- return *value;
- };
- // FIXME: We cannot return ErrorOr<T> from Core::ArgsParser::Option::accept_value(), fix the MUST's here whenever that function is ErrorOr-aware.
- auto commit = [&] {
- return current.visit(
- [&](Core::ArgsParser::Option& option) {
- if (!option.long_name && !option.short_name) {
- warnln("Defined option must have at least one of --long-name or --short-name");
- return false;
- }
- option.accept_value = [&, current_variable, treat_arg_as_list, type](StringView value) {
- auto result = MUST(try_convert(value, type));
- if (result.has_value()) {
- auto value = result.release_value();
- if (treat_arg_as_list)
- value = MUST(enlist(current_variable, move(value)));
- this->set_local_variable(current_variable, move(value), true);
- return true;
- }
- return false;
- };
- user_parser.add_option(move(option));
- type = Type::String;
- treat_arg_as_list = false;
- return true;
- },
- [&](Core::ArgsParser::Arg& arg) {
- if (!arg.name) {
- warnln("Defined positional argument must have a name");
- return false;
- }
- arg.accept_value = [&, current_variable, treat_arg_as_list, type](StringView value) {
- auto result = MUST(try_convert(value, type));
- if (result.has_value()) {
- auto value = result.release_value();
- if (treat_arg_as_list)
- value = MUST(enlist(current_variable, move(value)));
- this->set_local_variable(current_variable, move(value), true);
- return true;
- }
- return false;
- };
- user_parser.add_positional_argument(move(arg));
- type = Type::String;
- treat_arg_as_list = false;
- return true;
- },
- [&](Empty) {
- return true;
- });
- };
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::None,
- .help_string = "Stop processing descriptors after a non-argument parameter is seen",
- .long_name = "stop-on-first-non-option",
- .accept_value = [&](auto) {
- user_parser.set_stop_on_first_non_option(true);
- return true;
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
- .help_string = "Set the general help string for the parser",
- .long_name = "general-help",
- .value_name = "string",
- .accept_value = [&](StringView value) {
- VERIFY(strlen(value.characters_without_null_termination()) == value.length());
- user_parser.set_general_help(value.characters_without_null_termination());
- return true;
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
- .help_string = "Start describing an option",
- .long_name = "add-option",
- .value_name = "variable-name",
- .accept_value = [&](auto name) {
- if (!commit())
- return false;
- current = Core::ArgsParser::Option {};
- current_variable = name;
- if (current_variable.is_empty() || !all_of(current_variable, [](auto ch) { return ch == '_' || isalnum(ch); })) {
- warnln("Option variable name must be a valid identifier");
- return false;
- }
- return true;
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::None,
- .help_string = "Accept multiple of the current option being given",
- .long_name = "list",
- .accept_value = [&](auto) {
- if (!current.has<Core::ArgsParser::Option>()) {
- warnln("Must be defining an option to use --list");
- return false;
- }
- treat_arg_as_list = true;
- return true;
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
- .help_string = "Define the type of the option or argument being described",
- .long_name = "type",
- .value_name = "type",
- .accept_value = [&](StringView ty) {
- if (current.has<Empty>()) {
- warnln("Must be defining an argument or option to use --type");
- return false;
- }
- if (ty == "bool") {
- if (auto option = current.get_pointer<Core::ArgsParser::Option>()) {
- if (option->value_name != nullptr) {
- warnln("Type 'bool' does not apply to options with a value (value name is set to {})", option->value_name);
- return false;
- }
- }
- type = Type::Bool;
- } else if (ty == "string") {
- type = Type::String;
- } else if (ty == "i32") {
- type = Type::I32;
- } else if (ty == "u32") {
- type = Type::U32;
- } else if (ty == "double") {
- type = Type::Double;
- } else if (ty == "size") {
- type = Type::Size;
- } else {
- warnln("Invalid type '{}', expected one of bool | string | i32 | u32 | double | size", ty);
- return false;
- }
- if (type == Type::Bool) {
- set_local_variable(
- current_variable,
- make_ref_counted<AST::StringValue>(MUST("false"_string)),
- true);
- }
- return true;
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
- .help_string = "Set the help string of the option or argument being defined",
- .long_name = "help-string",
- .value_name = "string",
- .accept_value = [&](StringView value) {
- return current.visit(
- [](Empty) {
- warnln("Must be defining an option or argument to use --help-string");
- return false;
- },
- [&](auto& option) {
- help_string_storage = value;
- option.help_string = help_string_storage.characters();
- return true;
- });
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
- .help_string = "Set the long name of the option being defined",
- .long_name = "long-name",
- .value_name = "name",
- .accept_value = [&](StringView value) {
- auto option = current.get_pointer<Core::ArgsParser::Option>();
- if (!option) {
- warnln("Must be defining an option to use --long-name");
- return false;
- }
- if (option->long_name) {
- warnln("Repeated application of --long-name is not allowed, current option has long name set to \"{}\"", option->long_name);
- return false;
- }
- long_name_storage = value;
- option->long_name = long_name_storage.characters();
- return true;
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
- .help_string = "Set the short name of the option being defined",
- .long_name = "short-name",
- .value_name = "char",
- .accept_value = [&](StringView value) {
- auto option = current.get_pointer<Core::ArgsParser::Option>();
- if (!option) {
- warnln("Must be defining an option to use --short-name");
- return false;
- }
- if (value.length() != 1) {
- warnln("Option short name ('{}') must be exactly one character long", value);
- return false;
- }
- if (option->short_name) {
- warnln("Repeated application of --short-name is not allowed, current option has short name set to '{}'", option->short_name);
- return false;
- }
- option->short_name = value[0];
- return true;
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
- .help_string = "Set the value name of the option being defined",
- .long_name = "value-name",
- .value_name = "string",
- .accept_value = [&](StringView value) {
- return current.visit(
- [](Empty) {
- warnln("Must be defining an option or a positional argument to use --value-name");
- return false;
- },
- [&](Core::ArgsParser::Option& option) {
- if (option.value_name) {
- warnln("Repeated application of --value-name is not allowed, current option has value name set to \"{}\"", option.value_name);
- return false;
- }
- if (type == Type::Bool) {
- warnln("Options of type bool cannot have a value name");
- return false;
- }
- value_name_storage = value;
- option.value_name = value_name_storage.characters();
- return true;
- },
- [&](Core::ArgsParser::Arg& arg) {
- if (arg.name) {
- warnln("Repeated application of --value-name is not allowed, current argument has value name set to \"{}\"", arg.name);
- return false;
- }
- name_storage = value;
- arg.name = name_storage.characters();
- return true;
- });
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
- .help_string = "Start describing a positional argument",
- .long_name = "add-positional-argument",
- .value_name = "variable",
- .accept_value = [&](auto value) {
- if (!commit())
- return false;
- current = Core::ArgsParser::Arg {};
- current_variable = value;
- if (current_variable.is_empty() || !all_of(current_variable, [](auto ch) { return ch == '_' || isalnum(ch); })) {
- warnln("Argument variable name must be a valid identifier");
- return false;
- }
- return true;
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
- .help_string = "Set the minimum required number of positional descriptors for the argument being described",
- .long_name = "min",
- .value_name = "n",
- .accept_value = [&](StringView value) {
- auto arg = current.get_pointer<Core::ArgsParser::Arg>();
- if (!arg) {
- warnln("Must be describing a positional argument to use --min");
- return false;
- }
- auto number = value.to_uint();
- if (!number.has_value()) {
- warnln("Invalid value for --min: '{}', expected a non-negative number", value);
- return false;
- }
- if (static_cast<unsigned>(arg->max_values) < *number) {
- warnln("Invalid value for --min: {}, min must not be larger than max ({})", *number, arg->max_values);
- return false;
- }
- arg->min_values = *number;
- treat_arg_as_list = arg->max_values > 1 || arg->min_values < 1;
- return true;
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::Required,
- .help_string = "Set the maximum required number of positional descriptors for the argument being described",
- .long_name = "max",
- .value_name = "n",
- .accept_value = [&](StringView value) {
- auto arg = current.get_pointer<Core::ArgsParser::Arg>();
- if (!arg) {
- warnln("Must be describing a positional argument to use --max");
- return false;
- }
- auto number = value.to_uint();
- if (!number.has_value()) {
- warnln("Invalid value for --max: '{}', expected a non-negative number", value);
- return false;
- }
- if (static_cast<unsigned>(arg->min_values) > *number) {
- warnln("Invalid value for --max: {}, max must not be smaller than min ({})", *number, arg->min_values);
- return false;
- }
- arg->max_values = *number;
- treat_arg_as_list = arg->max_values > 1 || arg->min_values < 1;
- return true;
- },
- });
- parser.add_option(Core::ArgsParser::Option {
- .argument_mode = Core::ArgsParser::OptionArgumentMode::None,
- .help_string = "Mark the positional argument being described as required (shorthand for --min 1)",
- .long_name = "required",
- .accept_value = [&](auto) {
- auto arg = current.get_pointer<Core::ArgsParser::Arg>();
- if (!arg) {
- warnln("Must be describing a positional argument to use --required");
- return false;
- }
- arg->min_values = 1;
- if (arg->max_values < arg->min_values)
- arg->max_values = 1;
- treat_arg_as_list = arg->max_values > 1 || arg->min_values < 1;
- return true;
- },
- });
- parser.add_positional_argument(descriptors, "Arguments to parse via the described ArgsParser configuration", "arg", Core::ArgsParser::Required::No);
- if (!parser.parse(arguments, Core::ArgsParser::FailureBehavior::Ignore))
- return 2;
- if (!commit())
- return 2;
- if (!user_parser.parse(descriptors, Core::ArgsParser::FailureBehavior::Ignore))
- return 1;
- return 0;
- }
- bool Shell::has_builtin(StringView name) const
- {
- if (name == ":"sv)
- return true;
- #define __ENUMERATE_SHELL_BUILTIN(builtin) \
- if (name == #builtin) { \
- return true; \
- }
- ENUMERATE_SHELL_BUILTINS();
- #undef __ENUMERATE_SHELL_BUILTIN
- return false;
- }
- }
|