src/developer/cmd/tasks/unsetenv_task.cc

// 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/unsetenv_task.h"

#include <stdlib.h>

namespace cmd {

UnsetenvTask::UnsetenvTask(async_dispatcher_t* dispatcher) : Task(dispatcher) {}

UnsetenvTask::~UnsetenvTask() = default;

zx_status_t UnsetenvTask::Execute(Command command, Task::CompletionCallback callback) {
  if (command.args().size() == 2) {
    const std::string& name = command.args()[1];
    if (name.find('=') != std::string::npos) {
      fprintf(stderr, "unsetenv: Environment variable name cannot contain '=': %s\n", name.c_str());
    } else {
      unsetenv(name.c_str());
    }
  } else {
    fprintf(stderr, "unsetenv: Invalid number of arguments. Expected 1, got %zu.\n",
            command.args().size() - 1);
  }
  return ZX_ERR_NEXT;
}

void UnsetenvTask::Complete(Autocomplete* autocomplete) {
  if (autocomplete->tokens().size() == 1) {
    autocomplete->CompleteAsEnvironmentVariable();
  }
}

}  // namespace cmd