Support serializing shadow DOM to MHTML

third_party/WebKit/Source/web/LocalFrameClientImpl.cpp

 /*
  * Copyright (C) 2009, 2012 Google Inc. All rights reserved.
  * Copyright (C) 2011 Apple Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
@@ skipping 194 matching lines @@
     web_frame_->Client()->DidCreateDocumentElement(web_frame_);
 }
 
 void LocalFrameClientImpl::RunScriptsAtDocumentElementAvailable() {
   if (web_frame_->Client())
     web_frame_->Client()->RunScriptsAtDocumentElementAvailable(web_frame_);
   // The callback might have deleted the frame, do not use |this|!
 }
 
 void LocalFrameClientImpl::RunScriptsAtDocumentReady(bool document_is_empty) {
+  if (!document_is_empty && IsLoadedAsMHTMLArchive(web_frame_->GetFrame())) {
+    // For MHTML pages, recreate the shadow DOM contents from the templates that
+    // are captured from the shadow DOM trees at serialization.
+    // Note that the MHTML page is loaded in sandboxing mode with script
+    // execution disabled and thus only the following script will be executed.
+    // Any other scripts and event handlers outside the scope of the following
+    // script, including those that may be inserted in shadow DOM templates,
+    // will NOT be run.
+    String script = R"(
+function createShadowRootWithin(node) {
+  var nodes = node.querySelectorAll('template[shadowmode]');
+  for (var i = 0; i < nodes.length; ++i) {
+    var template = nodes[i];
+    var mode = template.getAttribute('shadowmode');
+    var parent = template.parentNode;
+    parent.removeChild(template);

[Dmitry Titov, 2017/06/06 04:42:09]

Just to be on a safe side, lets check parent for null...

[jianli, 2017/06/07 00:53:42]

Done.

+    var shadowRoot;
+    if (mode == 'v0') {
+      shadowRoot = parent.createShadowRoot();
+    } else if (mode == 'open' || mode == 'closed') {
+      var delegatesFocus = template.hasAttribute('shadowdelegatesfocus');
+      shadowRoot = parent.attachShadow({'mode': mode,
+                                        'delegatesFocus': delegatesFocus});
+    }
+    if (!shadowRoot)
+      continue;
+    var clone = document.importNode(template.content, true);
+    shadowRoot.appendChild(clone);
+    createShadowRootWithin(shadowRoot);
+  }
+}
+createShadowRootWithin(document.body);
+)";
+    web_frame_->GetFrame()->GetScriptController().ExecuteScriptInMainWorld(
+        script, ScriptController::kExecuteScriptWhenScriptsDisabled);
+  }
+
   if (web_frame_->Client()) {
     web_frame_->Client()->RunScriptsAtDocumentReady(document_is_empty);
   }
   // The callback might have deleted the frame, do not use |this|!
 }
 
 void LocalFrameClientImpl::RunScriptsAtDocumentIdle() {
   if (web_frame_->Client())
     web_frame_->Client()->RunScriptsAtDocumentIdle();
   // The callback might have deleted the frame, do not use |this|!
@@ skipping 815 matching lines @@
 
 TextCheckerClient& LocalFrameClientImpl::GetTextCheckerClient() const {
   return web_frame_->GetTextCheckerClient();
 }
 
 std::unique_ptr<blink::WebURLLoader> LocalFrameClientImpl::CreateURLLoader() {
   return web_frame_->CreateURLLoader();
 }
 
 }  // namespace blink