Mirrors/fish-shell
2
0
Fork 0
mirror of https://github.com/fish-shell/fish-shell synced 2025-01-15 14:34:05 +00:00
Code Issues Projects Releases Packages Wiki Activity

builtin: correctly flush streams after running Rust builtin

Browse source
This commit is contained in:
Xiretza 2023-02-05 23:52:25 +01:00 committed by Johannes Altmanninger
parent 0b160ebe71
commit cfb5bb2505
1 changed files with 36 additions and 36 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
src/builtin.cpp
View file

@ -82,7 +82,7 @@
#include "wutil.h" // IWYU pragma: keep #include "wutil.h" // IWYU pragma: keep
static maybe_t<RustBuiltin> try_get_rust_builtin(const wcstring &cmd); static maybe_t<RustBuiltin> try_get_rust_builtin(const wcstring &cmd);
static proc_status_t builtin_run_rust(parser_t &parser, io_streams_t &streams, static maybe_t<int> builtin_run_rust(parser_t &parser, io_streams_t &streams,
const wcstring_list_t &argv, RustBuiltin builtin); const wcstring_list_t &argv, RustBuiltin builtin);
/// Counts the number of arguments in the specified null-terminated array /// Counts the number of arguments in the specified null-terminated array
@ -452,11 +452,6 @@ proc_status_t builtin_run(parser_t &parser, const wcstring_list_t &argv, io_stre
if (argv.empty()) return proc_status_t::from_exit_code(STATUS_INVALID_ARGS); if (argv.empty()) return proc_status_t::from_exit_code(STATUS_INVALID_ARGS);
const wcstring &cmdname = argv.front(); const wcstring &cmdname = argv.front();
auto rust_builtin = try_get_rust_builtin(cmdname);
if (rust_builtin.has_value()) {
return builtin_run_rust(parser, streams, argv, *rust_builtin);
}
// We can be handed a keyword by the parser as if it was a command. This happens when the user // We can be handed a keyword by the parser as if it was a command. This happens when the user
// follows the keyword by `-h` or `--help`. Since it isn't really a builtin command we need to // follows the keyword by `-h` or `--help`. Since it isn't really a builtin command we need to
// handle displaying help for it here. // handle displaying help for it here.
@ -465,10 +460,19 @@ proc_status_t builtin_run(parser_t &parser, const wcstring_list_t &argv, io_stre
return proc_status_t::from_exit_code(STATUS_CMD_OK); return proc_status_t::from_exit_code(STATUS_CMD_OK);
} }
if (const builtin_data_t *data = builtin_lookup(cmdname)) { maybe_t<int> builtin_ret;
auto rust_builtin = try_get_rust_builtin(cmdname);
if (rust_builtin.has_value()) {
builtin_ret = builtin_run_rust(parser, streams, argv, *rust_builtin);
} else if (const builtin_data_t *data = builtin_lookup(cmdname)) {
// Construct the permutable argv array which the builtin expects, and execute the builtin. // Construct the permutable argv array which the builtin expects, and execute the builtin.
null_terminated_array_t<wchar_t> argv_arr(argv); null_terminated_array_t<wchar_t> argv_arr(argv);
maybe_t<int> builtin_ret = data->func(parser, streams, argv_arr.get()); builtin_ret = data->func(parser, streams, argv_arr.get());
} else {
FLOGF(error, UNKNOWN_BUILTIN_ERR_MSG, cmdname.c_str());
return proc_status_t::from_exit_code(STATUS_CMD_ERROR);
}
// Flush our out and error streams, and check for their errors. // Flush our out and error streams, and check for their errors.
int out_ret = streams.out.flush_and_check_error(); int out_ret = streams.out.flush_and_check_error();
@ -493,10 +497,6 @@ proc_status_t builtin_run(parser_t &parser, const wcstring_list_t &argv, io_stre
FLOGF(warning, "builtin %ls returned invalid exit code %d", cmdname.c_str(), code); FLOGF(warning, "builtin %ls returned invalid exit code %d", cmdname.c_str(), code);
} }
return proc_status_t::from_exit_code(code); return proc_status_t::from_exit_code(code);
}
FLOGF(error, UNKNOWN_BUILTIN_ERR_MSG, cmdname.c_str());
return proc_status_t::from_exit_code(STATUS_CMD_ERROR);
} }
/// Returns a list of all builtin names. /// Returns a list of all builtin names.
@ -542,5 +542,5 @@ static proc_status_t builtin_run_rust(parser_t &parser, io_streams_t &streams,
rust_argv.emplace_back(arg.c_str()); rust_argv.emplace_back(arg.c_str());
} }
rust_run_builtin(parser, streams, rust_argv, builtin); rust_run_builtin(parser, streams, rust_argv, builtin);
return proc_status_t{}; return none();
} }