garnet/bin/iquery/main.cc

// Copyright 2018 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 <lib/async-loop/cpp/loop.h>
#include <lib/async_promise/executor.h>
#include <lib/fit/promise.h>
#include <src/lib/fxl/command_line.h>
#include <src/lib/fxl/log_settings_command_line.h>
#include <unistd.h>

#include <iostream>

#include "garnet/bin/iquery/modes.h"

int main(int argc, const char** argv) {
  async::Loop loop(&kAsyncLoopConfigAttachToThread);
  async::Executor executor(loop.dispatcher());

  auto command_line = fxl::CommandLineFromArgcArgv(argc, argv);
  if (!fxl::SetLogSettingsFromCommandLine(command_line)) {
    return 1;
  }

  iquery::Options options(command_line);
  if (!options.Valid()) {
    return 1;
  }

  if (!options.chdir.empty()) {
    if (chdir(options.chdir.c_str()) != 0) {
      FXL_LOG(ERROR) << "Failed to chdir to " << options.chdir;
    }
  }

  if (command_line.HasOption("help") || options.paths.size() == 0) {
    options.Usage(command_line.argv0());
    return 0;
  }

  fit::promise<std::vector<inspect::Source>> results;
  // Dispatch to the correct mode.
  if (options.mode == iquery::Options::Mode::CAT) {
    results = iquery::RunCat(&options);
  } else if (options.mode == iquery::Options::Mode::FIND) {
    results = iquery::RunFind(&options);
  } else if (options.mode == iquery::Options::Mode::LS) {
    results = iquery::RunLs(&options);
  } else {
    FXL_LOG(ERROR) << "Unsupported mode";
    return 1;
  }

  executor.schedule_task(
      results
          .and_then([&options, &loop](std::vector<inspect::Source>& results) {
            // Sort the hierarchies if requested.
            if (options.sort) {
              for (auto& source : results) {
                source.SortHierarchy();
              }
            }

            if (options.mode == iquery::Options::Mode::CAT) {
              std::cout << options.formatter->FormatSourcesRecursive(results);
            } else if (options.mode == iquery::Options::Mode::FIND) {
              std::cout << options.formatter->FormatSourceLocations(results);
            } else if (options.mode == iquery::Options::Mode::LS) {
              std::cout << options.formatter->FormatChildListing(results);
            }

            loop.Quit();
          })
          .or_else([&loop] {
            FXL_LOG(ERROR) << "An error occurred";
            loop.Quit();
          }));

  loop.Run();

  return 0;
}