Worklet: Enable module script loading for main thread worklets

third_party/WebKit/Source/core/dom/Modulator.cpp

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "core/dom/Modulator.h"
 
 #include "bindings/core/v8/V8BindingForCore.h"
 #include "core/dom/Document.h"
 #include "core/dom/ModulatorImpl.h"
 #include "core/frame/LocalFrame.h"
+#include "core/workers/MainThreadWorkletGlobalScope.h"
 #include "platform/bindings/ScriptState.h"
 #include "platform/bindings/V8PerContextData.h"
 
 namespace blink {
 
 namespace {
 const char kPerContextDataKey[] = "Modulator";
 }  // namespace
 
 Modulator* Modulator::From(ScriptState* script_state) {
   if (!script_state)
     return nullptr;
 
   V8PerContextData* per_context_data = script_state->PerContextData();
   if (!per_context_data)
     return nullptr;
 
   Modulator* modulator =
       static_cast<Modulator*>(per_context_data->GetData(kPerContextDataKey));
-  if (!modulator) {
-    if (Document* document = ToDocument(ExecutionContext::From(script_state))) {
-      modulator = ModulatorImpl::Create(script_state, document->Fetcher());
-      Modulator::SetModulator(script_state, modulator);
-    }
+  if (modulator)
+    return modulator;
+  ExecutionContext* execution_context = ExecutionContext::From(script_state);
+  if (execution_context->IsDocument()) {
+    Document* document = ToDocument(execution_context);
+    modulator = ModulatorImpl::Create(script_state, document->Fetcher());
+    Modulator::SetModulator(script_state, modulator);
+  } else if (execution_context->IsMainThreadWorkletGlobalScope()) {
+    MainThreadWorkletGlobalScope* global_scope =
+        ToMainThreadWorkletGlobalScope(execution_context);
+    modulator = ModulatorImpl::Create(
+        script_state, global_scope->GetFrame()->GetDocument()->Fetcher());
+    Modulator::SetModulator(script_state, modulator);
+  } else {
+    NOTREACHED();
   }
   return modulator;
 }
 
 Modulator::~Modulator() {}
 
 void Modulator::SetModulator(ScriptState* script_state, Modulator* modulator) {
   DCHECK(script_state);
   V8PerContextData* per_context_data = script_state->PerContextData();
   DCHECK(per_context_data);
@@ skipping 27 matching lines @@
   // script's base URL as the base URL.
   DCHECK(base_url.IsValid());
   KURL absolute_url(base_url, module_request);
   if (absolute_url.IsValid())
     return absolute_url;
 
   return KURL();
 }
 
 }  // namespace blink