IOS Reading distillation: Handle Google AMP iFrame.

components/dom_distiller/ios/distiller_page_ios.mm

 // Copyright 2014 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 "components/dom_distiller/ios/distiller_page_ios.h"
 
 #import <UIKit/UIKit.h>
 
 #include <utility>
 
@@ skipping 145 matching lines @@
   }
 }
 
 std::unique_ptr<web::WebState> DistillerPageIOS::DetachWebState() {
   std::unique_ptr<web::WebState> old_web_state = std::move(web_state_);
   web_state_observer_.reset();
   web_state_.reset();
   return old_web_state;
 }
 
+web::WebState* DistillerPageIOS::CurrentWebState() {
+  return web_state_.get();
+}
+
 void DistillerPageIOS::DistillPageImpl(const GURL& url,
                                        const std::string& script) {
   if (!url.is_valid() || !script.length())
     return;
   url_ = url;
   script_ = script;
 
   if (!web_state_) {
     const web::WebState::CreateParams web_state_create_params(browser_state_);
     std::unique_ptr<web::WebState> web_state_unique =
@@ skipping 37 matching lines @@
   }
   OnDistillationDone(url_, resultValue.get());
 }
 
 std::unique_ptr<base::Value> DistillerPageIOS::ValueResultFromScriptResult(
     id wk_result) {
   return ::ValueResultFromScriptResult(wk_result,
                                        kMaximumParsingRecursionDepth);
 }
 }  // namespace dom_distiller