Reading List iOS: Handle multiple pages.

ios/chrome/browser/reading_list/reading_list_distiller_page.h

 // 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.
 
 #ifndef IOS_CHROME_BROWSER_READING_LIST_READING_LIST_DISTILLER_PAGE_H_
 #define IOS_CHROME_BROWSER_READING_LIST_READING_LIST_DISTILLER_PAGE_H_
 
 #include <memory>
 #include <string>
 
@@ skipping 27 matching lines @@
   ReadingListDistillerPageDelegate();
   DISALLOW_COPY_AND_ASSIGN(ReadingListDistillerPageDelegate);
 };
 
 // An DistillerPageIOS that will retain WebState to allow favicon download and
 // and add a 2 seconds delay between loading and distillation.
 class ReadingListDistillerPage : public dom_distiller::DistillerPageIOS {
  public:
   typedef base::Callback<void(const GURL&, const GURL&)> RedirectionCallback;
 
-  // Creates a ReadingListDistillerPage. WebStates to download the pages will
-  // be provided by web_state_dispatcher.
+  // Creates a ReadingListDistillerPage to distill |url|. WebStates to download
+  // the pages will be provided by web_state_dispatcher.
   // |browser_state|, |web_state_dispatcher| and |delegate| must not be null.
   explicit ReadingListDistillerPage(
+      const GURL& url,
       web::BrowserState* browser_state,
       FaviconWebStateDispatcher* web_state_dispatcher,
       ReadingListDistillerPageDelegate* delegate);
   ~ReadingListDistillerPage() override;
 
  protected:
   void DistillPageImpl(const GURL& url, const std::string& script) override;

[gambard, 2017/03/17 16:14:16]

Are you sure this won't be called for another "origin" page?

[Olivier, 2017/03/17 16:32:42]

Discussed offline.
Yes, we control this code.

   void OnDistillationDone(const GURL& page_url,
                           const base::Value* value) override;
   void OnLoadURLDone(
       web::PageLoadCompletionStatus load_completion_status) override;
 
  private:
   // Returns wether there is the loading has no error and if the distillation
   // can continue.
   bool IsLoadingSuccess(web::PageLoadCompletionStatus load_completion_status);
   // Work around the fact that articles opened from Google Search page and
@@ skipping 21 matching lines @@
 
   // Continue the distillation on the page that is currently loaded in
   // |CurrentWebState()|.
   void ContinuePageDistillation();
   // Starts the fetching of |page_url|'s favicon.
   void FetchFavicon(const GURL& page_url);
 
   // Continues distillation by calling superclass |OnLoadURLDone|.
   void DelayedOnLoadURLDone(int delayed_task_id);
   GURL original_url_;
+  bool distilling_main_page_;
 
   FaviconWebStateDispatcher* web_state_dispatcher_;
   ReadingListDistillerPageDelegate* delegate_;
   int delayed_task_id_;
   base::WeakPtrFactory<ReadingListDistillerPage> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(ReadingListDistillerPage);
 };
 
 }  // namespace reading_list
 
 #endif  // IOS_CHROME_BROWSER_READING_LIST_READING_LIST_DISTILLER_PAGE_H_