1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
/* -*- 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/. */
#ifndef mozilla_dom_worklet_WorkletModuleLoader_h
#define mozilla_dom_worklet_WorkletModuleLoader_h
#include "js/loader/LoadContextBase.h"
#include "js/loader/ModuleLoaderBase.h"
#include "js/loader/ResolveResult.h" // For ResolveError
#include "mozilla/dom/WorkletFetchHandler.h"
namespace mozilla::dom {
namespace loader {
class WorkletScriptLoader : public JS::loader::ScriptLoaderInterface {
public:
NS_DECL_CYCLE_COLLECTING_ISUPPORTS
NS_DECL_CYCLE_COLLECTION_CLASS(WorkletScriptLoader)
nsIURI* GetBaseURI() const override { return nullptr; }
void ReportErrorToConsole(ScriptLoadRequest* aRequest,
nsresult aResult) const override {}
void ReportWarningToConsole(
ScriptLoadRequest* aRequest, const char* aMessageName,
const nsTArray<nsString>& aParams) const override {}
nsresult FillCompileOptionsForRequest(
JSContext* cx, ScriptLoadRequest* aRequest, JS::CompileOptions* aOptions,
JS::MutableHandle<JSScript*> aIntroductionScript) override {
aOptions->setIntroductionType("Worklet");
aOptions->setFileAndLine(aRequest->mURL.get(), 1);
aOptions->setIsRunOnce(true);
aOptions->setNoScriptRval(true);
return NS_OK;
}
private:
~WorkletScriptLoader() = default;
};
class WorkletModuleLoader : public JS::loader::ModuleLoaderBase {
public:
NS_DECL_ISUPPORTS_INHERITED
NS_DECL_CYCLE_COLLECTION_CLASS_INHERITED(WorkletModuleLoader,
JS::loader::ModuleLoaderBase)
WorkletModuleLoader(WorkletScriptLoader* aScriptLoader,
nsIGlobalObject* aGlobalObject);
void InsertRequest(nsIURI* aURI, JS::loader::ModuleLoadRequest* aRequest);
void RemoveRequest(nsIURI* aURI);
JS::loader::ModuleLoadRequest* GetRequest(nsIURI* aURI) const;
bool HasSetLocalizedStrings() const { return (bool)mLocalizedStrs; }
void SetLocalizedStrings(const nsTArray<nsString>* aStrings) {
mLocalizedStrs = aStrings;
}
private:
~WorkletModuleLoader() = default;
already_AddRefed<JS::loader::ModuleLoadRequest> CreateStaticImport(
nsIURI* aURI, JS::loader::ModuleLoadRequest* aParent) override;
already_AddRefed<JS::loader::ModuleLoadRequest> CreateDynamicImport(
JSContext* aCx, nsIURI* aURI, LoadedScript* aMaybeActiveScript,
JS::Handle<JS::Value> aReferencingPrivate,
JS::Handle<JSString*> aSpecifier,
JS::Handle<JSObject*> aPromise) override;
bool CanStartLoad(JS::loader::ModuleLoadRequest* aRequest,
nsresult* aRvOut) override;
nsresult StartFetch(JS::loader::ModuleLoadRequest* aRequest) override;
nsresult CompileFetchedModule(
JSContext* aCx, JS::Handle<JSObject*> aGlobal,
JS::CompileOptions& aOptions, JS::loader::ModuleLoadRequest* aRequest,
JS::MutableHandle<JSObject*> aModuleScript) override;
void OnModuleLoadComplete(JS::loader::ModuleLoadRequest* aRequest) override;
nsresult GetResolveFailureMessage(JS::loader::ResolveError aError,
const nsAString& aSpecifier,
nsAString& aResult) override;
// A hashtable to map a nsIURI(from main thread) to a ModuleLoadRequest(in
// worklet thread).
nsRefPtrHashtable<nsURIHashKey, JS::loader::ModuleLoadRequest>
mFetchingRequests;
// We get the localized strings on the main thread, and pass it to
// WorkletModuleLoader.
const nsTArray<nsString>* mLocalizedStrs = nullptr;
};
} // namespace loader
class WorkletLoadContext : public JS::loader::LoadContextBase {
public:
explicit WorkletLoadContext(
const nsMainThreadPtrHandle<WorkletFetchHandler>& aHandlerRef)
: JS::loader::LoadContextBase(JS::loader::ContextKind::Worklet),
mHandlerRef(aHandlerRef) {}
const nsMainThreadPtrHandle<WorkletFetchHandler>& GetHandlerRef() const {
return mHandlerRef;
}
private:
~WorkletLoadContext() = default;
nsMainThreadPtrHandle<WorkletFetchHandler> mHandlerRef;
};
} // namespace mozilla::dom
#endif // mozilla_dom_worklet_WorkletModuleLoader_h
|
