diff options
author | Andreas Kling <kling@serenityos.org> | 2021-01-12 11:53:14 +0100 |
---|---|---|
committer | Andreas Kling <kling@serenityos.org> | 2021-01-12 12:04:07 +0100 |
commit | c4e2fd8123b83a6b890b410741f1d416a1162fdb (patch) | |
tree | be0ae8ba44fec67223809237807597273c092282 /Userland/Shell/Builtin.cpp | |
parent | 07c7e353722d880b565188dba9f81a0449baf80c (diff) | |
download | serenity-c4e2fd8123b83a6b890b410741f1d416a1162fdb.zip |
Shell: Move to Userland/Shell/
Diffstat (limited to 'Userland/Shell/Builtin.cpp')
-rw-r--r-- | Userland/Shell/Builtin.cpp | 929 |
1 files changed, 929 insertions, 0 deletions
diff --git a/Userland/Shell/Builtin.cpp b/Userland/Shell/Builtin.cpp new file mode 100644 index 0000000000..d3a48b5089 --- /dev/null +++ b/Userland/Shell/Builtin.cpp @@ -0,0 +1,929 @@ +/* + * Copyright (c) 2020, The SerenityOS developers. + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the above copyright notice, this + * list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright notice, + * this list of conditions and the following disclaimer in the documentation + * and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE + * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE + * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL + * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER + * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, + * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include "AST.h" +#include "Shell.h" +#include <AK/LexicalPath.h> +#include <AK/ScopeGuard.h> +#include <LibCore/ArgsParser.h> +#include <LibCore/EventLoop.h> +#include <LibCore/File.h> +#include <inttypes.h> +#include <signal.h> +#include <sys/wait.h> +#include <unistd.h> + +extern char** environ; + +namespace Shell { + +int Shell::builtin_alias(int argc, const char** argv) +{ + Vector<const char*> arguments; + + Core::ArgsParser parser; + parser.add_positional_argument(arguments, "List of name[=values]'s", "name[=value]", Core::ArgsParser::Required::No); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + if (arguments.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 : arguments) { + auto parts = String { argument }.split_limit('=', 2, true); + 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(parts[0]); + } + } + + return fail ? 1 : 0; +} + +int Shell::builtin_bg(int argc, const char** argv) +{ + int job_id = -1; + + Core::ArgsParser parser; + parser.add_positional_argument(job_id, "Job ID to run in background", "job-id", Core::ArgsParser::Required::No); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + if (job_id == -1 && !jobs.is_empty()) + job_id = find_last_job_id(); + + auto* job = const_cast<Job*>(find_job(job_id)); + + if (!job) { + if (job_id == -1) { + fprintf(stderr, "bg: no current job\n"); + } else { + fprintf(stderr, "bg: job with id %d not found\n", job_id); + } + return 1; + } + + job->set_running_in_background(true); + job->set_should_announce_exit(true); + job->set_is_suspended(false); + + dbgln("Resuming {} ({})", job->pid(), job->cmd()); + warnln("Resuming job {} - {}", job->job_id(), job->cmd().characters()); + + // 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; +} + +int Shell::builtin_cd(int argc, const char** argv) +{ + const char* arg_path = nullptr; + + Core::ArgsParser parser; + parser.add_positional_argument(arg_path, "Path to change to", "path", Core::ArgsParser::Required::No); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + String new_path; + + if (!arg_path) { + new_path = home; + } else { + if (strcmp(arg_path, "-") == 0) { + char* oldpwd = getenv("OLDPWD"); + if (oldpwd == nullptr) + return 1; + new_path = oldpwd; + } else if (arg_path[0] == '/') { + new_path = argv[1]; + } else { + StringBuilder builder; + builder.append(cwd); + builder.append('/'); + builder.append(arg_path); + new_path = builder.to_string(); + } + } + + auto real_path = Core::File::real_path_for(new_path); + if (real_path.is_empty()) { + fprintf(stderr, "Invalid path '%s'\n", new_path.characters()); + return 1; + } + + if (cd_history.is_empty() || cd_history.last() != real_path) + cd_history.enqueue(real_path); + + const char* path = real_path.characters(); + + int rc = chdir(path); + if (rc < 0) { + if (errno == ENOTDIR) { + fprintf(stderr, "Not a directory: %s\n", path); + } else { + fprintf(stderr, "chdir(%s) failed: %s\n", path, strerror(errno)); + } + return 1; + } + setenv("OLDPWD", cwd.characters(), 1); + cwd = real_path; + setenv("PWD", cwd.characters(), 1); + return 0; +} + +int Shell::builtin_cdh(int argc, const char** argv) +{ + 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(argc, const_cast<char**>(argv), false)) + return 1; + + if (index == -1) { + if (cd_history.is_empty()) { + fprintf(stderr, "cdh: no history available\n"); + return 0; + } + + for (ssize_t i = cd_history.size() - 1; i >= 0; --i) + printf("%lu: %s\n", cd_history.size() - i, cd_history.at(i).characters()); + return 0; + } + + if (index < 1 || (size_t)index > cd_history.size()) { + fprintf(stderr, "cdh: history index out of bounds: %d not in (0, %zu)\n", index, cd_history.size()); + return 1; + } + + const char* path = cd_history.at(cd_history.size() - index).characters(); + const char* cd_args[] = { "cd", path, nullptr }; + return Shell::builtin_cd(2, cd_args); +} + +int Shell::builtin_dirs(int argc, const char** argv) +{ + // 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<const char*> 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(argc, const_cast<char**>(argv), false)) + return 1; + + // -v implies -p + print = print || number_when_printing; + + if (print) { + if (!paths.is_empty()) { + fprintf(stderr, "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; +} + +int Shell::builtin_exec(int argc, const char** argv) +{ + if (argc < 2) { + fprintf(stderr, "Shell: No command given to exec\n"); + return 1; + } + + Vector<const char*> argv_vector; + argv_vector.append(argv + 1, argc - 1); + argv_vector.append(nullptr); + + execute_process(move(argv_vector)); +} + +int Shell::builtin_exit(int argc, const char** argv) +{ + int exit_code = 0; + Core::ArgsParser parser; + parser.add_positional_argument(exit_code, "Exit code", "code", Core::ArgsParser::Required::No); + if (!parser.parse(argc, const_cast<char**>(argv))) + return 1; + + if (m_is_interactive) { + if (!jobs.is_empty()) { + if (!m_should_ignore_jobs_on_next_exit) { + fprintf(stderr, "Shell: You have %zu active job%s, run 'exit' again to really exit.\n", jobs.size(), jobs.size() > 1 ? "s" : ""); + m_should_ignore_jobs_on_next_exit = true; + return 1; + } + } + } + stop_all_jobs(); + m_editor->save_history(get_history_path()); + if (m_is_interactive) + printf("Good-bye!\n"); + exit(exit_code); + return 0; +} + +int Shell::builtin_export(int argc, const char** argv) +{ + Vector<const char*> vars; + + Core::ArgsParser parser; + parser.add_positional_argument(vars, "List of variable[=value]'s", "values", Core::ArgsParser::Required::No); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + 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 = String { value }.split_limit('=', 2); + + if (parts.size() == 1) { + auto value = lookup_local_variable(parts[0]); + if (value) { + auto values = value->resolve_as_list(*this); + StringBuilder builder; + builder.join(" ", values); + parts.append(builder.to_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; +} + +int Shell::builtin_glob(int argc, const char** argv) +{ + Vector<const char*> globs; + Core::ArgsParser parser; + parser.add_positional_argument(globs, "Globs to resolve", "glob"); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + for (auto& glob : globs) { + for (auto& expanded : expand_globs(glob, cwd)) + outln("{}", expanded); + } + + return 0; +} + +int Shell::builtin_fg(int argc, const char** argv) +{ + int job_id = -1; + + Core::ArgsParser parser; + parser.add_positional_argument(job_id, "Job ID to bring to foreground", "job-id", Core::ArgsParser::Required::No); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + if (job_id == -1 && !jobs.is_empty()) + job_id = find_last_job_id(); + + RefPtr<Job> job = find_job(job_id); + + if (!job) { + if (job_id == -1) { + fprintf(stderr, "fg: no current job\n"); + } else { + fprintf(stderr, "fg: job with id %d not found\n", job_id); + } + return 1; + } + + job->set_running_in_background(false); + job->set_is_suspended(false); + + dbgln("Resuming {} ({})", job->pid(), job->cmd()); + warnln("Resuming job {} - {}", job->job_id(), job->cmd().characters()); + + 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; +} + +int Shell::builtin_disown(int argc, const char** argv) +{ + Vector<const char*> str_job_ids; + + Core::ArgsParser parser; + parser.add_positional_argument(str_job_ids, "Id of the jobs to disown (omit for current job)", "job_ids", Core::ArgsParser::Required::No); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + Vector<size_t> job_ids; + for (auto& job_id : str_job_ids) { + auto id = StringView(job_id).to_uint(); + if (id.has_value()) + job_ids.append(id.value()); + else + fprintf(stderr, "disown: Invalid job id %s\n", job_id); + } + + if (job_ids.is_empty()) + job_ids.append(find_last_job_id()); + + Vector<const Job*> jobs_to_disown; + + for (auto id : job_ids) { + auto job = find_job(id); + if (!job) + fprintf(stderr, "disown: job with id %zu not found\n", id); + else + jobs_to_disown.append(job); + } + + if (jobs_to_disown.is_empty()) { + if (str_job_ids.is_empty()) + fprintf(stderr, "disown: no current job\n"); + // 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()) + fprintf(stderr, "disown warning: job %" PRIu64 " is currently not running, 'kill -%d %d' to make it continue\n", job->job_id(), SIGCONT, job->pid()); + + jobs.remove(job->pid()); + } + + return 0; +} + +int Shell::builtin_history(int, const char**) +{ + for (size_t i = 0; i < m_editor->history().size(); ++i) { + printf("%6zu %s\n", i, m_editor->history()[i].entry.characters()); + } + return 0; +} + +int Shell::builtin_jobs(int argc, const char** argv) +{ + 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(argc, const_cast<char**>(argv), false)) + 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; +} + +int Shell::builtin_popd(int argc, const char** argv) +{ + if (directory_stack.size() <= 1) { + fprintf(stderr, "Shell: popd: directory stack empty\n"); + return 1; + } + + bool should_not_switch = false; + String path = directory_stack.take_last(); + + Core::ArgsParser parser; + parser.add_option(should_not_switch, "Do not switch dirs", "no-switch", 'n'); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + bool should_switch = !should_not_switch; + + // When no arguments are given, popd removes the top directory from the stack and performs a cd to the new top directory. + if (argc == 1) { + int rc = chdir(path.characters()); + if (rc < 0) { + fprintf(stderr, "chdir(%s) failed: %s\n", path.characters(), strerror(errno)); + return 1; + } + + cwd = path; + return 0; + } + + LexicalPath lexical_path(path.characters()); + if (!lexical_path.is_valid()) { + fprintf(stderr, "LexicalPath failed to canonicalize '%s'\n", path.characters()); + return 1; + } + + const char* real_path = lexical_path.string().characters(); + + struct stat st; + int rc = stat(real_path, &st); + if (rc < 0) { + fprintf(stderr, "stat(%s) failed: %s\n", real_path, strerror(errno)); + return 1; + } + + if (!S_ISDIR(st.st_mode)) { + fprintf(stderr, "Not a directory: %s\n", real_path); + return 1; + } + + if (should_switch) { + int rc = chdir(real_path); + if (rc < 0) { + fprintf(stderr, "chdir(%s) failed: %s\n", real_path, strerror(errno)); + return 1; + } + + cwd = lexical_path.string(); + } + + return 0; +} + +int Shell::builtin_pushd(int argc, const char** argv) +{ + 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 (argc == 1) { + if (directory_stack.size() < 2) { + fprintf(stderr, "pushd: no other directory\n"); + return 1; + } + + String dir1 = directory_stack.take_first(); + String dir2 = directory_stack.take_first(); + directory_stack.insert(0, dir2); + directory_stack.insert(1, dir1); + + int rc = chdir(dir2.characters()); + if (rc < 0) { + fprintf(stderr, "chdir(%s) failed: %s\n", dir2.characters(), strerror(errno)); + return 1; + } + + cwd = dir2; + + return 0; + } + + // Let's assume the user's typed in 'pushd <dir>' + if (argc == 2) { + directory_stack.append(cwd.characters()); + if (argv[1][0] == '/') { + path_builder.append(argv[1]); + } else { + path_builder.appendf("%s/%s", cwd.characters(), argv[1]); + } + } else if (argc == 3) { + directory_stack.append(cwd.characters()); + for (int i = 1; i < argc; i++) { + const char* arg = argv[i]; + + if (arg[0] != '-') { + if (arg[0] == '/') { + path_builder.append(arg); + } else + path_builder.appendf("%s/%s", cwd.characters(), arg); + } + + if (!strcmp(arg, "-n")) + should_switch = false; + } + } + + LexicalPath lexical_path(path_builder.to_string()); + if (!lexical_path.is_valid()) { + fprintf(stderr, "LexicalPath failed to canonicalize '%s'\n", path_builder.to_string().characters()); + return 1; + } + + const char* real_path = lexical_path.string().characters(); + + struct stat st; + int rc = stat(real_path, &st); + if (rc < 0) { + fprintf(stderr, "stat(%s) failed: %s\n", real_path, strerror(errno)); + return 1; + } + + if (!S_ISDIR(st.st_mode)) { + fprintf(stderr, "Not a directory: %s\n", real_path); + return 1; + } + + if (should_switch) { + int rc = chdir(real_path); + if (rc < 0) { + fprintf(stderr, "chdir(%s) failed: %s\n", real_path, strerror(errno)); + return 1; + } + + cwd = lexical_path.string(); + } + + return 0; +} + +int Shell::builtin_pwd(int, const char**) +{ + print_path(cwd); + fputc('\n', stdout); + return 0; +} + +int Shell::builtin_setopt(int argc, const char** argv) +{ + if (argc == 1) { +#define __ENUMERATE_SHELL_OPTION(name, default_, description) \ + if (options.name) \ + fprintf(stderr, #name "\n"); + + 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(argc, const_cast<char**>(argv), false)) + 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; +} + +int Shell::builtin_shift(int argc, const char** argv) +{ + int count = 1; + + Core::ArgsParser parser; + parser.add_positional_argument(count, "Shift count", "count", Core::ArgsParser::Required::No); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + if (count < 1) + return 0; + + auto argv_ = lookup_local_variable("ARGV"); + if (!argv_) { + fprintf(stderr, "shift: ARGV is unset\n"); + return 1; + } + + if (!argv_->is_list()) + argv_ = adopt(*new AST::ListValue({ argv_.release_nonnull() })); + + auto& values = static_cast<AST::ListValue*>(argv_.ptr())->values(); + if ((size_t)count > values.size()) { + fprintf(stderr, "shift: shift count must not be greater than %zu\n", values.size()); + return 1; + } + + for (auto i = 0; i < count; ++i) + values.take_first(); + + return 0; +} + +int Shell::builtin_source(int argc, const char** argv) +{ + const char* file_to_source = nullptr; + Vector<const char*> 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(argc, const_cast<char**>(argv))) + return 1; + + Vector<String> string_argv; + for (auto& arg : args) + string_argv.append(arg); + + auto previous_argv = lookup_local_variable("ARGV"); + ScopeGuard guard { [&] { + if (!args.is_empty()) + set_local_variable("ARGV", move(previous_argv)); + } }; + + if (!args.is_empty()) + set_local_variable("ARGV", AST::create<AST::ListValue>(move(string_argv))); + + if (!run_file(file_to_source, true)) + return 126; + + return 0; +} + +int Shell::builtin_time(int argc, const char** argv) +{ + Vector<const char*> args; + + Core::ArgsParser parser; + parser.add_positional_argument(args, "Command to execute with arguments", "command", Core::ArgsParser::Required::Yes); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + AST::Command command; + for (auto& arg : args) + command.argv.append(arg); + + auto commands = expand_aliases({ move(command) }); + + Core::ElapsedTimer timer; + int exit_code = 1; + timer.start(); + for (auto& job : run_commands(commands)) { + block_on_job(job); + exit_code = job.exit_code(); + } + fprintf(stderr, "Time: %d ms\n", timer.elapsed()); + return exit_code; +} + +int Shell::builtin_umask(int argc, const char** argv) +{ + const char* mask_text = nullptr; + + 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(argc, const_cast<char**>(argv), false)) + return 1; + + if (!mask_text) { + mode_t old_mask = umask(0); + printf("%#o\n", old_mask); + umask(old_mask); + return 0; + } + + unsigned mask; + int matches = sscanf(mask_text, "%o", &mask); + if (matches == 1) { + umask(mask); + return 0; + } + + fprintf(stderr, "umask: Invalid mask '%s'\n", mask_text); + return 1; +} + +int Shell::builtin_wait(int argc, const char** argv) +{ + Vector<const char*> job_ids; + + Core::ArgsParser parser; + parser.add_positional_argument(job_ids, "Job IDs to wait for, defaults to all jobs if missing", "jobs", Core::ArgsParser::Required::No); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + Vector<NonnullRefPtr<Job>> jobs_to_wait_for; + + if (job_ids.is_empty()) { + for (auto it : jobs) + jobs_to_wait_for.append(it.value); + } else { + for (String id_s : job_ids) { + auto id_opt = id_s.to_uint(); + if (id_opt.has_value()) { + if (auto job = find_job(id_opt.value())) { + jobs_to_wait_for.append(*job); + continue; + } + } + + warnln("wait: invalid or nonexistent job id {}", id_s); + return 1; + } + } + + for (auto& job : jobs_to_wait_for) { + job->set_running_in_background(false); + block_on_job(job); + } + + return 0; +} + +int Shell::builtin_unset(int argc, const char** argv) +{ + Vector<const char*> vars; + + Core::ArgsParser parser; + parser.add_positional_argument(vars, "List of variables", "variables", Core::ArgsParser::Required::Yes); + + if (!parser.parse(argc, const_cast<char**>(argv), false)) + return 1; + + for (auto& value : vars) { + if (lookup_local_variable(value)) { + unset_local_variable(value); + } else { + unsetenv(value); + } + } + + return 0; +} + +bool Shell::run_builtin(const AST::Command& command, const NonnullRefPtrVector<AST::Rewiring>& rewirings, int& retval) +{ + if (command.argv.is_empty()) + return false; + + if (!has_builtin(command.argv.first())) + return false; + + Vector<const char*> argv; + for (auto& arg : command.argv) + argv.append(arg.characters()); + + argv.append(nullptr); + + 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(); + +#define __ENUMERATE_SHELL_BUILTIN(builtin) \ + if (name == #builtin) { \ + retval = builtin_##builtin(argv.size() - 1, argv.data()); \ + if (!has_error(ShellError::None)) \ + raise_error(m_error, m_error_description, command.position); \ + return true; \ + } + + ENUMERATE_SHELL_BUILTINS(); + +#undef __ENUMERATE_SHELL_BUILTIN + return false; +} + +bool Shell::has_builtin(const StringView& name) const +{ +#define __ENUMERATE_SHELL_BUILTIN(builtin) \ + if (name == #builtin) { \ + return true; \ + } + + ENUMERATE_SHELL_BUILTINS(); + +#undef __ENUMERATE_SHELL_BUILTIN + return false; +} + +} |