[WIP Prototype] ES6 https://html.spec.whatwg.org/#fetch-a-single-module-script implementation

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

 /*
  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
  *           (C) 2001 Dirk Mueller (mueller@kde.org)
  * Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008 Apple Inc. All rights
  * reserved.
  * Copyright (C) 2008 Nikolas Zimmermann <zimmermann@kde.org>
  *
  * This library is free software; you can redistribute it and/or
  * modify it under the terms of the GNU Library General Public
  * License as published by the Free Software Foundation; either
  * version 2 of the License, or (at your option) any later version.
  *
  * This library is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  * Library General Public License for more details.
  *
  * You should have received a copy of the GNU Library General Public License
  * along with this library; see the file COPYING.LIB.  If not, write to
  * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
  * Boston, MA 02110-1301, USA.
  */
 
 #include "core/dom/ScriptLoader.h"
 
 #include "bindings/core/v8/ScriptController.h"
 #include "bindings/core/v8/ScriptSourceCode.h"
+#include "bindings/core/v8/ScriptState.h"
 #include "core/HTMLNames.h"
 #include "core/SVGNames.h"
 #include "core/dom/Document.h"
 #include "core/dom/DocumentParserTiming.h"
 #include "core/dom/IgnoreDestructiveWriteCountIncrementer.h"
+#include "core/dom/Modulator.h"
+#include "core/dom/ModuleScript.h"
 #include "core/dom/ScriptLoaderClient.h"
 #include "core/dom/ScriptRunner.h"
 #include "core/dom/ScriptableDocumentParser.h"
 #include "core/dom/Text.h"
 #include "core/events/Event.h"
 #include "core/frame/LocalFrame.h"
 #include "core/frame/SubresourceIntegrity.h"
 #include "core/frame/UseCounter.h"
 #include "core/frame/csp/ContentSecurityPolicy.h"
 #include "core/html/CrossOriginAttribute.h"
 #include "core/html/HTMLScriptElement.h"
 #include "core/html/imports/HTMLImport.h"
 #include "core/html/parser/HTMLParserIdioms.h"
 #include "core/inspector/ConsoleMessage.h"
+#include "core/loader/modulescript/ModuleScriptFetchRequest.h"
 #include "core/svg/SVGScriptElement.h"
 #include "platform/WebFrameScheduler.h"
 #include "platform/loader/fetch/AccessControlStatus.h"
 #include "platform/loader/fetch/FetchRequest.h"
 #include "platform/loader/fetch/MemoryCache.h"
 #include "platform/loader/fetch/ResourceFetcher.h"
 #include "platform/network/mime/MIMETypeRegistry.h"
 #include "platform/weborigin/SecurityOrigin.h"
 #include "public/platform/WebCachePolicy.h"
 #include "wtf/StdLibExtras.h"
 #include "wtf/text/StringBuilder.h"
 #include "wtf/text/StringHash.h"
 
 namespace blink {
 
+class ScriptElementModuleClient
+    : public GarbageCollectedFinalized<ScriptElementModuleClient>,
+      public ModuleTreeClient {
+  USING_GARBAGE_COLLECTED_MIXIN(ScriptElementModuleClient);
+
+ public:
+  static ScriptElementModuleClient* create(Element* element) {
+    return new ScriptElementModuleClient(element);
+  }
+  virtual ~ScriptElementModuleClient() = default;
+
+  DECLARE_TRACE();
+
+ private:
+  ScriptElementModuleClient(Element* element) : m_element(element) {}
+
+  // Implements ModuleTreeClient
+  void notifyModuleTreeLoadFinished(ModuleScript*) override;
+
+  Member<Element> m_element;
+};
+
+void ScriptElementModuleClient::notifyModuleTreeLoadFinished(
+    ModuleScript* moduleScript) {
+  printf("notifyFinishedModuleTree!!! moduleScript: %p\n", moduleScript);
+  if (!moduleScript)
+    return;
+  DCHECK_EQ(moduleScript->instantiationState(),
+            ModuleInstantiationState::Instantiated);
+
+  LocalFrame* frame = m_element->document().frame();
+
+  // TODO(kouhei): Actually we should just return w/ the prepared script here.
+  // In this prototype, we will execute the module immediately just for fun. :)
+  Modulator::from(frame)->executeModule(moduleScript->record());
+}
+
+DEFINE_TRACE(ScriptElementModuleClient) {
+  visitor->trace(m_element);
+  ModuleTreeClient::trace(visitor);
+}
+
 ScriptLoader::ScriptLoader(Element* element,
                            bool parserInserted,
                            bool alreadyStarted,
                            bool createdDuringDocumentWrite)
     : m_element(element),
       m_startLineNumber(WTF::OrdinalNumber::beforeFirst()),
       m_haveFiredLoad(false),
       m_willBeParserExecuted(false),
       m_willExecuteWhenDocumentFinishedParsing(false),
       m_createdDuringDocumentWrite(createdDuringDocumentWrite),
@@ skipping 27 matching lines @@
       !element->document().isInDocumentWrite())
     m_startLineNumber =
         element->document().scriptableDocumentParser()->lineNumber();
 }
 
 ScriptLoader::~ScriptLoader() {}
 
 DEFINE_TRACE(ScriptLoader) {
   visitor->trace(m_element);
   visitor->trace(m_resource);
+  visitor->trace(m_elementModuleClient);
   visitor->trace(m_pendingScript);
   PendingScriptClient::trace(visitor);
 }
 
 void ScriptLoader::setFetchDocWrittenScriptDeferIdle() {
   DCHECK(!m_createdDuringDocumentWrite);
   m_documentWriteIntervention =
       DocumentWriteIntervention::FetchDocWrittenScriptDeferIdle;
 }
 
@@ skipping 192 matching lines @@
 
   // 21. "If the element has a src content attribute, run these substeps:"
   if (client->hasSourceAttribute()) {
     FetchRequest::DeferOption defer = FetchRequest::NoDefer;
     if (!m_parserInserted || client->asyncAttributeValue() ||
         client->deferAttributeValue())
       defer = FetchRequest::LazyLoad;
     if (m_documentWriteIntervention ==
         DocumentWriteIntervention::FetchDocWrittenScriptDeferIdle)
       defer = FetchRequest::IdleLoad;
+
     if (!fetchScript(client->sourceAttributeValue(), encoding, defer))
       return false;
   }
 
   // 22. "If the element does not have a src content attribute,
   //      run these substeps:"
 
   // 22.1. "Let source text be the value of the text IDL attribute."
   // This step is done later:
   // - in ScriptLoader::pendingScript() (Step 23, 6th Clause),
@@ skipping 183 matching lines @@
                                FetchRequest::DeferOption defer) {
   DCHECK(m_element);
 
   Document* elementDocument = &(m_element->document());
   if (!m_element->isConnected() || m_element->document() != elementDocument)
     return false;
 
   DCHECK(!m_resource);
   // 21. "If the element has a src content attribute, run these substeps:"
   if (!stripLeadingAndTrailingHTMLSpaces(sourceUrl).isEmpty()) {
-    // 21.4. "Parse src relative to the element's node document."
-    FetchRequest request(
-        ResourceRequest(elementDocument->completeURL(sourceUrl)),
-        m_element->localName());
+    KURL url = elementDocument->completeURL(sourceUrl);
 
     // 15. "Let CORS setting be the current state of the element's
     //      crossorigin content attribute."
     CrossOriginAttributeValue crossOrigin = crossOriginAttributeValue(
         m_element->fastGetAttribute(HTMLNames::crossoriginAttr));
 
-    // 16. "Let module script credentials mode be determined by switching
-    //      on CORS setting:"
-    // TODO(hiroshige): Implement this step for "module".
+    if (RuntimeEnabledFeatures::moduleScriptsEnabled() &&
+        client()->typeAttributeValue() == "module") {
+      m_elementModuleClient = ScriptElementModuleClient::create(m_element);
+
+      String nonce;
+      if (ContentSecurityPolicy::isNonceableElement(m_element.get())) {
+        nonce = m_element->fastGetAttribute(HTMLNames::nonceAttr);
+      }
+      ParserDisposition parserState =
+          isParserInserted() ? ParserInserted : NotParserInserted;
+
+      // 16. "Let module script credentials mode be determined by switching
+      //      on CORS setting:"
+      WebURLRequest::FetchCredentialsMode credentialsMode;
+      switch (crossOrigin) {
+        case CrossOriginAttributeNotSet:
+          credentialsMode = WebURLRequest::FetchCredentialsModeOmit;
+          break;
+        case CrossOriginAttributeAnonymous:
+          credentialsMode = WebURLRequest::FetchCredentialsModeSameOrigin;
+          break;
+        case CrossOriginAttributeUseCredentials:
+          credentialsMode = WebURLRequest::FetchCredentialsModeInclude;
+          break;
+        default:
+          NOTREACHED();
+      }
+
+      ModuleScriptFetchRequest moduleRequest(url, nonce, parserState,
+                                             credentialsMode);
+      Modulator::from(elementDocument->frame())
+          ->fetchTree(moduleRequest, m_elementModuleClient);
+      return true;
+    }
+
+    // 21.4. "Parse src relative to the element's node document."
+    FetchRequest request(
+        ResourceRequest(elementDocument->completeURL(sourceUrl)),
+        m_element->localName());
 
     // 21.6, "classic": "Fetch a classic script given ... CORS setting
     //                   ... and encoding."
     if (crossOrigin != CrossOriginAttributeNotSet)
       request.setCrossOriginAccessControl(elementDocument->getSecurityOrigin(),
                                           crossOrigin);
 
     request.setCharset(encoding);
 
     // 17. "If the script element has a nonce attribute,
@@ skipping 342 matching lines @@
   if (isHTMLScriptLoader(element))
     return toHTMLScriptElement(element)->loader();
 
   if (isSVGScriptLoader(element))
     return toSVGScriptElement(element)->loader();
 
   return 0;
 }
 
 }  // namespace blink