// Copyright 2019 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 <fuchsia/modular/cpp/fidl.h>
#include <fuchsia/modular/testing/cpp/fidl.h>
#include <lib/fit/function.h>
#include <lib/fsl/vmo/strings.h>
#include <lib/modular_test_harness/cpp/fake_module.h>
#include <lib/modular_test_harness/cpp/test_harness_fixture.h>
#include <sdk/lib/sys/cpp/component_context.h>
#include <sdk/lib/sys/cpp/service_directory.h>
#include <src/lib/fxl/logging.h>

#include "peridot/lib/testing/session_shell_impl.h"

namespace {

constexpr zx::duration kTimeout = zx::sec(30);

// The name of the intent parameter which contains a string which the second
// module is to append to its signal.
constexpr char kIntentParameterName[] = "intent_parameter";

// The initial module's intent parameter data. This needs to be JSON formatted.
constexpr char kInitialIntentParameterData[] = "\"initial\"";
constexpr char kModuleName[] = "mod_name";
constexpr char kStoryName[] = "story";
constexpr char kIntentAction[] = "action";

class IntentsTest : public modular::testing::TestHarnessFixture {
 public:
  void SetUp() override {
    intent_handled_ = false;
    test_module_ = std::make_unique<modular::testing::FakeModule>(
        [this](fuchsia::modular::Intent intent) {
          latest_handled_intent_ = std::move(intent);
          intent_handled_ = true;
        });
    test_module_url_ = builder_.GenerateFakeUrl();
    builder_.InterceptComponent(
        test_module_->GetOnCreateHandler(),
        {.url = test_module_url_,
         .sandbox_services =
             modular::testing::FakeModule::GetSandboxServices()});

    test_harness().events().OnNewComponent =
        builder_.BuildOnNewComponentHandler();
    test_harness()->Run(builder_.BuildSpec());
  }

  // Create an Intent with the given handler, parameter name, and parameter
  // data.
  fuchsia::modular::Intent CreateIntent(std::string handler,
                                        std::string parameter_name,
                                        std::string parameter_data) {
    fuchsia::modular::Intent intent;
    intent.handler = handler;
    intent.action = kIntentAction;

    fuchsia::modular::IntentParameter intent_parameter;
    intent_parameter.name = parameter_name;
    intent_parameter.data = fuchsia::modular::IntentParameterData();
    fsl::SizedVmo vmo;
    FXL_CHECK(fsl::VmoFromString(parameter_data, &vmo));
    intent_parameter.data.set_json(std::move(vmo).ToTransport());
    intent.parameters.push_back(std::move(intent_parameter));

    return intent;
  }

  // Uses the PuppetMaster service to lauch an initial module.
  // This initial module is required to test the reissuing of intents to
  // additional modules.
  void ExecutePuppetMasterAddMod(fuchsia::modular::Intent intent) {
    // Create an AddMod command
    fuchsia::modular::AddMod add_mod;
    add_mod.mod_name = {kModuleName};
    add_mod.intent = std::move(intent);
    add_mod.surface_relation = fuchsia::modular::SurfaceRelation{};

    fuchsia::modular::StoryCommand cmd;
    cmd.set_add_mod(std::move(add_mod));

    std::vector<fuchsia::modular::StoryCommand> cmds;
    cmds.push_back(std::move(cmd));

    // Connect to PuppetMaster Service
    fuchsia::modular::PuppetMasterPtr puppet_master;
    fuchsia::modular::testing::ModularService svc;
    svc.set_puppet_master(puppet_master.NewRequest());
    test_harness()->ConnectToModularService(std::move(svc));

    // Create a story
    fuchsia::modular::StoryPuppetMasterPtr story_master;
    puppet_master->ControlStory(kStoryName, story_master.NewRequest());

    // Add the initial module to the story
    story_master->Enqueue(std::move(cmds));
    story_master->Execute([&](fuchsia::modular::ExecuteResult result) {});
  }

  // Starts a second module by calling AddModuleToStory() using the
  // ModuleContext of the original module. The intent is expected to be handled
  // by the original module if the modules' intent handlers match.
  void AddModuleToStory(
      fuchsia::modular::ModuleContext* const module_context,
      fuchsia::modular::Intent intent,
      fidl::InterfaceRequest<fuchsia::modular::ModuleController> request,
      bool* started) {
    module_context->AddModuleToStory(
        kModuleName, std::move(intent), std::move(request), nullptr,
        [started](const fuchsia::modular::StartModuleStatus) mutable {
          *started = true;
        });
  }

  // Checks that the given intent's paramter name and data matches expectations.
  bool IntentMatchesExpectations(fuchsia::modular::Intent* intent,
                                 std::string expected_param_name,
                                 std::string expected_param_data) {
    for (const auto& parameter : *intent->parameters) {
      if (parameter.data.is_json()) {
        if (parameter.name == expected_param_name) {
          std::string parameter_data;
          FXL_CHECK(fsl::StringFromVmo(parameter.data.json(), &parameter_data));
          if (expected_param_data == parameter_data) {
            return true;
          }
        }
      }
    }
    return false;
  }

  std::unique_ptr<modular::testing::FakeModule> test_module_;
  modular::testing::TestHarnessBuilder builder_;
  std::string test_module_url_;
  fuchsia::modular::Intent latest_handled_intent_;
  bool intent_handled_;
};

// Launches a single module with an intent. Checks that the module exposes an
// intent handler and gets notified of the intent by the framework.
TEST_F(IntentsTest, ModuleUsesIntentHandler) {
  // Launch initial module
  auto initial_module_intent = CreateIntent(
      test_module_url_, kIntentParameterName, kInitialIntentParameterData);
  ExecutePuppetMasterAddMod(std::move(initial_module_intent));
  ASSERT_TRUE(RunLoopWithTimeoutOrUntil(
      [&] { return intent_handled_; }, kTimeout));

  // Check that the intent handler received the intent
  EXPECT_TRUE(IntentMatchesExpectations(&latest_handled_intent_,
                                        kIntentParameterName,
                                        kInitialIntentParameterData));
}

// Launches a module that exposes an intent handler service then tests that a
// second intent sent to an already running module with the same parameters but
// different data notifies the intent handler of the new intent.
TEST_F(IntentsTest, DISABLED_ReuseIntentHandlerSameParamName) {
  // Launch initial module
  auto initial_module_intent = CreateIntent(
      test_module_url_, kIntentParameterName, kInitialIntentParameterData);
  ExecutePuppetMasterAddMod(std::move(initial_module_intent));
  ASSERT_TRUE(RunLoopWithTimeoutOrUntil(
      [&] { return test_module_->is_running(); }, kTimeout));

  // Launch second module using first module's |module_context|
  fuchsia::modular::ModuleControllerPtr second_module_controller;
  bool module_started{false};

  // Use different param data
  auto second_module_param_data = "\"second_module_param_data\"";
  AddModuleToStory(test_module_->module_context(),
                   CreateIntent(test_module_url_, kIntentParameterName,
                                second_module_param_data),
                   second_module_controller.NewRequest(), &module_started);

  intent_handled_ = false;
  ASSERT_TRUE(
      RunLoopWithTimeoutOrUntil([&] { return intent_handled_; }, kTimeout));

  EXPECT_TRUE(IntentMatchesExpectations(
      &latest_handled_intent_, kIntentParameterName, second_module_param_data));
}

// Launches a module that exposes an intent handler service then tests that a
// second intent with different parameters is delivered to to the already
// running intent handler.
TEST_F(IntentsTest, DISABLED_ReuseIntentHandlerDifferentParam) {
  // Launch initial module
  auto initial_module_intent = CreateIntent(
      test_module_url_, kIntentParameterName, kInitialIntentParameterData);
  ExecutePuppetMasterAddMod(std::move(initial_module_intent));
  ASSERT_TRUE(RunLoopWithTimeoutOrUntil(
      [&] { return test_module_->is_running(); }, kTimeout));

  // Launch second module using first module's |module_context|
  fuchsia::modular::ModuleControllerPtr second_module_controller;
  bool module_started{false};

  // Use different param name and data
  auto second_module_param_name = "different_param_name";
  auto second_module_param_data = "\"second_module_param_data\"";
  AddModuleToStory(test_module_->module_context(),
                   CreateIntent(test_module_url_, second_module_param_name,
                                second_module_param_data),
                   second_module_controller.NewRequest(), &module_started);

  intent_handled_ = false;
  ASSERT_TRUE(
      RunLoopWithTimeoutOrUntil([&] { return intent_handled_; }, kTimeout));

  EXPECT_TRUE(IntentMatchesExpectations(&latest_handled_intent_,
                                        second_module_param_name,
                                        second_module_param_data));
}

// Launches a module that exposes an intent handler service then tests that a
// second intent with different handler is not delivered to the running
// intent handler.
TEST_F(IntentsTest, DISABLED_DifferentHandler) {
  // Launch initial module
  auto initial_module_intent = CreateIntent(
      test_module_url_, kIntentParameterName, kInitialIntentParameterData);
  ExecutePuppetMasterAddMod(std::move(initial_module_intent));
  ASSERT_TRUE(RunLoopWithTimeoutOrUntil(
      [&] { return test_module_->is_running(); }, kTimeout));

  // Launch second module using first module's |module_context|
  fuchsia::modular::ModuleControllerPtr second_module_controller;
  bool module_started{false};

  // Use different handler
  auto different_module_url = builder_.GenerateFakeUrl();
  auto different_intent_param_name = "different_param_name";
  auto different_intent_param_data = "\"different_param_data\"";
  auto different_intent =
      CreateIntent(different_module_url, different_intent_param_name,
                   different_intent_param_data);
  AddModuleToStory(test_module_->module_context(), std::move(different_intent),
                   second_module_controller.NewRequest(), &module_started);

  ASSERT_TRUE(
      RunLoopWithTimeoutOrUntil([&] { return module_started; }, kTimeout));

  // Check that the intercepted_module_'s latest handled intent matches the
  // initial module
  EXPECT_TRUE(IntentMatchesExpectations(&latest_handled_intent_,
                                        kIntentParameterName,
                                        kInitialIntentParameterData));
}

}  // namespace