// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/android/webapk/webapk_database.h"

#include <memory>
#include <string>

#include "base/strings/to_string.h"
#include "base/test/bind.h"
#include "base/test/test_future.h"
#include "chrome/browser/android/webapk/test/fake_data_type_store_service.h"
#include "chrome/browser/android/webapk/webapk_helpers.h"
#include "chrome/browser/android/webapk/webapk_registrar.h"
#include "chrome/browser/android/webapk/webapk_registry_update.h"
#include "chrome/browser/android/webapk/webapk_sync_bridge.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chrome/test/base/testing_profile_manager.h"
#include "components/sync/model/metadata_batch.h"
#include "components/sync/protocol/web_apk_specifics.pb.h"
#include "components/webapps/common/web_app_id.h"
#include "content/public/test/browser_task_environment.h"
#include "content/public/test/test_web_contents_factory.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"

namespace webapk {

// Note: this only compares basic sync attributes, it doesn't do a full deep
// comparison.
bool IsRegistryEqual(const Registry& registry, const Registry& registry2) {
  if (registry.size() != registry2.size()) {
    return false;
  }

  for (auto& kv : registry) {
    const WebApkProto* web_app = kv.second.get();
    const WebApkProto* web_app2 = registry2.at(kv.first).get();

    const sync_pb::WebApkSpecifics& specifics = web_app->sync_data();
    const sync_pb::WebApkSpecifics& specifics2 = web_app2->sync_data();

    if (web_app->is_locally_installed() != web_app2->is_locally_installed() ||
        specifics.manifest_id() != specifics2.manifest_id() ||
        specifics.start_url() != specifics2.start_url() ||
        specifics.name() != specifics2.name()) {
      return false;
    }
  }

  return true;
}

std::unique_ptr<RegistryUpdateData> RegistryToRegistryUpdateData(
    Registry* registry) {
  std::unique_ptr<RegistryUpdateData> update_data =
      std::make_unique<RegistryUpdateData>();
  for (auto& entry : *registry) {
    update_data->apps_to_create.emplace_back(std::move(entry.second));
  }
  return update_data;
}

class WebApkDatabaseTest : public ::testing::Test {
 public:
  WebApkDatabaseTest()
      : task_environment_(content::BrowserTaskEnvironment::IO_MAINLOOP) {}

  void SetUp() override {
    ASSERT_TRUE(testing_profile_manager_.SetUp());
    profile_ = testing_profile_manager_.CreateTestingProfile(
        TestingProfile::kDefaultProfileUserName);

    data_type_store_service_ = std::make_unique<FakeDataTypeStoreService>();

    web_contents_ = web_contents_factory_.CreateWebContents(profile());
  }

  // Creates a random WebApkProto based off of a suffix.
  static std::unique_ptr<WebApkProto> CreateWebApkProto(
      int suffix,
      bool is_locally_installed) {
    const std::string start_url =
        "https://example.com/" + base::ToString(suffix);
    const std::string manifest_id =
        "https://example.com/id/" + base::ToString(suffix);
    const std::string name = "App Name " + base::ToString(suffix);

    std::unique_ptr<WebApkProto> proto = std::make_unique<WebApkProto>();
    sync_pb::WebApkSpecifics* sync_proto = proto->mutable_sync_data();

    sync_proto->set_manifest_id(manifest_id);
    sync_proto->set_start_url(GURL(start_url).spec());
    sync_proto->set_name(name);
    proto->set_is_locally_installed(is_locally_installed);

    return proto;
  }

  void WriteBatch(
      std::unique_ptr<syncer::DataTypeStore::WriteBatch> write_batch) {
    base::test::TestFuture<const std::optional<syncer::ModelError>&> error;
    data_type_store_service().GetStore()->CommitWriteBatch(
        std::move(write_batch), error.GetCallback());
    EXPECT_FALSE(error.Get());
  }

  Registry CreateWebApps(uint32_t num_apps) {
    Registry registry;

    for (uint32_t i = 0; i < num_apps; ++i) {
      std::unique_ptr<WebApkProto> proto = CreateWebApkProto(i, false);
      const webapps::AppId app_id =
          GenerateAppIdFromManifestId(GURL(proto->sync_data().manifest_id()));

      registry.emplace(app_id, std::move(proto));
    }

    return registry;
  }

  Registry CreateAndWriteWebApps(uint32_t num_apps) {
    Registry registry = CreateWebApps(num_apps);

    auto write_batch = data_type_store_service().GetStore()->CreateWriteBatch();
    for (const auto& entry : registry) {
      write_batch->WriteData(entry.first, entry.second->SerializeAsString());
    }
    WriteBatch(std::move(write_batch));

    return registry;
  }

  TestingProfile* profile() { return profile_.get(); }

  FakeDataTypeStoreService& data_type_store_service() {
    return *data_type_store_service_;
  }
  FakeDataTypeStoreService* data_type_store_service_ptr() {
    return data_type_store_service_.get();
  }

 private:
  content::BrowserTaskEnvironment task_environment_;
  std::unique_ptr<FakeDataTypeStoreService> data_type_store_service_;
  TestingProfileManager testing_profile_manager_{
      TestingBrowserProcess::GetGlobal()};

  raw_ptr<TestingProfile> profile_;

  content::TestWebContentsFactory web_contents_factory_;
  raw_ptr<content::WebContents>
      web_contents_;  // Owned by `web_contents_factory_`.
};

TEST_F(WebApkDatabaseTest, OpenDatabaseAndReadRegistry) {
  Registry registry = CreateAndWriteWebApps(100);

  std::unique_ptr<WebApkDatabase> web_apk_database =
      std::make_unique<WebApkDatabase>(
          data_type_store_service_ptr(),
          base::BindRepeating([](const syncer::ModelError& error) {
            ASSERT_TRUE(false);  // should not be reached
          }));

  base::test::TestFuture<Registry, std::unique_ptr<syncer::MetadataBatch>>
      registry_and_batch;
  web_apk_database->OpenDatabase(registry_and_batch.GetCallback());
  EXPECT_TRUE(IsRegistryEqual(registry_and_batch.Get<0>(), registry));
}

TEST_F(WebApkDatabaseTest, OpenDatabaseAndWriteRegistry) {
  Registry registry = CreateWebApps(100);
  std::unique_ptr<RegistryUpdateData> update_data =
      RegistryToRegistryUpdateData(&registry);

  std::unique_ptr<WebApkDatabase> web_apk_database =
      std::make_unique<WebApkDatabase>(
          data_type_store_service_ptr(),
          base::BindRepeating([](const syncer::ModelError& error) {
            ASSERT_TRUE(false);  // should not be reached
          }));

  base::test::TestFuture<Registry, std::unique_ptr<syncer::MetadataBatch>>
      open_result;
  web_apk_database->OpenDatabase(open_result.GetCallback());
  ASSERT_TRUE(open_result.Wait());

  base::test::TestFuture<bool> success;
  web_apk_database->Write(
      *update_data,
      syncer::DataTypeStore::WriteBatch::CreateMetadataChangeList(),
      success.GetCallback());
  EXPECT_TRUE(success.Get());
  EXPECT_TRUE(IsRegistryEqual(data_type_store_service().ReadRegistry(),
                              CreateWebApps(100)));
}

TEST_F(WebApkDatabaseTest, OpenDatabaseAndDeleteFromRegistry) {
  Registry registry = CreateAndWriteWebApps(100);

  std::unique_ptr<WebApkDatabase> web_apk_database =
      std::make_unique<WebApkDatabase>(
          data_type_store_service_ptr(),
          base::BindRepeating([](const syncer::ModelError& error) {
            ASSERT_TRUE(false);  // should not be reached
          }));

  base::test::TestFuture<Registry, std::unique_ptr<syncer::MetadataBatch>>
      open_result;
  web_apk_database->OpenDatabase(open_result.GetCallback());
  ASSERT_TRUE(open_result.Wait());

  RegistryUpdateData update_data;
  update_data.apps_to_delete.push_back(
      ManifestIdStrToAppId("https://example.com/id/95"));
  update_data.apps_to_delete.push_back(
      ManifestIdStrToAppId("https://example.com/id/96"));
  update_data.apps_to_delete.push_back(
      ManifestIdStrToAppId("https://example.com/id/97"));
  update_data.apps_to_delete.push_back(
      ManifestIdStrToAppId("https://example.com/id/98"));
  update_data.apps_to_delete.push_back(
      ManifestIdStrToAppId("https://example.com/id/99"));

  base::test::TestFuture<bool> success;
  web_apk_database->Write(
      update_data,
      syncer::DataTypeStore::WriteBatch::CreateMetadataChangeList(),
      success.GetCallback());
  EXPECT_TRUE(success.Get());

  EXPECT_TRUE(IsRegistryEqual(data_type_store_service().ReadRegistry(),
                              CreateWebApps(95)));
}

TEST_F(WebApkDatabaseTest, OpenDatabaseAndOverwriteRegistry) {
  Registry registry = CreateAndWriteWebApps(1);

  std::unique_ptr<WebApkDatabase> web_apk_database =
      std::make_unique<WebApkDatabase>(
          data_type_store_service_ptr(),
          base::BindRepeating([](const syncer::ModelError& error) {
            ASSERT_TRUE(false);  // should not be reached
          }));

  base::test::TestFuture<Registry, std::unique_ptr<syncer::MetadataBatch>>
      open_result;
  web_apk_database->OpenDatabase(open_result.GetCallback());
  ASSERT_TRUE(open_result.Wait());

  std::unique_ptr<WebApkProto> replacement =
      CreateWebApkProto(0 /* suffix */, true /* is_locally_installed */);
  sync_pb::WebApkSpecifics* replacement_sync_proto =
      replacement->mutable_sync_data();
  replacement_sync_proto->set_name("asfd1234");

  RegistryUpdateData update_data;
  update_data.apps_to_create.emplace_back(std::move(replacement));

  base::test::TestFuture<bool> success;
  web_apk_database->Write(
      update_data,
      syncer::DataTypeStore::WriteBatch::CreateMetadataChangeList(),
      success.GetCallback());
  EXPECT_TRUE(success.Get());

  std::unique_ptr<WebApkProto> final_proto =
      CreateWebApkProto(0 /* suffix */, true /* is_locally_installed */);
  sync_pb::WebApkSpecifics* final_sync_proto = final_proto->mutable_sync_data();
  final_sync_proto->set_name("asfd1234");
  const webapps::AppId app_id =
      ManifestIdStrToAppId(final_proto->sync_data().manifest_id());

  Registry final_registry;
  final_registry.emplace(app_id, std::move(final_proto));

  EXPECT_TRUE(IsRegistryEqual(data_type_store_service().ReadRegistry(),
                              final_registry));
}

}  // namespace webapk