| // Copyright 2019 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "src/developer/cmd/app.h" |
| |
| #include <lib/cmdline/args_parser.h> |
| #include <unistd.h> |
| #include <zircon/status.h> |
| |
| #include <memory> |
| #include <utility> |
| |
| namespace cmd { |
| |
| const char kHelpIntro[] = R"(cmd [-c <command> ] |
| |
| A command line interface for Fuchsia. |
| |
| Options |
| |
| )"; |
| |
| const char kHelpHelp[] = R"( --help |
| -h |
| Prints all command-line switches.)"; |
| |
| const char kCommandHelp[] = R"( --command |
| -c |
| Execute the given command.)"; |
| |
| App::App(async_dispatcher_t* dispatcher) |
| : console_(this, dispatcher, STDIN_FILENO), executor_(dispatcher) {} |
| |
| App::~App() = default; |
| |
| bool App::Init(int argc, const char** argv, QuitCallback quit_callback) { |
| quit_callback_ = std::move(quit_callback); |
| |
| cmdline::ArgsParser<Options> parser; |
| parser.AddSwitch("command", 'c', kCommandHelp, &Options::command); |
| |
| // Special --help switch which doesn't exist in the options structure. |
| bool requested_help = false; |
| parser.AddGeneralSwitch("help", 'h', kHelpHelp, [&requested_help]() { requested_help = true; }); |
| |
| std::vector<std::string> params; |
| cmdline::Status parser_status = parser.Parse(argc, argv, &options_, ¶ms); |
| if (parser_status.has_error()) { |
| fprintf(stderr, "error: %s\n", parser_status.error_message().c_str()); |
| return false; |
| } |
| |
| // Handle --help switch since we're the one that knows about the switches. |
| if (requested_help) { |
| printf("%s", (kHelpIntro + parser.GetHelp()).c_str()); |
| Quit(); |
| return true; |
| } |
| |
| // Initialize the PWD environment variable, which is how we signal to child |
| // processes what their current working directory ought to be. We update this |
| // value in cd_task.cc. |
| char cwd[PATH_MAX]; |
| getcwd(cwd, sizeof(cwd)); |
| setenv("PWD", cwd, 1); |
| |
| console_.Init("% "); |
| |
| if (options_.command) { |
| Command command; |
| command.Parse(options_.command.value()); |
| zx_status_t status = OnConsoleCommand(std::move(command)); |
| if (status == ZX_ERR_NEXT) { |
| Quit(); |
| } |
| return true; |
| } else { |
| console_.GetNextCommand(); |
| } |
| return true; |
| } |
| |
| zx_status_t App::OnConsoleCommand(Command command) { |
| if (!command.parse_error().empty()) { |
| fprintf(stderr, "error: Invalid command: %s\n", command.parse_error().c_str()); |
| return ZX_ERR_NEXT; |
| } |
| zx_status_t status = executor_.Execute(std::move(command), [this]() { |
| if (options_.command) { |
| Quit(); |
| } else { |
| console_.GetNextCommand(); |
| } |
| }); |
| if (status == ZX_ERR_STOP) { |
| Quit(); |
| return status; |
| } |
| if (status != ZX_ERR_NEXT && status != ZX_ERR_ASYNC) { |
| fprintf(stderr, "error: Failed to execute command: %d (%s)\n", status, |
| zx_status_get_string(status)); |
| return ZX_ERR_NEXT; |
| } |
| return status; |
| } |
| |
| void App::OnConsoleInterrupt() { |
| executor_.KillForegroundTask(); |
| console_.GetNextCommand(); |
| } |
| |
| void App::OnConsoleError(zx_status_t status) { |
| fprintf(stderr, "error: Failed to read console: %d (%s)\n", status, zx_status_get_string(status)); |
| Quit(); |
| } |
| |
| void App::Quit() { |
| auto quit_callback = std::move(quit_callback_); |
| quit_callback(); |
| } |
| |
| void App::OnConsoleAutocomplete(Autocomplete* autocomplete) { |
| return executor_.Complete(autocomplete); |
| } |
| |
| } // namespace cmd |