| // 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/tasks/setenv_task.h" |
| |
| #include <stdlib.h> |
| |
| namespace cmd { |
| |
| SetenvTask::SetenvTask(async_dispatcher_t* dispatcher) : Task(dispatcher) {} |
| |
| SetenvTask::~SetenvTask() = default; |
| |
| zx_status_t SetenvTask::Execute(Command command, Task::CompletionCallback callback) { |
| if (command.args().size() == 3) { |
| const std::string& name = command.args()[1]; |
| const std::string& value = command.args()[2]; |
| if (name.find('=') != std::string::npos) { |
| fprintf(stderr, "setenv: Environment variable name cannot contain '=': %s\n", name.c_str()); |
| } else { |
| setenv(name.c_str(), value.c_str(), 1); |
| } |
| } else { |
| fprintf(stderr, "setenv: Invalid number of arguments. Expected 2, got %zu.\n", |
| command.args().size() - 1); |
| } |
| return ZX_ERR_NEXT; |
| } |
| |
| void SetenvTask::Complete(Autocomplete* autocomplete) { |
| if (autocomplete->tokens().size() == 1) { |
| autocomplete->CompleteAsEnvironmentVariable(); |
| } |
| } |
| |
| } // namespace cmd |