// Copyright 2021 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/web_applications/web_app_origin_association_manager.h"

#include <utility>

#include "base/functional/bind.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/web_applications/proto/web_app.equal.h"
#include "chrome/browser/web_applications/web_app_origin_association_task.h"
#include "components/webapps/services/web_app_origin_association/web_app_origin_association_fetcher.h"
#include "content/public/browser/storage_partition.h"

namespace web_app {

WebAppOriginAssociationManager::WebAppOriginAssociationManager(Profile& profile)
    : profile_(profile) {}

WebAppOriginAssociationManager::~WebAppOriginAssociationManager() = default;

OriginAssociations::OriginAssociations() = default;
OriginAssociations::OriginAssociations(const OriginAssociations&) = default;
OriginAssociations::OriginAssociations(OriginAssociations&&) = default;
OriginAssociations& OriginAssociations::operator=(const OriginAssociations&) =
    default;
OriginAssociations& OriginAssociations::operator=(OriginAssociations&&) =
    default;
OriginAssociations::~OriginAssociations() = default;

bool OriginAssociations::operator==(const OriginAssociations& other) const {
  return scope_extensions == other.scope_extensions &&
         migration_sources == other.migration_sources;
}

void WebAppOriginAssociationManager::GetWebAppOriginAssociations(
    const GURL& web_app_identity,
    OriginAssociations origin_associations,
    OnDidGetWebAppOriginAssociations callback) {
  if (origin_associations.scope_extensions.empty() &&
      origin_associations.migration_sources.empty()) {
    std::move(callback).Run(OriginAssociations());
    return;
  }

  auto task =
      std::make_unique<Task>(web_app_identity, std::move(origin_associations),
                             *this, std::move(callback));
  pending_tasks_.push_back(std::move(task));
  MaybeStartNextTask();
}

void WebAppOriginAssociationManager::MaybeStartNextTask() {
  if (task_in_progress_ || pending_tasks_.empty())
    return;

  task_in_progress_ = true;
  pending_tasks_.front().get()->Start();
}

void WebAppOriginAssociationManager::OnTaskCompleted() {
  DCHECK(!pending_tasks_.empty());
  task_in_progress_ = false;
  pending_tasks_.pop_front();
  MaybeStartNextTask();
}

void WebAppOriginAssociationManager::SetFetcherForTest(
    std::unique_ptr<webapps::WebAppOriginAssociationFetcher> fetcher) {
  fetcher_ = std::move(fetcher);
}

webapps::WebAppOriginAssociationFetcher&
WebAppOriginAssociationManager::GetFetcherForTest() {
  DCHECK(fetcher_);
  return *fetcher_;
}

webapps::WebAppOriginAssociationFetcher&
WebAppOriginAssociationManager::GetFetcher() {
  if (!fetcher_) {
    fetcher_ = std::make_unique<webapps::WebAppOriginAssociationFetcher>(
        profile_->GetDefaultStoragePartition()
            ->GetURLLoaderFactoryForBrowserProcess());
  }
  return *fetcher_;
}

}  // namespace web_app