Worklet: Enable module script loading for main thread worklets

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

 // Copyright 2016 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/TaskRunnerHelper.h"
 
 #include "core/dom/Document.h"
 #include "core/dom/ExecutionContext.h"
 #include "core/frame/LocalFrame.h"
+#include "core/workers/MainThreadWorkletGlobalScope.h"
 #include "core/workers/WorkerOrWorkletGlobalScope.h"
 #include "core/workers/WorkerThread.h"
 #include "platform/WebFrameScheduler.h"
 #include "platform/WebTaskRunner.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebThread.h"
 
 namespace blink {
 
 RefPtr<WebTaskRunner> TaskRunnerHelper::Get(TaskType type, LocalFrame* frame) {
@@ skipping 44 matching lines @@
   return Get(type, document ? document->GetFrame() : nullptr);
 }
 
 RefPtr<WebTaskRunner> TaskRunnerHelper::Get(
     TaskType type,
     ExecutionContext* executionContext) {
   if (!executionContext)
     return Get(type, ToDocument(executionContext));
   if (executionContext->IsDocument())
     return Get(type, ToDocument(executionContext));
+  if (executionContext->IsMainThreadWorkletGlobalScope()) {
+    return Get(type,
+               ToMainThreadWorkletGlobalScope(executionContext)->GetFrame());
+  }
   if (executionContext->IsWorkerOrWorkletGlobalScope())
     return Get(type, ToWorkerOrWorkletGlobalScope(executionContext));
   executionContext = nullptr;
   return Get(type, ToDocument(executionContext));
 }
 
 RefPtr<WebTaskRunner> TaskRunnerHelper::Get(TaskType type,
                                             ScriptState* script_state) {
   return Get(type,
              script_state ? ExecutionContext::From(script_state) : nullptr);
@@ skipping 37 matching lines @@
       // TODO(nhiroki): Identify which tasks can be throttled / suspendable and
       // move them into other task runners. See also comments in
       // Get(LocalFrame). (https://crbug.com/670534)
       return worker_thread->GetGlobalScopeScheduler()->UnthrottledTaskRunner();
   }
   NOTREACHED();
   return nullptr;
 }
 
 }  // namespace blink