// Copyright 2017 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 "peridot/bin/ledger/testing/ledger_app_instance_factory.h"

#include <lib/fidl/cpp/clone.h>
#include <lib/fsl/io/fd.h>
#include <lib/fxl/memory/ref_ptr.h>

#include "garnet/public/lib/callback/capture.h"
#include "gtest/gtest.h"

namespace ledger {

LedgerAppInstanceFactory::LedgerAppInstance::LedgerAppInstance(
    LoopController* loop_controller, std::vector<uint8_t> test_ledger_name,
    ledger_internal::LedgerRepositoryFactoryPtr ledger_repository_factory)
    : loop_controller_(loop_controller),
      test_ledger_name_(std::move(test_ledger_name)),
      ledger_repository_factory_(std::move(ledger_repository_factory)) {
  ledger_repository_factory_.set_error_handler([](zx_status_t status) {
    if (status != ZX_ERR_PEER_CLOSED) {
      ADD_FAILURE() << "|LedgerRepositoryFactory| failed with an error: "
                    << status;
    }
  });
}

LedgerAppInstanceFactory::LedgerAppInstance::~LedgerAppInstance() {}

ledger_internal::LedgerRepositoryFactory*
LedgerAppInstanceFactory::LedgerAppInstance::ledger_repository_factory() {
  return ledger_repository_factory_.get();
}

ledger_internal::LedgerRepositoryPtr
LedgerAppInstanceFactory::LedgerAppInstance::GetTestLedgerRepository() {
  ledger_internal::LedgerRepositoryPtr repository;
  repository.set_error_handler([](zx_status_t status) {
    if (status != ZX_ERR_PEER_CLOSED) {
      ADD_FAILURE() << "|LedgerRepository| failed with an error: " << status;
    }
  });
  ledger_repository_factory_->GetRepository(
      fsl::CloneChannelFromFileDescriptor(tmpfs_.root_fd()),
      MakeCloudProvider(), GetUserId(), repository.NewRequest());
  return repository;
}

LedgerPtr LedgerAppInstanceFactory::LedgerAppInstance::GetTestLedger() {
  LedgerPtr ledger;
  ledger.set_error_handler([](zx_status_t status) {
    if (status != ZX_ERR_PEER_CLOSED) {
      ADD_FAILURE() << "|Ledger| failed with an error: " << status;
    }
  });

  auto repository = GetTestLedgerRepository();
  repository->GetLedger(fidl::Clone(test_ledger_name_), ledger.NewRequest());
  auto waiter = loop_controller_->NewWaiter();
  repository->Sync(waiter->GetCallback());
  if (!waiter->RunUntilCalled()) {
    ADD_FAILURE() << "|GetLedger| failed to call back.";
    return nullptr;
  }
  return ledger;
}

PagePtr LedgerAppInstanceFactory::LedgerAppInstance::GetTestPage() {
  fidl::InterfaceHandle<Page> page;
  Status status;
  LedgerPtr ledger = GetTestLedger();
  auto waiter = loop_controller_->NewWaiter();
  ledger->GetPage(nullptr, page.NewRequest(),
                  callback::Capture(waiter->GetCallback(), &status));
  if (!waiter->RunUntilCalled()) {
    ADD_FAILURE() << "|GetPage| failed to call back.";
    return nullptr;
  }
  EXPECT_EQ(Status::OK, status);

  return page.Bind();
}

PagePtr LedgerAppInstanceFactory::LedgerAppInstance::GetPage(
    const PageIdPtr& page_id, Status expected_status) {
  PagePtr page_ptr;
  Status status;
  LedgerPtr ledger = GetTestLedger();
  auto waiter = loop_controller_->NewWaiter();
  ledger->GetPage(fidl::Clone(page_id), page_ptr.NewRequest(),
                  callback::Capture(waiter->GetCallback(), &status));
  if (!waiter->RunUntilCalled()) {
    ADD_FAILURE() << "|GetPage| failed to call back.";
    return nullptr;
  }
  EXPECT_EQ(expected_status, status);

  return page_ptr;
}

}  // namespace ledger