content/public/test/service_worker_host_interceptor.cc

// Copyright 2019 The Chromium 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 "content/public/test/service_worker_host_interceptor.h"

#include <utility>

#include "base/callback.h"
#include "base/location.h"
#include "base/run_loop.h"
#include "content/browser/service_worker/service_worker_context_wrapper.h"
#include "content/browser/service_worker/service_worker_version.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/storage_partition.h"

namespace content {

ServiceWorkerHostInterceptor::ServiceWorkerHostInterceptor() = default;

ServiceWorkerHostInterceptor::~ServiceWorkerHostInterceptor() = default;

blink::ServiceWorkerStatusCode
ServiceWorkerHostInterceptor::InterceptServiceWorkerHostWithScope(
    BrowserContext* browser_context,
    const GURL& scope,
    int* service_worker_process_id_out) {
  base::RunLoop run_loop;
  scoped_refptr<ServiceWorkerContextWrapper> context =
      static_cast<ServiceWorkerContextWrapper*>(
          BrowserContext::GetDefaultStoragePartition(browser_context)
              ->GetServiceWorkerContext());
  BrowserThread::ID run_done_thread;
  bool is_known_thread =
      BrowserThread::GetCurrentThreadIdentifier(&run_done_thread);
  DCHECK(is_known_thread);
  RunOrPostTaskOnThread(
      FROM_HERE, ServiceWorkerContext::GetCoreThreadId(),
      base::BindOnce(&ServiceWorkerHostInterceptor::
                         FindRegistrationOnServiceWorkerCoreThread,
                     base::Unretained(this), context, scope, run_done_thread,
                     run_loop.QuitClosure()));
  run_loop.Run();
  *service_worker_process_id_out = service_worker_process_id_;
  return status_;
}

bool ServiceWorkerHostInterceptor::WillOpenPaymentHandlerWindow(GURL* url) {
  return true;
}

blink::mojom::ServiceWorkerHost*
ServiceWorkerHostInterceptor::GetForwardingInterface() {
  return service_worker_version_;
}

void ServiceWorkerHostInterceptor::OpenPaymentHandlerWindow(
    const GURL& url,
    OpenPaymentHandlerWindowCallback callback) {
  GURL url_copy = url;
  if (WillOpenPaymentHandlerWindow(&url_copy)) {
    GetForwardingInterface()->OpenPaymentHandlerWindow(url_copy,
                                                       std::move(callback));
  }
}

void ServiceWorkerHostInterceptor::FindRegistrationOnServiceWorkerCoreThread(
    scoped_refptr<ServiceWorkerContextWrapper> context,
    const GURL& scope,
    BrowserThread::ID run_done_thread,
    base::OnceClosure done) {
  context->FindRegistrationForScope(
      scope,
      base::BindOnce(&ServiceWorkerHostInterceptor::
                         OnFoundRegistrationOnServiceWorkerCoreThread,
                     base::Unretained(this), run_done_thread, std::move(done)));
}

void ServiceWorkerHostInterceptor::OnFoundRegistrationOnServiceWorkerCoreThread(
    BrowserThread::ID run_done_thread,
    base::OnceClosure done,
    blink::ServiceWorkerStatusCode status,
    scoped_refptr<ServiceWorkerRegistration> registration) {
  status_ = status;
  service_worker_version_ = registration->active_version();
  service_worker_process_id_ =
      service_worker_version_->embedded_worker()->process_id();
  forwarding_interface_ =
      service_worker_version_->service_worker_host_receiver_for_testing()
          .SwapImplForTesting(this);
  RunOrPostTaskOnThread(FROM_HERE, run_done_thread, std::move(done));
}

}  // namespace content