IOS Reading distillation: Handle Google AMP iFrame.

components/dom_distiller/ios/distiller_page_ios.h

 // 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.
 
 #ifndef COMPONENTS_DOM_DISTILLER_IOS_DISTILLER_PAGE_IOS_H_
 #define COMPONENTS_DOM_DISTILLER_IOS_DISTILLER_PAGE_IOS_H_
 
 #include <memory>
 #include <string>
 
@@ skipping 19 matching lines @@
   ~DistillerPageIOS() override;
 
  protected:
   bool StringifyOutput() override;
   void DistillPageImpl(const GURL& url, const std::string& script) override;
 
   // Sets the WebState that will be used for the distillation. Do not call
   // between |DistillPageImpl| and |OnDistillationDone|.
   virtual void AttachWebState(std::unique_ptr<web::WebState> web_state);
 
+  // Returns the WebState handling the current distillation.
+  web::WebState* CurrentWebState();
+
   // Release the WebState used for distillation. Do not call between
   // |DistillPageImpl| and |OnDistillationDone|.
   virtual std::unique_ptr<web::WebState> DetachWebState();
 
   // Called by |web_state_observer_| once the page has finished loading.
   virtual void OnLoadURLDone(
       web::PageLoadCompletionStatus load_completion_status);
 
  private:
   friend class DistillerWebStateObserver;
   // Called once the |script_| has been evaluated on the page.
   void HandleJavaScriptResult(id result);
 
   // Converts result of WKWebView script evaluation to base::Value
   std::unique_ptr<base::Value> ValueResultFromScriptResult(id wk_result);
 
   GURL url_;
   std::string script_;
   web::BrowserState* browser_state_;
   std::unique_ptr<web::WebState> web_state_;
   std::unique_ptr<DistillerWebStateObserver> web_state_observer_;
   base::WeakPtrFactory<DistillerPageIOS> weak_ptr_factory_;
 };
 
 }  // namespace dom_distiller
 
 #endif  // COMPONENTS_DOM_DISTILLER_IOS_DISTILLER_PAGE_IOS_H_