chrome/browser/web_applications/externally_managed_app_registration_task.h

// 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.

#ifndef CHROME_BROWSER_WEB_APPLICATIONS_EXTERNALLY_MANAGED_APP_REGISTRATION_TASK_H_
#define CHROME_BROWSER_WEB_APPLICATIONS_EXTERNALLY_MANAGED_APP_REGISTRATION_TASK_H_

#include "base/callback.h"
#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/observer_list_types.h"
#include "base/timer/timer.h"
#include "chrome/browser/web_applications/web_app_url_loader.h"
#include "content/public/browser/service_worker_context_observer.h"

class GURL;

namespace content {
enum class ServiceWorkerCapability;
class ServiceWorkerContext;
class WebContents;
}  // namespace content

namespace web_app {

enum class RegistrationResultCode;
class WebAppUrlLoader;

class ExternallyManagedAppRegistrationTaskBase
    : public content::ServiceWorkerContextObserver {
 public:
  ~ExternallyManagedAppRegistrationTaskBase() override;

  const GURL& install_url() const { return install_url_; }

 protected:
  explicit ExternallyManagedAppRegistrationTaskBase(GURL install_url);

 private:
  const GURL install_url_;
};

class ExternallyManagedAppRegistrationTask
    : public ExternallyManagedAppRegistrationTaskBase {
 public:
  using RegistrationCallback = base::OnceCallback<void(RegistrationResultCode)>;

  ExternallyManagedAppRegistrationTask(GURL install_url,
                                       WebAppUrlLoader* url_loader,
                                       content::WebContents* web_contents,
                                       RegistrationCallback callback);
  ExternallyManagedAppRegistrationTask(
      const ExternallyManagedAppRegistrationTask&) = delete;
  ExternallyManagedAppRegistrationTask& operator=(
      const ExternallyManagedAppRegistrationTask&) = delete;
  ~ExternallyManagedAppRegistrationTask() override;

  // ServiceWorkerContextObserver:
  void OnRegistrationCompleted(const GURL& scope) override;
  void OnDestruct(content::ServiceWorkerContext* context) override;

  static void SetTimeoutForTesting(int registration_timeout_in_seconds);

 private:
  // Check to see if there is already a service worker for the install url.
  void CheckHasServiceWorker();

  void OnDidCheckHasServiceWorker(content::ServiceWorkerCapability capability);

  void OnWebContentsReady(WebAppUrlLoader::Result result);

  void OnRegistrationTimeout();

  const raw_ptr<WebAppUrlLoader> url_loader_;
  const raw_ptr<content::WebContents> web_contents_;
  RegistrationCallback callback_;
  raw_ptr<content::ServiceWorkerContext> service_worker_context_;

  base::OneShotTimer registration_timer_;

  base::WeakPtrFactory<ExternallyManagedAppRegistrationTask> weak_ptr_factory_{
      this};

  static int registration_timeout_in_seconds_;
};

}  // namespace web_app

#endif  // CHROME_BROWSER_WEB_APPLICATIONS_EXTERNALLY_MANAGED_APP_REGISTRATION_TASK_H_