bin/ui/screencap/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 <trace-provider/provider.h>
#include <memory>

#include "garnet/lib/ui/gfx/gfx_system.h"
#include "garnet/lib/ui/scenic/scenic.h"
#include "garnet/lib/ui/scenic/system.h"
#include "lib/app/cpp/application_context.h"
#include "lib/fsl/tasks/message_loop.h"
#include "lib/fxl/command_line.h"
#include "lib/fxl/log_settings_command_line.h"
#include "lib/fxl/logging.h"

int main(int argc, const char** argv) {
  auto command_line = fxl::CommandLineFromArgcArgv(argc, argv);
  if (!fxl::SetLogSettingsFromCommandLine(command_line))
    return 1;

  const auto& positional_args = command_line.positional_args();
  if (positional_args.empty()) {
    FXL_LOG(ERROR) << "screencap requires a path for where to save "
                   << "the screenshot.";
    return 1;
  }
  const std::string filename = positional_args[0];

  fsl::MessageLoop loop;
  trace::TraceProvider trace_provider(loop.async());
  std::unique_ptr<component::ApplicationContext> app_context(
      component::ApplicationContext::CreateFromStartupInfo());
  // Connect to the SceneManager service.
  auto scenic = app_context->ConnectToEnvironmentService<ui::Scenic>();
  scenic.set_error_handler([&loop] {
    FXL_LOG(ERROR) << "Lost connection to Scenic service.";
    loop.QuitNow();
  });
  auto done_cb = [&loop](bool status) {
    if (!status) {
      FXL_LOG(ERROR) << "TakeScreenshot failed";
    }
    loop.QuitNow();
  };
  scenic->TakeScreenshot(filename, done_cb);
  loop.Run();

  return 0;
}