Source code
Revision control
Copy as Markdown
Other Tools
/* -*- 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
#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