// Copyright 2023 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 <fidl/fuchsia.logger/cpp/fidl.h>
#include <lib/async/default.h>
#include <lib/component/incoming/cpp/protocol.h>
#include <lib/driver/testing/cpp/internal/test_environment.h>

namespace fdf_testing::internal {
const char TestEnvironment::kTestEnvironmentThreadSafetyDescription[] =
    "|fdf_testing::internal::TestEnvironment| is thread-unsafe.";

TestEnvironment::TestEnvironment(fdf_dispatcher_t* dispatcher)
    : dispatcher_(dispatcher ? dispatcher : fdf::Dispatcher::GetCurrent()->get()),
      incoming_directory_server_(dispatcher_),
      checker_(fdf_dispatcher_get_async_dispatcher(dispatcher_),
               kTestEnvironmentThreadSafetyDescription) {}

zx::result<> TestEnvironment::Initialize(
    fidl::ServerEnd<fuchsia_io::Directory> incoming_directory_server_end) {
  std::lock_guard guard(checker_);

  zx::result result = incoming_directory_server_.Serve(std::move(incoming_directory_server_end));
  if (result.is_error()) {
    return result.take_error();
  }

  if (!logsink_added_) {
    // Forward the LogSink protocol that we have from our own incoming namespace.
    result = incoming_directory_server_.component().AddUnmanagedProtocol<fuchsia_logger::LogSink>(
        [](fidl::ServerEnd<fuchsia_logger::LogSink> server_end) {
          ZX_ASSERT(component::Connect<fuchsia_logger::LogSink>(std::move(server_end)).is_ok());
        });
    if (result.is_error()) {
      return result.take_error();
    }

    logsink_added_ = true;
  }

  return zx::ok();
}

}  // namespace fdf_testing::internal