1 files changed, 122 insertions, 0 deletions
diff --git a/dom/worklet/Worklet.cpp b/dom/worklet/Worklet.cpp
new file mode 100644
index 0000000000..7c37bdbba9
--- /dev/null
+++ b/dom/worklet/Worklet.cpp
@@ -0,0 +1,122 @@
+/* -*- 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 "Worklet.h"
+#include "WorkletThread.h"
+
+#include "mozilla/dom/WorkletFetchHandler.h"
+#include "mozilla/dom/WorkletImpl.h"
+#include "xpcprivate.h"
+
+using JS::loader::ResolveError;
+using JS::loader::ResolveErrorInfo;
+
+namespace mozilla::dom {
+// ---------------------------------------------------------------------------
+// Worklet
+
+NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE_CLASS(Worklet)
+
+NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN(Worklet)
+  NS_IMPL_CYCLE_COLLECTION_UNLINK(mWindow)
+  NS_IMPL_CYCLE_COLLECTION_UNLINK(mOwnedObject)
+  NS_IMPL_CYCLE_COLLECTION_UNLINK(mImportHandlers)
+  tmp->mImpl->NotifyWorkletFinished();
+  NS_IMPL_CYCLE_COLLECTION_UNLINK_PRESERVED_WRAPPER
+NS_IMPL_CYCLE_COLLECTION_UNLINK_END
+
+NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN(Worklet)
+  NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mWindow)
+  NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mOwnedObject)
+  NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mImportHandlers)
+NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END
+
+NS_IMPL_CYCLE_COLLECTING_ADDREF(Worklet)
+NS_IMPL_CYCLE_COLLECTING_RELEASE(Worklet)
+
+NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(Worklet)
+  NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
+  NS_INTERFACE_MAP_ENTRY(nsISupports)
+NS_INTERFACE_MAP_END
+
+Worklet::Worklet(nsPIDOMWindowInner* aWindow, RefPtr<WorkletImpl> aImpl,
+                 nsISupports* aOwnedObject)
+    : mWindow(aWindow), mOwnedObject(aOwnedObject), mImpl(std::move(aImpl)) {
+  MOZ_ASSERT(aWindow);
+  MOZ_ASSERT(mImpl);
+  MOZ_ASSERT(NS_IsMainThread());
+}
+
+Worklet::~Worklet() { mImpl->NotifyWorkletFinished(); }
+
+JSObject* Worklet::WrapObject(JSContext* aCx,
+                              JS::Handle<JSObject*> aGivenProto) {
+  return mImpl->WrapWorklet(aCx, this, aGivenProto);
+}
+
+static bool LoadLocalizedStrings(nsTArray<nsString>& aStrings) {
+  // All enumes in ResolveError.
+  ResolveError errors[] = {ResolveError::Failure,
+                           ResolveError::FailureMayBeBare,
+                           ResolveError::BlockedByNullEntry,
+                           ResolveError::BlockedByAfterPrefix,
+                           ResolveError::BlockedByBacktrackingPrefix,
+                           ResolveError::InvalidBareSpecifier};
+
+  static_assert(
+      ArrayLength(errors) == static_cast<size_t>(ResolveError::Length),
+      "The array 'errors' has missing entries in the enum class ResolveError.");
+
+  for (auto i : errors) {
+    nsAutoString message;
+    nsresult rv = nsContentUtils::GetLocalizedString(
+        nsContentUtils::eDOM_PROPERTIES, ResolveErrorInfo::GetString(i),
+        message);
+    if (NS_WARN_IF(NS_FAILED(rv))) {
+      if (NS_WARN_IF(!aStrings.AppendElement(EmptyString(), fallible))) {
+        return false;
+      }
+    } else {
+      if (NS_WARN_IF(!aStrings.AppendElement(message, fallible))) {
+        return false;
+      }
+    }
+  }
+
+  return true;
+}
+
+already_AddRefed<Promise> Worklet::AddModule(JSContext* aCx,
+                                             const nsAString& aModuleURL,
+                                             const WorkletOptions& aOptions,
+                                             CallerType aCallerType,
+                                             ErrorResult& aRv) {
+  MOZ_ASSERT(NS_IsMainThread());
+  if (mLocalizedStrings.IsEmpty()) {
+    bool result = LoadLocalizedStrings(mLocalizedStrings);
+    if (!result) {
+      return nullptr;
+    }
+  }
+
+  return WorkletFetchHandler::AddModule(this, aCx, aModuleURL, aOptions, aRv);
+}
+
+WorkletFetchHandler* Worklet::GetImportFetchHandler(const nsACString& aURI) {
+  MOZ_ASSERT(NS_IsMainThread());
+  return mImportHandlers.GetWeak(aURI);
+}
+
+void Worklet::AddImportFetchHandler(const nsACString& aURI,
+                                    WorkletFetchHandler* aHandler) {
+  MOZ_ASSERT(aHandler);
+  MOZ_ASSERT(!mImportHandlers.GetWeak(aURI));
+  MOZ_ASSERT(NS_IsMainThread());
+
+  mImportHandlers.InsertOrUpdate(aURI, RefPtr{aHandler});
+}
+
+}  // namespace mozilla::dom