dom/serviceworkers/NavigationPreloadManager.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "NavigationPreloadManager.h"
   8 #include "ServiceWorkerUtils.h"
   9 #include "nsNetUtil.h"
  10 #include "mozilla/StaticPrefs_dom.h"
  11 #include "mozilla/dom/NavigationPreloadManagerBinding.h"
  12 #include "mozilla/dom/Promise.h"
  13 #include "mozilla/dom/ServiceWorker.h"
  14 #include "mozilla/ipc/MessageChannel.h"
  15
  16 namespace mozilla::dom {
  17
  18 NS_IMPL_CYCLE_COLLECTING_ADDREF(NavigationPreloadManager)
  19 NS_IMPL_CYCLE_COLLECTING_RELEASE(NavigationPreloadManager)
  20
  21 NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(NavigationPreloadManager)
  22   NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
  23   NS_INTERFACE_MAP_ENTRY(nsISupports)
  24 NS_INTERFACE_MAP_END
  25
  26 NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(NavigationPreloadManager,
  27                                       mServiceWorkerRegistration)
  28
  29 /* static */
  30 bool NavigationPreloadManager::IsValidHeader(const nsACString& aHeader) {
  31   return NS_IsReasonableHTTPHeaderValue(aHeader);
  32 }
  33
  34 bool NavigationPreloadManager::IsEnabled(JSContext* aCx, JSObject* aGlobal) {
  35   return StaticPrefs::dom_serviceWorkers_navigationPreload_enabled() &&
  36          ServiceWorkerVisible(aCx, aGlobal);
  37 }
  38
  39 NavigationPreloadManager::NavigationPreloadManager(
  40     RefPtr<ServiceWorkerRegistration>& aServiceWorkerRegistration)
  41     : mServiceWorkerRegistration(aServiceWorkerRegistration) {}
  42
  43 JSObject* NavigationPreloadManager::WrapObject(
  44     JSContext* aCx, JS::Handle<JSObject*> aGivenProto) {
  45   return NavigationPreloadManager_Binding::Wrap(aCx, this, aGivenProto);
  46 }
  47
  48 already_AddRefed<Promise> NavigationPreloadManager::SetEnabled(
  49     bool aEnabled, ErrorResult& aError) {
  50   RefPtr<Promise> promise = Promise::Create(GetParentObject(), aError);
  51
  52   if (NS_WARN_IF(aError.Failed())) {
  53     return nullptr;
  54   }
  55
  56   if (!mServiceWorkerRegistration) {
  57     promise->MaybeReject(NS_ERROR_DOM_INVALID_STATE_ERR);
  58     return promise.forget();
  59   }
  60
  61   mServiceWorkerRegistration->SetNavigationPreloadEnabled(
  62       aEnabled,
  63       [promise](bool aSuccess) {
  64         if (aSuccess) {
  65           promise->MaybeResolveWithUndefined();
  66           return;
  67         }
  68         promise->MaybeReject(NS_ERROR_DOM_INVALID_STATE_ERR);
  69       },
  70       [promise](ErrorResult&& aRv) { promise->MaybeReject(std::move(aRv)); });
  71
  72   return promise.forget();
  73 }
  74
  75 already_AddRefed<Promise> NavigationPreloadManager::Enable(
  76     ErrorResult& aError) {
  77   return SetEnabled(true, aError);
  78 }
  79
  80 already_AddRefed<Promise> NavigationPreloadManager::Disable(
  81     ErrorResult& aError) {
  82   return SetEnabled(false, aError);
  83 }
  84
  85 already_AddRefed<Promise> NavigationPreloadManager::SetHeaderValue(
  86     const nsACString& aHeader, ErrorResult& aError) {
  87   RefPtr<Promise> promise = Promise::Create(GetParentObject(), aError);
  88
  89   if (NS_WARN_IF(aError.Failed())) {
  90     return nullptr;
  91   }
  92
  93   if (!IsValidHeader(aHeader)) {
  94     promise->MaybeRejectWithTypeError<MSG_INVALID_HEADER_VALUE>(aHeader);
  95     return promise.forget();
  96   }
  97
  98   if (!mServiceWorkerRegistration) {
  99     promise->MaybeReject(NS_ERROR_DOM_INVALID_STATE_ERR);
 100     return promise.forget();
 101   }
 102
 103   mServiceWorkerRegistration->SetNavigationPreloadHeader(
 104       nsAutoCString(aHeader),
 105       [promise](bool aSuccess) {
 106         if (aSuccess) {
 107           promise->MaybeResolveWithUndefined();
 108           return;
 109         }
 110         promise->MaybeReject(NS_ERROR_DOM_INVALID_STATE_ERR);
 111       },
 112       [promise](ErrorResult&& aRv) { promise->MaybeReject(std::move(aRv)); });
 113
 114   return promise.forget();
 115 }
 116
 117 already_AddRefed<Promise> NavigationPreloadManager::GetState(
 118     ErrorResult& aError) {
 119   RefPtr<Promise> promise = Promise::Create(GetParentObject(), aError);
 120
 121   if (NS_WARN_IF(aError.Failed())) {
 122     return nullptr;
 123   }
 124
 125   if (!mServiceWorkerRegistration) {
 126     promise->MaybeReject(NS_ERROR_DOM_INVALID_STATE_ERR);
 127     return promise.forget();
 128   }
 129
 130   mServiceWorkerRegistration->GetNavigationPreloadState(
 131       [promise](NavigationPreloadState&& aState) {
 132         promise->MaybeResolve(std::move(aState));
 133       },
 134       [promise](ErrorResult&& aRv) { promise->MaybeReject(std::move(aRv)); });
 135
 136   return promise.forget();
 137 }
 138
 139 }  // namespace mozilla::dom