// Copyright 2016 The Chromium 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 "flutter/content_handler/app.h"

#include <thread>
#include <utility>

#include "apps/tracing/lib/trace/provider.h"
#include "flutter/common/settings.h"
#include "flutter/common/threads.h"
#include "flutter/runtime/runtime_init.h"
#include "flutter/sky/engine/platform/fonts/fuchsia/FontCacheFuchsia.h"
#include "lib/ftl/macros.h"
#include "lib/ftl/tasks/task_runner.h"
#include "lib/mtl/tasks/message_loop.h"
#include "lib/mtl/threading/create_thread.h"

namespace flutter_runner {
namespace {

void QuitMessageLoop() {
  mtl::MessageLoop::GetCurrent()->QuitNow();
}

}  // namespace

App::App() {
  context_ = modular::ApplicationContext::CreateFromStartupInfo();

  tracing::InitializeTracer(context_.get(), {});

  ftl::RefPtr<ftl::TaskRunner> gpu_task_runner;
  gpu_thread_ = mtl::CreateThread(&gpu_task_runner);

  ftl::RefPtr<ftl::TaskRunner> ui_task_runner(
      mtl::MessageLoop::GetCurrent()->task_runner());

  ftl::RefPtr<ftl::TaskRunner> io_task_runner;
  io_thread_ = mtl::CreateThread(&io_task_runner);

  // Notice that the Platform and UI threads are actually the same.
  blink::Threads::Set(blink::Threads(ui_task_runner, gpu_task_runner,
                                     ui_task_runner, io_task_runner));
  blink::Settings::Set(blink::Settings());
  blink::InitRuntime();

  blink::SetFontProvider(
      context_->ConnectToEnvironmentService<fonts::FontProvider>());

  context_->outgoing_services()->AddService<modular::ApplicationRunner>(
      [this](fidl::InterfaceRequest<modular::ApplicationRunner> request) {
        runner_bindings_.AddBinding(this, std::move(request));
      });
}

App::~App() {
  StopThreads();
}

void App::StartApplication(
    modular::ApplicationPackagePtr application,
    modular::ApplicationStartupInfoPtr startup_info,
    fidl::InterfaceRequest<modular::ApplicationController> controller) {
  std::unique_ptr<ApplicationControllerImpl> impl =
      std::make_unique<ApplicationControllerImpl>(this, std::move(application),
                                                  std::move(startup_info),
                                                  std::move(controller));
  ApplicationControllerImpl* key = impl.get();
  controllers_.emplace(key, std::move(impl));
}

void App::Destroy(ApplicationControllerImpl* controller) {
  auto it = controllers_.find(controller);
  if (it == controllers_.end())
    return;
  controllers_.erase(it);
}

void App::StopThreads() {
  blink::Threads::Gpu()->PostTask(QuitMessageLoop);
  blink::Threads::IO()->PostTask(QuitMessageLoop);
  gpu_thread_.join();
  io_thread_.join();
}

}  // namespace flutter_runner