1 files changed, 139 insertions, 0 deletions
diff --git a/dom/serviceworkers/NavigationPreloadManager.cpp b/dom/serviceworkers/NavigationPreloadManager.cpp
new file mode 100644
index 0000000000..93c17353ab
--- /dev/null
+++ b/dom/serviceworkers/NavigationPreloadManager.cpp
@@ -0,0 +1,139 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "NavigationPreloadManager.h"
+#include "ServiceWorkerUtils.h"
+#include "nsNetUtil.h"
+#include "mozilla/StaticPrefs_dom.h"
+#include "mozilla/dom/NavigationPreloadManagerBinding.h"
+#include "mozilla/dom/Promise.h"
+#include "mozilla/dom/ServiceWorker.h"
+#include "mozilla/ipc/MessageChannel.h"
+
+namespace mozilla::dom {
+
+NS_IMPL_CYCLE_COLLECTING_ADDREF(NavigationPreloadManager)
+NS_IMPL_CYCLE_COLLECTING_RELEASE(NavigationPreloadManager)
+
+NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(NavigationPreloadManager)
+  NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
+  NS_INTERFACE_MAP_ENTRY(nsISupports)
+NS_INTERFACE_MAP_END
+
+NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(NavigationPreloadManager,
+                                      mServiceWorkerRegistration)
+
+/* static */
+bool NavigationPreloadManager::IsValidHeader(const nsACString& aHeader) {
+  return NS_IsReasonableHTTPHeaderValue(aHeader);
+}
+
+bool NavigationPreloadManager::IsEnabled(JSContext* aCx, JSObject* aGlobal) {
+  return StaticPrefs::dom_serviceWorkers_navigationPreload_enabled() &&
+         ServiceWorkerVisible(aCx, aGlobal);
+}
+
+NavigationPreloadManager::NavigationPreloadManager(
+    RefPtr<ServiceWorkerRegistration>& aServiceWorkerRegistration)
+    : mServiceWorkerRegistration(aServiceWorkerRegistration) {}
+
+JSObject* NavigationPreloadManager::WrapObject(
+    JSContext* aCx, JS::Handle<JSObject*> aGivenProto) {
+  return NavigationPreloadManager_Binding::Wrap(aCx, this, aGivenProto);
+}
+
+already_AddRefed<Promise> NavigationPreloadManager::SetEnabled(
+    bool aEnabled, ErrorResult& aError) {
+  RefPtr<Promise> promise = Promise::Create(GetParentObject(), aError);
+
+  if (NS_WARN_IF(aError.Failed())) {
+    return nullptr;
+  }
+
+  if (!mServiceWorkerRegistration) {
+    promise->MaybeReject(NS_ERROR_DOM_INVALID_STATE_ERR);
+    return promise.forget();
+  }
+
+  mServiceWorkerRegistration->SetNavigationPreloadEnabled(
+      aEnabled,
+      [promise](bool aSuccess) {
+        if (aSuccess) {
+          promise->MaybeResolveWithUndefined();
+          return;
+        }
+        promise->MaybeReject(NS_ERROR_DOM_INVALID_STATE_ERR);
+      },
+      [promise](ErrorResult&& aRv) { promise->MaybeReject(std::move(aRv)); });
+
+  return promise.forget();
+}
+
+already_AddRefed<Promise> NavigationPreloadManager::Enable(
+    ErrorResult& aError) {
+  return SetEnabled(true, aError);
+}
+
+already_AddRefed<Promise> NavigationPreloadManager::Disable(
+    ErrorResult& aError) {
+  return SetEnabled(false, aError);
+}
+
+already_AddRefed<Promise> NavigationPreloadManager::SetHeaderValue(
+    const nsACString& aHeader, ErrorResult& aError) {
+  RefPtr<Promise> promise = Promise::Create(GetParentObject(), aError);
+
+  if (NS_WARN_IF(aError.Failed())) {
+    return nullptr;
+  }
+
+  if (!IsValidHeader(aHeader)) {
+    promise->MaybeRejectWithTypeError<MSG_INVALID_HEADER_VALUE>(aHeader);
+    return promise.forget();
+  }
+
+  if (!mServiceWorkerRegistration) {
+    promise->MaybeReject(NS_ERROR_DOM_INVALID_STATE_ERR);
+    return promise.forget();
+  }
+
+  mServiceWorkerRegistration->SetNavigationPreloadHeader(
+      nsAutoCString(aHeader),
+      [promise](bool aSuccess) {
+        if (aSuccess) {
+          promise->MaybeResolveWithUndefined();
+          return;
+        }
+        promise->MaybeReject(NS_ERROR_DOM_INVALID_STATE_ERR);
+      },
+      [promise](ErrorResult&& aRv) { promise->MaybeReject(std::move(aRv)); });
+
+  return promise.forget();
+}
+
+already_AddRefed<Promise> NavigationPreloadManager::GetState(
+    ErrorResult& aError) {
+  RefPtr<Promise> promise = Promise::Create(GetParentObject(), aError);
+
+  if (NS_WARN_IF(aError.Failed())) {
+    return nullptr;
+  }
+
+  if (!mServiceWorkerRegistration) {
+    promise->MaybeReject(NS_ERROR_DOM_INVALID_STATE_ERR);
+    return promise.forget();
+  }
+
+  mServiceWorkerRegistration->GetNavigationPreloadState(
+      [promise](NavigationPreloadState&& aState) {
+        promise->MaybeResolve(std::move(aState));
+      },
+      [promise](ErrorResult&& aRv) { promise->MaybeReject(std::move(aRv)); });
+
+  return promise.forget();
+}
+
+}  // namespace mozilla::dom